/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * 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 <thrift/compiler/ast/t_enum.h>

#include <stdexcept>

namespace apache {
namespace thrift {
namespace compiler {

void t_enum::set_values(t_enum_value_list values) {
  values_.clear();
  values_raw_.clear();
  constants_.clear();
  value_map_.clear();
  unused_ = t_enum::default_unused;
  for (auto& value : values) {
    append(std::move(value));
  }
}

void t_enum::append_value(std::unique_ptr<t_enum_value> enum_value) {
  if (!enum_value->has_value()) {
    auto last_value = values_.empty() ? -1 : values_.back()->get_value();
    if (last_value == INT32_MAX) {
      throw std::runtime_error(
          "enum value overflow: " + enum_value->get_name());
    }
    enum_value->set_implicit_value(last_value + 1);
  }
  auto const_val = std::make_unique<t_const_value>(enum_value->get_value());
  const_val->set_is_enum();
  const_val->set_enum(this);
  const_val->set_enum_value(enum_value.get());
  auto tconst = std::make_unique<t_const>(
      program_,
      &t_primitive_type::t_i32(),
      enum_value->get_name(),
      std::move(const_val));
  append(std::move(enum_value), std::move(tconst));
}

const t_enum_value* t_enum::find_value(int32_t value) const {
  auto itr = value_map_.find(value);
  if (itr != value_map_.end()) {
    return itr->second;
  }
  return nullptr;
}

const t_const* t_enum::find_const_by_name(std::string_view name) const {
  auto itr = consts_by_name_.find(name);
  if (itr != consts_by_name_.end()) {
    return itr->second;
  }
  return nullptr;
}

void t_enum::update_unused(int32_t val) {
  if (unused_ != val) {
    return;
  }
  // Find next unused value in O(NlnN) worst case total time.
  for (auto itr = value_map_.upper_bound(unused_++);
       itr != value_map_.end() && itr->first == unused_;
       ++itr, ++unused_) {
  }
}

} // namespace compiler
} // namespace thrift
} // namespace apache