summaryrefslogtreecommitdiff
path: root/bot/include/http_webhook_server.hpp
blob: 8ba684162b1b703a7c2d28312b3affe4b462d640 (plain)
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
#pragma once

#include <functional>
#include <string>
#include <unordered_map>
#include <system_error>
#include <sstream>
#include <string>
#include <netinet/in.h>

#ifdef __linux__
    #include <sys/epoll.h>
#elif defined(__APPLE__)
    #include <sys/event.h>
#endif

class HttpWebhookServer {
public:
    struct HttpRequest {
        std::string method;
        std::string path;
        std::unordered_map<std::string, std::string> headers;
        std::string body;
    };

    struct HttpResponse {
        int status_code = 200;
        std::unordered_map<std::string, std::string> headers;
        std::string body;
    };

    using Handler = std::function<HttpResponse(const HttpRequest&)>;

    HttpWebhookServer(uint16_t port, Handler handler);
    ~HttpWebhookServer();
    void setupEpoll();
    void handleEvents();
    void start();
    void stop();

private:
    struct ClientContext {
        std::string input_buffer;
        std::string output_buffer;
        size_t bytes_written = 0;
    };

    void setupSocket();
    void setupEventLoop();
    void handleEvent(int fd, uint32_t events);
    void handleClient(int fd);
    void closeClient(int fd);
    void registerFdForRead(int fd);
    void modifyFdToWrite(int fd);
    void sendToClient(int fd);
    void parseHttpRequest(ClientContext& ctx, HttpRequest& req);
    void buildHttpResponse(const HttpResponse& res, std::string& output);

    int server_fd = -1;
    int event_fd = -1;
    int epoll_fd = -1;
    bool running = false;
    uint16_t port;
    Handler request_handler;
    std::unordered_map<int, ClientContext> clients;

#ifdef __linux__
    static constexpr int MAX_EVENTS = 64;
    struct epoll_event events[MAX_EVENTS];
#elif defined(__APPLE__)
    static constexpr int MAX_EVENTS = 64;
    struct kevent events[MAX_EVENTS];
#endif
};