#include "mnn_tts_config.hpp"

MNNTTSConfig::MNNTTSConfig(const std::string &config_json_path)
{
  // 检查文件是否存在且是常规文件
  if (!fs::exists(config_json_path) || !fs::is_regular_file(config_json_path))
  {
    throw std::runtime_error("Config file not found or is not a regular file: " + config_json_path);
  }

  std::ifstream file(config_json_path);
  if (!file.is_open())
  {
    throw std::runtime_error("Failed to open config file: " + config_json_path);
  }

  try
  {
    file >> raw_config_data_; // 直接从文件流解析JSON
  }
  catch (const nlohmann::json::parse_error &e)
  {
    throw std::runtime_error("Error parsing config.json (" + config_json_path + "): " + e.what());
  }
  catch (const std::exception &e)
  {
    throw std::runtime_error("Error reading config.json (" + config_json_path + "): " + e.what());
  }

  try
  {
    model_type_ = get_value_from_json<std::string>(raw_config_data_, "model_type");
    model_path_ = get_value_from_json<std::string>(raw_config_data_, "model_path");
    asset_folder_ = get_value_from_json<std::string>(raw_config_data_, "asset_folder");
    cache_folder_ = get_value_from_json<std::string>(raw_config_data_, "cache_folder");
    sample_rate_ = get_value_from_json<int>(raw_config_data_, "sample_rate");
  }
  catch (const std::runtime_error &e)
  {
    // 捕获 get_value_from_json 抛出的异常，并添加更多上下文信息
    throw std::runtime_error("Error in config file " + config_json_path + ": " + e.what());
  }
}

// 新增：支持参数覆盖的构造函数
MNNTTSConfig::MNNTTSConfig(const std::string &config_file_path, 
                           const std::map<std::string, std::string> &overrides)
    : MNNTTSConfig(config_file_path)  // 委托给原构造函数
{
  // 应用参数覆盖
  applyOverrides(overrides);
}

// 应用参数覆盖
void MNNTTSConfig::applyOverrides(const std::map<std::string, std::string> &overrides) {
  if (overrides.empty()) {
    return;
  }
  
  for (const auto& [key, value] : overrides) {
    try {
      if (key == "model_type") {
        model_type_ = value;
      } else if (key == "sample_rate") {
        sample_rate_ = std::stoi(value);
      }
      // 可以继续添加其他参数的覆盖逻辑
    } catch (const std::exception &e) {
      // 忽略无法转换的参数，使用配置文件中的默认值
      std::cerr << "Warning: Failed to override parameter '" << key 
                << "' with value '" << value << "': " << e.what() << std::endl;
    }
  }
}