[ui] Add auto-mount on first launch functionality #52

This commit is contained in:
2025-09-06 12:07:07 -05:00
parent 30633aa4ff
commit 1904bd868d
8 changed files with 142 additions and 3 deletions

View File

@@ -426,6 +426,7 @@ inline constexpr auto JSON_MAX_CONNECTIONS{"MaxConnections"};
inline constexpr auto JSON_MAX_UPLOAD_COUNT{"MaxUploadCount"};
inline constexpr auto JSON_MED_FREQ_INTERVAL_SECS{"MedFreqIntervalSeconds"};
inline constexpr auto JSON_META{"Meta"};
inline constexpr auto JSON_MOUNT_AUTO_START{"MountAutoStart"};
inline constexpr auto JSON_MOUNT_LOCATIONS{"MountLocations"};
inline constexpr auto JSON_ONLINE_CHECK_RETRY_SECS{"OnlineCheckRetrySeconds"};
inline constexpr auto JSON_PATH{"Path"};

View File

@@ -104,6 +104,9 @@ private:
void handle_post_mount(const httplib::Request &req, httplib::Response &res);
void handle_put_mount_auto_start(const httplib::Request &req,
httplib::Response &res) const;
void handle_put_mount_location(const httplib::Request &req,
httplib::Response &res) const;

View File

@@ -42,6 +42,8 @@ private:
std::unordered_map<provider_type,
std::unordered_map<std::string, std::string>>
locations_;
std::unordered_map<provider_type, std::unordered_map<std::string, bool>>
mount_auto_start_;
mutable std::recursive_mutex mtx_;
private:
@@ -60,6 +62,9 @@ public:
[[nodiscard]] auto get_auto_start() const -> bool { return auto_start_; }
[[nodiscard]] auto get_auto_start(provider_type prov,
std::string_view name) const -> bool;
[[nodiscard]] auto get_hidden() const -> bool { return hidden_; }
[[nodiscard]] auto get_launch_only() const -> bool { return launch_only_; }
@@ -76,6 +81,9 @@ public:
void set_auto_start(bool auto_start);
void set_auto_start(provider_type prov, std::string_view name,
bool auto_start);
void set_hidden(bool hidden);
void set_launch_only(bool launch_only);

View File

@@ -255,6 +255,10 @@ handlers::handlers(mgmt_app_config *config, httplib::Server *server)
server->Post("/api/v1/mount",
[this](auto &&req, auto &&res) { handle_post_mount(req, res); });
server->Put("/api/v1/mount_auto_start", [this](auto &&req, auto &&res) {
handle_put_mount_auto_start(req, res);
});
server->Put("/api/v1/mount_location", [this](auto &&req, auto &&res) {
handle_put_mount_location(req, res);
});
@@ -382,6 +386,21 @@ void handlers::generate_config(provider_type prov, std::string_view name,
}
}
void handlers::handle_put_mount_auto_start(const httplib::Request &req,
httplib::Response &res) const {
auto prov = provider_type_from_string(req.get_param_value("type"));
auto name = req.get_param_value("name");
if (not data_directory_exists(prov, name)) {
res.status = http_error_codes::not_found;
return;
}
auto auto_start = utils::string::to_bool(req.get_param_value("auto_start"));
config_->set_auto_start(prov, name, auto_start);
res.status = http_error_codes::ok;
}
void handlers::handle_put_mount_location(const httplib::Request &req,
httplib::Response &res) const {
auto prov = provider_type_from_string(req.get_param_value("type"));
@@ -523,6 +542,8 @@ void handlers::handle_get_mount_status(const httplib::Request &req,
result.at("Location").get<std::string>());
}
result["AutoStart"] = config_->get_auto_start(prov, name);
res.set_content(result.dump(), "application/json");
res.status = http_error_codes::ok;
}

View File

@@ -29,11 +29,12 @@
#include "utils/unix.hpp"
namespace {
template <typename data_t>
[[nodiscard]] auto from_json(const nlohmann::json &json)
-> std::unordered_map<repertory::provider_type,
std::unordered_map<std::string, std::string>> {
std::unordered_map<std::string, data_t>> {
std::unordered_map<repertory::provider_type,
std::unordered_map<std::string, std::string>>
std::unordered_map<std::string, data_t>>
map_of_maps{
{repertory::provider_type::encrypt, nlohmann::json::object()},
{repertory::provider_type::remote, nlohmann::json::object()},
@@ -97,7 +98,8 @@ mgmt_app_config::mgmt_app_config(bool hidden, bool launch_only)
auto_start_ = data.contains(JSON_AUTO_START)
? data.at(JSON_AUTO_START).get<bool>()
: false;
locations_ = from_json(data.at(JSON_MOUNT_LOCATIONS));
mount_auto_start_ = from_json<bool>(data.at(JSON_MOUNT_AUTO_START));
locations_ = from_json<std::string>(data.at(JSON_MOUNT_LOCATIONS));
if (not data.contains(JSON_AUTO_START)) {
save();
@@ -119,6 +121,17 @@ mgmt_app_config::mgmt_app_config(bool hidden, bool launch_only)
}
}
auto mgmt_app_config::get_auto_start(provider_type prov,
std::string_view name) const -> bool {
recur_mutex_lock lock(mtx_);
if (mount_auto_start_.contains(prov) &&
mount_auto_start_.at(prov).contains(std::string{name})) {
return mount_auto_start_.at(prov).at(std::string{name});
}
return false;
}
auto mgmt_app_config::get_mount_location(provider_type prov,
std::string_view name) const
-> std::string {
@@ -294,6 +307,22 @@ void mgmt_app_config::set_auto_start(bool auto_start) {
save();
}
void mgmt_app_config::set_auto_start(provider_type prov, std::string_view name,
bool auto_start) {
if (name.empty()) {
return;
}
recur_mutex_lock lock(mtx_);
if (mount_auto_start_[prov][std::string{name}] == auto_start) {
return;
}
mount_auto_start_[prov][std::string{name}] = auto_start;
save();
}
auto mgmt_app_config::to_json() const -> nlohmann::json {
nlohmann::json data;
data[JSON_AUTO_START] = auto_start_;
@@ -301,6 +330,7 @@ auto mgmt_app_config::to_json() const -> nlohmann::json {
data[JSON_API_PORT] = api_port_;
data[JSON_API_USER] = api_user_;
data[JSON_MOUNT_LOCATIONS] = map_to_json(locations_);
data[JSON_MOUNT_AUTO_START] = map_to_json(mount_auto_start_);
return data;
}
} // namespace repertory::ui