//SPDX-FileCopyrightText: 2024 Yury Gubich <blue@macaw.me>
//SPDX-License-Identifier: GPL-3.0-or-later

#include "addasset.h"

#include <map>

#include "server/server.h"
#include "server/session.h"
#include "database/exceptions.h"

Handler::AddAsset::AddAsset (std::weak_ptr<Server> server):
    Handler("addAsset", Request::Method::post),
    server(server)
{}

void Handler::AddAsset::handle (Request& request) {
    std::string access = request.getAuthorizationToken();
    if (access.empty())
        return error(request, Response::Status::unauthorized);

    if (access.size() != 32)
        return error(request, Response::Status::badRequest);

    std::shared_ptr<Server> srv = server.lock();
    if (!srv)
        return error(request, Response::Status::internalError);

    std::map form = request.getForm();
    std::map<std::string, std::string>::const_iterator itr = form.find("currency");
    if (itr == form.end())
        return error(request, Response::Status::badRequest);

    DB::Asset asset;
    asset.currency = std::stoi(itr->second);
    //TODO validate the currency

    itr = form.find("title");
    if (itr == form.end())
        return error(request, Response::Status::badRequest);

    asset.title = itr->second;

    itr = form.find("icon");
    if (itr == form.end())
        return error(request, Response::Status::badRequest);

    asset.icon = itr->second;

    try {
        Session& session = srv->getSession(access);

        asset.owner = session.owner;
        asset = srv->getDatabase()->addAsset(asset);

        Response& res = request.createResponse(Response::Status::ok);
        res.send();

        session.assetAdded(asset);

    } catch (const DB::NoSession& e) {
        return error(request, Response::Status::unauthorized);
    } catch (const std::exception& e) {
        std::cerr << "Exception on poll:\n\t" << e.what() << std::endl;
        return error(request, Response::Status::internalError);
    }  catch (...) {
        std::cerr << "Unknown exception on poll" << std::endl;
        return error(request, Response::Status::internalError);
    }
}

void Handler::AddAsset::error (Request& request, Response::Status status) {
    Response& res = request.createResponse(status);
    res.send();
}