Distribution_Service/CC_SDK/Include/CCServlet/CCWebServlet.h

#ifndef CCWeb_Servlet_H
#define CCWeb_Servlet_H
#pragma once

#include <CCThreadPool.h>
#include "CCResponse.h"
#include "WebSocket/CCWebSocket.h"
#include "HttpSocketChannel.h"
#include "Socket/CCServerSocket.h"

// 定义 WebServlet 类格式的回调函数宏，用于简化成员函数回调的编写
#define CC_ServletFunClass(Fun) [this](auto && PH1, auto && PH2) { Fun(std::forward<decltype(PH1)>(PH1), std::forward<decltype(PH2)>(PH2)); }
// 定义 WebServlet 普通函数格式的回调函数宏，用于简化普通函数回调的编写
#define CC_ServletFun(Fun) Fun

namespace CTL {
    struct StaticResource {
        // 资源类型
        String Type;
        // 资源内容
        CCVector<char> Content;
        // 资源内容GZip
        CCVector<char> ContentGZip;
        // 资源名称
        String Name;
        // 资源路径
        String Path;
    };
    struct Event_Client {
        SOCKET fd{};
        ByteArray Data{};
    };
    // WebServlet 类，用于处理网络请求和管理服务器状态
    class WebServlet{
    public:
        // 使用 std::function 定义请求处理函数类型
        using RequestFunc = std::function<void(Request& ,Response&)>;
    private:
        // 网络服务器对象
        ServerSocket* server = nullptr;
        // 服务器运行标志
        bool ServerFlag = true;
        EventLoop event_loop_t{};
        // 线程池对象，用于处理并发请求
        ThreadPool M_S_POOl{};
        // 线程池大小配置
        size_t M_Pool_Size = 128,M_Pool_Max = 1024;
        // 线程池中线程的保活时间
        int M_Pool_KeepAliveTime = 1000;
        // 静态资源的 URL 映射
        CCList<CTL::String> StaticPath;
        // 静态资源内容的映射
        CTL::Map<CTL::String,StaticResource> StaticResources;
        // Web 信号处理函数
        RequestFunc WebSignalFun;
        // 请求处理函数映射
        CTL::Map<std::string,RequestFunc> RequestFun;
        // CORS 头对象指针，用于处理跨域请求
        CORS* CORS_Header = nullptr;
        // SSL 上下文对象指针，用于 HTTPS 连接
        CTX_SSL SSL_CTX;
        // WebSocket 对象，用于处理 WebSocket 连接
        WebSocket m_WebSocket;
        // WebSocket 连接标志
        bool WebSocketFlag = false;
        // 请求超时时间 单位/毫秒
        unsigned long TimingTime = 1000;
        // 设置请求头缓冲区大小
        unsigned int ReadHBuff = 4096;
        IPVX IPvx = IPV4;
        bool source_map_ = false;
        unsigned long CacheTime = 86400 * 7; // 浏览器缓存时间
        time_t CacheTime_t = {};
        bool GZipFlag = true;
    public:
        // 构造函数
        WebServlet();
        // 析构函数
        ~WebServlet();
        // 初始化服务器
        bool Init(const InetAddress &address,int Port,bool ChokeUp = true);
        // 初始化服务器
        bool Init(const String& address,int Port,bool ChokeUp = true);
        // 设置 WebSocket 配置
        bool SetWebSocket(const WebSocketBind& bind);
        // 初始化 SSL 配置
        bool InitSSL();
        // 加载 SSL 证书
        bool LoadSSLCertificate(const CTL::String& CertPath) const;
        // 加载 SSL 私钥
        bool LoadSSLKey(const CTL::String &KeyPath, const CTL::String &Passwd) const;
        // 加载 SSL 客户端验证证书
        bool LoadSSLVerify(const CTL::String& CA_Path) const;
        // 运行服务器
        int Running(bool Ws = false,bool InitStaticFlag = true);
        // 停止服务器
        void Stop();
        // 添加静态资源路径 请填写相对路径
        bool AddStaticPath(const CTL::String& Path);
        // 设置 WebServlet 请求处理函数
        void SetWebServlet(const CTL::String& UrlPath,const RequestFunc& RFunc);
        // 设置 WebServlet 普通请求处理函数
        void SetWebServletFun(RequestFunc RFunc);
        // 设置 CORS 配置
        void SetWebCORS(const CORS& cors);
        // 获取 WebSocket 对象
        WebSocket* GetWebSocket();
        // 设置请求超时时间
        void SetTimeout(unsigned long MS);
        // 设置接收数据头缓冲区大小
        void SetReadHeaderBufferSize(unsigned int Size);
        // 设置线程池参数
        void SetThreadPool(size_t Size,size_t Max,int KeepAliveTime);
        // 设置IP类型 必须在Init之前设置
        void SetIPType(IPVX IPvx);
        // 初始化静态资源
        void InitStatic();
        // 设置最大文件上传数量
        static void SetMaxConcurrentFileUploads(int Num);
        // 设置资源调试
        void SetSourceMap(bool flag);
        // 设置静态资源缓存时间
        void SetCacheTime(unsigned long CacheTime);
        // 静态资源GZip
        void SetGZipFlag(bool Flag);
    private:
        // 打印消息
        static void MessagePrint(const CTL::String& str);
        // 检查客户端权限
        bool Permission(const HttpSocketChannel* client_channel) const;
        // 接受客户端连接
        void AcceptingClient();
        // 处理客户端请求
        void ProcessingClientRequests(ClientSocket* client);
        // 处理数据类型
        int ProcessingDataType(HttpSocketChannel* client_channel,Request& request);
        // 处理请求
        void HandleRequest(HttpSocketChannel* client_channel,Request& request);
        // 处理路径
        CTL::String ProcessPath(const CTL::String& path);
        // 初始化目录静态资源
        void DirInitStatic(const CTL::String& Path);
        // 检查路径是否包含映射后缀
        bool hasMapSuffix(const std::string& path);
        // 移除路径中的映射后缀
        std::string removeMapSuffix(const std::string& path);
        // 获取 Application/form-data 类型的数据
        static CTL::String GetApp_Form_Data(const HttpSocketChannel* client_channel,Request& HTTPRequests);
        // 获取 Multipart/form-data 类型的数据信息
        CCMap<CTL::String,CTL::String> GetMultipart_Form_DataInfo(const HttpSocketChannel* client_channel,CTL::String& RecvMessage);
        void ErrorHandling(HttpSocketChannel* client_channel,Request& request) const;
        static String FormatRequest(const String& str);
    };

}

#endif