0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-11-29 10:12:39 +01:00
construct/charybdis/console.cc
2017-10-17 00:49:33 -07:00

1222 lines
24 KiB
C++

/*
* Copyright (C) 2016 Charybdis Development Team
* Copyright (C) 2016 Jason Volk <jason@zemos.net>
*
* 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 is present in all copies.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR oPROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
* STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
* IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
#include <ircd/ircd.h>
#include <ircd/asio.h>
#include <ircd/m.h>
#include "charybdis.h"
const char *const generic_message
{R"(
*** - To end the console session: type ctrl-d -> EOF
*** - To exit cleanly: type exit, die, or ctrl-\ -> SIGQUIT
*** - To generate a coredump for developers, type ABORT -> abort()
***
)"};
bool console_active;
ircd::ctx::ctx *console_ctx;
boost::asio::posix::stream_descriptor *console_in;
static bool handle_line(const std::string &line);
static void console();
void
console_spawn()
{
if(console_active)
return;
// The console function is executed asynchronously.
// The DETACH indicates it will clean itself up.
ircd::context("console", std::bind(&console), ircd::context::DETACH);
}
void
console_cancel()
try
{
if(!console_active)
return;
if(!console_in)
return;
console_in->cancel();
console_in->close();
}
catch(const std::exception &e)
{
ircd::log::error("Interrupting console: %s", e.what());
}
void
console_hangup()
try
{
using namespace ircd;
using log::console_quiet;
console_cancel();
static console_quiet *quieted;
if(!quieted)
{
log::notice("Suppressing console log output after terminal hangup");
quieted = new console_quiet;
return;
}
log::notice("Reactivating console logging after second hangup");
delete quieted;
quieted = nullptr;
}
catch(const std::exception &e)
{
ircd::log::error("console_hangup(): %s", e.what());
}
const char *const termstop_message
{R"(
***
*** The server has been paused and will resume when you hit enter.
*** This is a client and your commands will originate from the server itself.
***)"};
void
console_termstop()
try
{
console_cancel();
std::cout << termstop_message << generic_message;
std::string line;
std::cout << "\n> " << std::flush;
std::getline(std::cin, line);
if(std::cin.eof())
{
std::cout << std::endl;
std::cin.clear();
return;
}
handle_line(line);
}
catch(const std::exception &e)
{
ircd::log::error("console_termstop(): %s", e.what());
}
ircd::m::session *moi;
const char *const console_message
{R"(
***
*** The server is still running in the background. A command line is now available below.
*** This is a client and your commands will originate from the server itself.
***)"};
void
console()
try
{
using namespace ircd;
const unwind atexit([]
{
console_active = false;
console_in = nullptr;
delete moi; moi = nullptr;
});
console_active = true;
console_ctx = &ctx::cur();
std::cout << console_message << generic_message;
boost::asio::posix::stream_descriptor in{*::ios, dup(STDIN_FILENO)};
console_in = &in;
boost::asio::streambuf buf{BUFSIZE};
std::istream is{&buf};
std::string line;
while(1)
{
std::cout << "\n> " << std::flush;
// Suppression scope ends after the command is entered
// so the output of the command (if log messages) can be seen.
{
const log::console_quiet quiet(false);
boost::asio::async_read_until(in, buf, '\n', yield_context{to_asio{}});
}
std::getline(is, line);
if(line.empty())
continue;
if(!handle_line(line))
break;
}
}
catch(const std::exception &e)
{
std::cout << std::endl;
std::cout << "***\n";
std::cout << "*** The console session has ended: " << e.what() << "\n";
std::cout << "***" << std::endl;
ircd::log::debug("The console session has ended: %s", e.what());
return;
}
bool
handle_line(const std::string &line)
try
{
using namespace ircd;
if(line == "ABORT")
abort();
if(line == "EXIT")
exit(0);
if(line == "exit" || line == "die")
{
ircd::stop();
return false;
}
switch(hash(token(line, " ", 0)))
{
case hash("reload"):
{
module matrix("matrix");
auto &root(matrix.get<module *>("root_module"));
root->reset();
*root = module("root");
break;
}
case hash("show"):
{
const auto args(tokens_after(line, " ", 0));
const params token{args, " ", {"what"}};
const std::string what(token.at(0));
switch(hash(what))
{
case hash("dbs"):
{
const auto dirs(ircd::db::available());
for(const auto &dir : dirs)
std::cout << dir << ", ";
std::cout << std::endl;
break;
}
}
break;
}
case hash("reconnect"):
{
handle_line("disconnect");
handle_line("connect");
break;
}
case hash("messages"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args(tokens_after(line, " ", 0));
const params token{args, " ", {"room_id"}};
const string_view &room_id
{
token.at(0, ircd::m::my_room.room_id)
};
char room_id_encoded[512];
char url[512]; const auto url_len
{
fmt::sprintf(url, "_matrix/client/r0/rooms/%s/messages",
urlencode(room_id, room_id_encoded))
};
char query[1024];
snprintf(query, sizeof(query), "%s=%s",
"access_token",
moi->access_token.c_str());
m::request request
{
"GET", string_view{url, size_t(url_len)}, query, {}
};
static char buf[65536];
ircd::parse::buffer pb{buf};
const json::object response{(*moi)(pb, request)};
const json::array chunk(response["chunk"]);
for(const string_view &chunk : chunk)
std::cout << chunk << std::endl;
break;
}
case hash("members"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args(tokens_after(line, " ", 0));
const params token{args, " ", {"room_id"}};
const string_view &room_id
{
token.at(0, ircd::m::my_room.room_id)
};
char room_id_encoded[512];
char url[512]; const auto url_len
{
fmt::sprintf(url, "_matrix/client/r0/rooms/%s/members",
urlencode(room_id, room_id_encoded))
};
char query[1024];
snprintf(query, sizeof(query), "%s=%s",
"access_token",
moi->access_token.c_str());
m::request request
{
"GET", string_view{url, size_t(url_len)}, query, {}
};
static char buf[65536];
ircd::parse::buffer pb{buf};
const json::object response{(*moi)(pb, request)};
const json::array chunk(response["chunk"]);
for(const string_view &chunk : chunk)
std::cout << chunk << std::endl;
break;
}
case hash("context"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args(tokens_after(line, " ", 0));
const params token{args, " ", {"room_id", "event_id"}};
char room_id_buf[768], event_id_buf[768];
const auto room_id{urlencode(token.at(0), room_id_buf)};
const auto event_id{urlencode(token.at(1), event_id_buf)};
char url[512]; const auto url_len
{
fmt::snprintf(url, sizeof(url), "_matrix/client/r0/rooms/%s/context/%s",
room_id,
event_id)
};
m::request request
{
"GET", string_view{url, size_t(url_len)}, {}
};
static char buf[65536];
ircd::parse::buffer pb{buf};
const json::object response{(*moi)(pb, request)};
for(const auto &member : response)
std::cout << member.first << " " << member.second << std::endl;
break;
}
case hash("state"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args(tokens_after(line, " ", 0));
const params token{args, " ", {"room_id", "event_type", "state_key"}};
char room_id_buf[512];
const auto room_id{urlencode(token.at(0), room_id_buf)};
const auto event_type{token[1]};
char state_key_buf[512];
const auto state_key
{
urlencode(token[2], state_key_buf)
};
char url[512]; const auto url_len
{
event_type && state_key?
fmt::snprintf(url, sizeof(url), "_matrix/client/r0/rooms/%s/state/%s/%s",
room_id,
event_type,
state_key):
event_type?
fmt::snprintf(url, sizeof(url), "_matrix/client/r0/rooms/%s/state/%s",
room_id,
event_type):
fmt::snprintf(url, sizeof(url), "_matrix/client/r0/rooms/%s/state",
room_id)
};
static char query[512]; const auto query_len
{
fmt::snprintf(query, sizeof(query), "%s=%s",
"access_token",
moi->access_token)
};
m::request request
{
"GET", string_view{url, size_t(url_len)}, string_view{query, size_t(query_len)}
};
static char buf[65536];
ircd::parse::buffer pb{buf};
const json::array response{(*moi)(pb, request)};
for(const auto &event : response)
std::cout << event << std::endl << std::endl;
break;
}
case hash("connect"):
{
if(moi)
{
std::cerr << "Already have session." << std::endl;
break;
}
const auto args(tokens_after(line, " ", 0));
const params token{args, " ", {"host", "port"}};
const std::string host{token.at(0, "127.0.0.1"s)};
const auto port{token.at<uint16_t>(1, 8448)};
moi = new m::session{{host, port}};
break;
}
case hash("disconnect"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
delete moi; moi = nullptr;
break;
}
case hash("versions"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
m::request request
{
"GET", "_matrix/client/versions"
};
static char buf[1024];
ircd::parse::buffer pb{buf};
const auto doc((*moi)(pb, request));
std::cout << doc << std::endl;
break;
}
case hash("register"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args(tokens_after(line, " ", 0));
const params token{args, " ", {"username", "password"}};
m::request request
{
"POST", "_matrix/client/r0/register?kind=user", {},
{
{ "username", token.at(0) },
{ "password", token.at(1) },
{
"auth",
{
{ "type", "m.login.dummy" }
}
}
}
};
static char buf[4096];
ircd::parse::buffer pb{buf};
const string_view doc((*moi)(pb, request));
std::cout << doc << std::endl;
break;
}
case hash("login"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args(tokens_after(line, " ", 0));
if(args.empty())
{
m::request request
{
"GET", "_matrix/client/r0/login", {}, {}
};
static char buf[4096];
ircd::parse::buffer pb{buf};
const json::object doc((*moi)(pb, request));
const json::array flows(doc.at("flows"));
size_t i(0);
for(const auto &flow : flows)
std::cout << i++ << ": " << flow << std::endl;
break;
}
const params token
{
args, " ",
{
"username", "password"
}
};
m::request request
{
"POST", "_matrix/client/r0/login", {},
{
{ "user", token.at(0) },
{ "password", token.at(1) },
{ "type", "m.login.password" },
}
};
static char buf[4096];
ircd::parse::buffer pb{buf};
const json::object doc((*moi)(pb, request));
std::cout << doc << std::endl;
moi->access_token = std::string(unquote(doc.at("access_token")));
break;
}
case hash("sync"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args
{
tokens_after(line, " ", 0)
};
const params token
{
args, " ",
{
"timeout", "filter_id", "full_state", "set_presence"
}
};
const time_t timeout
{
token.at(0, 0)
};
static char query[2048];
snprintf(query, sizeof(query), "%s=%s&timeout=%zd",
"access_token",
moi->access_token.c_str(),
timeout * 1000);
while(1)
{
m::request request
{
"GET", "_matrix/client/r0/sync", query,
{
}
};
static char buf[8192];
ircd::parse::buffer pb{buf};
const json::object doc((*moi)(pb, request));
const auto since(doc.at("next_batch"));
for(const auto &member : doc)
std::cout << string_view{member.first} << " => " << string_view{member.second} << std::endl;
fmt::snprintf(query, sizeof(query), "%s=%s&since=%s&timeout=%zd",
"access_token",
moi->access_token,
since,
timeout * 1000);
}
break;
}
case hash("createroom"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args(tokens_after(line, " ", 0));
const params token{args, " ", {"name"}};
char query[1024];
snprintf(query, sizeof(query), "%s=%s",
"access_token",
moi->access_token.c_str());
m::request request
{
"POST", "_matrix/client/r0/createRoom", query,
{
{ "name", token.at(0) },
}
};
static char buf[4096];
ircd::parse::buffer pb{buf};
const auto doc((*moi)(pb, request));
std::cout << doc << std::endl;
break;
}
/*
case hash("GET"):
case hash("POST"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto raw(line);
m::request::quote q
{
"GET", "/foo", "{}"
};
moi->quote(q);
break;
}
*/
case hash("privmsg"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
static uint txnid;
const auto args(tokens_after(line, " ", 0));
const params token{args, " ", {"room_id", "msgtype"}};
const auto &room_id{token.at(0)};
const auto &msgtype{token.at(1)};
const auto &event_type{"m.room.message"};
const auto text(tokens_after(line, " ", 2));
static char query[512]; const auto query_len
{
fmt::snprintf(query, sizeof(query), "%s=%s",
"access_token",
moi->access_token)
};
static char url[512]; const auto url_len
{
fmt::snprintf(url, sizeof(url), "_matrix/client/r0/rooms/%s/send/%s/%u",
room_id,
event_type,
txnid++)
};
m::request request
{
"PUT", url, query, json::members
{
{ "msgtype", msgtype },
{ "body", text }
}
};
static char buf[4096];
ircd::parse::buffer pb{buf};
const json::object response{(*moi)(pb, request)};
std::cout << string_view{response} << std::endl;
break;
}
case hash("password"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args(tokens_after(line, " ", 0));
const params token{args, " ", {"new_password"}};
static char query[512]; const auto query_len
{
fmt::snprintf(query, sizeof(query), "%s=%s",
"access_token",
moi->access_token)
};
m::request request
{
"POST", "_matrix/client/r0/account/password", query, json::members
{
{ "new_password", token.at(0) },
{ "auth", json::members
{
{ "type", "m.login.password" }
}},
}
};
static char buf[4096];
ircd::parse::buffer pb{buf};
const json::object response{(*moi)(pb, request)};
std::cout << string_view{response} << std::endl;
break;
}
case hash("deactivate"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args(tokens_after(line, " ", 0));
static char query[512]; const auto query_len
{
fmt::snprintf(query, sizeof(query), "%s=%s",
"access_token",
moi->access_token)
};
m::request request
{
"POST", "_matrix/client/r0/account/deactivate", query, json::members
{
{ "auth", json::members
{
{ "type", "m.login.password" }
}},
}
};
static char buf[4096];
ircd::parse::buffer pb{buf};
const json::object response{(*moi)(pb, request)};
std::cout << string_view{response} << std::endl;
break;
}
case hash("setfilter"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args
{
tokens_after(line, " ", 0)
};
const auto user_id
{
token(args, " ", 0)
};
const json::object filter
{
tokens_after(args, " ", 0)
};
static char url[128]; const auto url_len
{
fmt::snprintf(url, sizeof(url), "_matrix/client/r0/user/%s/filter", user_id)
};
static char query[512]; const auto query_len
{
fmt::snprintf(query, sizeof(query), "%s=%s",
"access_token",
moi->access_token)
};
m::request request
{
"POST", url, query, filter
};
static char buf[4096];
ircd::parse::buffer pb{buf};
const json::object response{(*moi)(pb, request)};
std::cout << string_view{response} << std::endl;
break;
}
case hash("getfilter"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args
{
tokens_after(line, " ", 0)
};
const auto user_id
{
token(args, " ", 0)
};
const auto filter_id
{
tokens_after(args, " ", 0)
};
static char url[128]; const auto url_len
{
fmt::snprintf(url, sizeof(url), "_matrix/client/r0/user/%s/filter/%s",
user_id,
filter_id)
};
static char query[512]; const auto query_len
{
fmt::snprintf(query, sizeof(query), "%s=%s",
"access_token",
moi->access_token)
};
m::request request
{
"GET", url, query, {}
};
static char buf[4096];
ircd::parse::buffer pb{buf};
const json::object response{(*moi)(pb, request)};
std::cout << string_view{response} << std::endl;
break;
}
case hash("keys"):
{
const auto args
{
tokens_after(line, " ", 0)
};
const auto argc
{
token_count(args, " ")
};
const string_view server_name
{
argc >= 1? token(args, " ", 0) : ""
};
const string_view key_id
{
argc >= 2? token(args, " ", 1) : ""
};
const auto got
{
m::keys::get(server_name, key_id, []
(const auto &key)
{
std::cout << key.verify() << std::endl;
std::cout << key << std::endl;
})
};
if(!got)
std::cerr << "failed" << std::endl;
break;
}
case hash("backfill"):
{
const auto args
{
tokens_after(line, ' ', 0)
};
const m::room room
{
m::room::id{token(args, ' ', 0)}
};
const m::event::id event_id
{
token(args, ' ', 1)
};
const auto limit
{
token_count(args, ' ') >= 2? lex_cast<uint>(token(args, ' ', 2)) : 0U
};
m::backfill(room, event_id, limit);
break;
}
case hash("pull"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args
{
tokens_after(line, " ", 0)
};
const auto origin_token
{
token(args, " ", 0)
};
char origin_buf[512];
const auto origin
{
urlencode(origin_token, origin_buf)
};
const auto event_id_token
{
token(args, " ", 0)
};
char event_id_buf[512];
const auto event_id
{
urlencode(event_id_token, event_id_buf)
};
static char url[128]; const auto url_len
{
fmt::snprintf(url, sizeof(url), "_matrix/federation/v1/pull/")
};
static char query[512]; const auto query_len
{
fmt::snprintf(query, sizeof(query), "origin=%s&v=%s",
origin,
event_id)
};
m::request request
{
"GET", url, string_view{query, size_t(query_len)}, {}
};
static char buf[65536];
ircd::parse::buffer pb{buf};
const string_view response{(*moi)(pb, request)};
std::cout << response << std::endl;
break;
}
case hash("sjoin"):
{
const auto args
{
tokens_after(line, ' ', 0)
};
const m::room::id room_id
{
token(args, ' ', 0)
};
const m::user::id user_id
{
token(args, ' ', 1)
};
m::room room
{
room_id
};
json::iov content{};
room.join(user_id, content);
break;
}
case hash("fedstate"):
{
const auto args
{
tokens_after(line, ' ', 0)
};
const m::room room
{
m::room::id{token(args, ' ', 0)}
};
const m::event::id event_id
{
token(args, ' ', 1)
};
m::state(room, event_id);
break;
}
case hash("fetch"):
{
const auto args
{
tokens_after(line, ' ', 0)
};
const m::event::id event_id
{
token(args, ' ', 0)
};
static char buf[65536];
std::cout << m::get(event_id, buf) << std::endl;
break;
}
case hash("trace"):
{
const auto args
{
tokens_after(line, ' ', 0)
};
m::event::id event_id
{
token(args, ' ', 0)
};
m::trace(event_id, []
(const auto &event, auto &next)
{
std::cout << event << std::endl << std::endl;
return true;
});
break;
}
case hash("directory"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args
{
tokens_after(line, " ", 0)
};
const auto query_type
{
"directory"
};
const auto room_alias_token
{
token(args, " ", 0)
};
char room_alias_buf[512];
const auto room_alias
{
urlencode(room_alias_token, room_alias_buf)
};
static char url[128]; const auto url_len
{
fmt::snprintf(url, sizeof(url), "_matrix/federation/v1/query/%s",
query_type)
};
static char query[512]; const auto query_len
{
fmt::snprintf(query, sizeof(query), "%s=%s",
"room_alias",
room_alias)
};
m::request request
{
"GET", url, string_view{query, size_t(query_len)}, {}
};
static char buf[65536];
ircd::parse::buffer pb{buf};
const json::object response{(*moi)(pb, request)};
std::cout << response << std::endl;
break;
}
case hash("profile"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args
{
tokens_after(line, " ", 0)
};
const auto query_type
{
"profile"
};
const auto user_id_token
{
token(args, " ", 0)
};
char user_id_buf[512];
const auto user_id
{
urlencode(user_id_token, user_id_buf)
};
static char url[128]; const auto url_len
{
fmt::snprintf(url, sizeof(url), "_matrix/federation/v1/query/%s",
query_type)
};
static char query[512]; const auto query_len
{
fmt::snprintf(query, sizeof(query), "%s=%s",
"user_id",
user_id)
};
m::request request
{
"GET", url, string_view{query, size_t(query_len)}, {}
};
static char buf[65536];
ircd::parse::buffer pb{buf};
const json::object response{(*moi)(pb, request)};
std::cout << response << std::endl;
break;
}
default:
std::cerr << "Bad command or filename" << std::endl;
}
return true;
}
catch(const std::out_of_range &e)
{
std::cerr << "missing required arguments." << std::endl;
return true;
}
catch(const ircd::http::error &e)
{
ircd::log::error("%s %s", e.what(), e.content);
return true;
}
catch(const std::exception &e)
{
ircd::log::error("%s", e.what());
return true;
}