filewebserver.cpp

#include "httpconnection.h"
#include "filewebserver.h"
#include "generator.h"

#include <boost/beast/version.hpp>
#include <boost/beast/core.hpp>
#include <boost/beast/http.hpp>
#include <boost/asio.hpp>

#include <iostream>

namespace
{
    const size_t SUCCESS = 0;
    const size_t ERROR_UNHANDLED_EXCEPTION = 2;
}

FileWebServer::FileWebServer()
    : Configurable()
    , mServerIP{"127.0.0.1"}
    , mPort{8080}
    , mPath{"d://"}
    , mGenerator{nullptr}
{
}

FileWebServer::~FileWebServer()
{
}

void http_server(boost::asio::ip::tcp::acceptor &acceptor, boost::asio::ip::tcp::socket &socket, const std::string &response_html, FileSystemModel &fileSystemModel, Generator &htmlContentGenerator)
{
    acceptor.async_accept(socket, [&, response_html](boost::beast::error_code ec)
    {
        if (!ec)
        {
            auto connection=std::make_shared<http_connection>(std::move(socket), response_html);
            connection->setFileSystemModel(&fileSystemModel);
            connection->setHTMLContentGenerator(&htmlContentGenerator);
            connection->start();


        }
//        std::cout << "IP address: " << socket.remote_endpoint().address().to_string() << std::endl;
        http_server(acceptor, socket, response_html, fileSystemModel, htmlContentGenerator);
    });
}

int FileWebServer::run()
{
    try
    {
        try
        {
            auto const address = boost::asio::ip::make_address(mServerIP);

            boost::asio::io_context ioc{1};

            boost::asio::ip::tcp::acceptor acceptor{ioc, {address,
                                                          static_cast<unsigned short>(mPort)}};
            boost::asio::ip::tcp::socket socket{ioc};

            std::cout << "Server started on " << mServerIP << ":" << mPort <<std::endl;
            std::cout << "Working directory is a " << mPath <<std::endl;
            if(mGenerator)
            {
                http_server(acceptor, socket, mGenerator->generate(mPath), *mFileSysemModel, *mGenerator);
            }
            ioc.run();
        }
        catch (std::exception const &e)
        {
            std::cerr << "Error: " << e.what() << std::endl;
            return EXIT_FAILURE;
        }
    }
    catch(std::exception& e)
    {
        std::cerr << "Unhandled Exception reached the top of main: "
                  << e.what() << ", application will now exit" << std::endl;
        return ERROR_UNHANDLED_EXCEPTION;
    }

    return SUCCESS;
}

void FileWebServer::setConfiguration(Configuration config)
{
    mPath = config.dir();
    mPort = static_cast<unsigned>(config.port());
}

void FileWebServer::setGenerator(Generator *generator)
{
    mGenerator = generator;
}

void FileWebServer::setFileSysemModel(FileSystemModel *fileSysemModel)
{
    mFileSysemModel = fileSysemModel;
}