mergerfs ui

#include "mergerfs_webui.hpp" #include "fs_exists.hpp" #include "fs_mounts.hpp" #include "mergerfs_api.hpp" #include "str.hpp" #include "CLI11.hpp" #include "httplib.h" #include #include static void _get_root(const httplib::Request &req_, httplib::Response &res_) { std::string html; if(fs::exists("index.html")) { res_.set_file_content("index.html"); return; } html = R"html( mergerfs ui

Select Mountpoint:

)html"; res_.set_content(html, "text/html"); } #define IERT(S) if(type_ == (S)) return true; static bool _valid_fs_type(const std::string &type_) { IERT("ext2"); IERT("ext3"); IERT("ext4"); IERT("xfs"); IERT("jfs"); IERT("btrfs"); IERT("zfs"); IERT("reiserfs"); IERT("f2fs"); IERT("ntfs"); IERT("vfat"); IERT("exfat"); if(str::startswith(type_,"fuse.")) return true; return false; } static void _get_mounts(const httplib::Request &req_, httplib::Response &res_) { json json_array; std::string type; fs::MountVec mounts; fs::mounts(mounts); json_array = json::array(); for(const auto &mount : mounts) { if(not ::_valid_fs_type(mount.type)) continue; json obj; obj["path"] = mount.dir; obj["type"] = mount.type; json_array.push_back(obj); } res_.set_content(json_array.dump(), "application/json"); } static void _get_mounts_mergerfs(const httplib::Request &req_, httplib::Response &res_) { json json_array; std::string type; fs::MountVec mounts; fs::mounts(mounts); json_array = json::array(); for(const auto &mount : mounts) { if(mount.type != "fuse.mergerfs") continue; json_array.push_back(mount.dir); } res_.set_content(json_array.dump(), "application/json"); } static void _get_kvs(const httplib::Request &req_, httplib::Response &res_) { if(not req_.has_param("mount")) { res_.status = 400; res_.set_content("mount param not set", "text/plain"); return; } std::string mount; std::map kvs; bool first = true; mount = req_.get_param_value("mount"); mergerfs::api::get_kvs(mount,&kvs); for(const auto &[key, value] : kvs) { if(!first) response += ","; response += "\"" + key + "\":\"" + value + "\""; first = false; } response += "}"; res_.set_content(response, "application/json"); } static void _get_kvs_key(const httplib::Request &req_, httplib::Response &res_) { if(not req_.has_param("mount")) { res_.status = 400; res_.set_content("mount param not set", "text/plain"); return; } fs::path mount; std::string key; std::string val; key = req_.path_params.at("key"); mount = req_.get_param_value("mount"); mergerfs::api::get_kv(mount,key,&val); res_.set_content(val, "text/plain"); } // New endpoints for enhanced UI static void _get_branches(const httplib::Request &req_, httplib::Response &res_) { if(not req_.has_param("mount")) { res_.status = 400; res_.set_content("{\"error\":\"mount param not set\"}", "application/json"); return; } fs::path mount; std::string val; mount = req_.get_param_value("mount"); if(mergerfs::api::get_kv(mount,"branches",&val) >= 0) { res_.set_content(val, "text/plain"); } else { res_.set_content("", "text/plain"); } } static void _post_branches(const httplib::Request &req_, httplib::Response &res_) { if(not req_.has_param("mount")) { res_.status = 400; res_.set_content("{\"error\":\"mount param not set\"}", "application/json"); return; } fs::path mount; std::string val; mount = req_.get_param_value("mount"); val = req_.body; int rv = mergerfs::api::set_kv(mount,"branches",val); if(rv >= 0) { res_.set_content("{\"result\":\"success\"}", "application/json"); } else { res_.status = 400; res_.set_content("{\"result\":\"error\",\"message\":\"Failed to set branches\"}", "application/json"); } } static void _get_xattr(const httplib::Request &req_, httplib::Response &res_) { if(not req_.has_param("mount")) { res_.status = 400; res_.set_content("{\"error\":\"mount param not set\"}", "application/json"); return; } fs::path mount; std::string key; std::string val; mount = req_.get_param_value("mount"); key = req_.path_params.at("key"); int rv = mergerfs::api::get_kv(mount,key,&val); if(rv >= 0) { res_.set_content("\"" + val + "\"", "application/json"); } else { res_.status = 404; res_.set_content("{\"error\":\"xattr not found\"}", "application/json"); } } static void _post_xattr(const httplib::Request &req_, httplib::Response &res_) { if(not req_.has_param("mount")) { res_.status = 400; res_.set_content("{\"error\":\"mount param not set\"}", "application/json"); return; } fs::path mount; std::string key; std::string val; mount = req_.get_param_value("mount"); key = req_.path_params.at("key"); val = req_.body; int rv = mergerfs::api::set_kv(mount,key,val); if(rv >= 0) { res_.set_content("{\"result\":\"success\"}", "application/json"); } else { res_.status = 400; res_.set_content("{\"result\":\"error\",\"message\":\"Failed to set xattr\"}", "application/json"); } } static void _execute_command(const httplib::Request &req_, httplib::Response &res_) { if(not req_.has_param("mount")) { res_.status = 400; res_.set_content("{\"error\":\"mount param not set\"}", "application/json"); return; } fs::path mount; std::string cmd; mount = req_.get_param_value("mount"); cmd = req_.path_params.at("command"); int rv = mergerfs::api::set_kv(mount,"cmd." + cmd,""); if(rv >= 0) { res_.set_content("{\"result\":\"success\",\"message\":\"Command executed\"}", "application/json"); } else { res_.status = 400; res_.set_content("{\"result\":\"error\",\"message\":\"Failed to execute command\"}", "application/json"); } } static void _post_kvs_key(const httplib::Request &req_, httplib::Response &res_) { if(not req_.has_param("mount")) { res_.status = 400; res_.set_content("{\"error\":\"mount param not set\"}", "application/json"); return; } fs::path mount; std::string key; std::string val; key = req_.path_params.at("key"); val = req_.body; mount = req_.get_param_value("mount"); int rv = mergerfs::api::set_kv(mount,key,val); if(rv >= 0) { res_.set_content("{\"result\":\"success\"}", "application/json"); } else { res_.status = 400; res_.set_content("{\"result\":\"error\",\"message\":\"Failed to set configuration\"}", "application/json"); } } int mergerfs::webui::main(const int argc_, char **argv_) { CLI::App app; std::string host; int port; std::string password; app.description("mergerfs.webui:" " A simple webui to configure mergerfs instances"); app.name("USAGE: mergerfs.webui"); app.add_option("--host",host) ->description("Interface to bind to") ->default_val("0.0.0.0"); app.add_option("--port",port) ->description("TCP port to use") ->default_val(8080); app.add_option("--password",password) ->description(""); try { app.parse(argc_,argv_); } catch(const CLI::ParseError &e) { return app.exit(e); } // TODO: Warn if uid of server is not root but mergerfs is. httplib::Server http_server; http_server.Get("/",::_get_root); http_server.Get("/mounts",::_get_mounts); http_server.Get("/mounts/mergerfs",::_get_mounts_mergerfs); http_server.Get("/kvs",::_get_kvs); http_server.Get("/kvs/:key",::_get_kvs_key); http_server.Post("/kvs/:key",::_post_kvs_key); // New endpoints for enhanced UI http_server.Get("/branches",::_get_branches); http_server.Post("/branches",::_post_branches); http_server.Get("/xattr/:key",::_get_xattr); http_server.Post("/xattr/:key",::_post_xattr); http_server.Post("/cmd/:command",::_execute_command); std::cout << "host:port = http://" << host << ":" << port << std::endl; http_server.listen(host,port); return 0; }