#pragma once
#include <iostream>
#include <string>
#include <sstream>
#include <vector>
#include <memory>
#include <ctime>
#include "TcpServer.hpp"
const std::string HttpSep = "\r\n";
const std::string homepage = "index.html";
const std::string wwwroot = "./wwwroot";
class HttpRequest {
public:
HttpRequest() : _req_blank(HttpSep), _path(wwwroot) {}
bool GetLine(std::string &str, std::string *line) {
auto pos = str.find(HttpSep);
if (pos == std::string::npos) return false;
*line = str.substr(0, pos);
str.erase(0, pos + HttpSep.size());
return true;
}
bool Deserialize(std::string &request) {
std::string line;
bool ok = GetLine(request, &line);
if (!ok) return false;
_req_line = line;
while (true) {
bool ok = GetLine(request, &line);
if (ok && line.empty()) { _req_content = request; break; }
else if (ok && !line.empty()) { _req_header.push_back(line); }
else { break; }
}
return true;
}
~HttpRequest() {}
private:
std::string _req_line;
std::vector<std::string> _req_header;
std::string _req_blank;
std::string _req_content;
std::string _method;
std::string _url;
std::string _http_version;
std::string _path;
std::string _suffix;
};
const std::string BlankSep = " ";
const std::string LineSep = "\r\n";
class HttpResponse {
public:
HttpResponse() : _http_version("HTTP/1.0"), _status_code(200), _status_code_desc("OK"), _resp_blank(LineSep) {}
void SetCode(int code) { _status_code = code; }
void SetDesc(const std::string &desc) { _status_code_desc = desc; }
void MakeStatusLine() { _status_line = _http_version + BlankSep + std::to_string(_status_code) + BlankSep + _status_code_desc + LineSep; }
void AddHeader(const std::string &header) { _resp_header.push_back(header + LineSep); }
void AddContent(const std::string &content) { _resp_content = content; }
std::string Serialize() {
MakeStatusLine();
std::string response_str = _status_line;
for (auto &header : _resp_header) { response_str += header; }
response_str += _resp_blank;
response_str += _resp_content;
return response_str;
}
~HttpResponse() {}
private:
std::string _status_line;
std::vector<std::string> _resp_header;
std::string _resp_blank;
std::string _resp_content;
std::string _http_version;
int _status_code;
std::string _status_code_desc;
};
class Http {
private:
std::string GetMonthName(int month) {
std::vector<std::string> months = {"Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"};
return months[month];
}
std::string GetWeekDayName(int day) {
std::vector<std::string> weekdays = {"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"};
return weekdays[day];
}
std::string ExpireTimeUseRfc1123(int t)
{
time_t timeout = time(nullptr) + t;
struct tm *tm = gmtime(&timeout);
char timebuffer[1024];
snprintf(timebuffer, sizeof(timebuffer), "%s, %02d %s %d %02d:%02d:%02d UTC", GetWeekDayName(tm->tm_wday).c_str(), tm->tm_mday, GetMonthName(tm->tm_mon).c_str(), tm->tm_year + 1900, tm->tm_hour, tm->tm_min, tm->tm_sec);
return timebuffer;
}
public:
Http(uint16_t port) { _tsvr = std::make_unique<TcpServer>(port, std::bind(&Http::HandlerHttp, this, std::placeholders::_1)); _tsvr->Init(); }
std::string ProveCookieWrite()
{
return "Set-Cookie: username=zhangsan;";
}
std::string ProveCookieTimeOut() { return "Set-Cookie: username=zhangsan; expires=" + ExpireTimeUseRfc1123(60) + ";"; }
std::string ProvePath() { return "Set-Cookie: username=zhangsan; path=/a/b;"; }
std::string ProveOtherCookie() { return "Set-Cookie: passwd=1234567890; path=/a/b;"; }
std::string HandlerHttp(std::string request) {
HttpRequest req;
req.Deserialize(request);
req.DebugHttp();
lg.LogMessage(Debug, "%s\n", ExpireTimeUseRfc1123(60).c_str());
HttpResponse resp;
resp.SetCode(200);
resp.SetDesc("OK");
resp.AddHeader("Content-Type: text/html");
resp.AddHeader(ProvePath());
resp.AddHeader(ProveOtherCookie());
resp.AddContent("<html><h1>helloworld</h1></html>");
return resp.Serialize();
}
void Run() { _tsvr->Start(); }
~Http() {}
private:
std::unique_ptr<TcpServer> _tsvr;
};
