mithril_aggregator/commands/
mod.rs

1mod database_command;
2mod era_command;
3mod genesis_command;
4mod serve_command;
5mod tools_command;
6
7use anyhow::anyhow;
8use clap::{CommandFactory, Parser, Subcommand};
9use config::{builder::DefaultState, ConfigBuilder, Map, Source, Value};
10use mithril_cli_helper::{register_config_value, register_config_value_option};
11use mithril_common::StdResult;
12use mithril_doc::{Documenter, DocumenterDefault, StructDoc};
13use slog::{debug, Level, Logger};
14use std::path::PathBuf;
15
16use crate::{Configuration, DefaultConfiguration};
17use mithril_doc::GenerateDocCommands;
18
19/// Main command selector
20#[derive(Debug, Clone, Subcommand)]
21pub enum MainCommand {
22    Genesis(genesis_command::GenesisCommand),
23    Era(era_command::EraCommand),
24    Serve(serve_command::ServeCommand),
25    Tools(tools_command::ToolsCommand),
26    Database(database_command::DatabaseCommand),
27    #[clap(alias("doc"), hide(true))]
28    GenerateDoc(GenerateDocCommands),
29}
30/// Identifies the type of command
31pub enum CommandType {
32    /// Command that runs a server
33    Server,
34
35    /// Command that outputs some result after execution
36    CommandLine,
37}
38
39impl MainCommand {
40    pub async fn execute(
41        &self,
42        root_logger: Logger,
43        config_builder: ConfigBuilder<DefaultState>,
44    ) -> StdResult<()> {
45        match self {
46            Self::Genesis(cmd) => cmd.execute(root_logger, config_builder).await,
47            Self::Era(cmd) => cmd.execute(root_logger, config_builder).await,
48            Self::Serve(cmd) => cmd.execute(root_logger, config_builder).await,
49            Self::Tools(cmd) => cmd.execute(root_logger, config_builder).await,
50            Self::Database(cmd) => cmd.execute(root_logger).await,
51            Self::GenerateDoc(cmd) => {
52                let config_infos = vec![Configuration::extract(), DefaultConfiguration::extract()];
53                cmd.execute_with_configurations(&mut MainOpts::command(), &config_infos)
54                    .map_err(|message| anyhow!(message))
55            }
56        }
57    }
58
59    pub fn command_type(&self) -> CommandType {
60        match self {
61            MainCommand::Serve(_) => CommandType::Server,
62            MainCommand::Genesis(_) => CommandType::CommandLine,
63            MainCommand::Era(_) => CommandType::CommandLine,
64            MainCommand::Tools(_) => CommandType::CommandLine,
65            MainCommand::Database(_) => CommandType::CommandLine,
66            MainCommand::GenerateDoc(_) => CommandType::CommandLine,
67        }
68    }
69}
70
71/// Mithril Aggregator Node
72#[derive(Documenter, Parser, Debug, Clone)]
73#[command(version)]
74pub struct MainOpts {
75    /// application main command
76    #[clap(subcommand)]
77    pub command: MainCommand,
78
79    /// Run Mode
80    #[clap(short, long, default_value = "dev")]
81    pub run_mode: String,
82
83    /// Verbosity level
84    #[clap(short, long, action = clap::ArgAction::Count)]
85    #[example = "Parsed from the number of occurrences: `-v` for `Warning`, `-vv` for `Info`, `-vvv` for `Debug` and `-vvvv` for `Trace`"]
86    pub verbose: u8,
87
88    /// Directory of the Cardano node files
89    #[clap(long)]
90    pub db_directory: Option<PathBuf>,
91
92    /// Directory where configuration file is located
93    #[clap(long, default_value = "./config")]
94    pub config_directory: PathBuf,
95}
96
97impl Source for MainOpts {
98    fn clone_into_box(&self) -> Box<dyn Source + Send + Sync> {
99        Box::new(self.clone())
100    }
101
102    fn collect(&self) -> Result<Map<String, Value>, config::ConfigError> {
103        let mut result = Map::new();
104        let namespace = "clap arguments".to_string();
105
106        register_config_value_option!(result, &namespace, self.db_directory, |v: PathBuf| format!(
107            "{}",
108            v.to_string_lossy()
109        ));
110
111        Ok(result)
112    }
113}
114
115impl MainOpts {
116    /// execute command
117    pub async fn execute(&self, root_logger: Logger) -> StdResult<()> {
118        let config_file_path = self
119            .config_directory
120            .join(format!("{}.json", self.run_mode));
121        let config_builder = config::Config::builder()
122            .add_source(DefaultConfiguration::default())
123            .add_source(
124                config::File::with_name(&config_file_path.to_string_lossy()).required(false),
125            )
126            .add_source(config::Environment::default().separator("__"))
127            .add_source(self.clone());
128        debug!(root_logger, "Started"; "run_mode" => &self.run_mode, "node_version" => env!("CARGO_PKG_VERSION"));
129
130        self.command.execute(root_logger, config_builder).await
131    }
132
133    /// get log level from parameters
134    pub fn log_level(&self) -> Level {
135        match self.verbose {
136            0 => Level::Error,
137            1 => Level::Warning,
138            2 => Level::Info,
139            3 => Level::Debug,
140            _ => Level::Trace,
141        }
142    }
143}
mithril_aggregator/commands/mod.rs

mithril_aggregator/commands/
mod.rs
