#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 GetDiariesExecutor : public IExecutor<Body, Allocator, ResponseType>
{
  mysqlx::Session& session_;
  const std::shared_ptr<IAuthDAO>& auth_dao_;
  const std::shared_ptr<IDiariesDAO>& diaries_dao_;

public:
  GetDiariesExecutor(
    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 =
      "GET /api/v1/User/Medications - Response 401: Unauthorized"s;

    BOOST_LOG_TRIVIAL(info) << "GET /api/v1/User/Medications - 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()};

    std::vector<diary_dto> diaries = diaries_dao_->GetDiariesByUserUUID(user_uuid);

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

    response_body.emplace_object();
    response_body.as_object().emplace("diaries", toJSONArray(diaries));

    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;
  }

  boost::json::array toJSONArray(const std::vector<diary_dto>& schemes)
  {
    using namespace boost;
    using namespace boost::json;
    using namespace boost::beast;
    using namespace std::string_literals;
    using namespace std::string_view_literals;

    json::array arr;

    for (const auto& m : schemes)
    {
      arr.emplace_back(ToJSON(m));
    }

    return std::move(arr);
  }
};
}