PresenceDetection/Application/PresenceDetection.cc

#include <iostream>
#include <string>
#include <signal.h>
#include <syslog.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <boost/lexical_cast.hpp>
#include <boost/program_options.hpp>
#include <boost/property_tree/ptree.hpp>
#include <boost/property_tree/ini_parser.hpp>
#include "UniFi/Device.h"
#include "Bluetooth/Device.h"


int main(int argc, char** argv)
{
	try
	{
		setlogmask(LOG_UPTO(LOG_INFO));
		openlog("PresenceDetection", LOG_CONS | LOG_NDELAY | LOG_PERROR | LOG_PID, LOG_USER);

		boost::program_options::options_description description("Options");
		description.add_options()
			("help,h", "Provide Help Message")
			("daemon,d", "Run the process as Daemon");

		boost::program_options::variables_map vm;
		boost::program_options::store(boost::program_options::parse_command_line(argc, argv, description), vm);
		boost::program_options::notify(vm);

		if (vm.count("help"))
		{
			std::cerr << description << std::endl;
			return 1;
		}

		bool daemon = false;
		if (vm.count("daemon"))
			daemon = true;

		boost::property_tree::ptree pt;
		boost::property_tree::ini_parser::read_ini("PresenceDetection.ini", pt);

		if (daemon)
		{
			syslog(LOG_INFO, "Starting Daemon");
			pid_t pid, sid;
			pid = fork();

			if (pid < 0)
				exit(EXIT_FAILURE);
			if (pid > 0)
				exit(EXIT_SUCCESS);

			umask(0);
			sid = setsid();
			if (sid < 0)
				exit(EXIT_FAILURE);

			if ((chdir("/")) < 0)
				exit(EXIT_FAILURE);

			close(STDIN_FILENO);
			close(STDOUT_FILENO);
			close(STDERR_FILENO);
		}

		int port;
		boost::optional<boost::property_tree::ptree&> presenceDetectionPort = pt.get_child_optional("PresenceDetection.Port");
		if (presenceDetectionPort)
			port = pt.get<int>("PresenceDetection.Port");
		else
			throw std::runtime_error("Port directive missing in ini file.");

		std::string target;
		boost::optional<boost::property_tree::ptree&> presenceDetectionTarget = pt.get_child_optional("PresenceDetection.Target");
		if (presenceDetectionTarget)
			target = pt.get<std::string>("PresenceDetection.Target");
		else
			throw std::runtime_error("Target directive missing in ini file.");

		bool unifi = false;
		boost::optional<boost::property_tree::ptree&> presenceDetectionUniFi = pt.get_child_optional("PresenceDetection.UniFi");
		if (presenceDetectionUniFi)
			unifi = (pt.get<std::string>("PresenceDetection.UniFi") == "On");

		bool bluetooth = false;
		boost::optional<boost::property_tree::ptree&> presenceDetectionBluetooth = pt.get_child_optional("PresenceDetection.Bluetooth");
		if (presenceDetectionBluetooth)
			bluetooth = (pt.get<std::string>("PresenceDetection.Bluetooth") == "On");

		std::shared_ptr<PresenceDetection::UniFi::Device> pUniFiDevice;
		if (unifi)
		{
			std::string hostname;
			boost::optional<boost::property_tree::ptree&> unifiHostname = pt.get_child_optional("UniFi.Hostname");
			if (unifiHostname)
				hostname = pt.get<std::string>("UniFi.Hostname");
			else
				throw std::runtime_error("UniFi Hostname directive missing in ini file.");

			std::string username;
			boost::optional<boost::property_tree::ptree&> unifiUsername = pt.get_child_optional("UniFi.Username");
			if (unifiUsername)
				username = pt.get<std::string>("UniFi.Username");
			else
				throw std::runtime_error("UniFi Username directive missing in ini file.");

			std::string password;
			boost::optional<boost::property_tree::ptree&> unifiPassword = pt.get_child_optional("UniFi.Password");
			if (unifiPassword)
				password = pt.get<std::string>("UniFi.Password");
			else
				throw std::runtime_error("UniFi Password directive missing in ini file.");

			std::string cookiefile;
			boost::optional<boost::property_tree::ptree&> unifiCookieFile = pt.get_child_optional("UniFi.CookieFile");
			if (unifiCookieFile)
				cookiefile = pt.get<std::string>("UniFi.CookieFile");
			else
				throw std::runtime_error("UniFi CookieFile directive missing in ini file.");

			pUniFiDevice = std::make_shared<PresenceDetection::UniFi::Device>(hostname, username, password, cookiefile, target);
		}

		std::shared_ptr<PresenceDetection::Bluetooth::Device> pBluetoothDevice;
		if (bluetooth)
		{
			std::string inventoryURL;
			boost::optional<boost::property_tree::ptree&> bluetoothInventoryURL = pt.get_child_optional("Bluetooth.Inventory");
			if (bluetoothInventoryURL)
				inventoryURL = pt.get<std::string>("Bluetooth.Inventory");
			else
				throw std::runtime_error("Bluetooth Inventory directive missing in ini file.");

			pBluetoothDevice = std::make_shared<PresenceDetection::Bluetooth::Device>(inventoryURL, target);
		}

		sigset_t wset;
		sigemptyset(&wset);
		sigaddset(&wset,SIGHUP);
		sigaddset(&wset,SIGINT);
		sigaddset(&wset,SIGTERM);
		int sig;
		sigwait(&wset,&sig);

		syslog(LOG_INFO, "Stopping...");

		if (pUniFiDevice)
			pUniFiDevice->Stop();

		if (pBluetoothDevice)
			pBluetoothDevice->Stop();

		if (pUniFiDevice)
			pUniFiDevice->Wait();

		if (pBluetoothDevice)
			pBluetoothDevice->Wait();

		closelog();
	}
	catch (const std::exception& e)
	{
		std::stringstream ss;
		ss << "ERROR: " << e.what() << std::endl;

		std::cerr << ss.str() << std::endl;
		syslog(LOG_ERR, "%s", ss.str().c_str());

		closelog();

		return -1;
	}

	return 0;
}