#include "set_couriers.hpp"

#include <fmt/format.h>

#include <userver/server/handlers/http_handler_base.hpp>
#include <userver/storages/postgres/cluster.hpp>
#include <userver/storages/postgres/component.hpp>
#include <userver/utils/assert.hpp>
#include <userver/formats/json/value.hpp>

namespace enrollment_couriers{
namespace {
class Couriers final : public userver::server::handlers::HttpHandlerBase{
 public:
  static constexpr std::string_view kName = "handler-set-couriers";

  Couriers(const userver::components::ComponentConfig& config,
       const userver::components::ComponentContext& component_context)
      : HttpHandlerBase(config, component_context),
        pg_cluster_(
            component_context
                .FindComponent<userver::components::Postgres>("postgres-db-1")
                .GetCluster()) {}

  std::string HandleRequestThrow(
    const userver::server::http::HttpRequest& request,
        userver::server::request::RequestContext&) const override {
      const auto& courier_list = userver::formats::json::FromString(request.RequestBody());
      int courier_count = 0;
      if(courier_list.IsEmpty())
        return "Nothing to add...";
      else{
        for(const auto& value : courier_list){
          ++courier_count;
          const auto& regions = value["regions"].As<std::vector<int>>();
          std::string type = value["courier_type"].As<std::string>();
          std::vector<std::string> hours = value["working_hours"].As<std::vector<std::string>>();
          std::vector<std::vector<std::string>> working_hours;
          for(auto& hour : hours){
              std::string hour1(hour.begin(), hour.begin() + hour.find('-')), hour2(hour.begin() + hour.find('-') + 1, hour.end());
              working_hours.push_back({hour1, hour2});
           }

          auto result = pg_cluster_->Execute(
              userver::storages::postgres::ClusterHostType::kMaster,
              "INSERT INTO hello_schema.couriers(courier_type, regions, working_hours)"
              "VALUES ($1, $2, $3)", type, regions, working_hours);
        }
        return SayAddCourier(courier_count);
      }
  }
  userver::storages::postgres::ClusterPtr pg_cluster_;

};
}

  std::string SayAddCourier(int value){
    return fmt::format("Added {} couriers", value);
  }

  void AppendCouriers(userver::components::ComponentList& component_list) {
    component_list.Append<Couriers>();
  }
}