// Spectre 2
//
// Douglas Thrift
//
// $Id$

#include "Mounter.hpp"
#include "Unmounter.hpp"
#include "Matcher/Matcher.hpp"

#include <menes-api/exename.hpp>
#include <menes-app/application.hpp>

#include <sys/types.h>
#include <signal.h>

struct Spectre2Command : public app::Application
{
	virtual int Run(const app::ArgumentList& args)
	{
		ext::String config("spectre.xml"), pid("spectre.pid");

		_foreach (app::ArgumentList, arg, args)
		{
			Matcher matcher;

			if (*arg == matcher("^-config=(.*)$")) config = matcher[1];
			else if (*arg == matcher("^-pid=(.*)$")) pid = matcher[1];
			else if (*arg == "-D") Spectre2::debug = true;
			else
			{
				api::Cerr << "Usage: " << Spectre2::program << " [-config=config] [-pid=pid] [-D]" << ios::NewLine;

				return 1;
			}
		}

		Spectre2(config, pid);

		return 0;
	}
} spectre;

extern "C"
{
	void reload(int signal, ::siginfo_t* info, void* uap);
	void stop(int signal, ::siginfo_t* info, void* uap);
}

Spectre2::Spectre2(const ext::String& config, const ext::String& pid) : pid(pid)
{
	{
		api::FileWriter out(pid);
		ios::FormatWriter fout(out);

		fout << ::getpid() << ios::NewLine;
	}

	Mounter mounter(config);
	Unmounter unmounter(config);

	daemons.InsertLast(dynamic_cast<Daemon*>(&mounter));
	daemons.InsertLast(dynamic_cast<Daemon*>(&unmounter));

	{
		struct ::sigaction action;

		action.sa_sigaction = reload;
		action.sa_flags = SA_SIGINFO;

		api::Posix::CheckError(::sigemptyset(&action.sa_mask));
		api::Posix::CheckError(::sigaction(SIGUSR1, &action, NULL));

		action.sa_sigaction = stop;

		api::Posix::CheckError(::sigaction(SIGTERM, &action, NULL));
	}

	_mforeach (ext::Vector<Daemon*>, daemon, daemons) (*daemon)->start();
	_mforeach (ext::Vector<Daemon*>, daemon, daemons) (*daemon)->wait();
}

Spectre2::~Spectre2()
{
	api::Posix::CheckError(::unlink(pid.NullTerminate()));
}

ext::String Spectre2::program(api::GetExecutablePath().GetName()), Spectre2::prefix(_Spectre2_prefix_), Spectre2::root(_Spectre2_root_);
bool Spectre2::debug(false);
ext::Vector<Daemon*> Spectre2::daemons;

extern "C"
{
	void reload(int signal, ::siginfo_t* info, void* uap)
	{
		_mforeach (ext::Vector<Daemon*>, daemon, Spectre2::daemons) (*daemon)->reload();
	}

	void stop(int signal, ::siginfo_t* info, void* uap)
	{
		_mforeach (ext::Vector<Daemon*>, daemon, Spectre2::daemons) (*daemon)->stop();
	}
}