/**
 * 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.
 **/

#include "query_optimizer/physical/TableReference.hpp"

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

#include "catalog/CatalogRelation.hpp"
#include "catalog/PartitionScheme.hpp"
#include "catalog/PartitionSchemeHeader.hpp"
#include "query_optimizer/OptimizerTree.hpp"
#include "query_optimizer/physical/PartitionSchemeHeader.hpp"
#include "query_optimizer/physical/Physical.hpp"
#include "utility/Cast.hpp"

namespace quickstep {
namespace optimizer {
namespace physical {

namespace E = ::quickstep::optimizer::expressions;

TableReferencePtr TableReference::Create(
    const CatalogRelation *relation,
    const std::string &alias,
    const std::vector<expressions::AttributeReferencePtr> &attribute_list) {
  std::unique_ptr<PartitionSchemeHeader> output_partition_scheme_header;
  const quickstep::PartitionScheme *partition_scheme = relation->getPartitionScheme();
  if (partition_scheme) {
    const quickstep::PartitionSchemeHeader &partition_scheme_header = partition_scheme->getPartitionSchemeHeader();

    PartitionSchemeHeader::PartitionType physical_partition_type;
    switch (partition_scheme_header.getPartitionType()) {
      case quickstep::PartitionSchemeHeader::PartitionType::kHash:
        physical_partition_type = PartitionSchemeHeader::PartitionType::kHash;
        break;
      case quickstep::PartitionSchemeHeader::PartitionType::kRange:
        physical_partition_type = PartitionSchemeHeader::PartitionType::kRange;
        break;
      default:
        return nullptr;
    }

    PartitionSchemeHeader::PartitionExprIds partition_expr_ids;
    for (const attribute_id part_attr : partition_scheme_header.getPartitionAttributeIds()) {
      partition_expr_ids.push_back({ attribute_list[part_attr]->id() });
    }

    output_partition_scheme_header =
        std::make_unique<PartitionSchemeHeader>(physical_partition_type,
                                                partition_scheme_header.getNumPartitions(),
                                                std::move(partition_expr_ids));
  }

  return TableReferencePtr(new TableReference(relation, alias, attribute_list,
                                              output_partition_scheme_header.release()));
}

PhysicalPtr TableReference::copyWithNewChildren(
    const std::vector<PhysicalPtr> &new_children) const {
  DCHECK_EQ(new_children.size(), children().size());
  std::unique_ptr<PartitionSchemeHeader> output_partition_scheme_header;
  return TableReferencePtr(new TableReference(relation_, alias_, attribute_list_,
                                              cloneOutputPartitionSchemeHeader()));
}

void TableReference::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<OptimizerTreeBaseNodePtr> *non_container_child_fields,
    std::vector<std::string> *container_child_field_names,
    std::vector<std::vector<OptimizerTreeBaseNodePtr>> *container_child_fields) const {
  inline_field_names->push_back("relation");
  inline_field_values->push_back(relation_->getName());

  if (alias_ != relation_->getName()) {
    inline_field_names->push_back("alias");
    inline_field_values->push_back(alias_);
  }

  if (partition_scheme_header_) {
    inline_field_names->push_back("output_partition_scheme_header");
    inline_field_values->push_back(partition_scheme_header_->toString());
  }

  container_child_field_names->push_back("");
  container_child_fields->push_back(CastSharedPtrVector<OptimizerTreeBase>(attribute_list_));
}

}  // namespace physical
}  // namespace optimizer
}  // namespace quickstep