xbot/main.cpp

66 lines
1.6 KiB
C++
Raw Normal View History

2023-11-25 09:22:55 -08:00
#include "connection.hpp"
2023-11-22 19:59:34 -08:00
#include "settings.hpp"
2023-11-25 09:22:55 -08:00
2023-11-29 13:13:48 -08:00
#include <boost/asio.hpp>
2023-11-22 19:59:34 -08:00
#include <fstream>
#include <iostream>
#include <memory>
#include <string>
2025-01-22 23:49:48 -08:00
#include "registration_thread.hpp"
#include "self_thread.hpp"
2025-01-25 21:24:33 -08:00
#include "bot.hpp"
2023-11-22 19:59:34 -08:00
2025-01-23 12:46:52 -08:00
using namespace std::chrono_literals;
2025-01-23 00:47:05 -08:00
2025-01-25 15:45:31 -08:00
auto start(boost::asio::io_context &io, const Settings &settings) -> void
2023-11-22 19:59:34 -08:00
{
2025-01-25 15:45:31 -08:00
const auto connection = std::make_shared<Connection>(io);
2023-11-22 19:59:34 -08:00
2025-01-25 15:45:31 -08:00
const auto selfThread = SelfThread::start(*connection);
2025-01-25 12:25:38 -08:00
RegistrationThread::start(*connection, settings, selfThread);
2025-01-25 21:24:33 -08:00
Bot::start(selfThread);
2025-01-22 23:49:48 -08:00
2025-01-24 14:48:15 -08:00
connection->sig_snote.connect([](auto &match) {
std::cout << "SNOTE " << static_cast<int>(match.get_tag()) << std::endl;
2025-01-22 23:49:48 -08:00
for (auto c : match.get_results())
2023-11-28 11:34:27 -08:00
{
std::cout << " " << std::string_view{c.first, c.second} << std::endl;
}
});
2023-11-22 19:59:34 -08:00
boost::asio::co_spawn(
2025-01-25 15:45:31 -08:00
io, connection->connect(io, settings.host, settings.service),
[&io, &settings](std::exception_ptr e) {
2023-11-25 09:22:55 -08:00
auto timer = std::make_shared<boost::asio::steady_timer>(io);
2025-01-22 23:49:48 -08:00
2025-01-22 20:33:17 -08:00
timer->expires_after(5s);
2025-01-25 15:45:31 -08:00
timer->async_wait(
[&io, &settings, timer](auto) { start(io, settings); }
);
}
);
2023-11-22 19:59:34 -08:00
}
auto get_settings() -> Settings
{
2025-01-25 15:45:31 -08:00
if (auto config_stream = std::ifstream{"config.toml"})
2023-11-22 19:59:34 -08:00
{
return Settings::from_stream(config_stream);
}
else
{
std::cerr << "Unable to open config.toml\n";
std::exit(1);
}
}
auto main() -> int
{
2025-01-25 15:45:31 -08:00
const auto settings = get_settings();
2023-11-22 19:59:34 -08:00
auto io = boost::asio::io_context{};
start(io, settings);
io.run();
}