128 lines
3.1 KiB
C++
128 lines
3.1 KiB
C++
#pragma once
|
|
|
|
#include <chrono>
|
|
#include <ctime>
|
|
#include <string>
|
|
|
|
#include "time_util.hpp"
|
|
|
|
namespace cinatra {
|
|
|
|
class cookie {
|
|
public:
|
|
cookie() = default;
|
|
cookie(const std::string &name, const std::string &value)
|
|
: name_(name), value_(value) {}
|
|
|
|
void set_version(int version) { version_ = version; }
|
|
|
|
void set_name(const std::string &name) { name_ = name; }
|
|
|
|
std::string get_name() const { return name_; }
|
|
|
|
void set_value(const std::string &value) { value_ = value; }
|
|
|
|
std::string get_value() const { return value_; }
|
|
|
|
void set_comment(const std::string &comment) { comment_ = comment; }
|
|
|
|
void set_domain(const std::string &domain) { domain_ = domain; }
|
|
|
|
void set_path(const std::string &path) { path_ = path; }
|
|
|
|
void set_priority(const std::string &priority) { priority_ = priority; }
|
|
|
|
void set_secure(bool secure) { secure_ = secure; }
|
|
|
|
void set_max_age(std::time_t seconds) { max_age_ = seconds; }
|
|
|
|
void set_http_only(bool http_only) { http_only_ = http_only; }
|
|
|
|
std::string to_string() const {
|
|
std::string result;
|
|
result.reserve(256);
|
|
result.append(name_);
|
|
result.append("=");
|
|
if (version_ == 0) {
|
|
// Netscape cookie
|
|
result.append(value_);
|
|
if (!domain_.empty()) {
|
|
result.append("; domain=");
|
|
result.append(domain_);
|
|
}
|
|
if (!path_.empty()) {
|
|
result.append("; path=");
|
|
result.append(path_);
|
|
}
|
|
if (!priority_.empty()) {
|
|
result.append("; Priority=");
|
|
result.append(priority_);
|
|
}
|
|
if (max_age_ != -1) {
|
|
result.append("; expires=");
|
|
result.append(get_gmt_time_str(std::chrono::system_clock::from_time_t(
|
|
max_age_ + std::time(nullptr))));
|
|
}
|
|
if (secure_) {
|
|
result.append("; secure");
|
|
}
|
|
if (http_only_) {
|
|
result.append("; HttpOnly");
|
|
}
|
|
}
|
|
else {
|
|
// RFC 2109 cookie
|
|
result.append("\"");
|
|
result.append(value_);
|
|
result.append("\"");
|
|
if (!comment_.empty()) {
|
|
result.append("; Comment=\"");
|
|
result.append(comment_);
|
|
result.append("\"");
|
|
}
|
|
if (!domain_.empty()) {
|
|
result.append("; Domain=\"");
|
|
result.append(domain_);
|
|
result.append("\"");
|
|
}
|
|
if (!path_.empty()) {
|
|
result.append("; Path=\"");
|
|
result.append(path_);
|
|
result.append("\"");
|
|
}
|
|
if (!priority_.empty()) {
|
|
result.append("; Priority=\"");
|
|
result.append(priority_);
|
|
result.append("\"");
|
|
}
|
|
|
|
if (max_age_ != -1) {
|
|
result.append("; Max-Age=\"");
|
|
result.append(std::to_string(max_age_));
|
|
result.append("\"");
|
|
}
|
|
if (secure_) {
|
|
result.append("; secure");
|
|
}
|
|
if (http_only_) {
|
|
result.append("; HttpOnly");
|
|
}
|
|
result.append("; Version=\"1\"");
|
|
}
|
|
return result;
|
|
}
|
|
|
|
private:
|
|
std::string name_ = "";
|
|
std::string value_ = "";
|
|
std::string comment_ = "";
|
|
std::string domain_ = "";
|
|
std::string priority_ = "";
|
|
std::string path_ = "";
|
|
std::time_t max_age_ = -1;
|
|
int version_ = 0;
|
|
bool secure_ = false;
|
|
bool http_only_ = false;
|
|
};
|
|
|
|
} // namespace cinatra
|