pica/server/server.cpp

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

#include "server.h"

constexpr uint8_t currentDbVesion = 1;

Server::Server():
    terminating(false),
    requestCount(0),
    serverName(std::nullopt),
    router(),
    db()
{
    std::cout << "Startig pica..." << std::endl;

    db = DBInterface::create(DBInterface::Type::mysql);
    std::cout << "Database type: MySQL" << std::endl;

    db->setCredentials("pica", "pica");
    db->setDatabase("pica");

    db->connect("/run/mysqld/mysqld.sock");
    db->migrate(currentDbVesion);

    router.addRoute("info", Server::info);
    router.addRoute("env", Server::printEnvironment);
}

Server::~Server() {}

void Server::run(int socketDescriptor) {
    while (!terminating) {
        std::unique_ptr<Request> request = std::make_unique<Request>();
        bool result = request->wait(socketDescriptor);
        if (!result) {
            std::cerr << "Error accepting a request" << std::endl;
            return;
        }
        handleRequest(std::move(request));
    }
}

void Server::handleRequest(std::unique_ptr<Request> request) {
    ++requestCount;
    if (!serverName) {
        try {
            serverName = request->getServerName();
            std::cout << "received server name " << serverName.value() << std::endl;
        } catch (...) {
            std::cerr << "failed to read server name" << std::endl;
            Response error(Response::Status::internalError);
            error.replyTo(*request.get());
            return;
        }
    }

    if (!request->isGet()) {
        static const Response methodNotAllowed(Response::Status::methodNotAllowed);
        methodNotAllowed.replyTo(*request.get());
        return;
    }

    try {
        std::string path = request->getPath(serverName.value());
        router.route(path.data(), std::move(request), this);
    } catch (const std::exception& e) {
        Response error(Response::Status::internalError);
        error.setBody(std::string(e.what()));
        error.replyTo(*request.get());
    }
}

bool Server::printEnvironment(Request* request, Server* server) {
    UNUSED(server);
    nlohmann::json body = nlohmann::json::object();
    request->printEnvironment(body);

    Response res;
    res.setBody(body);
    res.replyTo(*request);

    return true;
}

bool Server::info(Request* request, Server* server) {
    UNUSED(server);
    Response res;
    nlohmann::json body = nlohmann::json::object();
    body["type"] = PROJECT_NAME;
    body["version"] = PROJECT_VERSION;

    res.setBody(body);
    res.replyTo(*request);

    return true;
}