70 lines
1.7 KiB
C++
70 lines
1.7 KiB
C++
#include <thread>
|
|
#include <chrono>
|
|
|
|
#include <curl/curl.h>
|
|
#include <include/libplatform/libplatform.h>
|
|
#include <include/v8.h>
|
|
|
|
#include "ClientConnection.hpp"
|
|
#include "Logger.hpp"
|
|
#include "DiscordAPI.hpp"
|
|
|
|
std::string bot_token;
|
|
|
|
int main(int argc, char *argv[]) {
|
|
curl_global_init(CURL_GLOBAL_DEFAULT);
|
|
|
|
v8::V8::InitializeICUDefaultLocation(argv[0]);
|
|
v8::V8::InitializeExternalStartupData(argv[0]);
|
|
v8::Platform* platform = v8::platform::CreateDefaultPlatform();
|
|
v8::V8::InitializePlatform(platform);
|
|
v8::V8::Initialize();
|
|
|
|
Logger::write("Initialised V8 and curl", Logger::LogLevel::Debug);
|
|
|
|
if (argc == 2) {
|
|
bot_token = argv[1];
|
|
}
|
|
else {
|
|
std::cout << "Please enter your bot token: " << std::endl;
|
|
std::cin >> bot_token;
|
|
}
|
|
|
|
std::string args = "/?v=5&encoding=json";
|
|
std::string url = DiscordAPI::get_gateway().value("url", "wss://gateway.discord.gg");
|
|
|
|
bool retry = true;
|
|
int exit_code = 0;
|
|
while (retry) {
|
|
retry = false;
|
|
|
|
try {
|
|
ClientConnection conn;
|
|
conn.start(url + args);
|
|
}
|
|
catch (const std::exception &e) {
|
|
Logger::write("std exception: " + std::string(e.what()), Logger::LogLevel::Severe);
|
|
exit_code = 1;
|
|
}
|
|
catch (websocketpp::lib::error_code e) {
|
|
Logger::write("websocketpp exception: " + e.message(), Logger::LogLevel::Severe);
|
|
std::this_thread::sleep_for(std::chrono::seconds(10));
|
|
retry = true; // should just be an occasional connection issue
|
|
}
|
|
catch (...) {
|
|
Logger::write("other exception.", Logger::LogLevel::Severe);
|
|
exit_code = 2;
|
|
}
|
|
}
|
|
|
|
v8::V8::Dispose();
|
|
v8::V8::ShutdownPlatform();
|
|
delete platform;
|
|
|
|
curl_global_cleanup();
|
|
|
|
Logger::write("Cleaned up", Logger::LogLevel::Info);
|
|
|
|
return exit_code;
|
|
}
|