#include <string>
#include <vector>
#include <iostream>
#include <iomanip>
#include <sstream>
#include <unistd.h>

#if 0
#define UPDATE_TESTSUITECPP 1
#endif

std::string events;

#define RFC2045_ENTITY_PARSER_TEST(msg)		\
	do { events += msg "\n"; } while (0)

#define RFC2045_ENTITY_PARSER_DECL(d) d


#include "testsuitecpp.H"
#include "rfc2045.h"
#include "rfc2045cpp.C"

template<bool crlf>
rfc2045::entity runtest(std::string message,
			size_t chunksize)
{
	events.clear();

	rfc2045::entity_parser<crlf> parser;

	auto b=message.begin(), e=message.end();

	while (b != e)
	{
		if (static_cast<size_t>(e-b) < chunksize)
			chunksize=e-b;

		auto p=b;
		b += chunksize;
		parser.parse(p, b);
	}

	return parser.parsed_entity();
}

template<bool crlf>
void test1()
{
	const char *nl = crlf ? "\r\n":"\n";
	auto entity=runtest<crlf>(
		std::string{
			"Mime-Version: 1.0"
		}
		+ nl
		+ "Content-Type: text"
		+ nl
		+ nl
		+ std::string(40, '=')
		+ nl
		+ std::string(40, '=')
		+ nl
		+ std::string(40, '=')
		+ nl
		+ std::string(40, '=')
		+ nl
		+ std::string(40, '=')
		+ nl
		+ std::string(40, '=')
		+ nl
		+ std::string(40, '=')
		+ nl, 100);

#if UPDATE_TESTSUITECPP
	parsed_mime_info::dump_entity(entity, "");

	std::cout << events;
#else
	using cte=rfc2045::cte;

	static const parsed_mime_info expected_results{
		0    , // startpos
		38+crlf*3   , // startbody
		325+crlf*10  , // endbody
		10   , // nlines
		7    , // nbodylines
		1    , // mime1
		0, "text", "iso-8859-1",
		"", cte::sevenbit,
		0, // has8bitheader
		0, // has8bitbody
		0  // has8bitcontentchar
	};

	if (expected_results != entity)
	{
		std::cout << "test1 parsing error ("
			  << (crlf ? "\\r\\n":"\\n")
			  << "):\n";
		parsed_mime_info::dump(entity);
		exit(1);
	}

	if (events !=
	    "parser: chunk\n"
	    "retrieved next chunk\n"
	    "parser: chunk\n"
	    "retrieved next chunk\n"
	    "parser: chunk\n"
	    "retrieved next chunk\n"
	    "parser: chunk\n"
	    "retrieved next chunk\n"
	    "end_of_parse set\n"
	    "end_of_parse received\n"
	    "thread finished\n")
	{
		std::cout << "test1 unexpected event sequence:\n"
			  << events;
		exit(1);
	}
#endif
}

template<bool crlf>
void test2()
{
	const char *nl = crlf ? "\r\n":"\n";
	auto entity=runtest<crlf>(
		std::string{
			"Mime-Version: 1.0"
		}
		+ nl
		+ "Content-Type: text"
		+ nl
		+ "Content-Transfer-Encoding: something"
		+ nl
		+ nl
		+ std::string(40, 'A')
		+ nl
		+ std::string(40, 'A')
		+ nl
		+ std::string(40, 'A')
		+ nl
		+ std::string(40, 'A')
		+ nl
		+ std::string(40, 'A')
		+ nl
		+ std::string(40, 'A')
		+ nl
		+ std::string(40, 'A')
		+ nl
		+ std::string(40, 'A')
		+ nl
		+ std::string(40, 'A')
		+ nl
		+ std::string(40, 'A')
		+ nl, 100);

#if UPDATE_TESTSUITECPP
	parsed_mime_info::dump_entity(entity, "");

	std::cout << events;
#else

	if (events !=
	    "parser: chunk\n"
	    "retrieved next chunk\n"
	    "parser: chunk\n"
	    "retrieved next chunk\n"
	    "parser: chunk\n"
	    "retrieved next chunk\n"
	    "parser: chunk\n"
	    "retrieved next chunk\n"
	    "parser: chunk\n"
	    "retrieved next chunk\n"
	    "end_of_parse set\n"
	    "end_of_parse received\n"
	    "thread finished\n")
	{
		std::cout << "test2 unexpected event sequence:\n"
			  << events;
		exit(1);
	}

	if (!(entity.errors.code & RFC2045_ERRUNKNOWNTE))
	{
		std::cout << "test2 did not receive expected error\n";
		exit(1);
	}
#endif
}

void test34_setup(std::stringstream &ss, const char *nl)
{
	ss << "Mime-Version: 1.0" << nl
	   << "Content-Type: multipart/mixed; boundary=aaa" << nl
	   << nl
	   << "--aaa" << nl
	   << "Content-Type: text/plain" << nl
	   << "Content-Description: MIME" << nl
	   << " section 1" << nl
	   << nl
	   << "--aaa--" << nl;

	ss.seekg(0);
}

template<bool crlf>
void test3()
{
	const char *nl = crlf ? "\r\n":"\n";

	std::stringstream ss;

	test34_setup(ss, nl);

	auto b=std::istreambuf_iterator<char>{ss};
	auto e=std::istreambuf_iterator<char>{};

	typename rfc2045::entity::line_iter<crlf>::iter<
		std::istreambuf_iterator<char>,
		std::istreambuf_iterator<char>
		> iter{b, e};

	rfc2045::entity entity;

	entity.parse(iter);

	if (entity.subentities.size() != 1)
	{
		std::cout << "test2 (" << (crlf ? "\\r\\n":"\\n") << "): "
			" unexpected parsing error.\n";
		exit(1);
	}

	typename rfc2045::entity::line_iter<crlf>::
		template headers<std::streambuf> h{
		entity.subentities[0],
		*ss.rdbuf()
	};

#if UPDATE_TESTSUITECPP
	bool flag;

	do
	{
		const auto &[n, c]=h.name_content();

		std::cout << "{\n\t\"" << h.current_header() << "\",\n"
			"\t\"" << n << "\", \"" << c << "\", ";

		flag=h.next();

		std::cout << flag << "\n},\n";
	} while (flag);
#else
	static const struct {
		const char *header, *name, *content;
		bool next;
	} default_test[]={
		{
			"content-type: text/plain",
			"content-type", "text/plain", 1
		},
		{
			"content-description: MIME section 1",
			"content-description", "MIME section 1", 1
		},
		{
			"",
			"", "", 0
		},
	};

	for (const auto &t:default_test)
	{
		auto full=h.current_header();
		const auto &[n, c]=h.name_content();

		if (full != t.header || n != t.name || c != t.content)
		{
			std::cout << "test3: unexpected result (default_test)\n"
				"Got: " << full << "\n    " << n
				  << "\n    " << c << "\n"
				"Exp: " << t.header << "\n    " << t.name
				  << "\n    " << t.content << "\n";
			exit(1);
		}

		if (h.next() != t.next)
		{
			std::cout << "test3: unexpected result (default_test)\n"
				  << "Got next=" << !t.next
				  << " for \"" << t.header << "\"\n";
			exit(1);
		}
	}
#endif
}

template<bool crlf>
void test4()
{
	const char *nl = crlf ? "\r\n":"\n";

	std::stringstream ss;

	test34_setup(ss, nl);

	auto b=std::istreambuf_iterator<char>{ss};
	auto e=std::istreambuf_iterator<char>{};

	typename rfc2045::entity::line_iter<crlf>::iter<
		std::istreambuf_iterator<char>,
		std::istreambuf_iterator<char>
		> iter{b, e};

	rfc2045::entity entity;

	entity.parse(iter);

	if (entity.subentities.size() != 1)
	{
		std::cout << "test2 (" << (crlf ? "\\r\\n":"\\n") << "): "
			" unexpected parsing error.\n";
		exit(1);
	}

	typename rfc2045::entity::line_iter<crlf>::
		template headers<std::streambuf> h{
		entity.subentities[0],
		*ss.rdbuf()
	};

	h.name_lc=false;
	h.keep_eol=true;

#if UPDATE_TESTSUITECPP
	bool flag;

	do
	{
		const auto &[n, c]=h.name_content();

		std::cout << "{";

		for (const auto &str
			     : std::array<std::string_view, 3>{
			     h.current_header(), n, c})
		{
			std::cout << "\n\t";

			if (str.empty())
			{
				std::cout << "\"\"";
			}
			else
			{
				bool has_prev=false;
				bool prev_was_nl=true;

				for (char c:str)
				{
					if (c != '\n')
					{
						if (prev_was_nl)
						{
							if (has_prev)
							{
								std::cout <<
									" + ";
							}
							std::cout <<
								"std::string{"
								"\"";
						}
						std::cout << c;
						has_prev=true;
						prev_was_nl=false;
					}
					else
					{
						if (has_prev)
						{
							if (!prev_was_nl)
								std::cout <<
									"\"}";
							std::cout << " + ";
						}
						std::cout << "nl";
						has_prev=true;
						prev_was_nl=true;
					}
				}

				if (!prev_was_nl)
					std::cout << "\"}";
			}
			std::cout << ",";
		}

		flag=h.next();

		std::cout << " " << flag << "\n},\n";
	} while (flag);
#else
	const struct {
		std::string header, name, content;
		bool next;
	} default_test[]={
		{
			std::string{"Content-Type: text/plain"} + nl,
			std::string{"Content-Type"},
			std::string{"text/plain"} + nl, 1
		},
		{
			std::string{"Content-Description: MIME"} + nl + std::string{" section 1"} + nl,
			std::string{"Content-Description"},
			std::string{"MIME"} + nl + std::string{" section 1"} + nl, 1
		},
		{
			nl,
			nl,
			"", 0
		},
	};

	for (const auto &t:default_test)
	{
		auto full=h.current_header();
		const auto &[n, c]=h.name_content();

		if (full != t.header || n != t.name || c != t.content)
		{
			std::cout << "test3: unexpected result (default_test)\n"
				"Got: " << full << "\n    " << n
				  << "\n    " << c << "\n"
				"Exp: " << t.header << "\n    " << t.name
				  << "\n    " << t.content << "\n";
			exit(1);
		}

		if (h.next() != t.next)
		{
			std::cout << "test3: unexpected result (default_test)\n"
				  << "Got next=" << !t.next
				  << " for \"" << t.header << "\"\n";
			exit(1);
		}
	}
#endif
}

template<bool crlf>
void test5()
{
	std::string nl = crlf ? "\r\n":"\n";

	std::stringstream ss;

	ss << "Subject: =?utf-8?q?nob=c3=92dy?=" << nl
	   << " subject" << nl
	   << "From: Nobody1 <test1@example.com>, test2@example.com," << nl
	   << "  =?iso-8859-1?q?No?= =?iso-8859-1?q?b=D2dy?= "
		"<test3@xn--80akhbyknj4f.net>" << nl
	   << "Mime-Version: 1.0" << nl
	   << "Content-Type: multipart/mixed; boundary=aaa" << nl
	   << nl
	   << "preable" << nl
	   << "--aaa" << nl
	   << "Content-Type: text/plain; charset=iso-8859-1" << nl
	   << "Content-Description: =?utf-8?q?quoted-?= "
		"=?utf8?q?printable?=" << nl
	   << "Content-Transfer-Encoding: quoted-printable" << nl
	   << nl
	   << "H=E9llo =" << nl
	   << "H=E9llo" << nl
	   << nl
	   << "--aaa" << nl
	   << "Content-Type: text/plain; charset=iso-8859-1" << nl
	   << "Content-Description: base64 decoded" << nl
	   << "Content-Transfer-Encoding: base64" << nl
	   << nl
	   << "SOlsbG8=" << nl
	   << nl
	   << "--aaa--" << nl << nl;

	auto b=std::istreambuf_iterator<char>{ss};
	auto e=std::istreambuf_iterator<char>{};

	typename rfc2045::entity::line_iter<crlf>::iter<
		std::istreambuf_iterator<char>,
		std::istreambuf_iterator<char>
		> iter{b, e};

	rfc2045::entity entity;

	entity.parse(iter);

	std::string decoded;

	{
		rfc822::mime_decoder<
			std::function<void (const char *, size_t)>,
			std::streambuf> decoder{
			[&]
			(const char *ptr, size_t n)
			{
				decoded.insert(decoded.end(), ptr, ptr+n);
			}, *ss.rdbuf(), "utf-8"
		};

		decoder.decode<crlf>(entity);

		decoded.erase(std::remove(decoded.begin(), decoded.end(),
					  '\r'), decoded.end());
	}
#if UPDATE_TESTSUITECPP

	auto db=decoded.begin(), de=decoded.end();

	const char *str_plus="\t\t";

	while (db != de)
	{
		auto p=std::find(db, de, '\n');

		if (p != de)
			++p;

		std::cout << str_plus << "\"";

		for ( ; db != p; ++db)
		{
			if (*db == '\n')
				std::cout << "\\n";
			else if (*db == '"')
				std::cout << "\\\"";
			else
				std::cout << *db;
		}
		std::cout << "\"";
		str_plus="\n\t\t";
	}
	std::cout << "\n";
#else
	std::string expected =
		"subject: nobÒdy subject\n"
		"from: Nobody1 <test1@example.com>, test2@example.com, NobÒdy <test3@испытание.net>\n"
		"mime-version: 1.0\n"
		"content-type: multipart/mixed; boundary=aaa\n"
		"\n"
		"content-type: text/plain; charset=iso-8859-1\n"
		"content-description: quoted-printable\n"
		"content-transfer-encoding: quoted-printable\n"
		"\n"
		"Héllo Héllo\n"
		"content-type: text/plain; charset=iso-8859-1\n"
		"content-description: base64 decoded\n"
		"content-transfer-encoding: base64\n"
		"\n"
		"Héllo";

	if (expected != decoded)
	{
		std::cout << "test5 (" << (crlf ? "\\r\\n":"\\n") << "): "
			" unexpected decoding result:\n"
			  << decoded << "\n";
		exit(1);
	}
#endif
}

void test6()
{
	std::stringstream ss;

	std::string s="Mime-Version: 1.0\n"
		"Content-Type: multipart/digest; boundary=aaa\n"
		"\n"
		"--aaa\n"
		"\n"
		"Subject: first message\n"
		"\n"
		"one\n"
		"two\n"
		"\n"
		"--aaa\n"
		"\n"
		"Subject: second message\n"
		"\n"
		"one\n"
		"two\n"
		"--aaa---\n";

	ss << s;

	auto b=std::istreambuf_iterator<char>{ss};
	auto e=std::istreambuf_iterator<char>{};

	rfc2045::entity::line_iter<false>::iter<
		std::istreambuf_iterator<char>,
		std::istreambuf_iterator<char>
		> iter{b, e};

	rfc2045::entity entity;

	entity.parse(iter);

	for (auto &mime:entity.subentities)
	{
		if (mime.subentities.size() != 1)
		{
			std::cout << "test6: unexpected structure\n";
			exit(1);
		}
	}

	std::vector<std::array<size_t, 6>> actual;

	actual.push_back(std::array<size_t, 6>{
			{entity.startpos, entity.startbody,
			 entity.endbody,
			 entity.nlines,
			 entity.nbodylines,
			 entity.no_terminating_nl}});

	for (auto &mime:entity.subentities)
	{
		actual.push_back(std::array<size_t, 6>{
				{mime.startpos, mime.startbody,
				 mime.endbody,
				 mime.nlines,
				 mime.nbodylines,
				 mime.no_terminating_nl}});
		actual.push_back(std::array<size_t, 6>{
				{mime.subentities[0].startpos,
				 mime.subentities[0].startbody,
				 mime.subentities[0].endbody,
				 mime.subentities[0].nlines,
				 mime.subentities[0].nbodylines,
				 mime.subentities[0].no_terminating_nl}});
	}
#if 1
	std::vector<std::array<size_t, 6>> expected{
		/*
		 1 Mime-Version: 1.0
		 2 Content-Type: multipart/digest; boundary=aaa
		 3
		 4 --aaa
		 5
		 6 Subject: first message
		 7
		 8 one
		 9 two
		10
		11 --aaa
		12
		13 Subject: second message
		14
		15 one
		16 two
		17 --aaa---
		*/

		/* Header:

		[Mime-Version: 1.0
		Co...gest; boundary=aaa

		]

		Body:

		[--aaa

		Subject: firs...e

		one
		two
		--aaa---
		]

		*/
		{{0, 64, 153, 17, 14, 0}},
		/* Header:

		[
		]

		Body:

		[Subject: first message

		one
		two
		]

		*/
		{{70, 71, 103, 5, 4, 0}},
		/* Header:

		[Subject: first message

		]

		Body:

		[one
		two
		]

		*/
		{{71, 95, 103, 4, 2, 0}},
		/* Header:

		[
		]

		Body:

		[Subject: second message

		one
		two]

		*/
		{{110, 111, 143, 5, 4, 1}},
		/* Header:

		[Subject: second message

		]

		Body:

		[one
		two]

		*/
		{{111, 136, 143, 4, 2, 1}}
	};

	if (expected != actual)
	{
		std::cout << "test6 failed:\n";
		for (auto &a:actual)
		{
			std::cout << "\n";
			const auto &[startpos, startbody, endbody, nlines,
				     nbodylines, no_term_nl]=a;

			std::cout << "startpos:   " << startpos << "\n"
				  << "startbody:  " << startbody << "\n"
				  << "endbody:    " << endbody << "\n"
				  << "nlines:     " << nlines << "\n"
				  << "nbodylines: " << nbodylines << "\n"
				  << "no term nl: " << no_term_nl << "\n";
		}
		exit(1);
	}
#else

	std::cout << "\t\t/*\n";

	{
		std::istringstream i{s};
		std::string line;
		size_t n=0;

		while (std::getline(i, line))
			std::cout << "\t\t" << std::setw(2) << ++n
				  << std::setw(0) << " "
				  << line << "\n";
	}
	std::cout << "\t\t*/\n\n";

	const char *commanl="";

	for (auto &a:actual)
	{
		std::cout << commanl;

		const auto &[startpos, startbody, endbody, nlines, nbodylines,
			     no_term_nl]
			= a;

		std::cout << "\t\t/* Header:\n\n";

		std::string chunk=s.substr(startpos, startbody-startpos);

		if (chunk.size() > 50)
			chunk=chunk.substr(0, 20) + "..." +
				chunk.substr(chunk.size()-20);

		{
			std::istringstream i{"[" + chunk + "]"};

			std::string line;

			while (std::getline(i, line))
				std::cout << "\t\t" << line << "\n";
		}
		std::cout << "\n\t\tBody:\n\n";

		chunk=s.substr(startbody, endbody-startbody);

		if (chunk.size() > 50)
			chunk=chunk.substr(0, 20) + "..." +
				chunk.substr(chunk.size()-20);

		{
			std::istringstream i{"[" + chunk + "]"};

			std::string line;

			while (std::getline(i, line))
				std::cout << "\t\t" << line << "\n";
		}

		std::cout << "\n\t\t*/\n";

		std::cout << "\t\t{{";
		commanl="";

		for (auto v:a)
		{
			std::cout << commanl << v;
			commanl = ", ";
		}
		std::cout << "}}";
		commanl=",\n";
	}
	std::cout << "\n";
#endif
}

void test7()
{
	std::stringstream ss;
	ss << "Mime-Version: 1.0\n"
		"Content-Type: text/plain; charset=iso-8859-1\n"
		"Content-Transfer-Encoding: quoted-printable\n"
		"\n"
		"H=E9llo\n";

	auto b=std::istreambuf_iterator<char>{ss};
	auto e=std::istreambuf_iterator<char>{};

	rfc2045::entity::line_iter<false>::iter<
		std::istreambuf_iterator<char>,
		std::istreambuf_iterator<char>
		> iter{b, e};

	rfc2045::entity entity;

	entity.parse(iter);

	std::u32string us;

	{
		rfc822::mime_unicode_decoder decoder{
			[&]
			(const char32_t *ptr, size_t cnt)
			{
				us.insert(us.end(), ptr, ptr+cnt);
			},

			*ss.rdbuf()
		};

		decoder.decode_header=false;
		decoder.decode(entity);
	}

	{
		auto closure=
			[&]
			(const char32_t *ptr, size_t cnt)
			{
				us.insert(us.end(), ptr, ptr+cnt);
			};

		rfc822::mime_unicode_decoder decoder{
			closure,
			*ss.rdbuf()
		};

		decoder.decode_header=false;
		decoder.decode(entity);
	}

	if (us != U"Héllo\nHéllo\n")
	{
		std::cout << "test7 failed\n";
		exit(1);
	}
}

int main()
{
	alarm(60);
	rfc2045_setdefaultcharset("iso-8859-1");
#if UPDATE_TESTSUITECPP
	// test1<false>();
	// test1<true>();

	// test2<false>();

	// test3<false>();

	// test4<false>();

	test5<false>();
#else
	for (int i=0; i<100; ++i)
	{
		test1<false>();
		test1<true>();
	}

	test2<false>();
	test2<true>();

	test3<false>();
	test3<true>();

	test4<false>();
	test4<true>();

	test5<false>();
	test5<true>();
#endif

	test6();
	test7();
	return 0;
}