diff options
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/spotify/spotify.cpp | 185 |
1 files changed, 166 insertions, 19 deletions
diff --git a/modules/spotify/spotify.cpp b/modules/spotify/spotify.cpp index 93b66ac..8f44f2e 100644 --- a/modules/spotify/spotify.cpp +++ b/modules/spotify/spotify.cpp @@ -16,19 +16,29 @@ * ===================================================================================== */ #include <pcre.h> +#include <regex> #include <stdlib.h> #include "cpr/cpr.h" #include <142bot/modules.hpp> +#include <142bot/util.hpp> +#include <142bot/db.hpp> #include <fmt/format.h> +#include <chrono> + +using json = nlohmann::json; class SpotifyModule: public Module { std::string spotifyRegex; + std::string defaultSpotifyAccount; + std::string spotifyBaseUrl; public: SpotifyModule(Bot* creator, ModuleLoader* ml) : Module(creator, ml) { ml->attach({I_OnMessage, I_OnCommand}, this); this->spotifyRegex = "^https:\/\/open.spotify.com\/track\/([a-zA-Z0-9]+)(.*)$"; + this->defaultSpotifyAccount = "1"; + this->spotifyBaseUrl = "https://api.spotify.com/v1"; } virtual std::string version() { @@ -39,37 +49,174 @@ public: return "Manage spotify queues for 142"; } + void refreshSpotify(std::string refreshToken, pqxx::work &tx) { + bot->core->log(dpp::ll_debug, "Attempting to refresh spotify token..."); + cpr::Response r = cpr::Post(cpr::Url("https://accounts.spotify.com/api/token"), + cpr::Authentication{bot->cfg["spotify"]["id"], bot->cfg["spotify"]["secret"], cpr::AuthMode::BASIC}, cpr::Payload{{"grant_type", "refresh_token"}, {"refresh_token", refreshToken}}); + + bot->core->log(dpp::ll_trace, "Made request"); + + if (r.status_code != 200) { + bot->core->log(dpp::ll_error, r.text); + throw std::exception(); + } + bot->core->log(dpp::ll_trace, "Request successful"); + printf("%s\n", r.text.c_str()); + auto tmp = json::parse(r.text); + bot->core->log(dpp::ll_trace, "Parsed JSON"); + + uint64_t expires = tmp["expires_in"].get<uint64_t>(); + + asdf::timestamp parsed_expires = asdf::from_unix_time(time(0) + expires); + bot->core->log(dpp::ll_trace, "Got expires_in"); + std::string access = tmp["access_token"].get<std::string>(); + bot->core->log(dpp::ll_trace, "Got access token"); + tx.exec_params("UPDATE spotify SET spotify_token=$1, spotify_token_expires=$2 WHERE id=$3", access, parsed_expires, this->defaultSpotifyAccount); + bot->core->log(dpp::ll_trace, "Updated DB"); + tx.commit(); + bot->core->log(dpp::ll_debug, "Done refreshing spotify token"); + } + + // Obtains the (refreshed) token for the default spotify account + std::string get_spotify_token() { + bot->core->log(dpp::ll_debug, "Attempting to retrieve spotify token."); + std::string token; + pqxx::work tx(bot->conn); + try { + bot->core->log(dpp::ll_debug, fmt::format("Default spotify account: {}", this->defaultSpotifyAccount)); + auto res = tx.exec_params1("SELECT spotify_username,spotify_token,spotify_token_expires,spotify_refresh_token FROM spotify WHERE id=$1", atoi(this->defaultSpotifyAccount.c_str())); + bot->core->log(dpp::ll_trace, "Retrieved from DB."); + + auto ts = res[2].as<asdf::timestamp>(); + + if (ts < std::chrono::system_clock::now()) { + refreshSpotify(res[3].as<std::string>(), tx); + res = tx.exec_params1("SELECT spotify_username, spotify_token FROM spotify WHERE id=$1", atoi(this->defaultSpotifyAccount.c_str())); + bot->core->log(dpp::ll_trace, "Retrieved from database *again*"); + } + + token = res[1].as<std::string>(); + tx.commit(); + } catch (std::exception &e) { + std::string error_msg = "Error getting spotify token: " + *e.what(); + bot->core->log(dpp::ll_error, e.what()); + sentry_value_t event = sentry_value_new_event(); + sentry_value_t exc = sentry_value_new_exception("Exception", "Spotify SQL error"); + sentry_value_set_stacktrace(exc, NULL, 5); + sentry_event_add_exception(event, exc); + sentry_capture_event(event); + tx.abort(); + return nullptr; + } + bot->core->log(dpp::ll_debug, "Done retrieving spotify token."); + + return token; + } + + /** + * Performs a GET request to the Spotify API + */ + json spotify_get(const std::string route) { + bot->core->log(dpp::ll_debug, fmt::format("Making GET request to {}", route)); + + std::string token = get_spotify_token(); + bot->core->log(dpp::ll_trace, "Obtained token."); + + cpr::Response r = cpr::Get(cpr::Url{fmt::format("{}/{}", this->spotifyBaseUrl, route).c_str()}, + cpr::Bearer{token}); + + bot->core->log(dpp::ll_trace, fmt::format("Made request. Code: {}", r.status_code)); + + if (r.status_code != 200) { + bot->core->log(dpp::ll_error, fmt::format("Spotify API Error: {}", r.text)); + throw std::exception(); + } + + return json::parse(r.text); + } + virtual bool OnMessage(const dpp::message_create_t &message, const std::string& clean_message, bool mentioned, const std::vector<std::string> & mentions) { sentry_set_tag("module", "spotify"); bot->core->log(dpp::ll_debug, "Got message event"); - const char* pcre_error; - int pcre_error_ofs; + std::regex re("^https:\/\/open.spotify.com\/track\/([a-zA-Z0-9]+)(.*)$"); - auto comp = pcre_compile("^https:\/\/open.spotify.com\/track\/([a-zA-Z0-9]+)(.*)$", PCRE_CASELESS, &pcre_error, &pcre_error_ofs, NULL); - if (!comp) { - bot->core->log(dpp::ll_error, pcre_error); - } + std::smatch match; + + if (std::regex_search(clean_message, match, re) == true) { + try { + json res = spotify_get("tracks/" + match.str(1)); - int matcharr[90]; - int matchcount = pcre_exec(comp, NULL, clean_message.c_str(), clean_message.length(), 0, 0, matcharr, 90); + dpp::embed embed = dpp::embed() + .set_title(res["name"]) + .set_author(res["artists"][0]["name"], "", "") + .set_image(res["album"]["images"][0]["url"]); - if (matchcount > 0) { - // We found a spotify URL! - EmbedSimple("Found a spotify URL", message.msg.channel_id); - } else { - if (clean_message.starts_with(bot->prefix)) { - + bot->core->message_create(dpp::message(message.msg.channel_id, embed).set_reference(message.msg.id)); + return true; + } catch (std::exception &e) { + EmbedError(message.msg.channel_id, e); + return false; } - } + } return true; } - virtual bool OnCommand(const dpp::message_create_t &message, const std::string &command, std::vector<std::string>& params) { - bot->core->log(dpp::ll_debug, fmt::format("spotify Got command: {}", command)); - printf("Am i getting called?????\n"); - EmbedSimple("Got your command!", message.msg.channel_id); + bool OnCommand(const dpp::message_create_t &message, const std::string &command, const std::vector<std::string>& params) { + + if (command == "spotify") { + // We can process subcommands! + + if (params.size() <= 1) { + return true; // we don't have enough to process + } + + std::string subcommand = lowercase(params[1]); + + if (subcommand == "accounts") { + pqxx::work tx(bot->conn); + + auto res = tx.exec("SELECT id, spotify_username FROM spotify"); + tx.commit(); + dpp::embed embed = dpp::embed(). + set_color(dpp::colors::green) + .set_title("Spotify Accounts") + .set_description("List of Spotify accounts. Username is field ID and ID is field content.\n\nCurrent default account ID: " + this->defaultSpotifyAccount); + + for (int i = 0; i < res.size(); i++) { + embed.add_field(res[i][1].c_str(), res[i][0].c_str(), true); + } + + + bot->core->message_create(dpp::message(message.msg.channel_id, embed).set_reference(message.msg.id)); + return true; + } else if (subcommand == "default") { + pqxx::work tx(bot->conn); + + try { + auto res = tx.exec_params1("SELECT id FROM spotify WHERE id=$1", params[2]); + + this->defaultSpotifyAccount = params[2]; + + tx.commit(); + } catch (std::exception &e) { + bot->core->log(dpp::ll_error, e.what()); + sentry_value_t event = sentry_value_new_event(); + sentry_value_t exc = sentry_value_new_exception("Exception", "Spotify SQL error"); + sentry_value_set_stacktrace(exc, NULL, 5); + sentry_event_add_exception(event, exc); + sentry_capture_event(event); + tx.abort(); + EmbedError("Invalid ID", message.msg.channel_id); + return false; + } + + EmbedSuccess("Updated default account.", message.msg.channel_id); + } else { + EmbedError("Unknown Command", message.msg.channel_id); + } + } return true; } |