/*
 * charybdis: 21st Century IRC++d
 *
 * Copyright (C) 2016 Charybdis Development Team
 * Copyright (C) 2016 Jason Volk <jason@zemos.net>
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice is present in all copies.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
 * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 *
 */

#pragma once
#define HAVE_IRCD_HTTP_H

/// HyperText TransPort: formal grammars & tools
namespace ircd::http
{
	enum code :int;
	struct error;
	struct line;
	struct query;
	struct headers;
	struct content;
	struct request;
	struct response;

	string_view status(const enum code &);
	enum code status(const string_view &);

	string_view urlencode(const string_view &url, const mutable_buffer &);
	string_view urldecode(const string_view &url, const mutable_buffer &);
}

namespace ircd
{
	using http::urldecode;
	using http::urlencode;
}

//
// Add more as you go...
//
enum ircd::http::code
:int
{
	CONTINUE                                = 100,
	SWITCHING_PROTOCOLS                     = 101,

	OK                                      = 200,
	CREATED                                 = 201,
	ACCEPTED                                = 202,
	NON_AUTHORITATIVE_INFORMATION           = 203,
	NO_CONTENT                              = 204,
	PARTIAL_CONTENT                         = 206,

	MOVED_PERMANENTLY                       = 301,
	FOUND                                   = 302,
	SEE_OTHER                               = 303,
	NOT_MODIFIED                            = 304,
	TEMPORARY_REDIRECT                      = 305,
	PERMANENT_REDIRECT                      = 306,

	BAD_REQUEST                             = 400,
	UNAUTHORIZED                            = 401,
	FORBIDDEN                               = 403,
	NOT_FOUND                               = 404,
	METHOD_NOT_ALLOWED                      = 405,
	REQUEST_TIMEOUT                         = 408,
	CONFLICT                                = 409,
	REQUEST_URI_TOO_LONG                    = 414,
	EXPECTATION_FAILED                      = 417,
	IM_A_TEAPOT                             = 418,
	UNPROCESSABLE_ENTITY                    = 422,
	TOO_MANY_REQUESTS                       = 429,
	REQUEST_HEADER_FIELDS_TOO_LARGE         = 431,

	INTERNAL_SERVER_ERROR                   = 500,
	NOT_IMPLEMENTED                         = 501,
	SERVICE_UNAVAILABLE                     = 503,
	HTTP_VERSION_NOT_SUPPORTED              = 505,
	INSUFFICIENT_STORAGE                    = 507,
};

struct ircd::http::error
:ircd::error
{
	enum code code;
	std::string content;

	error(const enum code &, std::string content = {});
};

struct ircd::http::line
:string_view
{
	struct request;
	struct response;
	struct header;

	using string_view::string_view;
	line(parse::capstan &);
};

struct ircd::http::line::request
{
	string_view method;
	string_view path;
	string_view query;
	string_view fragment;
	string_view version;

	request(const line &);
	request() = default;
};

struct ircd::http::line::response
{
	string_view version;
	string_view status;
	string_view reason;

	response(const line &);
	response() = default;
};

struct ircd::http::query
:std::pair<string_view, string_view>
{
	struct string;

	bool operator<(const string_view &s) const   { return iless(first, s);                         }
	bool operator==(const string_view &s) const  { return iequals(first, s);                       }

	using std::pair<string_view, string_view>::pair;
	query() = default;
};

// Query string is read as a complete string off the tape (into request.query) and
// not parsed further. To make queries into that string use this class to view it.
struct ircd::http::query::string
:string_view
{
	void for_each(const std::function<void (const query &)> &) const;
	bool until(const std::function<bool (const query &)> &) const;

	string_view at(const string_view &key) const;
	string_view operator[](const string_view &key) const;

	using string_view::string_view;
};

struct ircd::http::line::header
:std::pair<string_view, string_view>
{
	bool operator<(const string_view &s) const   { return iless(first, s);                         }
	bool operator==(const string_view &s) const  { return iequals(first, s);                       }

	using std::pair<string_view, string_view>::pair;
	header(const line &);
	header() = default;
};

// HTTP headers are read once off the tape and proffered to the closure.
struct ircd::http::headers
{
	using header = line::header;
	using closure = std::function<void (const header &)>;

	headers(parse::capstan &, const closure & = {});
};

// Use the request::content / response::content wrappers. They ensure the proper amount
// of content is read and the tape is in the right position for the next request
// with exception safety.
struct ircd::http::content
:string_view
{
	IRCD_OVERLOAD(discard)
	IRCD_OVERLOAD(chunked)

	content(parse::capstan &, const size_t &length, discard_t);
	content(parse::capstan &, const size_t &length);
	content(parse::capstan &, chunked_t);
	content() = default;
};

struct ircd::http::response
{
	struct head;
	struct content;
	struct chunked;

	using write_closure = std::function<void (const ilist<const_buffer> &)>;
	using proffer = std::function<void (const head &)>;
	using header = line::header;

	response(const code &,
	         const string_view &content,
	         const write_closure &,
	         const vector_view<const header> & = {});

	response(parse::capstan &,
	         content *const &          = nullptr,
	         const proffer &           = nullptr,
	         const headers::closure &  = {});

	response() = default;
};

struct ircd::http::response::chunked
:response
{
	struct chunk;

	write_closure closure;

	chunked(const code &,
	        const write_closure &,
	        const vector_view<const header> &headers);

	chunked(const chunked &) = delete;
	~chunked() noexcept;
};

struct ircd::http::response::chunked::chunk
{
	chunk(chunked &, const const_buffer &);
};

struct ircd::http::response::head
:line::response
{
	size_t content_length {0};
	string_view transfer_encoding;

	head(parse::capstan &pc, const headers::closure &c = {});
};

struct ircd::http::response::content
:http::content
{
	content(parse::capstan &pc, const head &h, discard_t)
	:http::content{pc, h.content_length, discard}
	{}

	content(parse::capstan &pc, const head &h, chunked_t)
	:http::content{pc, chunked}
	{}

	content(parse::capstan &pc, const head &h)
	:http::content{pc, h.content_length}
	{}

	content() = default;
};

struct ircd::http::request
{
	struct head;
	struct content;

	using write_closure = std::function<void (const ilist<const_buffer> &)>;
	using proffer = std::function<void (const head &)>;
	using header = line::header;

	request(const string_view &host            = {},
	        const string_view &method          = "GET",
	        const string_view &path            = "/",
	        const string_view &query           = {},
	        const string_view &content         = {},
	        const write_closure &              = nullptr,
	        const vector_view<const header> &  = {});

	request(parse::capstan &,
	        content *const &          = nullptr,
	        const write_closure &     = nullptr,
	        const proffer &           = nullptr,
	        const headers::closure &  = {});
};

struct ircd::http::request::head
:line::request
{
	string_view host;
	string_view expect;
	string_view te;
	size_t content_length {0};

	head(parse::capstan &pc, const headers::closure &c = {});
};

struct ircd::http::request::content
:http::content
{
	content(parse::capstan &pc, const head &h, discard_t)
	:http::content{pc, h.content_length, discard}
	{}

	content(parse::capstan &pc, const head &h)
	:http::content{pc, h.content_length}
	{}
};