1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
#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)
{
if (!std::filesystem::exists(path))
{
logger->error("Couldn't load config file! Was trying : " + path);
exit(1);
}
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);
int len = WideCharToMultiByte(CP_UTF8, 0, documents_widestr.c_str(), (int)documents_widestr.size(), nullptr, 0, nullptr, nullptr);
documents_dir = std::string(len, '\0');
WideCharToMultiByte(CP_UTF8, 0, documents_widestr.c_str(), (int)documents_widestr.size(), (LPSTR)documents_dir.data(), (int)documents_dir.size(), nullptr, nullptr);
}
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
try
{
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());
}
catch (std::exception &e)
{
logger->error("Cannot create directory for user config: %s", e.what());
return;
}
logger->info("Saving user config at " + user_cfg_path);
saveJsonFile(user_cfg_path, diff_json);
}
}
}
|
