discordlistforbots/controllers/views.cc

#include <string>
#include <unordered_map>
#include <fstream>
#include <sstream>
#include <iomanip>
#include <memory>
#include <exception>
#include <functional>
#include <fmt/core.h>
#include <drogon/drogon.h>
#include "../config.h"
#include "views.h"

#define authenticate(cb) cb(HttpResponse::newRedirectionResponse(OAUTH_URL)); return
#define toStartPage(cb) cb(HttpResponse::newRedirectionResponse("/")); return
#define voteID(uid, bid) std::to_string(uid)+'-'+std::to_string(bid)
#define cantVote(vid, ...) {auto _cantvoteres = last_votes.find(vid); auto now = trantor::Date::date(); if (_cantvoteres != last_votes.end() and now < _cantvoteres->second.after(43200)) __VA_ARGS__}



Bot deserializeBot(orm::Row row) {
    Bot thisbot;
#   define sf(f) thisbot.f = row[#f]
    sf(name).as<std::string>();
    sf(short_description).as<std::string>();
    sf(long_description).as<std::string>();
    sf(avatar_url).as<std::string>();
    sf(owner).as<std::string>();
    sf(support_server).as<std::string>();
    sf(prefix).as<std::string>();
    sf(votes).as<uint32_t>();
    sf(approved).as<bool>();
    thisbot.owner_id = std::stoul(row["owner_id"].as<std::string>());
    thisbot.app_id = std::stoul(row["app_id"].as<std::string>());
    return thisbot;
}

std::string dbEsc(const std::string& src) {
    std::ostringstream fres;
    for (const auto &character : src) {
        if (character == '\'') {
            fres << "''";
        } else {
            fres << character;
        }
    }
    return fres.str();
}

auto errPage(const std::exception& e) {
    HttpViewData data;
    data.insert("message", e.what());

    auto resp = HttpResponse::newHttpViewResponse("exception.csp", data);
    resp->setStatusCode(HttpStatusCode::k500InternalServerError);
    return resp;
}

SessionDataPtr getSessionData(SessionPtr session) {
    return session->getOptional<SessionDataPtr>("data").value_or(nullptr);
}
void setSessionData(SessionPtr session, SessionDataPtr data) {
    session->insert("data", data);
}
void resetSessionData(SessionPtr session) {
    session->erase("data");
}

HttpViewData HttpViewDataPrep(SessionDataPtr sessionData) {
    HttpViewData fres;
    fres.insert("sessionData", sessionData);
    return fres;
}
HttpViewData HttpViewDataPrep(SessionPtr session) {
    return HttpViewDataPrep(getSessionData(session));
}
HttpViewData HttpViewDataPrep(const HttpRequestPtr& request) {
    return HttpViewDataPrep(request->session());
}

void getUser(uint64_t user_id, std::function<void (const Json::Value&)> callback) {
    auto discordapi = HttpClient::newHttpClient("https://discord.com");
    auto req = HttpRequest::newHttpRequest();
    req->setPath("/api/v8/users/"+std::to_string(user_id));
    req->setMethod(HttpMethod::Get);
    req->addHeader("Authorization", "Bot " BOT_TOKEN);
    discordapi->sendRequest(req, [callback, user_id] (ReqResult, const HttpResponsePtr &response) {
        if (response->getStatusCode() == HttpStatusCode::k200OK) {
            auto &json = *response->getJsonObject().get();
            auto avatar_hash = json["avatar"].asString();
            if (avatar_hash.empty()) {
                json["avatar_url"] = "https://cdn.discordapp.com/embed/avatars/"+std::to_string(std::stoi(json["discriminator"].asString()) % 5)+".png";
            } else {
                json["avatar_url"] = "https://cdn.discordapp.com/avatars/"+std::to_string(user_id)+"/"+avatar_hash+".png";
            }
            callback(json);
        } else {
            callback({});
        }
    });
}

auto dbErr = [](const orm::DrogonDbException &) {};

std::string SessionData::discord_fullname() {
    return discord_username+'#'+discord_discriminator;
}

void LoginFilter::doFilter(const HttpRequestPtr &req, FilterCallback &&fcb, FilterChainCallback &&fccb) {
    if (getSessionData(req->session())) {
        fccb();
    } else {
        authenticate(fcb);
    }
}

views::views() {
    db = drogon::app().getDbClient();
    moderators = MODERATORS;
}

void views::start(
        const HttpRequestPtr&, std::function<void (const HttpResponsePtr &)> &&callback
        )
{
    callback(HttpResponse::newRedirectionResponse("/bots/@all", HttpStatusCode::k301MovedPermanently));
}

void views::menu(
        const HttpRequestPtr& req, std::function<void (const HttpResponsePtr &)> &&callback
        )
{
    auto data = HttpViewDataPrep(req);
    data.insert("ref", req->getHeader("Referer"));

    callback(HttpResponse::newHttpViewResponse("menu.csp", data));
}

void views::botlist(
        const HttpRequestPtr& req, std::function<void (const HttpResponsePtr &)> &&callback
        )
{
    auto sessionData = getSessionData(req->session());
    auto justMine = req->getPath()=="/bots/@me";
    auto modView = req->getPath()=="/bots/@unapproved";

    std::string q = "SELECT * FROM bots WHERE ";
    if (justMine) {
        q.append("owner_id = '"+std::to_string(sessionData->discord_id)+"'");
    } else {
        q.append("approved = "+std::string(modView?"false":"true"));
    }
    q.append(" ORDER BY votes");

    db->execSqlAsync(q,
                     [justMine, sessionData, callback] (const orm::Result &rows) {
        std::map<uint64_t, Bot> bot_list;
        for (const auto& r : rows) {
            Bot bot = deserializeBot(r);
            bot_list[bot.app_id] = bot;
        }

        auto data = HttpViewDataPrep(sessionData);
        data.insert("modView", false);
        data.insert("justMine", justMine);
        data.insert("authed", bool(sessionData));
        data.insert("bots", bot_list);

        callback(HttpResponse::newHttpViewResponse("botlist.csp", data));
    }, dbErr);
}

void views::botdetail(
        const HttpRequestPtr& req, std::function<void (const HttpResponsePtr &)> &&callback,
        uint64_t bot_id)
{
    db->execSqlAsync("SELECT * FROM bots WHERE app_id = '"+std::to_string(bot_id)+"'",
                     [this, req, callback, bot_id] (const orm::Result &rows) {
        if (rows.empty()) {
            // Bot not found
            callback(HttpResponse::newNotFoundResponse());
        } else {
            // Bot found
            auto sessionData = getSessionData(req->session());
            auto bot = deserializeBot(rows[0]);
            auto data = HttpViewDataPrep(sessionData);
            data.insert("bot", bot);
            if (not sessionData) {
                // Vote button is not grezed out if user isn"t logged in
                data.insert("canVote", bot.approved);
            } else cantVote(voteID(sessionData->discord_id, bot_id), {
                data.insert("canVote", false);
            } else {
                data.insert("canVote", bot.approved);
            })

            callback(HttpResponse::newHttpViewResponse("botdetail.csp", data));
        }
    }, dbErr);
}

void views::botvote(const HttpRequestPtr& req, std::function<void (const HttpResponsePtr &)> &&callback,
             uint64_t bot_id) {
    // Check if user is able to vote again
    auto user_id = getSessionData(req->session())->discord_id;
    auto vote_id = voteID(user_id, bot_id);
    cantVote(vote_id, {
        callback(HttpResponse::newRedirectionResponse("detail"));
        return;
    })
    // Register vote
    db->execSqlAsync("UPDATE bots SET votes = votes + 1 WHERE app_id = '"+std::to_string(bot_id)+"'",
                     [this, vote_id, callback] (const orm::Result &rows) {
        if (rows.affectedRows() == 0) {
            // Bot not found
            callback(HttpResponse::newNotFoundResponse());
        } else {
            last_votes[vote_id] = trantor::Date::date();
            // Redirect back
            callback(HttpResponse::newRedirectionResponse("detail"));
        }
    }, [callback] (const orm::DrogonDbException &) {
        callback(HttpResponse::newNotFoundResponse());
    });
}

void views::botedit(const HttpRequestPtr& req, std::function<void (const HttpResponsePtr &)> &&callback,
             uint64_t bot_id, const std::string& action) {
    auto sessionData = getSessionData(req->session());
    db->execSqlAsync("SELECT * FROM bots WHERE app_id = '"+std::to_string(bot_id)+"'",
                     [sessionData, callback, action] (const orm::Result &rows) {
        if (rows.empty()) {
            callback(HttpResponse::newNotFoundResponse());
        } else {
            auto bot = deserializeBot(rows[0]);
            // Check if user is botowner or moderator
            if (sessionData->discord_id == bot.owner_id or sessionData->moderator) {
                // Get action
                if (action == "edit") {
                    // TODO
                } else if (action == "delete") {
                    // TODO
                } else if (not sessionData->moderator) {
                    goto else_part;
                } else if (action == "approve") {
                    // TODO
                } else if (action == "decline") {
                    // TODO
                } else {
                    else_part:
                    callback(HttpResponse::newNotFoundResponse());
                }
            }
        }
    }, dbErr);
}

void views::botregister_view(const HttpRequestPtr& req, std::function<void (const HttpResponsePtr &)> &&callback,
                             const std::string& error) {
    // Display page
    auto data = HttpViewDataPrep(req);
    data.insert("error", error);

    callback(HttpResponse::newHttpViewResponse("botregister.csp", data));
}
void views::botregister_submit(const HttpRequestPtr& req, std::function<void (const HttpResponsePtr &)> &&callback
                               ) {
    auto session = req->session();
    // Set error handler
    auto onError = [callback] (const std::string& e) {
       callback(HttpResponse::newRedirectionResponse("register?error="+e));
    };
    // Get and check parameters
    string short_description, long_description, support_server, prefix;
    uint64_t app_id;
    try {
        app_id = std::stoul(req->getParameter("app_id"));
        short_description = HttpViewData::htmlTranslate(req->getParameter("short_description"));
        long_description = HttpViewData::htmlTranslate(req->getParameter("long_description"));
        support_server = HttpViewData::htmlTranslate(req->getParameter("support_server"));
        prefix = HttpViewData::htmlTranslate(req->getParameter("prefix"));
    } catch (std::exception& e) {
        onError(e.what());
    }
    // Check if bot already exists
    db->execSqlAsync("select 1 from bots where app_id ='"+std::to_string(app_id)+"'",
                     [=] (const orm::Result &r) {
        if (not r.empty()) {
            onError("Bot%20has%20already%20been%20registered");
            return;
        }
        // Get bots avatar
        getUser(app_id, [=] (const Json::Value& botuser) {
            // Check result
            if (botuser.empty() or not botuser["bot"].asBool()) {
                onError("Invalid%20client%20ID");
                return;
            }
            // Get session data
            auto sessionData = getSessionData(session);
            // Perform database operation
            db->execSqlAsync(fmt::format("INSERT INTO bots (name, short_description, long_description, avatar_url, owner, support_server, prefix, owner_id, app_id, votes, approved) "
                                         "VALUES('{}', '{}', '{}', '{}', '{}', '{}', '{}', '{}', '{}', 0, 'f')",
                             dbEsc(botuser["username"].asString()), dbEsc(short_description), dbEsc(long_description), dbEsc(botuser["avatar_url"].asString()), sessionData->discord_fullname(), dbEsc(support_server), dbEsc(prefix), sessionData->discord_id, app_id),
                             [app_id, callback] (const orm::Result &) {
                callback(HttpResponse::newRedirectionResponse(std::to_string(app_id)+"/detail"));
            }, dbErr);
        });
    }, dbErr);
}

void views::discorddeauth(
        const HttpRequestPtr& req, std::function<void (const HttpResponsePtr &)> &&callback
        ) {
    req->session()->clear();
    toStartPage(callback);
}

void views::discordauth(
        const HttpRequestPtr& client_req, std::function<void (const HttpResponsePtr &)> &&callback,
        const std::string& code)
{
    if (code.empty()) {
        authenticate(callback);
    } else {
        // Get token from API
        auto discordapi = HttpClient::newHttpClient("https://discord.com");
        auto req = HttpRequest::newHttpRequest();
        req->setPath("/api/v8/oauth2/token");
        req->setContentTypeCode(ContentType::CT_APPLICATION_X_FORM);
        req->setMethod(HttpMethod::Post);
        {
            req->setParameter("client_id", CLIENT_ID);
            req->setParameter("client_secret", CLIENT_SECRET);
            req->setParameter("redirect_uri", REDIRECT_URI);
            req->setParameter("grant_type", "authorization_code");
            req->setParameter("scope", "identify");
            req->setParameter("code", code);
        }
        discordapi->sendRequest(req, [this, discordapi, client_req, callback] (ReqResult, const HttpResponsePtr &response) {
            // Check for success
            if (response->getStatusCode() == HttpStatusCode::k200OK) {
                // Auth success
                auto &data = *response->getJsonObject().get();
                // Prepare session
                auto session = client_req->session();
                resetSessionData(session);
                auto sessionData = std::make_shared<SessionData>();
                sessionData->discord_access_token = data["access_token"].asString();
                // Get user data
                auto req = HttpRequest::newHttpRequest();
                req->setPath("/api/v8/users/@me");
                req->setMethod(HttpMethod::Get);
                req->addHeader("Authorization", "Bearer "+sessionData->discord_access_token);
                discordapi->sendRequest(req, [this, client_req, callback, session, sessionData] (ReqResult, const HttpResponsePtr &response) {
                    if (response->getStatusCode() == HttpStatusCode::k200OK) {
                        // Getting user data success
                        auto &userdata = *response->getJsonObject().get();
                        sessionData->discord_id = std::stoul(userdata["id"].asString());
                        sessionData->discord_username = userdata["username"].asString();
                        sessionData->discord_discriminator = userdata["discriminator"].asString();
                        sessionData->moderator = std::find(moderators.begin(), moderators.end(), sessionData->discord_id) != moderators.end();
                        setSessionData(session, sessionData);
                        // Show success page
                        auto data = HttpViewDataPrep(sessionData);
                        callback(HttpResponse::newHttpViewResponse("authsuccess.csp", data));
                    } else {
                        toStartPage(callback);
                    }
                });
            } else {
                toStartPage(callback);
            }
        });
    }
}