WebServer.h

/*
 * WebServer.h
 *
 *  Created on: May 19, 2017
 *      Author: kolban
 */

#ifndef CPP_UTILS_WEBSERVER_H_
#define CPP_UTILS_WEBSERVER_H_
#include <string>
#include <vector>
#include <regex>
#include <map>
#include "sdkconfig.h"

#ifdef CONFIG_MONGOOSE_PRESENT
#include <mongoose.h>

#define MAX_CHUNK_LENGTH 4090 // 4 kilobytes

class WebServer;

class WebServer {
public:
    class HTTPRequest {
        public:
            HTTPRequest(struct http_message* message);
            const char* getMethod() const;
            const char* getPath() const;
            const char* getBody() const;
            size_t getMethodLen() const;
            size_t getPathLen() const;
            size_t getBodyLen() const;
            std::map<std::string, std::string> getQuery() const;
            std::vector<std::string> pathSplit() const;

        private:
            struct http_message* m_message;

    }; // HTTPRequest

    class HTTPResponse {
        public:
            HTTPResponse(struct mg_connection* nc);
            void addHeader(const std::string& name, const std::string& value);
            void addHeader(std::string&& name, std::string&& value);
            void setStatus(int status);
            void setHeaders(const std::map<std::string, std::string>& headers);
            void setHeaders(std::map<std::string, std::string>&& headers);
            void sendData(const std::string& data);
            void sendData(const uint8_t* pData, size_t length);
            void sendData(const char* pData, size_t length);
            const std::string& getRootPath() const;
            void setRootPath(const std::string& path);
            void setRootPath(std::string&& path);
            void sendChunkHead();
            void sendChunk(const char* pData, size_t length);
            void closeConnection();

        private:
            struct mg_connection* m_nc;
            std::string m_rootPath;
            int m_status;
            std::map<std::string, std::string> m_headers;
            bool m_dataSent;
            std::string buildHeaders();

    }; // HTTPResponse

    class HTTPMultiPart {
    public:
        virtual ~HTTPMultiPart() = default;
        virtual void begin(const std::string& varName, const std::string& fileName);
        virtual void end();
        virtual void data(const std::string& data);
        virtual void multipartEnd();
        virtual void multipartStart();

    }; // HTTPMultiPart

    class HTTPMultiPartFactory {
    public:
        virtual HTTPMultiPart* newInstance() = 0;

    };

    class PathHandler {
        public:
            PathHandler(const std::string& method, const std::string& pathPattern, void (*webServerRequestHandler) (WebServer::HTTPRequest* pHttpRequest, WebServer::HTTPResponse* pHttpResponse));
            PathHandler(std::string&& method, const std::string& pathPattern, void (*webServerRequestHandler) (WebServer::HTTPRequest* pHttpRequest, WebServer::HTTPResponse* pHttpResponse));
            bool match(const char* method, size_t method_len, const char* path);
            void invoke(HTTPRequest* request, HTTPResponse* response);

        private:
            std::string m_method;
            std::regex m_pattern;
            void (*m_requestHandler)(WebServer::HTTPRequest* pHttpRequest, WebServer::HTTPResponse* pHttpResponse);

    }; // PathHandler

    class WebSocketHandler {
    public:
        void onCreated();
        virtual void onMessage(const std::string& message);
        void onClosed();
        void sendData(const std::string& message);
        void sendData(const uint8_t* data, uint32_t size);
        void close();

    private:
        struct mg_connection* m_mgConnection;

    };

    class WebSocketHandlerFactory {
    public:
        virtual WebSocketHandler* newInstance() = 0;

    };

    WebServer();
    virtual ~WebServer();
    void addPathHandler(const std::string& method, const std::string& pathExpr, void (*webServerRequestHandler) (WebServer::HTTPRequest* pHttpRequest, WebServer::HTTPResponse* pHttpResponse));
    void addPathHandler(std::string&& method, const std::string& pathExpr, void (*webServerRequestHandler) (WebServer::HTTPRequest* pHttpRequest, WebServer::HTTPResponse* pHttpResponse));
    const std::string& getRootPath();
    void setMultiPartFactory(HTTPMultiPartFactory* pMultiPartFactory);
    void setRootPath(const std::string& path);
    void setRootPath(std::string&& path);
    void setWebSocketHandlerFactory(WebSocketHandlerFactory* pWebSocketHandlerFactory);
    void start(unsigned short port = 80);
    void processRequest(struct mg_connection* mgConnection, struct http_message* message);
    void continueConnection(struct mg_connection* mgConnection);
    HTTPMultiPartFactory* m_pMultiPartFactory;
    WebSocketHandlerFactory* m_pWebSocketHandlerFactory;

private:
    std::string m_rootPath;
    std::vector<PathHandler> m_pathHandlers;
    std::map<sock_t, FILE*> unfinishedConnection;

};

#endif // CONFIG_MONGOOSE_PRESENT
#endif /* CPP_UTILS_WEBSERVER_H_ */