#include "Net.hpp"

#include "Caps.hpp"

namespace molo::example::fw {

RestClient::RestClient(RestTransport transport)
    : transport_(std::move(transport)) {}

Response RestClient::get(const std::string& path) {
  Request req;
  req.method = Method::Get;
  req.path = path;
  return transport_(std::move(req));
}

Response RestClient::post_json(const std::string& path,
                               const std::string& json) {
  Request req;
  req.method = Method::Post;
  req.path = path;
  req.body = json;
  req.headers.set("Content-Type", "application/json");
  return transport_(std::move(req));
}

void OpenApi::add(const std::string& path, const std::string& method,
                  const std::string& summary) {
  if (routes_.size() >= caps::kMaxOpenApiRoutes) return;
  routes_.push_back({path, method, summary});
}

std::string OpenApi::to_json() const {
  std::string out =
      "{\"openapi\":\"3.0.0\",\"info\":{\"title\":\"living-example\","
      "\"version\":\"0.1.0\"},\"paths\":{";
  bool first = true;
  for (const auto& r : routes_) {
    if (!first) out += ",";
    first = false;
    out += "\"" + json_escape(r.path) + "\":{\"" + json_escape(r.method) +
           "\":{\"summary\":\"" + json_escape(r.summary) + "\"}}";
  }
  return out + "}}";
}

std::size_t OpenApi::size() const { return routes_.size(); }

std::string redact_secrets(const std::string& text) {
  std::string out = text;
  const char* keys[] = {"DEVPLACE_API_KEY", "API_KEY", "SECRET", "TOKEN",
                        "PASSWORD"};
  for (const char* k : keys) {
    std::string key(k);
    std::size_t pos = 0;
    while ((pos = out.find(key, pos)) != std::string::npos) {
      std::size_t eol = out.find('\n', pos);
      if (eol == std::string::npos) eol = out.size();
      std::size_t eq = out.find('=', pos);
      if (eq != std::string::npos && eq < eol) {
        out.replace(eq + 1, eol - eq - 1, "[REDACTED]");
        pos = eq + 1;
      } else {
        pos += key.size();
      }
    }
  }
  return out;
}

AiSample ai_harness_sample(const std::string& prompt) {
  AiSample s;
  s.base_url = "https://devplace.net/openai/v1";
  s.model = "molodetz";
  s.redacted_prompt = redact_secrets(prompt);
  s.stream_chunks = {"data: {\"delta\":\"plan: add route\"}\n\n",
                     "data: {\"delta\":\"apply: 1 file\"}\n\n",
                     "data: [DONE]\n\n"};
  return s;
}

}  // namespace molo::example::fw