// Google Tron
//
// Douglas Thrift
//
// $Id$

#include <cxx/standard.hh>

#include <api/exename.hpp>
#include <api/pcre/regex.hpp>
#include <api/process.hpp>
#include <app/simple.hpp>
#include <hop/bind.hpp>
#include <xml/textwriter.hpp>

#include "GoogleTron.hpp"
#include "Sleep.hpp"
#include "Zlib/GzipWriter.hpp"

int Main(const app::Options &options)
{
	cse::String sitemap(_B("sitemap.gz"));

	api::Pcre::RegEx sitemap_(_B("^-sitemap=(.+)$"));

	_foreach (const app::ArgumentList, arg, app::GetArguments())
	{
		api::Pcre::RegEx::Match match;

		if (match = sitemap_(*arg))
			sitemap = match[1];
		else
		{
			api::Cout << _B("Usage: ") << api::GetExecutablePath().GetName() << _B(" [-sitemap=sitemap]") << ios::NewLine;

			return 1;
		}
	}

	GoogleTron tron(sitemap);

	return 0;
}

GoogleTron::GoogleTron(const cse::String &sitemap) : working(true), sitemap(sitemap), output(new _H<api::Thread>(hop::BindAll(&GoogleTron::Output, this)))
{
	// XXX: start some threads
}

GoogleTron::~GoogleTron()
{
	// XXX: wait on threads
	
	working = false;

	output->Join();
}

int GoogleTron::Output()
{
	_S<Zlib::GzipWriter> gzip(sitemap);
	_S<xml::TextWriter> writer(gzip);
	xml::ScopeElement urlset(writer, _B("urlset"));

	writer.SetAttribute(_B("xmlns"), _B("http://www.google.com/schemas/sitemap/0.84"));
	writer.SetAttribute(_B("xmlns:xsi"), _B("http://www.w3.org/2001/XMLSchema-instance"));
	writer.SetAttribute(_B("xsi:schemaLocation"), _B("http://www.google.com/schemas/sitemap/0.84 http://www.google.com/schemas/sitemap/0.84/sitemap.xsd"));

	do
	{
		Sleep();

		_synchronized (queueLock)
			if (queue.size())
			{
				xml::ScopeElement url_(writer, _B("url"));
				const Url &url(queue.front());

				{
					xml::ScopeElement loc(writer, _B("loc"));

					writer.OutputText(url.GetLocation());
				}

				{
					xml::ScopeElement lastmod(writer, _B("lastmod"));

					writer.OutputText(url.GetModified());
				}

				{
					xml::ScopeElement changefreq(writer, _B("changefreq"));

					writer.OutputText(url.GetFrequency());
				}

				{
					xml::ScopeElement priority(writer, _B("priority"));

					writer.OutputText(url.GetPriority());
				}

				queue.pop();
			}
	}
	while (working);

	return 0;
}