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 <string>

#include <vector>

#include <iostream>

#include <signal.h>

#include <stdio.h>

#include <stdlib.h>

#include <pwd.h>

#include <boost/lexical_cast.hpp>

#include <boost/thread/mutex.hpp>

#include <boost/thread/condition.hpp>

#include <boost/thread/locks.hpp>

#include <boost/algorithm/string.hpp>

#include <boost/program_options.hpp>

#include "config.h"

#include "net/Manager.h"

#include "net/Subscriber.h"

#include "net/types.h"

#include "common/common.h"

#include <readline/readline.h>

#include <readline/history.h>

using namespace atom;

bool finish = false;

common::StringList autocomplete_list;

char* buffer = NULL;

boost::condition on_message_condition;

struct termios original_flags;

std::string history_filename;

void Handler(int status);

void CleanUp();

char* AutoCompleteGet(const char* text, int state);

static char** AutoComplete(const char* text, int start, int end);

std::string GetUserHomeDirectory()

{

    return std::string(getpwuid(getuid())->pw_dir);

}

class ConsoleClient : public net::Subscriber

{

public:

    typedef boost::shared_ptr<ConsoleClient> Pointer;

    ConsoleClient(std::string address, unsigned int port)

    {

        this->client_id_ = net::Manager::Instance()->Connect(net::PROTOCOL_TCP, address, port);

    }

    virtual ~ConsoleClient()

    {

        net::Manager::Instance()->Disconnect(this->client_id_);

        this->io_service_.stop();

    }

    std::string GetPrompt()

    {

        return this->prompt_;

    }

    void SendResponse(std::string payload)

    {

        std::string packet = "RESP";

        packet += common::PadNumber(payload.length() + 1, 4);

        packet += payload;

        net::Manager::Instance()->SendTo(this->client_id_, packet);

    }

    void AutoCompleteRequest(unsigned int arg_index, std::string commandline)

    {

        std::string payload = common::PadNumber(arg_index, 4);

        payload += commandline;

        std::string packet = "COMP";

        packet += common::PadNumber(payload.length() + 1, 4);

        packet += payload;

        net::Manager::Instance()->SendTo(this->client_id_, packet);

    }

private:

    net::SocketId client_id_;

    std::string prompt_;

    std::vector<unsigned char> buffer_;

    void SlotOnNewStateHandler(net::SocketId client_id, net::ClientState client_state)

    {

        if (client_state != net::CLIENT_STATE_CONNECTED)

        {

            std::cout << "Disconnected from server." << std::endl;

            this->client_id_ = 0;

            finish = true;

            on_message_condition.notify_all();

        }

    }

    void SlotOnNewClientHandler(net::SocketId id, net::SocketId server_id)

    {

    }

    void SlotOnNewDataHandler(net::SocketId client_id, common::Byteset data)

    {

        for (unsigned int n = 0; n < data.GetSize(); n++)

        {

            this->buffer_.push_back(data[n]);

        }

        while (this->buffer_.size() >= 8)

        {

            std::string command = "";

            command += (char)this->buffer_[0];

            command += (char)this->buffer_[1];

            command += (char)this->buffer_[2];

            command += (char)this->buffer_[3];

            std::string payload_length_str = "";

            payload_length_str += (char)this->buffer_[4];

            payload_length_str += (char)this->buffer_[5];

            payload_length_str += (char)this->buffer_[6];

            payload_length_str += (char)this->buffer_[7];

            unsigned int payload_length = boost::lexical_cast<unsigned int>(payload_length_str);

            if (this->buffer_.size() - 8 < payload_length)

            {

                return;

            }

            std::string payload = "";

            for (unsigned int n = 8; n < payload_length + 8; n++)

            {

                payload += (char)this->buffer_[n];

            }

            this->buffer_.erase(this->buffer_.begin(), this->buffer_.begin() + payload_length + 8);

            if (command == "TEXT")

            {

                std::cout << payload << std::flush;

            }

            else if (command == "PROM")

            {

                this->prompt_ = payload;

                on_message_condition.notify_all();

            }

            else if (command == "COMP")

            {

                autocomplete_list.clear();

                if (payload.length() > 0)

                {

                    boost::algorithm::split(autocomplete_list, payload, boost::is_any_of("\n"), boost::algorithm::token_compress_on);

                }

                on_message_condition.notify_all();

            }

            else

            {

                std::cerr << "Could not parse package." << std::endl;

                finish = true;

                on_message_condition.notify_all();

                break;

            }

        }

    }

};

ConsoleClient::Pointer cc;

int main(int argc, char **argv)

{

    // Signal handlers

    signal(SIGTERM, Handler);

    signal(SIGINT, Handler);

    signal(SIGQUIT, Handler);

    signal(SIGABRT, Handler);

    signal(SIGPIPE, Handler);

    boost::mutex guard_mutex;

    // Setup readline

    history_filename = GetUserHomeDirectory() + "/.atomic_history";

    read_history(history_filename.data());

    rl_attempted_completion_function = AutoComplete;

    // Save command line state

    tcgetattr(fileno(stdin), &original_flags);

    // Parse commandline

    boost::program_options::options_description command_line;

    boost::program_options::variables_map variable_map;

    command_line.add_options()

    ("help,h",    "produce help message")

    ("command,c", boost::program_options::value<std::string>()->default_value(""), "command")

    ("server,s",  boost::program_options::value<std::string>()->default_value("localhost"), "server address")

    ("port,p",    boost::program_options::value<unsigned int>()->default_value(1202), "server port");

    try

    {

        boost::program_options::store(boost::program_options::command_line_parser(argc, argv).options(command_line).run(), variable_map);

    }

    catch (boost::program_options::unknown_option e)

    {

        std::cerr << e.what() << std::endl;

        std::cout << command_line << std::endl;

        CleanUp();

        return EXIT_FAILURE;

    }

    catch (boost::program_options::invalid_syntax e)

    {

        std::cerr << e.what() << std::endl;

        std::cout << command_line << std::endl;

        CleanUp();

        return EXIT_FAILURE;

    }

    if (variable_map.count("help") != 0)

    {

        std::cout << command_line << std::endl;

        CleanUp();

        return EXIT_SUCCESS;

    }

    net::Manager::Create();

    if (variable_map["command"].as<std::string>() == "")

    {

        std::cout << "\033[29;1mAtom Interactive Console, version " + std::string(VERSION) + " starting...\033[0m" << std::endl;

        std::cout << "\033[29;1mReleased under " + std::string(LICENSE) + ".\033[0m" << std::endl;

        std::cout << "Written by Mattias Runge 2010." << std::endl;

        std::cout << "Connecting to " << variable_map["server"].as<std::string>().data() << ":" << variable_map["port"].as<unsigned int>() << "..." << std::endl;

    }

    try

    {

        cc = ConsoleClient::Pointer(new ConsoleClient(variable_map["server"].as<std::string>(), variable_map["port"].as<unsigned int>()));

    }

    catch (std::runtime_error& e)

    {

        std::cerr << "Connection error: " << e.what() << std::endl;

        CleanUp();

        return EXIT_FAILURE;

    }

    if (variable_map["command"].as<std::string>() != "")

    {

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

        on_message_condition.wait(guard);

        if (!finish)

        {

            cc->SendResponse(variable_map["command"].as<std::string>());

            if (!finish)

            {

                on_message_condition.wait(guard);

            }

        }

    }

    else

    {

        while (true)

        {

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

            on_message_condition.wait(guard);

            if (finish)

            {

                break;

            }

            while ((buffer = readline(cc->GetPrompt().data())) != NULL)

            {

                if (finish)

                {

                    break;

                }

                if (strlen(buffer) == 0)

                {

                    continue;

                }

                break;

            }

            if (finish)

            {

                break;

            }

            cc->SendResponse(buffer);

            add_history(buffer);

        }

    }

    CleanUp();

    return EXIT_SUCCESS;

}

static char** AutoComplete(const char* text, int start, int end)

{

    unsigned int count = 0;

    boost::mutex guard_mutex;

    for (unsigned int n = 0; n < (unsigned int)start; n++)

    {

        if (rl_line_buffer[n] == ' ')

        {

            count++;

        }

    }

    cc->AutoCompleteRequest(count, rl_line_buffer);

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

    on_message_condition.wait(guard);

    /*if (autocomplete_list.size() == 0)

    {

        return NULL;

    }*/

    return rl_completion_matches(text, &AutoCompleteGet);

}

char* AutoCompleteGet(const char* text, int state)

{

    static int index = 0;

    int length = strlen(text);

    std::string name;

    if (!state) // First run

    {

        index = 0;

    }

    while (autocomplete_list.size() > (unsigned int) index)

    {

        name = autocomplete_list[index];

        index++;

        if (strncmp(name.data(), text, length) == 0)

        {

            char *result = (char*)malloc(name.size() + 1);

            strcpy(result, name.data());

            return result;

        }

    }

    return NULL;

}

void CleanUp()

{

    std::cout << std::endl;

    //std::cout << "Cleaning up..." << std::endl;

    write_history(history_filename.data());

    cc.reset();

    net::Manager::Delete();

    if (buffer != NULL)

    {

        free(buffer);

    }

    //std::cout << "Thank you for using Atom. Goodbye!" << std::endl;

    tcsetattr(fileno(stdin), TCSANOW, &original_flags); // Restore

}

void Handler(int status)

{

    std::string signal_name = "Unknown";

    switch (status)

    {

        case SIGTERM:

        {

            signal_name = "Terminate";

            break;

        }  

        case SIGINT:

        {

            signal_name = "Interrupt";

            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;

        }

    }

    if (status != SIGPIPE)

    {

        CleanUp();

        exit(0);

    }

}