WebFrame/src/runtimes/server/runtime.cpp at 35d544ad30e15f8eb910cae8ef7fe70c073d2f23 · maxtek6/WebFrame · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#ifdef _WIN32
#include <winsock2.h>
#endif

#include <server.hpp>

#include <stdexcept>

#ifdef _WIN32
namespace
{
    class winsock_session
    {
    public:
        winsock_session()
        {
            const int result = WSAStartup(MAKEWORD(2, 2), &_data);
            if (result != 0)
            {
                throw std::runtime_error("WSAStartup failed");
            }
        }

        ~winsock_session()
        {
            WSACleanup();
        }

    private:
        WSADATA _data{};
    };
}
#endif

static void evhttp_callback(struct evhttp_request *req, void *arg)
{
    bool sent_response = false;
    int status = 200;
    webframe::router *runtime = static_cast<webframe::router *>(arg);
    webframe::server::request request(req);
    webframe::server::response response(req, sent_response, status);
    webframe::handler *handler = runtime->find_route(request.get_path());
    handler->handle_request(&request, &response);
    if (!sent_response)
    {
        evhttp_send_reply(req, status, nullptr, nullptr);
    }
}

namespace webframe
{
    namespace server
    {
        int runtime::dispatch(int argc, const char **argv, webframe::application *a, webframe::router *r)
        {
#ifdef _WIN32
            winsock_session winsock;
#endif
            webframe::server_config config;
            config.set_host("0.0.0.0");
            config.set_port(8080);
            a->configure_server(&config, argc, argv);
            a->configure_router(r);
            _event_base.reset(event_base_new());
            _http.reset(evhttp_new(_event_base.get()));
            _context = std::make_unique<context>(_event_base.get());
            evhttp_set_gencb(_http.get(), evhttp_callback, r);
            listen(config.get_host().c_str(), config.get_port());
            a->launch_server(_context.get());
            serve();
            return 0;
        }
        void runtime::listen(const char *address, int port)
        {
            evhttp_bind_socket(_http.get(), address, port);
        }
        void runtime::serve()
        {
            event_base_dispatch(_event_base.get());
        }
        void runtime::stop()
        {
            event_base_loopbreak(_event_base.get());
        }
    }
    runtime *create_runtime()
    {
        return new server::runtime();
    }
}