i2pd/daemon/Daemon.h

/*
* Copyright (c) 2013-2020, The PurpleI2P Project
*
* This file is part of Purple i2pd project and licensed under BSD3
*
* See full license text in LICENSE file at top of project tree
*/

#ifndef DAEMON_H__
#define DAEMON_H__

#include <memory>
#include <string>
#include <ostream>

namespace i2p
{
namespace util
{
	class Daemon_Singleton_Private;
	class Daemon_Singleton
	{
	public:

		virtual bool init (int argc, char* argv[], std::shared_ptr<std::ostream> logstream);
		virtual bool init (int argc, char* argv[]);
		virtual bool start ();
		virtual bool stop ();
		virtual void run () {};

		virtual void setDataDir (std::string path);

		bool isDaemon;
		bool running;

	protected:

		Daemon_Singleton ();
		virtual ~Daemon_Singleton ();

		bool IsService () const;

		// d-pointer for httpServer, httpProxy, etc.
		class Daemon_Singleton_Private;
		Daemon_Singleton_Private &d;

	private:

		std::string DaemonDataDir;
	};

#if defined(QT_GUI_LIB) // check if QT
#define Daemon i2p::util::DaemonQT::Instance()
	// dummy, invoked from RunQT
	class DaemonQT: public i2p::util::Daemon_Singleton
	{
		public:

			static DaemonQT& Instance()
			{
				static DaemonQT instance;
				return instance;
			}
	};

#elif defined(_WIN32)
#define Daemon i2p::util::DaemonWin32::Instance()
	class DaemonWin32 : public Daemon_Singleton
	{
		public:

			static DaemonWin32& Instance()
			{
				static DaemonWin32 instance;
				return instance;
			}

			bool init (int argc, char* argv[]);
			bool start ();
			bool stop ();
			void run ();

			void SetIsGraceful (bool state) { m_isGraceful = state; };
			const bool GetIsGraceful () { return m_isGraceful; };

		private:

			bool m_isGraceful;
	};
#elif (defined(ANDROID) && !defined(ANDROID_BINARY))
#define Daemon i2p::util::DaemonAndroid::Instance()
	// dummy, invoked from android/jni/DaemonAndroid.*
	class DaemonAndroid: public i2p::util::Daemon_Singleton
	{
		public:

			static DaemonAndroid& Instance()
			{
				static DaemonAndroid instance;
				return instance;
			}
	};
#else
#define Daemon i2p::util::DaemonLinux::Instance()
	class DaemonLinux : public Daemon_Singleton
	{
		public:

			static DaemonLinux& Instance()
			{
				static DaemonLinux instance;
				return instance;
			}

			bool start ();
			bool stop ();
			void run ();

		private:

			std::string pidfile;
			int pidFH;

		public:

			int gracefulShutdownInterval; // in seconds
	};
#endif
}
}

#endif // DAEMON_H__