initial commit

src/settings.cpp

@@ -0,0 +1,162 @@
+/*****************************************************************************
+* dtranslatebot Discord Translate Bot
+* Copyright (C) 2023-2024 Syping
+*
+* Redistribution and use in source and binary forms, with or without modification,
+* are permitted provided that the following conditions are met:
+*
+* 1. Redistributions of source code must retain the above copyright notice,
+* this list of conditions and the following disclaimer.
+*
+* 2. Redistributions in binary form must reproduce the above copyright notice,
+* this list of conditions and the following disclaimer in the documentation
+* and/or other materials provided with the distribution.
+*
+* This software is provided as-is, no warranties are given to you, we are not
+* responsible for anything with use of the software, you are self responsible.
+*****************************************************************************/
+
+#include <dpp/dpp.h>
+#include <iostream>
+#include "settings.h"
+
+bot::settings::channel* bot::settings::settings::get_channel(bot::settings::guild *guild, uint64_t channel_id)
+{
+    for (auto channel = guild->channel.begin(); channel != guild->channel.end(); channel++) {
+        if (channel->id == channel_id)
+            return &(*channel);
+    }
+    return nullptr;
+}
+
+bot::settings::guild* bot::settings::settings::get_guild(uint64_t guild_id)
+{
+    for (auto guild = m_guilds.begin(); guild != m_guilds.end(); guild++) {
+        if (guild->id == guild_id)
+            return &(*guild);
+    }
+    return nullptr;
+}
+
+bot::settings::translate* bot::settings::settings::get_translate()
+{
+    return &m_translate_settings;
+}
+
+const std::string bot::settings::settings::get_token()
+{
+    return m_token;
+}
+
+void bot::settings::settings::lock()
+{
+    m_mutex.lock();
+}
+
+bool bot::settings::settings::parse(const std::string &filename)
+{
+    std::ifstream ifs(filename, std::ios::in | std::ios::binary);
+    if (!ifs.is_open()) {
+        std::cerr << "Failed to open JSON configuration file located at " << filename << std::endl;
+        return false;
+    }
+
+    std::string sdata(std::istreambuf_iterator<char>{ifs}, {});
+    ifs.close();
+
+    dpp::json json = dpp::json::parse(sdata);
+    if (!json.is_object()) {
+        std::cerr << "JSON configuration file is corrupt" << std::endl;
+        return false;
+    }
+
+    auto json_token = json.find("token");
+    if (json_token == json.end()) {
+        std::cerr << "Bot token can not be found" << std::endl;
+        return false;
+    }
+
+    m_mutex.lock();
+    m_token = json_token.value();
+
+    auto json_translate = json.find("translate");
+    if (json_translate == json.end()) {
+        std::cerr << "Translate settings can not be found" << std::endl;
+        m_mutex.unlock();
+        return false;
+    }
+
+    auto json_translate_hostname = json_translate.value().find("hostname");
+    if (json_translate_hostname == json_translate.value().end()) {
+        std::cerr << "\"hostname\" can not be found in Translate settings" << std::endl;
+        m_mutex.unlock();
+        return false;
+    }
+    m_translate_settings.hostname = json_translate_hostname.value();
+
+    auto json_translate_port = json_translate.value().find("port");
+    if (json_translate_port == json_translate.value().end()) {
+        std::cerr << "\"port\" can not be found in Translate settings" << std::endl;
+        m_mutex.unlock();
+        return false;
+    }
+    m_translate_settings.port = json_translate_port.value();
+
+    auto json_translate_url = json_translate.value().find("url");
+    if (json_translate_url == json_translate.value().end()) {
+        std::cerr << "\"url\" can not be found in Translate settings" << std::endl;
+        m_mutex.unlock();
+        return false;
+    }
+    m_translate_settings.url = json_translate_url.value();
+
+    auto json_translate_tls = json_translate.value().find("tls");
+    if (json_translate_tls != json_translate.value().end())
+        m_translate_settings.tls = json_translate_tls.value();
+    else
+        m_translate_settings.tls = false;
+
+    auto json_translate_apiKey = json_translate.value().find("apiKey");
+    if (json_translate_apiKey != json_translate.value().end())
+        m_translate_settings.apiKey = json_translate_apiKey.value();
+    else
+        m_translate_settings.apiKey.clear();
+
+    auto json_guilds = json.find("guilds");
+    if (json_guilds != json.end()) {
+        for (auto json_guild = json_guilds.value().begin(); json_guild != json_guilds.value().end(); json_guild++) {
+            if (json_guild.value().is_object()) {
+                bot::settings::guild guild;
+                guild.id = std::stoull(json_guild.key());
+                for (auto json_channel = json_guild.value().begin(); json_channel != json_guild.value().end(); json_channel++) {
+                    bot::settings::channel channel;
+                    channel.id = std::stoull(json_channel.key());
+
+                    auto json_channel_source = json_channel.value().find("source");
+                    if (json_channel_source != json_channel.value().end())
+                        channel.source = json_channel_source.value();
+
+                    auto json_channel_target = json_channel.value().find("target");
+                    if (json_channel_target != json_channel.value().end())
+                        channel.target = json_channel_target.value();
+
+                    auto json_channel_webhook = json_channel.value().find("webhook");
+                    if (json_channel_webhook != json_channel.value().end())
+                        channel.webhook = json_channel_webhook.value();
+
+                    if (!channel.source.empty() && !channel.target.empty() && !channel.webhook.empty())
+                        guild.channel.push_back(channel);
+                }
+                m_guilds.push_back(guild);
+            }
+        }
+    }
+
+    m_mutex.unlock();
+    return true;
+}
+
+void bot::settings::settings::unlock()
+{
+    m_mutex.unlock();
+}