PlayerLink/src/utils.hpp

#ifndef _UTILS_
#define _UTILS_
#include <curl/include/curl/curl.h>
#include <wx/mstream.h>
#include <wx/wx.h>

#include <filesystem>
#include <fstream>
#include <nlohmann-json/single_include/nlohmann/json.hpp>
#include <sstream>
#include <string>
#include <vector>

#define DEFAULT_CLIENT_ID "1301849203378622545"
#define DEFAULT_APP_NAME "Music"
#define CONFIG_FILENAME "settings.json"

namespace utils {
    struct App {
        bool enabled;
        std::string appName;
        std::string clientId;
        std::string searchEndpoint;
        std::vector<std::string> processNames;
    };

    inline wxIcon loadIconFromMemory(const unsigned char* data, size_t size) {
        wxMemoryInputStream stream(data, size);
        wxImage img(stream, wxBITMAP_TYPE_PNG);
        if (img.IsOk()) {
            wxBitmap bmp(img);
            wxIcon icon;
            icon.CopyFromBitmap(bmp);
            return icon;
        }
        return wxNullIcon;
    }

    inline std::string ltrim(std::string& s) {
        s.erase(s.begin(), std::find_if(s.begin(), s.end(), [](unsigned char ch) { return !std::isspace(ch); }));
        return s;
    }

    inline std::string rtrim(std::string& s) {
        s.erase(std::find_if(s.rbegin(), s.rend(), [](unsigned char ch) { return !std::isspace(ch); }).base(), s.end());
        return s;
    }

    inline std::string trim(std::string& s) {
        ltrim(s);
        rtrim(s);
        return s;
    }

    inline std::string urlEncode(std::string str) {
        std::string new_str = "";
        char c;
        int ic;
        const char* chars = str.c_str();
        char bufHex[10];
        int len = strlen(chars);

        for (int i = 0; i < len; i++) {
            c = chars[i];
            ic = c;
            if (c == ' ')
                new_str += '+';
            else if (isalnum(c) || c == '-' || c == '_' || c == '.' || c == '~')
                new_str += c;
            else {
                snprintf(bufHex, sizeof(bufHex), "%X", c);
                if (ic < 16)
                    new_str += "%0";
                else
                    new_str += "%";
                new_str += bufHex;
            }
        }
        return new_str;
    }

    inline size_t curlWriteCallback(char* contents, size_t size, size_t nmemb, void* userp) {
        ((std::string*)userp)->append((char*)contents, size * nmemb);
        return size * nmemb;
    }

    inline std::string getRequest(std::string url) {
        CURL* curl;
        CURLcode res;
        std::string buf;
        curl_global_init(CURL_GLOBAL_ALL);
        curl = curl_easy_init();
        if (curl) {
            curl_easy_setopt(curl, CURLOPT_URL, url.c_str());
            curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, curlWriteCallback);
            curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0L);
            curl_easy_setopt(curl, CURLOPT_WRITEDATA, &buf);
            res = curl_easy_perform(curl);
            curl_easy_cleanup(curl);
        }
        curl_global_cleanup();
        return buf;
    }

    inline std::string getArtworkURL(std::string query) {
        std::string response =
            getRequest("https://itunes.apple.com/search?media=music&entity=song&term=" + urlEncode(query));
        nlohmann::json j = nlohmann::json::parse(response);
        auto results = j["results"];
        if (results.size() > 0) {
            return results[0]["artworkUrl100"].get<std::string>();
        }
        return "";
    }

    inline std::vector<App> getAllApps() {
        std::vector<App> results;
        if (!std::filesystem::exists(CONFIG_FILENAME))
            return results;

        std::ifstream i(CONFIG_FILENAME);
        std::stringstream s;
        s << i.rdbuf();
        i.close();

        try {
            nlohmann::json j = nlohmann::json::parse(s.str());
            auto apps = j["apps"];
            for (auto app : apps) {
                App a;
                a.appName = app["name"].get<std::string>();
                a.clientId = app["client_id"].get<std::string>();
                a.searchEndpoint = app["search_endpoint"].get<std::string>();
                a.enabled = app["enabled"].get<bool>();
                auto processNames = app["process_names"];
                for (auto process : processNames) a.processNames.push_back(process.get<std::string>());
                results.push_back(a);
            }
        } catch (nlohmann::json::parse_error& ex) {
        }  // TODO: handle parse errors
        return results;
    }

    inline App getApp(std::string processName) {
        auto apps = getAllApps();
        for (auto app : apps) {
            for(auto procName : app.processNames) {
                if(procName == processName)
                    return app;
            }
        }
        App a;
        a.clientId = DEFAULT_CLIENT_ID;
        a.appName = DEFAULT_APP_NAME;
        a.enabled = true;
        a.searchEndpoint = "";
        return a;
    }

    inline std::string getClientID(std::string processName) {
        auto app = getApp(processName);
        return app.clientId;
    }

    inline std::string getAppName(std::string processName) {
        auto app = getApp(processName);
        return app.appName;
    }

    inline std::string getSearchEndpoint(std::string processName) {
        auto app = getApp(processName);
        return app.searchEndpoint;
    }
}  // namespace utils

#undef DEFAULT_APP_NAME
#undef CONFIG_FILENAME
#undef DEFAULT_CLIENT_ID
#endif