#pragma once
#include <boost/log/trivial.hpp>

#include <regex>
#include <boost/json.hpp>
#include <boost/mpl/vector/vector0.hpp>
#include <mysqlx/xdevapi.h>

#include "IExecutor.h"
#include "../DAO/IAuthDAO.h"
#include "../DAO/IDiariesDAO.h"
#include "../DAO/IUserTreatmentSchemesDAO.h"
#include "../exceptions/session_exception.h"

namespace uad
{
template <class Body, class Allocator, class ResponseType>
class PostDiaryExecutor : public IExecutor<Body, Allocator, ResponseType>
{
  mysqlx::Session& session_;
  const std::shared_ptr<IAuthDAO>& auth_dao_;
  const std::shared_ptr<IDiariesDAO>& diaries_dao_;

public:
  PostDiaryExecutor(
    mysqlx::Session& session,
    const std::shared_ptr<IAuthDAO>& auth_dao,
    const std::shared_ptr<IDiariesDAO>& diaries_dao
  ) : session_(session), auth_dao_(auth_dao), diaries_dao_(diaries_dao)
  {
  }

  [[nodiscard]] boost::beast::http::response<ResponseType> operator ()(
    boost::beast::http::request<Body, boost::beast::http::basic_fields<Allocator>>&& req
  ) override
  {
    using namespace boost;
    using namespace boost::json;
    using namespace boost::beast;
    using namespace std::string_literals;
    using namespace std::string_view_literals;

    constexpr std::string_view auth_prefix = "Bearer "sv;
    static const std::string invalid_token_message =
      "POST /api/v1/Diary - Response 401: Unauthorized"s;

    BOOST_LOG_TRIVIAL(info) << "POST /api/v1/Diary - Request";

    if (req[http::field::authorization].size() <= auth_prefix.size())
    {
      BOOST_LOG_TRIVIAL(error) << invalid_token_message;
      throw session_exception(http::status::unauthorized, "Unauthorized");
    }

    const std::string auth_token = {
        req[http::field::authorization].begin() + auth_prefix.size(),
        req[http::field::authorization].end()
      };

    if (!auth_dao_->HasAuthorized(auth_token))
    {
      BOOST_LOG_TRIVIAL(error) << invalid_token_message;
      throw session_exception(http::status::unauthorized, "Unauthorized");
    }

    std::string_view user_uuid_ref = auth_dao_->GetUUID(auth_token);
    const std::string user_uuid{user_uuid_ref.begin(), user_uuid_ref.end()};

    value val = json::parse(req.body());
    object& obj = val.as_object();

    diary_dto dto;

    dto.uuid = obj["uuid"].as_string().c_str();
    dto.time_ms = obj["time_ms"].as_int64();
    dto.mania_level = obj["mania_level"].as_int64();
    dto.depression_level = obj["depression_level"].as_int64();
    dto.mood_level = obj["mood_level"].as_int64();
    dto.activity_level = obj["activity_level"].as_int64();
    dto.appetite_level = obj["appetite_level"].as_int64();
    dto.dream_level = obj["dream_level"].as_int64();
    dto.anxiety_level = obj["anxiety_level"].as_int64();

    if (obj.contains("comment"))
      dto.comment = obj["comment"].as_string().c_str();

    if (obj.contains("user_treatment_scheme_uuid"))
      dto.user_treatment_scheme_uuid =
        obj["user_treatment_scheme_uuid"].as_string().c_str();

    try
    {
      session_.startTransaction();
      diaries_dao_->СreateDiary(user_uuid, dto);
      session_.commit();
    }
    catch (const mysqlx::Error& e)
    {
      session_.rollback();
      BOOST_LOG_TRIVIAL(error) << "MySQL Error: " << e.what();
      throw session_exception(http::status::internal_server_error, "Internal Server Error");
    }
    catch (const std::exception& e)
    {
      session_.rollback();
      BOOST_LOG_TRIVIAL(error) << "Unexpected Error: " << e.what();
      throw session_exception(http::status::internal_server_error, "Internal Server Error");
    }

    http::response<ResponseType> res{http::status::ok, req.version()};
    value response_body;

    response_body.emplace_object();
    response_body.as_object().emplace("diary", ToJSON(dto));

    res.body() = serialize(response_body);
    res.set(http::field::content_type, "application/json");
    res.content_length(res.body().size());

    return res;
  }

private:
  boost::json::object ToJSON(const diary_dto& diary)
  {
    boost::json::object diary_json;

    diary_json["uuid"] = diary.uuid;
    diary_json["time"] = diary.time_ms;
    diary_json["mania_level"] = diary.mania_level;
    diary_json["depression_level"] = diary.depression_level;
    diary_json["mood_level"] = diary.mood_level;
    diary_json["activity_level"] = diary.activity_level;
    diary_json["appetite_level"] = diary.appetite_level;
    diary_json["dream_level"] = diary.dream_level;
    diary_json["anxiety_level"] = diary.anxiety_level;
    diary_json["comment"] = diary.comment;
    diary_json["user_treatment_scheme_uuid"] =
      diary.user_treatment_scheme_uuid;

    return diary_json;
  }
};
}