#include <cstdio>
#include <cstdarg>
#include <string>
#include <list>

#include "BroDoc.h"
#include "BroDocObj.h"

BroDoc::BroDoc(const std::string& sourcename)
	{
#ifdef DEBUG
	fprintf(stdout, "Documenting source: %s\n", sourcename.c_str());
#endif
	source_filename = sourcename.substr(sourcename.find_last_of('/') + 1);

	size_t ext_pos = source_filename.find_last_of('.');
	std::string ext = source_filename.substr(ext_pos + 1);
	if ( ext_pos == std::string::npos || ext != "bro" )
		{
		if ( source_filename != "bro.init" )
			{
			fprintf(stderr,
			        "Warning: documenting file without .bro extension: %s\n",
			        sourcename.c_str());
			}
		else
			{
			// Force the reST documentation file to be "bro.init.rst"
			ext_pos = std::string::npos;
			}
		}

	reST_filename = source_filename.substr(0, ext_pos);
	reST_filename += ".rst";
	reST_file = fopen(reST_filename.c_str(), "w");

	if ( ! reST_file )
		fprintf(stderr, "Failed to open %s", reST_filename.c_str());
#ifdef DEBUG
	else
		fprintf(stdout, "Created reST document: %s\n", reST_filename.c_str());
#endif
	}

BroDoc::~BroDoc()
	{
	if ( reST_file )
		if ( fclose( reST_file ) )
			fprintf(stderr, "Failed to close %s", reST_filename.c_str());
	FreeBroDocObjPtrList(options);
	FreeBroDocObjPtrList(constants);
	FreeBroDocObjPtrList(state_vars);
	FreeBroDocObjPtrList(types);
	FreeBroDocObjPtrList(notices);
	FreeBroDocObjPtrList(events);
	FreeBroDocObjPtrList(functions);
	FreeBroDocObjPtrList(redefs);
	}

void BroDoc::AddImport(const std::string& s)
	{
	size_t ext_pos = s.find_last_of('.');

	if ( ext_pos == std::string::npos )
		imports.push_back(s);
	else
		{
		if ( s.substr(ext_pos + 1) == "bro" )
			imports.push_back(s.substr(0, ext_pos));
		else
			fprintf(stderr, "Warning: skipped documenting @load of file "
			                "without .bro extension: %s\n", s.c_str());
		}
	}

void BroDoc::SetPacketFilter(const std::string& s)
	{
	packet_filter = s;
	size_t pos1 = s.find("{\n");
	size_t pos2 = s.find("}");
	if ( pos1 != std::string::npos && pos2 != std::string::npos )
		packet_filter = s.substr(pos1 + 2, pos2 - 2);

	bool has_non_whitespace = false;
	for ( std::string::const_iterator it = packet_filter.begin();
	      it != packet_filter.end(); ++it )
		if ( *it != ' ' && *it != '\t' && *it != '\n' && *it != '\r' )
			{
			has_non_whitespace = true;
			break;
			}

	if ( ! has_non_whitespace )
		packet_filter.clear();
	}

void BroDoc::AddPortAnalysis(const std::string& analyzer,
                             const std::string& ports)
	{
	std::string reST_string = analyzer + "::\n" + ports + "\n\n";
	port_analysis.push_back(reST_string);
	}

void BroDoc::WriteDocFile() const
	{
	WriteToDoc(".. Automatically generated.  Do not edit.\n\n");

	WriteToDoc("%s\n", source_filename.c_str());
	for ( size_t i = 0; i < source_filename.length(); ++i )
		WriteToDoc("=");
	WriteToDoc("\n\n");

	WriteSectionHeading("Summary", '-');
	WriteStringList("%s\n", "%s\n\n", summary);

	if ( ! imports.empty() )
		{
		WriteToDoc(":Imports: ");
		WriteStringList(":doc:`%s`, ", ":doc:`%s`\n", imports);
		}

	WriteToDoc("\n");

	if ( ! modules.empty() )
		{
		WriteSectionHeading("Namespaces", '-');
		WriteStringList(".. bro:namespace:: %s\n", modules);
		WriteToDoc("\n");
		}

	if ( ! notices.empty() )
		WriteBroDocObjList(notices, "Notices", '-');

	WriteInterface("Public Interface", '-', '~', true);
	WriteInterface("Private Interface", '-', '~', false);

	if ( ! port_analysis.empty() )
		{
		WriteSectionHeading("Port Analysis", '-');
		WriteToDoc(":ref:`More Information <common_port_analysis_doc>`\n\n");
		WriteStringList("%s", port_analysis);
		}

	if ( ! packet_filter.empty() )
		{
		WriteSectionHeading("Packet Filter", '-');
		WriteToDoc(":ref:`More Information <common_packet_filter_doc>`\n\n");
		WriteToDoc("Filters added::\n\n");
		WriteToDoc("%s\n", packet_filter.c_str());
		}
	}

void BroDoc::WriteInterface(const char* heading, char underline,
                            char sub, bool isPublic) const
	{
	WriteSectionHeading(heading, underline);
	WriteBroDocObjList(options, isPublic, "Options", sub);
	WriteBroDocObjList(constants, isPublic, "Constants", sub);
	WriteBroDocObjList(state_vars, isPublic, "State Variables", sub);
	WriteBroDocObjList(types, isPublic, "Types", sub);
	WriteBroDocObjList(events, isPublic, "Events", sub);
	WriteBroDocObjList(functions, isPublic, "Functions", sub);
	WriteBroDocObjList(redefs, isPublic, "Redefinitions", sub);
	}

void BroDoc::WriteStringList(const char* format,
                             const char* last_format,
                             const std::list<std::string>& l) const
	{
	if ( l.empty() )
		{
		WriteToDoc("\n");
		return;
		}
	std::list<std::string>::const_iterator it;
	std::list<std::string>::const_iterator last = l.end();
	last--;
	for ( it = l.begin(); it != last; ++it )
		WriteToDoc(format, it->c_str());
	WriteToDoc(last_format, last->c_str());
	}

void BroDoc::WriteBroDocObjList(const BroDocObjList& l,
                                bool wantPublic,
                                const char* heading,
                                char underline) const
	{
	if ( l.empty() ) return;

	std::list<const BroDocObj*>::const_iterator it;
	bool (*f_ptr)(const BroDocObj* o) = 0;

	if ( wantPublic )
		f_ptr = IsPublicAPI;
	else
		f_ptr = IsPrivateAPI;

	it = std::find_if(l.begin(), l.end(), f_ptr);
	if ( it == l.end() ) return;
	WriteSectionHeading(heading, underline);
	while ( it != l.end() )
		{
		(*it)->WriteReST(reST_file);
		it = find_if(++it, l.end(), f_ptr);
		}
	}

void BroDoc::WriteBroDocObjList(const BroDocObjMap& m,
                                bool wantPublic,
                                const char* heading,
                                char underline) const
	{
	BroDocObjMap::const_iterator it;
	BroDocObjList l;
	for ( it = m.begin(); it != m.end(); ++it ) l.push_back(it->second);
	WriteBroDocObjList(l, wantPublic, heading, underline);
	}

void BroDoc::WriteBroDocObjList(const BroDocObjList& l,
                                const char* heading,
                                char underline) const
	{
	WriteSectionHeading(heading, underline);
	BroDocObjList::const_iterator it;
	for ( it = l.begin(); it != l.end(); ++it )
		(*it)->WriteReST(reST_file);
	}

void BroDoc::WriteBroDocObjList(const BroDocObjMap& m,
                                const char* heading,
                                char underline) const
	{
	BroDocObjMap::const_iterator it;
	BroDocObjList l;
	for ( it = m.begin(); it != m.end(); ++it ) l.push_back(it->second);
	WriteBroDocObjList(l, heading, underline);
	}

void BroDoc::WriteToDoc(const char* format, ...) const
	{
	va_list argp;
	va_start(argp, format);
	vfprintf(reST_file, format, argp);
	va_end(argp);
	}

void BroDoc::WriteSectionHeading(const char* heading, char underline) const
	{
	WriteToDoc("%s\n", heading);
	size_t len = strlen(heading);
	for ( size_t i = 0; i < len; ++i )
		WriteToDoc("%c", underline);
	WriteToDoc("\n");
	}

void BroDoc::FreeBroDocObjPtrList(BroDocObjList& l)
	{
	for ( BroDocObjList::const_iterator it = l.begin(); it != l.end(); ++it )
		delete *it;
	l.clear();
	}

void BroDoc::FreeBroDocObjPtrList(BroDocObjMap& l)
	{
	for ( BroDocObjMap::const_iterator it = l.begin(); it != l.end(); ++it )
		delete it->second;
	l.clear();
	}

void BroDoc::AddFunction(BroDocObj* o)
	{
	BroDocObjMap::const_iterator it = functions.find(o->Name());
	if ( it == functions.end() )
		functions[o->Name()] = o;
	else
		functions[o->Name()]->Combine(o);
	}