firecgi/request.h

#pragma once

#include <functional>
#include <mutex>
#include <unordered_map>

#include "firebuf/buffer.h"

#include "parse.h"

namespace firecgi {

class Connection;

class Request {
  public:
	Request(Connection *conn);

	void NewRequest(uint16_t request_id);

	uint16_t RequestId() const;

	void AddParam(const std::string_view& key, const std::string_view& value);
	void SetBody(const std::string_view& in);

	const std::string_view& GetParam(const std::string_view& key) const;
	const std::string_view& GetBody() const;

	void WriteHeader(const std::string_view& name, const std::string_view& value);
	void WriteBody(const std::string_view& body);
	[[nodiscard]] bool Flush();
	bool End();

	template<typename...Args>
	void WriteBody(const std::string_view& first, Args... more);

	template<typename T>
	T InTransaction(const std::function<T()>& callback);

  private:
	Header OutputHeader();
	iovec OutputVec();

	Connection *conn_;
	uint16_t request_id_ = 0;

	std::unordered_map<std::string_view, std::string_view> params_;
	std::string_view body_;

	firebuf::Buffer out_buf_;
	bool body_written_;
	std::recursive_mutex output_mu_;
};

template<typename...Args>
void Request::WriteBody(const std::string_view& first, Args... more) {
	std::lock_guard<std::recursive_mutex> l(output_mu_);
	WriteBody(first);
	WriteBody(more...);
}

template<typename T>
T Request::InTransaction(const std::function<T()>& callback) {
	std::lock_guard<std::recursive_mutex> l(output_mu_);
	return callback();
}

} // namespace firecgi