Subversion Repositories HomeAutomation

/*
 *
 *  Copyright (C) 2010  Mattias Runge
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License along
 *  with this program; if not, write to the Free Software Foundation, Inc.,
 *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 */

#include <signal.h>

#include <boost/program_options.hpp>
#include <boost/thread/mutex.hpp>
#include <boost/thread/condition.hpp>
#include <boost/thread/locks.hpp>

#include "config.h"

#include "logging/Logger.h"
#include "net/Manager.h"
#include "timer/Manager.h"
#include "config/Manager.h"
#include "can/Protocol.h"
#include "broker/Manager.h"
#include "control/Manager.h"
#include "can/Network.h"
#include "can/Monitor.h"
#include "can/CanDaemon.h"
#include "vm/Manager.h"
#include "storage/Manager.h"
#include "mbb/Manager.h"

#include "vm/plugin/System.h"
#include "vm/plugin/Timer.h"
#include "vm/plugin/Module.h"
#include "vm/plugin/Node.h"
#include "vm/plugin/Console.h"
#include "vm/plugin/Storage.h"
#include "vm/plugin/Socket.h"

#ifdef USE_PLUGIN_XORG
#include "vm/plugin/Xorg.h"
#endif // USE_PLUGIN_XORG

using namespace atom;

logging::Logger LOG ("Main");
std::vector<broker::Subscriber::Pointer> subscribers;
boost::condition on_message_condition;
boost::mutex guard_mutex;

void Handler (int status);
void CleanUp();

int main (int argc, char **argv)
{
  LOG.Info ("\033[29;1mAtom Daemon, version " + std::string (VERSION) + " starting...\033[0m");
  LOG.Info ("\033[29;1mReleased under " + std::string (LICENSE) + ".\033[0m");
  LOG.Info ("Written by Mattias Runge 2010-2012.");

  signal (SIGTERM, Handler);
  signal (SIGINT, Handler);
  signal (SIGQUIT, Handler);
  signal (SIGABRT, Handler);
  signal (SIGPIPE, Handler);

  config::Manager::Create();

  if (!config::Manager::Instance()->Set (argc, argv))
  {
    CleanUp();
    return EXIT_SUCCESS;
  }

  if (config::Manager::Instance()->Exist ("LogFile"))
  {
    logging::Logger::OpenFile (config::Manager::Instance()->GetAsString ("LogFile"));
  }

  if (config::Manager::Instance()->Exist ("LogLevel"))
  {
    logging::Logger::SetLevel ( (logging::Logger::Level) config::Manager::Instance()->GetAsInt ("LogLevel"));
  }

  if (config::Manager::Instance()->Exist ("daemon"))
  {
    LOG.Info ("Entering daemon mode...");

    if (daemon (0, 0) == -1)
    {
      LOG.Error ("Could not enter daemon mode. Exiting...");
      CleanUp();
      return EXIT_FAILURE;
    }

    LOG.Info ("Deamon mode entered successfully!");
  }

  storage::Manager::Create();

  timer::Manager::Create();
  broker::Manager::Create();
  net::Manager::Create();
  can::Protocol::Create();
  control::Manager::Create();
  vm::Manager::Create();

  if (config::Manager::Instance()->Exist ("MbbPort"))
  {
    mbb::Manager::Create (config::Manager::Instance()->GetAsInt ("MbbPort"));
  }

  storage::Manager::Instance()->SetRootPath (config::Manager::Instance()->GetAsString ("StoragePath"));

  can::Protocol::Instance()->Load (config::Manager::Instance()->GetAsString ("ProtocolFile"));

  if (config::Manager::Instance()->Exist ("MonitorPort"))
  {
    subscribers.push_back (can::Monitor::Pointer (new can::Monitor (config::Manager::Instance()->GetAsInt ("MonitorPort"))));
  }

  if (config::Manager::Instance()->Exist ("DaemonPort"))
  {
    subscribers.push_back (can::CanDaemon::Pointer (new can::CanDaemon (config::Manager::Instance()->GetAsInt ("DaemonPort"))));
  }

  subscribers.push_back (control::Manager::Instance());

  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::System (vm::Manager::Instance()->GetIoService())));
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Console (vm::Manager::Instance()->GetIoService(), config::Manager::Instance()->GetAsInt ("CommandPort"))));
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Storage (vm::Manager::Instance()->GetIoService())));
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Timer (vm::Manager::Instance()->GetIoService())));
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Module (vm::Manager::Instance()->GetIoService())));
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Node (vm::Manager::Instance()->GetIoService())));
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Socket (vm::Manager::Instance()->GetIoService())));

#ifdef USE_PLUGIN_XORG
  vm::Manager::Instance()->AddPlugin (vm::Plugin::Pointer (new vm::plugin::Xorg (vm::Manager::Instance()->GetIoService())));
#endif // USE_PLUGIN_XORG

  vm::Manager::Instance()->Start (config::Manager::Instance()->GetAsString ("ScriptPath"), config::Manager::Instance()->GetAsString ("UserScriptPath"));

  if (config::Manager::Instance()->Exist ("CanNet"))
  {
    common::StringList cannetworks = config::Manager::Instance()->GetAsStringVector ("CanNet");

    for (uint n = 0; n < cannetworks.size(); n++)
    {
      subscribers.push_back (can::Network::Pointer (new can::Network (cannetworks[n])));
    }
  }

  boost::mutex::scoped_lock guard (guard_mutex);

  on_message_condition.wait (guard);

  LOG.Info ("Cleaning up...");

  CleanUp();

  LOG.Info ("Thank you for using Atom. Goodbye!");

  return EXIT_SUCCESS;
}

void CleanUp()
{
  net::Manager::Instance()->Stop();

  subscribers.clear();
  config::Manager::Delete();
  timer::Manager::Delete();
  control::Manager::Delete();
  mbb::Manager::Delete();
  can::Protocol::Delete();
  broker::Manager::Delete();
  vm::Manager::Delete();
  net::Manager::Delete();
  storage::Manager::Delete();
}

void Handler (int status)
{
  std::string signal_name = "Unknown";

  switch (status)
  {
    case SIGTERM:
    {
      signal_name = "Terminate";
      break;
    }

    case SIGINT:
    {
      signal_name = "Interupt";
      break;
    }

    case SIGQUIT:
    {
      signal_name = "Quit";
      break;
    }

    case SIGABRT:
    {
      signal_name = "Abort";
      break;
    }

    case SIGIO:
    {
      signal_name = "I/O";
      break;
    }

    case SIGPIPE:
    {
      signal_name = "Pipe";
      break;
    }
  }

  LOG.Debug ("Received signal " + signal_name + "(" + boost::lexical_cast<std::string> (status) + ").");

  if (status != SIGPIPE)
  {
    on_message_condition.notify_all();
  }
}