diff irccd-test/main.cpp @ 620:c79ae2987955

Irccd: create a brand new irccd-test executable, closes #569 @3h
author David Demelier <markand@malikania.fr>
date Thu, 21 Dec 2017 21:55:57 +0100
parents
children 1afefb4ffcf8
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/irccd-test/main.cpp	Thu Dec 21 21:55:57 2017 +0100
@@ -0,0 +1,538 @@
+/*
+ * main.cpp -- irccd-test main file
+ *
+ * Copyright (c) 2013-2017 David Demelier <markand@malikania.fr>
+ *
+ * Permission to use, copy, modify, and/or distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+
+#include <irccd/sysconfig.hpp>
+
+#include <algorithm>
+#include <functional>
+#include <iostream>
+#include <string>
+#include <unordered_map>
+
+#include <boost/algorithm/string/trim.hpp>
+#include <boost/filesystem/path.hpp>
+
+#if defined(HAVE_LIBEDIT)
+#   include <histedit.h>
+#endif
+
+#include <irccd/options.hpp>
+#include <irccd/string_util.hpp>
+
+#include <irccd/daemon/dynlib_plugin.hpp>
+#include <irccd/daemon/irccd.hpp>
+#include <irccd/daemon/plugin_service.hpp>
+#include <irccd/daemon/server_service.hpp>
+
+#include <irccd/test/debug_server.hpp>
+
+#if defined(HAVE_JS)
+#   include <irccd/js/directory_jsapi.hpp>
+#   include <irccd/js/elapsed_timer_jsapi.hpp>
+#   include <irccd/js/file_jsapi.hpp>
+#   include <irccd/js/irccd_jsapi.hpp>
+#   include <irccd/js/js_plugin.hpp>
+#   include <irccd/js/logger_jsapi.hpp>
+#   include <irccd/js/plugin_jsapi.hpp>
+#   include <irccd/js/server_jsapi.hpp>
+#   include <irccd/js/system_jsapi.hpp>
+#   include <irccd/js/timer_jsapi.hpp>
+#   include <irccd/js/unicode_jsapi.hpp>
+#   include <irccd/js/util_jsapi.hpp>
+#endif
+
+namespace irccd {
+
+namespace su = string_util;
+
+namespace {
+
+boost::asio::io_service io;
+
+std::unique_ptr<irccd> daemon;
+std::shared_ptr<plugin> plugin;
+
+void usage()
+{
+    std::cerr << "usage: irccd-test [-c config] plugin-name" << std::endl;
+    std::exit(1);
+}
+
+std::shared_ptr<server> get_server(std::string name)
+{
+    name = boost::algorithm::trim_copy(name);
+
+    if (name.empty())
+        name = "test";
+
+    auto s = daemon->servers().get(name);
+
+    if (!s) {
+        s = std::make_shared<debug_server>(io, std::move(name));
+        daemon->servers().add(s);
+    }
+
+    return s;
+}
+
+std::string get_arg(const std::vector<std::string>& args, unsigned index)
+{
+    if (index >= args.size())
+        return "";
+
+    return args[index];
+}
+
+/*
+ * onCommand server origin channel message
+ */
+void on_command(const std::string& data)
+{
+    auto args = su::split(data, " ", 4);
+
+    plugin->on_command(*daemon, {
+        get_server(get_arg(args, 0)),
+        get_arg(args, 1),
+        get_arg(args, 2),
+        get_arg(args, 3)
+    });
+}
+
+/*
+ * onConnect server
+ */
+void on_connect(const std::string& data)
+{
+    auto args = su::split(data, " ");
+
+    plugin->on_connect(*daemon, {get_server(get_arg(args, 0))});
+}
+
+/*
+ * onInvite server origin channel target
+ */
+void on_invite(const std::string& data)
+{
+    auto args = su::split(data, " ");
+
+    plugin->on_invite(*daemon, {
+        get_server(get_arg(args, 0)),
+        get_arg(args, 1),
+        get_arg(args, 2),
+        get_arg(args, 3),
+    });
+}
+
+/*
+ * onJoin server origin channel
+ */
+void on_join(const std::string& data)
+{
+    auto args = su::split(data, " ");
+
+    plugin->on_join(*daemon, {
+        get_server(get_arg(args, 0)),
+        get_arg(args, 1),
+        get_arg(args, 2)
+    });
+}
+
+/*
+ * onKick server origin channel reason
+ */
+void on_kick(const std::string& data)
+{
+    auto args = su::split(data, " ", 5);
+
+    plugin->on_kick(*daemon, {
+        get_server(get_arg(args, 0)),
+        get_arg(args, 1),
+        get_arg(args, 2),
+        get_arg(args, 3),
+        get_arg(args, 4),
+    });
+}
+
+/*
+ * onLoad
+ */
+void on_load(const std::string&)
+{
+    plugin->on_load(*daemon);
+}
+
+/*
+ * onMe server origin channel message
+ */
+void on_me(const std::string& data)
+{
+    auto args = su::split(data, " ", 4);
+
+    plugin->on_me(*daemon, {
+        get_server(get_arg(args, 0)),
+        get_arg(args, 1),
+        get_arg(args, 2),
+        get_arg(args, 3)
+    });
+}
+
+/*
+ * onMessage server origin channel message
+ */
+void on_message(const std::string& data)
+{
+    auto args = su::split(data, " ", 4);
+
+    plugin->on_message(*daemon, {
+        get_server(get_arg(args, 0)),
+        get_arg(args, 1),
+        get_arg(args, 2),
+        get_arg(args, 3)
+    });
+}
+
+/*
+ * onMode server origin channel mode limit user mask
+ */
+void on_mode(const std::string& data)
+{
+    auto args = su::split(data, " ", 7);
+
+    plugin->on_mode(*daemon, {
+        get_server(get_arg(args, 0)),
+        get_arg(args, 1),
+        get_arg(args, 2),
+        get_arg(args, 3),
+        get_arg(args, 4),
+        get_arg(args, 5),
+        get_arg(args, 6),
+    });
+}
+
+/*
+ * onNames server channel nick1 nick2 nickN
+ */
+void on_names(const std::string& data)
+{
+    auto args = su::split(data, " ");
+
+    names_event ev;
+
+    ev.server = get_server(get_arg(args, 0));
+    ev.channel = get_arg(args, 1);
+
+    if (args.size() >= 3U)
+        ev.names.insert(ev.names.begin(), args.begin() + 2, args.end());
+
+    plugin->on_names(*daemon, ev);
+}
+
+/*
+ * onNick server origin nickname
+ */
+void on_nick(const std::string& data)
+{
+    auto args = su::split(data, " ");
+
+    plugin->on_nick(*daemon, {
+        get_server(get_arg(args, 0)),
+        get_arg(args, 1),
+        get_arg(args, 2)
+    });
+}
+
+/*
+ * onNotice server origin channel nickname
+ */
+void on_notice(const std::string& data)
+{
+    auto args = su::split(data, " ", 4);
+
+    plugin->on_notice(*daemon, {
+        get_server(get_arg(args, 0)),
+        get_arg(args, 1),
+        get_arg(args, 2),
+        get_arg(args, 3)
+    });
+}
+
+/*
+ * onPart server origin channel reason
+ */
+void on_part(const std::string& data)
+{
+    auto args = su::split(data, " ", 4);
+
+    plugin->on_part(*daemon, {
+        get_server(get_arg(args, 0)),
+        get_arg(args, 1),
+        get_arg(args, 2),
+        get_arg(args, 3),
+    });
+}
+
+/*
+ * onReload
+ */
+void on_reload(const std::string&)
+{
+    plugin->on_reload(*daemon);
+}
+
+/*
+ * onTopic server origin channel topic
+ */
+void on_topic(const std::string& data)
+{
+    auto args = su::split(data, " ", 4);
+
+    plugin->on_topic(*daemon, {
+        get_server(get_arg(args, 0)),
+        get_arg(args, 1),
+        get_arg(args, 2),
+        get_arg(args, 3)
+    });
+}
+
+/*
+ * onUnload
+ */
+void on_unload(const std::string&)
+{
+    plugin->on_unload(*daemon);
+}
+
+/*
+ * onWhois server nick user host realname chan1 chan2 chanN
+ */
+void on_whois(const std::string& data)
+{
+    auto args = su::split(data, " ");
+
+    whois_event ev;
+
+    ev.server = get_server(get_arg(args, 0));
+    ev.whois.nick = get_arg(args, 1);
+    ev.whois.user = get_arg(args, 2);
+    ev.whois.host = get_arg(args, 3);
+    ev.whois.realname = get_arg(args, 4);
+
+    if (args.size() >= 5)
+        ev.whois.channels.insert(ev.whois.channels.begin(), args.begin() + 5, args.end());
+
+    plugin->on_whois(*daemon, ev);
+}
+
+/*
+ * Table of user functions.
+ */
+using function = std::function<void (const std::string&)>;
+using functions = std::unordered_map<std::string, function>;
+
+static const functions list{
+    { "onCommand",  &(on_command)   },
+    { "onConnect",  &(on_connect)   },
+    { "onInvite",   &(on_invite)    },
+    { "onJoin",     &(on_join)      },
+    { "onKick",     &(on_kick)      },
+    { "onLoad",     &(on_load)      },
+    { "onMe",       &(on_me)        },
+    { "onMessage",  &(on_message)   },
+    { "onMode",     &(on_mode)      },
+    { "onNames",    &(on_names)     },
+    { "onNick",     &(on_nick)      },
+    { "onNotice",   &(on_notice)    },
+    { "onPart",     &(on_part)      },
+    { "onReload",   &(on_reload)    },
+    { "onTopic",    &(on_topic)     },
+    { "onUnload",   &(on_unload)    },
+    { "onWhois",    &(on_whois)     }
+};
+
+void exec(const std::string& line)
+{
+    auto pos = line.find(' ');
+    auto it = list.find(line.substr(0, pos));
+
+    if (it != list.end())
+        it->second(pos == std::string::npos ? "" : line.substr(pos + 1));
+}
+
+#if defined(HAVE_LIBEDIT)
+
+const char* prompt(EditLine*)
+{
+    static const char* text = "> ";
+
+    return text;
+}
+
+std::string clean(std::string input)
+{
+    while (!input.empty() && (input.back() == '\n' || input.back() == '\r'))
+        input.pop_back();
+
+    return input;
+}
+
+std::vector<std::string> matches(const std::string& name)
+{
+    std::vector<std::string> result;
+
+    for (const auto& pair : list)
+        if (pair.first.compare(0U, name.size(), name) == 0U)
+            result.push_back(pair.first);
+
+    return result;
+}
+
+unsigned char complete(EditLine* el, int)
+{
+    const auto* lf = el_line(el);
+    const auto args = su::split(std::string(lf->buffer, lf->cursor), " ");
+
+    if (args.size() == 0U)
+        return CC_REFRESH;
+
+    const auto found = matches(args[0]);
+
+    if (found.size() != 1U)
+        return CC_REFRESH;
+
+    // Insert the missing text, e.g. onCom -> onCommand.
+    if (el_insertstr(el, &found[0].c_str()[args[0].size()]) < 0)
+        return CC_ERROR;
+
+    return CC_REFRESH;
+}
+
+void run()
+{
+    std::unique_ptr<EditLine, void (*)(EditLine*)> el(
+        el_init("irccd-test", stdin, stdout, stderr),
+        el_end
+    );
+    std::unique_ptr<History, void (*)(History*)> hist(
+        history_init(),
+        history_end
+    );
+    HistEvent hev;
+
+    history(hist.get(), &hev, H_SETSIZE, 1024);
+    el_set(el.get(), EL_EDITOR, "emacs");
+    el_set(el.get(), EL_PROMPT, prompt);
+    el_set(el.get(), EL_HIST, history, hist.get());
+    el_set(el.get(), EL_ADDFN, "ed-complete", "Complete command", complete);
+    el_set(el.get(), EL_BIND, "^I", "ed-complete", nullptr);
+
+    const char* s;
+    int size;
+
+    while ((s = el_gets(el.get(), &size)) && size >= 0) {
+        if (size > 0)
+            history(hist.get(), &hev, H_ENTER, s);
+
+        exec(clean(s));
+    }
+}
+
+#else
+
+void run()
+{
+    std::string line;
+
+    for (;;) {
+        std::cout << "> ";
+        std::getline(std::cin, line);
+        exec(line);
+    }
+}
+
+#endif
+
+void load_plugins(int argc, char** argv)
+{
+    if (argc <= 0)
+        usage();
+
+    daemon->plugins().load("test", boost::filesystem::exists(argv[0]) ? argv[0] : "");
+    plugin = daemon->plugins().get("test");
+}
+
+void load_options(int& argc, char**& argv)
+{
+    const option::options def{
+        { "-c",         true    },
+        { "--config",   true    }
+    };
+
+    auto result = option::read(argc, argv, def);
+    auto it = result.find("-c");
+
+    if (it == result.end())
+        it = result.find("--config");
+    if (it != result.end()) {
+        try {
+            daemon->set_config(it->second);
+        } catch (const std::exception& ex) {
+            throw std::runtime_error(su::sprintf("%s: %s", it->second, ex.what()));
+        }
+    }
+}
+
+void load(int argc, char** argv)
+{
+    daemon = std::make_unique<irccd>(io);
+
+#if defined(HAVE_JS)
+    auto loader = std::make_unique<js_plugin_loader>(*daemon);
+
+    loader->modules().push_back(std::make_unique<irccd_jsapi>());
+    loader->modules().push_back(std::make_unique<directory_jsapi>());
+    loader->modules().push_back(std::make_unique<elapsed_timer_jsapi>());
+    loader->modules().push_back(std::make_unique<file_jsapi>());
+    loader->modules().push_back(std::make_unique<logger_jsapi>());
+    loader->modules().push_back(std::make_unique<plugin_jsapi>());
+    loader->modules().push_back(std::make_unique<server_jsapi>());
+    loader->modules().push_back(std::make_unique<system_jsapi>());
+    loader->modules().push_back(std::make_unique<timer_jsapi>());
+    loader->modules().push_back(std::make_unique<unicode_jsapi>());
+    loader->modules().push_back(std::make_unique<util_jsapi>());
+
+    daemon->plugins().add_loader(std::move(loader));
+#endif
+
+    load_options(argc, argv);
+    load_plugins(argc, argv);
+}
+
+} // !namespace
+
+} // !irccd
+
+int main(int argc, char** argv)
+{
+    try {
+        irccd::load(--argc, ++argv);
+        irccd::run();
+    } catch (const std::exception& ex) {
+        std::cerr << "abort: " << ex.what() << std::endl;
+        return 1;
+    }
+}