0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-05-23 13:23:46 +02:00
construct/charybdis/console.cc

1622 lines
30 KiB
C++
Raw Normal View History

2018-02-04 03:22:01 +01:00
// Matrix Construct
//
// Copyright (C) Matrix Construct Developers, Authors & Contributors
// Copyright (C) 2016-2018 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. The
// full license for this software is available in the LICENSE file.
2017-03-21 03:24:41 +01:00
#include <ircd/ircd.h>
#include <ircd/asio.h>
2017-11-30 20:44:23 +01:00
#include <ircd/m/m.h>
#include <ircd/util/params.h>
2017-03-21 03:24:41 +01:00
#include "charybdis.h"
using namespace ircd;
2017-03-21 03:24:41 +01:00
const char *const generic_message
{R"(
2017-08-23 23:11:40 +02:00
*** - To end the console session: type ctrl-d -> EOF
*** - To exit cleanly: type exit, die, or ctrl-\ -> SIGQUIT
2017-03-21 03:24:41 +01:00
*** - To generate a coredump for developers, type ABORT -> abort()
***
)"};
const size_t stack_sz
{
2017-11-25 23:31:58 +01:00
8_MiB
};
2017-03-21 03:24:41 +01:00
bool console_active;
bool console_inwork;
2017-03-21 03:24:41 +01:00
ircd::ctx::ctx *console_ctx;
boost::asio::posix::stream_descriptor *console_in;
2017-08-23 23:11:40 +02:00
static bool handle_line(const std::string &line);
static void execute(const std::vector<std::string> lines);
2017-03-21 03:24:41 +01:00
static void console();
2017-11-25 23:31:58 +01:00
static
void check_console_active()
{
if(console_active)
2017-11-25 23:31:58 +01:00
throw ircd::error("Console is already active and cannot be reentered");
}
2017-11-25 23:31:58 +01:00
void
console_execute(const std::vector<std::string> &lines)
{
check_console_active();
ircd::context
{
"execute",
stack_sz,
std::bind(&execute, lines),
ircd::context::DETACH | ircd::context::POST
};
}
2017-03-21 03:24:41 +01:00
void
console_spawn()
{
2017-11-25 23:31:58 +01:00
check_console_active();
ircd::context
{
"console",
stack_sz,
std::bind(&console),
ircd::context::DETACH | ircd::context::POST
};
2017-03-21 03:24:41 +01:00
}
void
console_cancel()
try
2017-03-21 03:24:41 +01:00
{
if(!console_active)
return;
if(console_inwork && console_ctx)
{
interrupt(*console_ctx);
2017-03-21 03:24:41 +01:00
return;
}
2017-03-21 03:24:41 +01:00
if(console_in)
{
console_in->cancel();
console_in->close();
}
}
catch(const std::exception &e)
{
ircd::log::error("Interrupting console: %s", e.what());
2017-03-21 03:24:41 +01:00
}
void
console_hangup()
try
{
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());
}
2017-08-23 23:11:40 +02:00
ircd::m::session *moi;
2017-03-21 03:24:41 +01:00
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.
***)"};
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wframe-larger-than="
2017-03-21 03:24:41 +01:00
void
console()
try
{
if(ircd::runlevel != ircd::runlevel::RUN)
return;
2017-03-21 03:24:41 +01:00
const unwind atexit([]
2017-03-21 03:24:41 +01:00
{
console_active = false;
console_in = nullptr;
delete moi; moi = nullptr;
std::cin.clear();
2017-03-21 03:24:41 +01:00
});
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;
2017-08-23 23:11:40 +02:00
while(1)
2017-03-21 03:24:41 +01:00
{
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{}});
2017-03-21 03:24:41 +01:00
}
std::getline(is, line);
std::cin.clear();
2017-08-23 23:11:40 +02:00
if(line.empty())
continue;
if(!handle_line(line))
break;
2017-03-21 03:24:41 +01:00
}
}
catch(const std::exception &e)
{
std::cout << std::endl;
std::cout << "***" << std::endl;
std::cout << "*** The console session has ended: " << e.what() << std::endl;
std::cout << "***" << std::endl;
2017-03-21 03:24:41 +01:00
ircd::log::debug("The console session has ended: %s", e.what());
}
void
execute(const std::vector<std::string> lines)
try
{
if(ircd::runlevel != ircd::runlevel::RUN)
return;
const unwind atexit([]
{
console_active = false;
delete moi; moi = nullptr;
});
console_active = true;
console_ctx = &ctx::cur();
for(const auto &line : lines)
{
if(line.empty())
continue;
if(!handle_line(line))
break;
}
}
catch(const std::exception &e)
{
std::cout << std::endl;
std::cout << "***\n";
std::cout << "*** The execution aborted: " << e.what() << "\n";
std::cout << "***" << std::endl;
std::cout << std::flush;
std::cout.clear();
ircd::log::debug("The execution aborted: %s", e.what());
}
2017-08-23 23:11:40 +02:00
bool
2017-03-21 03:24:41 +01:00
handle_line(const std::string &line)
try
{
// _theirs is copied for recursive reentrance
const unwind outwork{[console_inwork_theirs(console_inwork)]
{
console_inwork = console_inwork_theirs;
}};
console_inwork = true;
2017-03-21 03:24:41 +01:00
if(line == "ABORT")
abort();
if(line == "EXIT")
exit(0);
2017-08-23 23:11:40 +02:00
if(line == "exit" || line == "die")
{
ircd::quit();
2017-08-23 23:11:40 +02:00
return false;
}
2017-03-21 03:24:41 +01:00
switch(hash(token(line, " ", 0)))
{
2017-08-23 23:11:40 +02:00
case hash("reload"):
2017-03-21 03:24:41 +01:00
{
2017-08-23 23:11:40 +02:00
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))
2017-03-21 03:24:41 +01:00
{
2017-08-23 23:11:40 +02:00
case hash("dbs"):
{
const auto dirs(ircd::db::available());
for(const auto &dir : dirs)
std::cout << dir << ", ";
std::cout << std::endl;
break;
}
case hash("front"):
{
for(const auto &front : ircd::m::vm::fronts.map)
for(const auto &map : front.second.map)
std::cout << std::setw(48) << std::right << front.first << " " << map.second << " " << map.first << std::endl;
break;
}
case hash("conf"):
{
const json::object &conf{ircd::conf};
for(const auto &kv : conf)
std::cout << std::setw(32) << std::right << kv.first << " " << kv.second << std::endl;
break;
}
}
if(startswith(what, '$'))
{
static char buf[65536];
const m::event event
{
m::event::id{what}, buf
};
std::cout << pretty(event) << std::endl;
2017-03-21 03:24:41 +01:00
}
2017-08-23 23:11:40 +02:00
break;
}
case hash("resolve"):
{
const auto args(tokens_after(line, " ", 0));
const params token{args, " ", {"what", "host", "port"}};
const auto &what(token.at(0));
const auto &host{token.at(1)};
const auto &port{token.at(2)};
switch(hash(what))
{
2018-01-04 22:34:07 +01:00
case hash("A"):
{
2018-01-04 22:34:07 +01:00
const net::hostport hostport
{
host, port
};
2018-01-04 22:34:07 +01:00
auto future
{
net::dns(hostport)
2018-01-04 22:34:07 +01:00
};
std::cout << future.get() << std::endl;
break;
}
2018-01-04 22:34:07 +01:00
case hash("PTR"):
{
2018-01-04 22:34:07 +01:00
const net::hostport hostport
{
host, port
};
const net::remote remote
{
2018-01-04 22:34:07 +01:00
hostport
};
2018-01-04 22:34:07 +01:00
auto future
{
net::dns(remote)
2018-01-04 22:34:07 +01:00
};
2018-01-04 22:34:07 +01:00
std::cout << future.get() << std::endl;
break;
}
}
break;
}
2017-08-23 23:11:40 +02:00
case hash("reconnect"):
{
handle_line("disconnect");
handle_line("connect");
2017-03-21 03:24:41 +01:00
break;
}
case hash("messages"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
2017-09-25 03:05:42 +02:00
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",
url::encode(room_id, room_id_encoded))
};
char query[1024];
snprintf(query, sizeof(query), "%s=%s",
"access_token",
moi->access_token.c_str());
2017-09-25 03:05:42 +02:00
m::request request
{
"GET", string_view{url, size_t(url_len)}, query, {}
};
2017-09-25 03:05:42 +02:00
static char buf[65536];
ircd::parse::buffer pb{buf};
2017-09-25 03:05:42 +02:00
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;
}
2017-10-17 09:49:33 +02:00
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",
url::encode(room_id, room_id_encoded))
2017-10-17 09:49:33 +02:00
};
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;
}
2017-09-25 03:05:42 +02:00
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"}};
2017-10-17 09:49:33 +02:00
char room_id_buf[768], event_id_buf[768];
const auto room_id{url::encode(token.at(0), room_id_buf)};
const auto event_id{url::encode(token.at(1), event_id_buf)};
2017-09-25 03:05:42 +02:00
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"}};
2017-10-17 09:49:33 +02:00
char room_id_buf[512];
const auto room_id{url::encode(token.at(0), room_id_buf)};
2017-10-17 09:49:33 +02:00
2017-09-25 03:05:42 +02:00
const auto event_type{token[1]};
2017-10-17 09:49:33 +02:00
char state_key_buf[512];
const auto state_key
{
url::encode(token[2], state_key_buf)
2017-10-17 09:49:33 +02:00
};
2017-09-25 03:05:42 +02:00
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)
};
2017-09-25 03:05:42 +02:00
m::request request
{
"GET", string_view{url, size_t(url_len)}, string_view{query, size_t(query_len)}
2017-09-25 03:05:42 +02:00
};
static char buf[65536];
ircd::parse::buffer pb{buf};
const json::array response{(*moi)(pb, request)};
for(const auto &event : response)
2017-10-17 09:49:33 +02:00
std::cout << event << std::endl << std::endl;
2017-09-25 03:05:42 +02:00
break;
}
2017-03-21 03:24:41 +01:00
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 auto &host{token.at(0, "127.0.0.1"s)};
const auto &port{token.at<uint16_t>(1, 8448)};
moi = new m::session{net::hostport{host, port}};
2017-03-21 03:24:41 +01:00
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;
}
2017-08-23 23:11:40 +02:00
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)
2017-03-21 03:24:41 +01:00
{
2017-08-23 23:11:40 +02:00
std::cerr << "No current session" << std::endl;
break;
}
const auto args(tokens_after(line, " ", 0));
const params token{args, " ", {"username", "password"}};
2017-08-23 23:11:40 +02:00
m::request request
{
"POST", "_matrix/client/r0/register?kind=user", {},
2017-03-21 03:24:41 +01:00
{
2017-08-23 23:11:40 +02:00
{ "username", token.at(0) },
{ "password", token.at(1) },
{
"auth",
{
{ "type", "m.login.dummy" }
}
}
2017-03-21 03:24:41 +01:00
}
};
2017-08-23 23:11:40 +02:00
static char buf[4096];
ircd::parse::buffer pb{buf};
2017-09-25 03:05:42 +02:00
const string_view doc((*moi)(pb, request));
2017-08-23 23:11:40 +02:00
std::cout << doc << std::endl;
2017-03-21 03:24:41 +01:00
break;
}
2017-08-23 23:11:40 +02:00
case hash("login"):
2017-03-21 03:24:41 +01:00
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
2017-08-23 23:11:40 +02:00
break;
2017-03-21 03:24:41 +01:00
}
const auto args(tokens_after(line, " ", 0));
2017-09-25 03:05:42 +02:00
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"
}
};
2017-08-23 23:11:40 +02:00
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};
2017-09-25 03:05:42 +02:00
const json::object doc((*moi)(pb, request));
2017-08-23 23:11:40 +02:00
std::cout << doc << std::endl;
moi->access_token = std::string(unquote(doc.at("access_token")));
2017-03-21 03:24:41 +01:00
break;
}
2017-08-23 23:11:40 +02:00
case hash("sync"):
2017-03-21 03:24:41 +01:00
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args
{
tokens_after(line, " ", 0)
};
2017-03-21 03:24:41 +01:00
const params token
2017-03-21 03:24:41 +01:00
{
args, " ",
2017-08-23 23:11:40 +02:00
{
"timeout", "filter_id", "full_state", "set_presence"
2017-08-23 23:11:40 +02:00
}
2017-03-21 03:24:41 +01:00
};
const time_t timeout
{
token.at(0, 0)
};
2017-03-21 03:24:41 +01:00
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);
}
2017-03-21 03:24:41 +01:00
break;
}
2017-08-23 23:11:40 +02:00
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"):
2017-03-21 03:24:41 +01:00
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
2017-08-23 23:11:40 +02:00
const auto raw(line);
m::request::quote q
{
"GET", "/foo", "{}"
};
moi->quote(q);
2017-03-21 03:24:41 +01:00
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
{
2017-11-25 23:31:58 +01:00
{ "body", text },
{ "msgtype", msgtype }
}
};
static char buf[4096];
ircd::parse::buffer pb{buf};
const json::object response{(*moi)(pb, request)};
std::cout << string_view{response} << std::endl;
break;
}
2017-08-23 23:11:40 +02:00
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
2017-08-23 23:11:40 +02:00
{
{ "new_password", token.at(0) },
{ "auth", json::members
{
{ "type", "m.login.password" }
}},
}
};
2017-08-23 23:11:40 +02:00
static char buf[4096];
ircd::parse::buffer pb{buf};
const json::object response{(*moi)(pb, request)};
std::cout << string_view{response} << std::endl;
2017-08-23 23:11:40 +02:00
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;
}
2017-09-25 03:05:42 +02:00
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)
};
2017-10-17 09:49:33 +02:00
const auto argc
{
2017-10-17 09:49:33 +02:00
token_count(args, " ")
};
2017-11-25 23:31:58 +01:00
size_t arg(0);
2017-10-17 09:49:33 +02:00
const string_view server_name
{
2017-11-25 23:31:58 +01:00
token(args, " ", arg++)
};
2017-10-17 09:49:33 +02:00
const string_view key_id
{
2017-11-25 23:31:58 +01:00
argc >= ++arg? token(args, " ", arg - 1) : ""
};
2017-11-25 23:31:58 +01:00
const string_view query_server
2017-10-17 09:49:33 +02:00
{
2017-11-25 23:31:58 +01:00
argc >= ++arg? token(args, " ", arg - 1) : ""
};
if(key_id && query_server)
{
m::keys::get(server_name, key_id, query_server, []
(const auto &keys)
{
std::cout << keys << std::endl;
});
}
else if(key_id)
{
const string_view key_id{token(args, " ", 1)};
2017-10-17 09:49:33 +02:00
m::keys::get(server_name, key_id, []
(const auto &key)
{
std::cout << key << std::endl;
2017-11-25 23:31:58 +01:00
});
}
else
{
m::keys::get(server_name, []
(const auto &keys)
{
std::cout << keys << std::endl;
});
}
break;
}
case hash("backfill"):
{
const auto args
{
2017-10-17 09:49:33 +02:00
tokens_after(line, ' ', 0)
};
2017-10-17 09:49:33 +02:00
const m::room room
{
2017-10-17 09:49:33 +02:00
m::room::id{token(args, ' ', 0)}
};
2017-10-17 09:49:33 +02:00
const m::event::id event_id
{
2017-10-17 09:49:33 +02:00
token(args, ' ', 1)
};
2017-10-17 09:49:33 +02:00
const auto limit
{
2017-10-17 09:49:33 +02:00
token_count(args, ' ') >= 2? lex_cast<uint>(token(args, ' ', 2)) : 0U
};
2017-10-25 18:47:03 +02:00
m::vm::backfill(room, event_id, limit);
break;
}
2017-11-25 23:31:58 +01:00
case hash("statefill"):
{
const auto args
{
tokens_after(line, ' ', 0)
};
const m::room::id room_id
{
token(args, ' ', 0)
};
const m::event::id event_id
{
token(args, ' ', 1)
};
m::vm::statefill(room_id, event_id);
break;
}
case hash("fedstate"):
{
const auto args
{
tokens_after(line, ' ', 0)
};
const m::room::id room_id
{
token(args, ' ', 0)
};
const m::event::id event_id
{
token(args, ' ', 1)
};
const unique_buffer<mutable_buffer> buf
{
64_MiB
};
const m::room::state state
{
room_id, event_id, buf
};
std::cout << pretty(state) << std::endl;
for_each(state, [](const auto &key, const auto &val)
{
if(bool(json::get<"event_id"_>(val)))
std::cout << pretty_oneline(val) << std::endl;
});
break;
}
2017-10-17 09:49:33 +02:00
case hash("pull"):
{
if(!moi)
{
std::cerr << "No current session" << std::endl;
break;
}
const auto args
{
tokens_after(line, " ", 0)
};
2017-10-17 09:49:33 +02:00
const auto origin_token
{
token(args, " ", 0)
};
2017-10-17 09:49:33 +02:00
char origin_buf[512];
const auto origin
{
url::encode(origin_token, origin_buf)
};
2017-10-17 09:49:33 +02:00
const auto event_id_token
{
2017-10-17 09:49:33 +02:00
token(args, " ", 0)
};
char event_id_buf[512];
const auto event_id
{
url::encode(event_id_token, event_id_buf)
};
static char url[128]; const auto url_len
{
2017-10-17 09:49:33 +02:00
fmt::snprintf(url, sizeof(url), "_matrix/federation/v1/pull/")
};
static char query[512]; const auto query_len
{
2017-10-17 09:49:33 +02:00
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};
2017-10-17 09:49:33 +02:00
const string_view response{(*moi)(pb, request)};
std::cout << response << std::endl;
break;
}
2017-10-17 09:49:33 +02:00
case hash("sjoin"):
{
2017-10-17 09:49:33 +02:00
const auto args
{
2017-10-17 09:49:33 +02:00
tokens_after(line, ' ', 0)
};
2017-10-17 09:49:33 +02:00
const m::room::id room_id
{
2017-10-17 09:49:33 +02:00
token(args, ' ', 0)
};
2017-10-17 09:49:33 +02:00
const m::user::id user_id
{
2017-10-17 09:49:33 +02:00
token(args, ' ', 1)
};
2017-11-25 23:31:58 +01:00
json::iov iov;
json::iov::push push[]
{
2017-11-25 23:31:58 +01:00
{ iov, { "sender", user_id } }
};
2017-11-25 23:31:58 +01:00
m::vm::join(room_id, iov);
2017-10-17 09:49:33 +02:00
break;
}
2017-10-25 18:47:03 +02:00
case hash("get"):
2017-10-17 09:49:33 +02:00
{
const auto args
{
tokens_after(line, ' ', 0)
};
const m::event::id event_id
{
token(args, ' ', 0)
};
static char buf[65536];
2017-11-25 23:31:58 +01:00
const m::event event
{
event_id, buf
};
std::cout << event << std::endl;
2017-10-17 09:49:33 +02:00
break;
}
2017-10-25 18:47:03 +02:00
case hash("fetch"):
{
const auto args
{
tokens_after(line, ' ', 0)
};
const m::event::id event_id
{
token(args, ' ', 0)
};
2017-11-25 23:31:58 +01:00
struct m::io::fetch::opts opts;
opts.hint = net::remote
2017-10-25 18:47:03 +02:00
{
token(args, ' ', 1)
};
static char buf[65536];
m::event::fetch fetch
{
2017-11-25 23:31:58 +01:00
event_id, buf, &opts
2017-10-25 18:47:03 +02:00
};
std::cout << m::io::acquire(fetch) << std::endl;
break;
}
case hash("mfetch"):
{
const auto args
{
tokens_after(line, ' ', 0)
};
const auto event_ids
{
tokens_after(line, ' ', 0)
};
// const net::remote remote
// {
// token(args, ' ', 0)
// };
string_view event_id[8];
const auto count
{
tokens(event_ids, ' ', event_id)
};
static char buf[8][65536];
m::event::fetch tab[count];
for(size_t i(0); i < count; ++i)
{
tab[i].event_id = event_id[i];
tab[i].buf = buf[i];
// tab[i].hint = remote;
}
m::io::acquire({tab, count});
for(size_t i(0); i < count; ++i)
{
std::cout << tab[i].event_id << ": size: " << string_view{tab[i].pdu}.size() << std::endl;
}
break;
}
2017-11-25 23:31:58 +01:00
case hash("eval"):
{
const auto args{tokens_after(line, ' ', 0)};
const params token{args, " ",
{
"event_id"
}};
m::event::id event_id
{
token.at(0)
};
if(m::vm::exists(event_id))
{
std::cout << "exists" << std::endl;
break;
}
static char buf[65536];
m::event::fetch fetch
{
event_id, buf
};
m::io::acquire(fetch);
if(bool(fetch.error))
std::rethrow_exception(fetch.error);
assert(bool(fetch.pdu));
const m::event event
{
fetch.pdu
};
m::vm::eval{event};
std::cout << pretty_oneline(event) << std::endl;
break;
}
case hash("exec"):
2017-10-25 18:47:03 +02:00
{
const auto args
{
tokens_after(line, ' ', 0)
};
2017-11-25 23:31:58 +01:00
const params token{args, " ",
2017-10-25 18:47:03 +02:00
{
2017-11-25 23:31:58 +01:00
"file path", "limit"
}};
const auto path
{
token.at(0)
2017-10-25 18:47:03 +02:00
};
2017-11-25 23:31:58 +01:00
const auto limit
{
token.at<size_t>(1)
};
const auto start
{
token[2]? lex_cast<size_t>(token[2]) : 0
};
//TODO: The file might be too large for a single read into ram for
// a big synapse server. We will need an iteration of the file
// instead.
const std::string data
{
ircd::fs::read(std::string(path))
};
std::cout << "read " << data.size() << " bytes " << std::endl;
const json::vector vector
{
data
};
struct ircd::m::vm::opts opts;
ircd::m::vm::eval eval
{
opts
};
size_t j(0), s(0);
static const auto max{1024};
for(auto it(begin(vector)); it != end(vector) && j < limit;)
{
if(s++ < start)
{
++it;
continue;
}
size_t i(0);
std::array<m::event, max> a;
for(; i < max && it != end(vector) && j < limit; ++it)
{
const json::object obj{*it};
{
a[i] = m::event{*it};
++i, ++j;
}
}
switch(eval(vector_view<const m::event>(a.data(), i)))
{
using fault = m::vm::fault;
case fault::ACCEPT:
continue;
case fault::EVENT:
std::cout << "EVENT FAULT " << eval.ef.size() << std::endl;
//for(const auto &eid : eval.ef)
// std::cout << eid << std::endl;
break;
case fault::STATE:
std::cout << "STATE FAULT " << std::endl;
break;
default:
std::cout << "FAULT " << std::endl;
break;
}
}
2017-10-25 18:47:03 +02:00
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
{
url::encode(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
{
url::encode(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;
}
2017-03-21 03:24:41 +01:00
default:
std::cerr << "Bad command or filename" << std::endl;
}
2017-08-23 23:11:40 +02:00
return true;
2017-03-21 03:24:41 +01:00
}
catch(const std::out_of_range &e)
{
std::cerr << "missing required arguments." << std::endl;
2017-08-23 23:11:40 +02:00
return true;
2017-03-21 03:24:41 +01:00
}
catch(const ircd::http::error &e)
{
2017-10-17 09:49:33 +02:00
ircd::log::error("%s %s", e.what(), e.content);
2017-08-23 23:11:40 +02:00
return true;
2017-03-21 03:24:41 +01:00
}
catch(const std::exception &e)
{
2017-10-17 09:49:33 +02:00
ircd::log::error("%s", e.what());
2017-08-23 23:11:40 +02:00
return true;
2017-03-21 03:24:41 +01:00
}
#pragma GCC diagnostic pop