//
// Created by 22100 on 2024/5/14.
//

#ifndef REACTORHTTPCPP_HTTPRESPONSE_H
#define REACTORHTTPCPP_HTTPRESPONSE_H
#include <string>
#include <unordered_map>
#include <functional>
#include "Buffer.h"

using std::function;
using std::string;

enum class StatusCode{
    Unknown,
    OK = 200,
    MovedPermanently = 301,
    MovedTemporarily = 302,
    BadRequest = 400,
    NotFound = 404
};


class HttpResponse {
public:
    HttpResponse();
    ~HttpResponse();
    function<void(const string, Buffer*, int)> sendDataFunc;
    // 添加响应头
    void addHeader(const string key, const string value);
    // 组织http响应数据
    void prepareMsg(Buffer* sendBuf, int socket);

    inline void setFileName(string name);
    inline void setStatusCode(StatusCode code);
private:
    // 状态行：状态码， 状态描述
    StatusCode m_statusCode;
    string m_fileName;
    // 响应头-键值对
    std::unordered_map<string, string> m_headers;
    // 定义状态码和描述的对应关系
    const std::unordered_map<int, string> m_info ={
            {200, "Ok"},
            {301, "MovedPermanently"},
            {302, "MovedTemporarily"},
            {400, "BadRequest"},
            {404, "NotFound"},
    };
};

void HttpResponse::setFileName(string name) {
    m_fileName = name;
}

void HttpResponse::setStatusCode(StatusCode code){
    m_statusCode = code;
}

#endif //REACTORHTTPCPP_HTTPRESPONSE_H