satdump/src-core/core/config.cpp

#define SATDUMP_DLL_EXPORT 1
#include "config.h"
#include "nlohmann/json_utils.h"
#include "logger.h"
#include <filesystem>

#if defined (__APPLE__)
#include <sysdir.h>
#include <glob.h>
#elif defined (_WIN32)
#include <direct.h>
#include <Shlobj.h>
#endif

namespace satdump
{
    namespace config
    {
        nlohmann::ordered_json master_cfg;
        nlohmann::ordered_json main_cfg;

        std::vector<PluginConfigHandler> plugin_config_handlers;

        std::string user_cfg_path;

        void loadConfig(std::string path, std::string user_path)
        {
            logger->info("Loading config " + path);
            master_cfg = loadJsonFile(path);
            main_cfg = master_cfg;

            loadUserConfig(user_path);
        }

        void checkOutputDirs()
        {
            std::string documents_dir = "";
#if defined (__APPLE__)
            //Always overwrite . on macOS since inside the app bundle can be writable
            char documents_glob[PATH_MAX];
            glob_t globbuf;
            sysdir_search_path_enumeration_state search_state = sysdir_start_search_path_enumeration(SYSDIR_DIRECTORY_DOCUMENT, SYSDIR_DOMAIN_MASK_USER);
            sysdir_get_next_search_path_enumeration(search_state, documents_glob);
            glob(documents_glob, GLOB_TILDE, nullptr, &globbuf);
            documents_dir = std::string(globbuf.gl_pathv[0]);
            globfree(&globbuf);

#elif defined (_WIN32)
            //Only set output to Documents if the current dir is not writable
            HANDLE test_handle = CreateFile(".", GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_BACKUP_SEMANTICS, NULL);
            if (test_handle == INVALID_HANDLE_VALUE)
            {
                PWSTR documents_wide;
                SHGetKnownFolderPath(FOLDERID_Documents, KF_FLAG_DEFAULT, NULL, &documents_wide);
                std::wstring documents_widestr = std::wstring(documents_wide);
                documents_dir = std::string(documents_widestr.begin(), documents_widestr.end());
            }
            else
                CloseHandle(test_handle);
#endif
            if (documents_dir == "")
                return;

            bool bad_path = false;
            if (config::main_cfg["satdump_directories"]["recording_path"]["value"].get<std::string>() == ".")
            {
                bad_path = true;
                config::main_cfg["satdump_directories"]["recording_path"]["value"] = documents_dir;
            }
            if (config::main_cfg["satdump_directories"]["live_processing_path"]["value"].get<std::string>() == "./live_output")
            {
                bad_path = true;
                config::main_cfg["satdump_directories"]["live_processing_path"]["value"] = documents_dir + "/live_output";
            }
            if (config::main_cfg["satdump_directories"]["default_input_directory"]["value"].get<std::string>() == ".")
            {
                bad_path = true;
                config::main_cfg["satdump_directories"]["default_input_directory"]["value"] = documents_dir;
            }
            if (config::main_cfg["satdump_directories"]["default_output_directory"]["value"].get<std::string>() == ".")
            {
                bad_path = true;
                config::main_cfg["satdump_directories"]["default_output_directory"]["value"] = documents_dir;
            }
            if (config::main_cfg["satdump_directories"]["default_image_output_directory"]["value"].get<std::string>() == ".")
            {
                bad_path = true;
                config::main_cfg["satdump_directories"]["default_image_output_directory"]["value"] = documents_dir;
            }
            if (config::main_cfg["satdump_directories"]["default_projection_output_directory"]["value"].get<std::string>() == ".")
            {
                bad_path = true;
                config::main_cfg["satdump_directories"]["default_projection_output_directory"]["value"] = documents_dir;
            }
            if (bad_path)
                saveUserConfig();
        }

        void loadUserConfig(std::string user_path)
        {
            std::string final_path = "";

            if (std::filesystem::exists("settings.json")) // First try loading in current folder
                final_path = "settings.json";
            else if (std::filesystem::exists(user_path + "/settings.json"))
                final_path = user_path + "/settings.json";

            if (final_path != "")
            {
                logger->info("Loading user settings " + final_path);
                nlohmann::ordered_json diff_json = loadJsonFile(final_path);
                main_cfg = merge_json_diffs(master_cfg, diff_json);
                user_cfg_path = final_path;
            }
            else
            {
                logger->warn("No user configuration found! Keeping defaults.");
                user_cfg_path = user_path + "/settings.json";
            }

            checkOutputDirs();
        }

        void saveUserConfig()
        {
            nlohmann::ordered_json diff_json = perform_json_diff(master_cfg, main_cfg); // Get diff between user settings and the master CFG

            if (!std::filesystem::exists(std::filesystem::path(user_cfg_path).parent_path()) && std::filesystem::path(user_cfg_path).has_parent_path())
                std::filesystem::create_directories(std::filesystem::path(user_cfg_path).parent_path());

            logger->info("Saving user config at " + user_cfg_path);
            saveJsonFile(user_cfg_path, diff_json);
        }
    }
}