/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 **/

#ifndef QUICKSTEP_PARSER_PARSE_SELECT_HPP_
#define QUICKSTEP_PARSER_PARSE_SELECT_HPP_

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

#include "parser/ParseGroupBy.hpp"
#include "parser/ParseHaving.hpp"
#include "parser/ParseLimit.hpp"
#include "parser/ParseOrderBy.hpp"
#include "parser/ParsePredicate.hpp"
#include "parser/ParseSelectionClause.hpp"
#include "parser/ParseTableReference.hpp"
#include "parser/ParseTreeNode.hpp"
#include "parser/ParseWindow.hpp"
#include "utility/Macros.hpp"
#include "utility/PtrList.hpp"

#include "glog/logging.h"

namespace quickstep {

/**
 * @brief The parsed representation of a (sub-query) SELECT statement.
 **/
class ParseSelect : public ParseTreeNode {
 public:
  /**
   * @brief Constructor.
   * @note Takes ownership of all pointers.
   *
   * @param line_number Line number of the first token of this node in the SQL statement.
   * @param column_number Column number of the first token of this node in the SQL statement.
   * @param selection The parsed selection, which becomes owned by this
   *        ParseStatementSelect.
   * @param from_list The parsed list of table references in the FROM clause,
   *        which becomes owned by this ParseStatementSelect.
   * @param where_predicate An optional predicate representing the parsed
   *        WHERE clause (may be NULL if no predicate). Becomes owned by this
   *        ParseStatementSelect if non-NULL.
   * @param group_by Optional parsed GROUP BY clause. Becomes owned by this ParseStatementSelect.
   * @param having_predicate Optional parsed HAVING clause. Becomes owned by this ParseStatementSelect.
   * @param order_by Optional parsed ORDER BY clause. Becomes owned by this ParseStatementSelect.
   * @param limit Optional parsed LIMIT clause. Becomes owned by this ParseStatementSelect.
   **/
  ParseSelect(const int line_number,
              const int column_number,
              ParseSelectionClause *selection,
              PtrList<ParseTableReference> *from_list,
              ParsePredicate *where_predicate,
              ParseGroupBy *group_by,
              ParseHaving *having,
              ParseOrderBy *order_by,
              ParseLimit *limit,
              PtrList<ParseWindow> *window_list)
      : ParseTreeNode(line_number, column_number),
        selection_(selection),
        from_list_(from_list),
        where_predicate_(where_predicate),
        group_by_(group_by),
        having_(having),
        order_by_(order_by),
        limit_(limit),
        window_list_(window_list) {
  }

  ~ParseSelect() override {
  }

  std::string getName() const override {
    return "Select";
  }

  /**
   * @brief Gets the selection.
   *
   * @return The selection.
   **/
  const ParseSelectionClause& selection() const {
    return *selection_;
  }

  /**
   * @brief Gets the FROM list.
   *
   * @return The list of table references in the FROM clause.
   **/
  const PtrList<ParseTableReference>& from_list() const {
    return *from_list_;
  }

  /**
   * @brief Determines whether this select statement has a WHERE predicate.
   *
   * @return Whether there is a WHERE predicate in this statement.
   **/
  bool hasWherePredicate() const {
    return where_predicate_.get() != nullptr;
  }

  /**
   * @brief Gets the WHERE predicate.
   * @warning Always call hasWherePredicate() first.
   *
   * @return The parsed WHERE predicate.
   **/
  const ParsePredicate& where_predicate() const {
    DCHECK(hasWherePredicate());
    return *where_predicate_;
  }

  /**
   * @brief Gets the parsed GROUP BY.
   *
   * @return The parsed GROUP BY.
   */
  const ParseGroupBy* group_by() const { return group_by_.get(); }

  /**
   * @brief Gets the parsed HAVING.
   *
   * @return The parsed HAVNING.
   */
  const ParseHaving* having() const { return having_.get(); }

  /**
   * @brief Gets the parsed ORDER BY.
   *
   * @return The parsed ORDER BY.
   */
  const ParseOrderBy* order_by() const { return order_by_.get(); }

  /**
   * @brief Gets the parsed LIMIT.
   *
   * @return The parsed LIMIT.
   */
  const ParseLimit* limit() const { return limit_.get(); }

  /**
   * @brief Gets the parsed WINDOW.
   *
   * @return The parsed WINDOW.
   */
  const PtrList<ParseWindow>* window_list() const { return window_list_.get(); }

 protected:
  void 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 override {
    non_container_child_field_names->push_back("select_clause");
    non_container_child_fields->push_back(selection_.get());

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

    if (from_list_ != nullptr) {
      container_child_field_names->push_back("from_clause");
      container_child_fields->emplace_back();
      for (const ParseTableReference &from_item : *from_list_) {
        container_child_fields->back().push_back(&from_item);
      }
    }

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

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

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

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

    if (window_list_ != nullptr) {
      container_child_field_names->push_back("window_list");
      container_child_fields->emplace_back();
      for (const ParseWindow &window : *window_list_) {
        container_child_fields->back().push_back(&window);
      }
    }
  }

 private:
  std::unique_ptr<ParseSelectionClause> selection_;
  std::unique_ptr<PtrList<ParseTableReference> > from_list_;
  std::unique_ptr<ParsePredicate> where_predicate_;
  std::unique_ptr<ParseGroupBy> group_by_;
  std::unique_ptr<ParseHaving> having_;
  std::unique_ptr<ParseOrderBy> order_by_;
  std::unique_ptr<ParseLimit> limit_;
  std::unique_ptr<PtrList<ParseWindow>> window_list_;

  DISALLOW_COPY_AND_ASSIGN(ParseSelect);
};

}  // namespace quickstep

#endif /* QUICKSTEP_PARSER_PARSESELECT_HPP_ */