206 lines
7.2 KiB
C++
206 lines
7.2 KiB
C++
#include "command_registry.hpp"
|
|
#include "config.hpp"
|
|
#include "utils/utils.hpp"
|
|
#include "utils/directories.hpp"
|
|
#include "shared_commands.hpp"
|
|
|
|
#include <unistd.h>
|
|
#include <cstring>
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <filesystem>
|
|
#include <libassert/assert.hpp>
|
|
|
|
namespace dropshell {
|
|
|
|
int edit_handler(const CommandContext& ctx);
|
|
|
|
static std::vector<std::string> edit_name_list={"edit"};
|
|
|
|
// Static registration
|
|
struct EditCommandRegister {
|
|
EditCommandRegister() {
|
|
CommandRegistry::instance().register_command({
|
|
edit_name_list,
|
|
edit_handler,
|
|
shared_commands::std_autocomplete,
|
|
false, // hidden
|
|
false, // requires_config
|
|
false, // requires_install
|
|
0, // min_args (after command)
|
|
2, // max_args (after command)
|
|
"edit [SERVER] [SERVICE]",
|
|
"Edit dropshell, server or service configuration",
|
|
// heredoc
|
|
R"(
|
|
Edit dropshell, server or service configuration.
|
|
edit edit the dropshell config.
|
|
edit <server> edit the server config.
|
|
edit <server> <service> edit the service config.
|
|
)"
|
|
});
|
|
}
|
|
} edit_command_register;
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
// edit command implementation
|
|
// ------------------------------------------------------------------------------------------------
|
|
|
|
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
// utility function to edit a file
|
|
// ------------------------------------------------------------------------------------------------
|
|
bool edit_file(const std::string &file_path, bool has_bb64)
|
|
{
|
|
// make sure parent directory exists.
|
|
std::string parent_dir = get_parent(file_path);
|
|
std::filesystem::create_directories(parent_dir);
|
|
|
|
std::string editor_cmd;
|
|
const char* editor_env = std::getenv("EDITOR");
|
|
|
|
if (editor_env && std::strlen(editor_env) > 0) {
|
|
editor_cmd = std::string(editor_env) + " " + quote(file_path);
|
|
} else if (isatty(STDIN_FILENO)) {
|
|
// Check if stdin is connected to a terminal if EDITOR is not set
|
|
editor_cmd = "nano -w " + quote(file_path);
|
|
} else {
|
|
error << "Standard input is not a terminal and EDITOR environment variable is not set." << std::endl;
|
|
info << "Try setting the EDITOR environment variable (e.g., export EDITOR=nano) or run in an interactive terminal." << std::endl;
|
|
info << "You can manually edit the file at: " << file_path << std::endl;
|
|
return false;
|
|
}
|
|
|
|
info << "Editing file: " << file_path << std::endl;
|
|
|
|
if (has_bb64) {
|
|
return execute_local_command("", editor_cmd, {}, nullptr, cMode::Interactive);
|
|
}
|
|
else {
|
|
// might not have bb64 at this early stage. Direct edit.
|
|
int ret = system(editor_cmd.c_str());
|
|
return EXITSTATUSCHECK(ret);
|
|
}
|
|
}
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
// edit config
|
|
// ------------------------------------------------------------------------------------------------
|
|
int edit_config()
|
|
{
|
|
if (!gConfig().is_config_set())
|
|
gConfig().save_config(false); // save defaults.
|
|
|
|
std::string config_file = localfile::dropshell_json();
|
|
if (!edit_file(config_file, false) || !std::filesystem::exists(config_file))
|
|
return die("Failed to edit config file.");
|
|
|
|
gConfig().load_config();
|
|
if (!gConfig().is_config_set())
|
|
return die("Failed to load and parse edited config file!");
|
|
|
|
gConfig().save_config(true);
|
|
|
|
std::cout << "Successfully edited config file at " << config_file << std::endl;
|
|
return 0;
|
|
}
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
// edit server
|
|
// ------------------------------------------------------------------------------------------------
|
|
int edit_server(const std::string &server_name)
|
|
{
|
|
if (localpath::server(server_name).empty()) {
|
|
error << "Server not found: " << server_name << std::endl;
|
|
return -1;
|
|
}
|
|
|
|
std::string config_file = localfile::server_json(server_name);
|
|
|
|
if (!edit_file(config_file, true)) {
|
|
error << "Failed to edit server config" << std::endl;
|
|
info << "You can manually edit this file at: " << config_file << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
info << "If you have changed DROPSHELL_DIR, you should manually move the files to the new location NOW." << std::endl;
|
|
info << "You can ssh in to the remote server with: dropshell ssh "<<server_name<< std::endl;
|
|
info << "Once moved, reinstall all services with: dropshell install " << server_name << std::endl;
|
|
return 0;
|
|
}
|
|
|
|
void list_directory(std::string dir, std::string msg)
|
|
{
|
|
bool first=true;
|
|
std::vector<std::string> directories;
|
|
for (const auto &file : std::filesystem::directory_iterator(dir))
|
|
{
|
|
if (first)
|
|
{
|
|
if (!msg.empty())
|
|
info << msg << std::endl;
|
|
first=false;
|
|
}
|
|
if (std::filesystem::is_directory(file.path()))
|
|
directories.push_back(file.path());
|
|
else
|
|
info << " " << file.path() << std::endl;
|
|
}
|
|
|
|
for (const auto &dir : directories)
|
|
list_directory(dir, "");
|
|
}
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
// edit service config
|
|
// ------------------------------------------------------------------------------------------------
|
|
int edit_service_config(const std::string &server, const std::string &service)
|
|
{
|
|
std::string config_file = localfile::service_env(server, service);
|
|
if (!std::filesystem::exists(config_file))
|
|
{
|
|
error << "Service config file not found: " << config_file << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
if (edit_file(config_file, true) && std::filesystem::exists(config_file))
|
|
info << "Successfully edited service config file at " << config_file << std::endl;
|
|
|
|
std::string service_dir = localpath::service(server, service);
|
|
list_directory(service_dir, "You may wish to edit the other files in " + service_dir);
|
|
|
|
info << "Then to apply your changes, run:" << std::endl;
|
|
info << " dropshell uninstall " + server + " " + service << std::endl;
|
|
info << " dropshell install " + server + " " + service << std::endl;
|
|
return 0;
|
|
}
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
// edit command handler
|
|
// ------------------------------------------------------------------------------------------------
|
|
int edit_handler(const CommandContext& ctx) {
|
|
// edit dropshell config
|
|
if (ctx.args.size() < 1)
|
|
return edit_config();
|
|
|
|
// edit server config
|
|
if (ctx.args.size() < 2) {
|
|
edit_server(safearg(ctx.args,0));
|
|
return 0;
|
|
}
|
|
|
|
// edit service config
|
|
if (ctx.args.size() < 3) {
|
|
edit_service_config(safearg(ctx.args,0), safearg(ctx.args,1));
|
|
return 0;
|
|
}
|
|
|
|
info << "Edit handler called with " << ctx.args.size() << " args\n";
|
|
return -1;
|
|
}
|
|
|
|
|
|
|
|
|
|
} // namespace dropshell
|