Toast/TriviaBot/bot/DiscordAPI.cpp

84 lines
2.7 KiB
C++
Raw Normal View History

#include "DiscordAPI.hpp"
#include <cstdio>
#include <thread>
#include <chrono>
2016-08-15 17:47:34 +01:00
#include "http/HTTP.hpp"
#include "Logger.hpp"
using namespace std::chrono_literals;
namespace DiscordAPI {
const std::string base_url = "https://discordapp.com/api";
const std::string channels_url = base_url + "/channels";
const std::string gateway_url = base_url + "/gateway";
const std::string json_mime_type = "application/json";
2016-08-15 17:47:34 +01:00
void send_message(std::string channel_id, std::string message, std::string token, std::string ca_location) {
if (message == "") {
Logger::write("[API] [send_message] Tried to send empty message", Logger::LogLevel::Warning);
return;
}
if (message.length() > 4000) {
Logger::write("[API] [send_message] Tried to send a message over 4000 characters", Logger::LogLevel::Warning);
return;
}
else if (message.length() > 2000) {
std::cout << message.length() << std::endl;
std::string first = message.substr(0, 2000);
std::string second = message.substr(2000);
2016-08-15 17:47:34 +01:00
send_message(channel_id, first, token, ca_location);
std::this_thread::sleep_for(50ms);
2016-08-15 17:47:34 +01:00
send_message(channel_id, second, token, ca_location);
return;
}
const std::string url = channels_url + "/" + channel_id + "/messages";
json data = {
{ "content", message }
};
std::string response;
long response_code = 0;
2016-08-15 17:47:34 +01:00
response = HTTP::post_request(url, json_mime_type, data.dump(), &response_code, token, ca_location);
int retries = 0;
while (response_code != 200 && retries < 2) {
Logger::write("[API] [send_message] Got non-200 response code, retrying", Logger::LogLevel::Warning);
std::this_thread::sleep_for(100ms);
// try 3 times. usually enough to prevent 502 bad gateway issues
2016-08-15 17:47:34 +01:00
response = HTTP::post_request(url, json_mime_type, data.dump(), &response_code, token, ca_location);
retries++;
}
if (response_code != 200) {
Logger::write("[API] [send_message] Giving up on sending message", Logger::LogLevel::Warning);
}
}
2016-08-15 17:47:34 +01:00
json get_gateway(std::string ca_location) {
std::string response;
long response_code;
2016-08-15 17:47:34 +01:00
response = HTTP::get_request(gateway_url, &response_code, "", ca_location);
int retries = 0;
while (response_code != 200 && retries < 4) {
Logger::write("[API] [get_gateway] Got non-200 response code, retrying", Logger::LogLevel::Warning);
std::this_thread::sleep_for(100ms);
// try 3 times. usually enough to prevent 502 bad gateway issues
2016-08-15 17:47:34 +01:00
response = HTTP::get_request(gateway_url, &response_code, "", ca_location);
retries++;
}
if (response_code != 200) {
Logger::write("[API] [get_gateway] Giving up on getting gateway url", Logger::LogLevel::Warning);
return json {};
}
return json::parse(response);
}
}