everest-framework/src/controller/command_api.cpp at 2b17dfb3f9990912ad4386228cd6df04d2e8ffb2 · EVerest/everest-framework · GitHub

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
// SPDX-License-Identifier: Apache-2.0
// Copyright 2020 - 2024 Pionix GmbH and Contributors to EVerest
#include "command_api.hpp"

#include <filesystem>
#include <fstream>

#include <fmt/core.h>

#include <ryml.hpp>
#include <ryml_std.hpp>

#include "rpc.hpp"
#include "transpile_config.hpp"

#include <utils/formatter.hpp>
#include <utils/yaml_loader.hpp>

using json = nlohmann::json;
namespace fs = std::filesystem;

CommandApi::CommandApi(const Config& config, RPC& rpc) : config(config), rpc(rpc) {
}

nlohmann::json CommandApi::handle(const std::string& cmd, const json& params) {
    // fmt::print("Handling command {}\n", cmd);

    if (cmd == "get_modules") {
        auto modules_list = json::object();

        for (const auto& item : fs::directory_iterator(this->config.module_dir)) {
            if (!fs::is_directory(item)) {
                continue;
            }
            const auto module_path = item.path();
            const auto module_name = module_path.filename().string();

            // fetch the manifest
            const auto manifest_path = module_path / "manifest.yaml";
            if (!fs::is_regular_file(manifest_path)) {
                continue;
            }

            modules_list[module_name] = Everest::load_yaml(manifest_path);
        }

        return modules_list;
    } else if (cmd == "get_configs") {
        auto config_list = json::object();

        for (const auto& item : fs::directory_iterator(this->config.configs_dir)) {
            if (!fs::is_regular_file(item)) {
                continue;
            }
            if (item.path().extension() != std::string(".yaml")) {
                continue;
            }

            const auto config_name = item.path().stem().string();

            config_list[config_name] = Everest::load_yaml(item.path());
        }

        return config_list;
    } else if (cmd == "get_interfaces") {
        auto interface_list = json::object();

        for (const auto& item : fs::directory_iterator(this->config.interface_dir)) {

            if (!fs::is_regular_file(item)) {
                continue;
            }
            if (item.path().extension() != std::string(".yaml")) {
                continue;
            }

            const auto interface_name = item.path().stem().string();

            interface_list[interface_name] = Everest::load_yaml(item.path());
        }

        return interface_list;
    } else if (cmd == "save_config") {
        // FIXME (aw): this is quite hacky
        if (!params.contains("name") || !params.at("name").is_string()) {
            throw CommandApiParamsError("The save_config needs a 'name' parameter for the config file of type string");
        }

        const auto name = params.at("name").get<std::string>();

        const json config_json = params.value("config", json::object());
        auto ryml_deserialized = transpile_config(config_json);

        const auto configs_path = fs::path(this->config.configs_dir);
        const auto check_config_file_path = configs_path / fmt::format("_{}.yaml", name);

        std::ofstream(check_config_file_path.string()) << ryml_deserialized;

        const auto result = this->rpc.ipc_request("check_config", check_config_file_path.string(), false);

        if (result.is_string()) {
            fs::remove(check_config_file_path);
            throw CommandApiParamsError(result);
        }

        fs::rename(check_config_file_path, configs_path / fmt::format("{}.yaml", name));

        return true;
    } else if (cmd == "restart_modules") {
        this->rpc.ipc_request("restart_modules", nullptr, true);

        return nullptr;
    } else if (cmd == "get_rpc_timeout") {
        return this->config.controller_rpc_timeout_ms;
    }

    throw CommandApiMethodNotFound(fmt::format("Command '{}' unknown", cmd));
}