/home/mhx/git/github/libmoost/include/moost/service/remote_shell.h

Go to the documentation of this file.

/* vim:set ts=3 sw=3 sts=3 et: */
#ifndef FM_LAST_MOOST_SERVICE_REMOTE_SHELL
#define FM_LAST_MOOST_SERVICE_REMOTE_SHELL

#include <iostream>
#include <boost/asio.hpp>
#include <boost/function.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/noncopyable.hpp>

#include "appender.h"

namespace moost { namespace service {

class remote_shell_iface
{
public:
   virtual bool handle_command(std::string& _return, const std::string& cmd, const std::string& args) = 0;

   virtual std::string get_prompt() const = 0;

   virtual std::string show_help() const = 0;
};

class remote_shell_server_impl;

class remote_shell_server : public boost::noncopyable
{
private:
   remote_shell_server_impl *m_impl;

public:
   remote_shell_server(boost::shared_ptr<boost::asio::io_service> ios);
   ~remote_shell_server();
   void run(remote_shell_iface *);
   void stop(const std::string&);
   void set_appender_factory(appender_factory_ptr);
   void set_default_stdout_state(bool);
   void set_default_stderr_state(bool);
   void set_listen_port(unsigned short);
   void set_pre_shutdown_function(boost::function0<void>&);
   void enable_local_shell(bool);
};

template <class HandlerType>
class remote_shell : private remote_shell_iface
{
private:
   HandlerType& m_hdl;
   remote_shell_server m_srv;

public:
   remote_shell(HandlerType& hdl)
      : m_hdl(hdl), m_srv(boost::shared_ptr<boost::asio::io_service>(new boost::asio::io_service))
   {
   }

   remote_shell(HandlerType& hdl, boost::shared_ptr<boost::asio::io_service> ios)
      : m_hdl(hdl), m_srv(ios)
   {
   }

   virtual std::string get_prompt() const
   {
      return m_hdl.get_prompt();
   }

   virtual std::string show_help() const
   {
      return m_hdl.show_help();
   }

   virtual bool handle_command(std::string& _return, const std::string& cmd, const std::string& args)
   {
      return m_hdl.handle_command(_return, cmd, args);
   }

   void run()
   {
      m_srv.run(this);
   }

   void set_appender_factory(appender_factory_ptr app_factory)
   {
      m_srv.set_appender_factory(app_factory);
   }

   void set_default_stdout_state(bool enabled)
   {
      m_srv.set_default_stdout_state(enabled);
   }

   void set_default_stderr_state(bool enabled)
   {
      m_srv.set_default_stderr_state(enabled);
   }

   void set_listen_port(unsigned short port)
   {
      m_srv.set_listen_port(port);
   }

   void set_pre_shutdown_function(boost::function0<void> func)
   {
      m_srv.set_pre_shutdown_function(func);
   }

   void enable_local_shell(bool enabled = true)
   {
      m_srv.enable_local_shell(enabled);
   }

   void stop(const std::string& msg)
   {
      m_srv.stop(msg);
   }
};

} }

#endif