/**
 *   Copyright 2016, Quickstep Research Group, Computer Sciences Department,
 *     University of Wisconsin—Madison.
 *
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 **/

#include "parser/ParseCaseExpressions.hpp"

#include <memory>
#include <sstream>
#include <string>
#include <vector>

#include "parser/ParseExpression.hpp"
#include "parser/ParsePredicate.hpp"
#include "parser/ParseTreeNode.hpp"

namespace quickstep {

void ParseSimpleWhenClause::getFieldStringItems(
    std::vector<std::string> *inline_field_names,
    std::vector<std::string> *inline_field_values,
    std::vector<std::string> *non_container_child_field_names,
    std::vector<const ParseTreeNode*> *non_container_child_fields,
    std::vector<std::string> *container_child_field_names,
    std::vector<std::vector<const ParseTreeNode*>> *container_child_fields) const {
  non_container_child_field_names->push_back("condition_operand");
  non_container_child_fields->push_back(condition_operand_.get());

  non_container_child_field_names->push_back("result_expression");
  non_container_child_fields->push_back(result_expression_.get());
}

void ParseSearchedWhenClause::getFieldStringItems(
    std::vector<std::string> *inline_field_names,
    std::vector<std::string> *inline_field_values,
    std::vector<std::string> *non_container_child_field_names,
    std::vector<const ParseTreeNode*> *non_container_child_fields,
    std::vector<std::string> *container_child_field_names,
    std::vector<std::vector<const ParseTreeNode*>> *container_child_fields) const {
  non_container_child_field_names->push_back("condition_predicate");
  non_container_child_fields->push_back(condition_predicate_.get());

  non_container_child_field_names->push_back("result_expression");
  non_container_child_fields->push_back(result_expression_.get());
}

std::string ParseSimpleCaseExpression::generateName() const {
  std::ostringstream out;
  out << "CASE " << case_operand_->generateName();
  for (const ParseSimpleWhenClause &when_clause : *when_clauses_) {
    out << " WHEN " << when_clause.condition_operand()->generateName()
        << " THEN " << when_clause.result_expression()->generateName();
  }
  if (else_result_expression_ != nullptr) {
    out << " ELSE " << else_result_expression_->generateName();
  }
  out << " END";
  return out.str();
}

void ParseSimpleCaseExpression::getFieldStringItems(
    std::vector<std::string> *inline_field_names,
    std::vector<std::string> *inline_field_values,
    std::vector<std::string> *non_container_child_field_names,
    std::vector<const ParseTreeNode*> *non_container_child_fields,
    std::vector<std::string> *container_child_field_names,
    std::vector<std::vector<const ParseTreeNode*>> *container_child_fields) const {
  non_container_child_field_names->push_back("case_operand");
  non_container_child_fields->push_back(case_operand_.get());

  container_child_field_names->push_back("when_clauses");
  container_child_fields->emplace_back();
  for (const ParseSimpleWhenClause &when_clause : *when_clauses_) {
    container_child_fields->back().push_back(&when_clause);
  }

  if (else_result_expression_ != nullptr) {
    non_container_child_field_names->push_back("else_result_expression");
    non_container_child_fields->push_back(else_result_expression_.get());
  }
}

std::string ParseSearchedCaseExpression::generateName() const {
  std::ostringstream out;
  out << "CASE";
  for (const ParseSearchedWhenClause &when_clause : *when_clauses_) {
    // TODO(jianqiao): implement generateName() for Predicate.
    out << " WHEN " << when_clause.condition_predicate()->getShortString()
        << " THEN " << when_clause.result_expression()->generateName();
  }
  if (else_result_expression_ != nullptr) {
    out << " ELSE " << else_result_expression_->generateName();
  }
  out << " END";
  return out.str();
}

void ParseSearchedCaseExpression::getFieldStringItems(
    std::vector<std::string> *inline_field_names,
    std::vector<std::string> *inline_field_values,
    std::vector<std::string> *non_container_child_field_names,
    std::vector<const ParseTreeNode*> *non_container_child_fields,
    std::vector<std::string> *container_child_field_names,
    std::vector<std::vector<const ParseTreeNode*>> *container_child_fields) const {
  container_child_field_names->push_back("when_clauses");
  container_child_fields->emplace_back();
  for (const ParseSearchedWhenClause &when_clause : *when_clauses_) {
    container_child_fields->back().push_back(&when_clause);
  }

  if (else_result_expression_ != nullptr) {
    non_container_child_field_names->push_back("else_result_expression");
    non_container_child_fields->push_back(else_result_expression_.get());
  }
}

}  // namespace quickstep