minotari_node 5.4.0-pre.0

//  Copyright 2022. The Tari Project
//
//  Redistribution and use in source and binary forms, with or without modification, are permitted provided that the
//  following conditions are met:
//
//  1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following
//  disclaimer.
//
//  2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the
//  following disclaimer in the documentation and/or other materials provided with the distribution.
//
//  3. Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote
//  products derived from this software without specific prior written permission.
//
//  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
//  INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
//  DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
//  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
//  SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
//  WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
//  USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

use std::path::PathBuf;

use clap::Parser;
use minotari_app_utilities::common_cli_args::CommonCliArgs;
use tari_common::configuration::{ConfigOverrideProvider, Network};

#[derive(Parser, Debug)]
#[clap(author, version, about, long_about = None)]
#[clap(propagate_version = true)]
#[allow(clippy::struct_excessive_bools)]
pub struct Cli {
    #[clap(flatten)]
    pub common: CommonCliArgs,
    /// Create a default configuration file if it doesn't exist
    #[clap(long)]
    pub init: bool,
    /// This will rebuild the db, adding block for block in
    #[clap(long, alias = "rebuild_db")]
    pub rebuild_db: bool,
    /// Run in non-interactive mode, with no UI.
    #[clap(short, long, alias = "non-interactive", env = "TARI_NON_INTERACTIVE")]
    pub non_interactive_mode: bool,
    /// Watch a command in the non-interactive mode.
    #[clap(long)]
    pub watch: Option<String>,
    /// Run in test profile mode
    #[clap(long, alias = "profile")]
    pub profile_with_tokio_console: bool,
    /// Enable mining
    #[clap(long, env = "MINOTARI_NODE_ENABLE_MINING", alias = "enable-mining")]
    pub mining_enabled: bool,
    /// Enable the gRPC server
    #[clap(long, env = "MINOTARI_NODE_ENABLE_GRPC", alias = "enable-grpc")]
    pub grpc_enabled: bool,
    /// GRPC address of base node
    #[clap(long, env = "MINOTARI_NODE_GRPC_ADDRESS")]
    pub grpc_address: Option<String>,
    /// Enable the second layer gRPC server
    #[clap(long, env = "MINOTARI_NODE_SECOND_LAYER_GRPC_ENABLED", alias = "enable-second-layer")]
    pub second_layer_grpc_enabled: bool,
    /// Disable the splash screen
    #[clap(long)]
    pub disable_splash_screen: bool,
    /// Print all TARI_* and MINOTARI_* environment variables and exit.
    /// Useful for verifying which environment variables are set before starting the node.
    #[clap(long)]
    pub print_env: bool,
    /// Path to the libtor data directory
    #[clap(short = 'z', long)]
    pub libtor_data_dir: Option<PathBuf>,
}

impl ConfigOverrideProvider for Cli {
    /// Get the configuration property overrides for the given network. In case of duplicates, the final override
    /// added to the list will have preference.
    fn get_config_property_overrides(&self, network: &Network) -> Vec<(String, String)> {
        // Config file overrides
        let mut overrides = vec![("base_node.network".to_string(), network.to_string())];
        overrides.push(("base_node.override_from".to_string(), network.to_string()));
        overrides.push(("p2p.seeds.override_from".to_string(), network.to_string()));
        overrides.push(("auto_update.override_from".to_string(), network.to_string()));
        overrides.push(("metrics.override_from".to_string(), network.to_string()));
        // Command-line overrides
        let command_line_overrides = self.common.get_config_property_overrides(network);
        command_line_overrides.iter().for_each(|(k, v)| {
            replace_or_add_override(&mut overrides, k, v);
        });
        // Logical overrides based on command-line flags
        if self.mining_enabled {
            replace_or_add_override(&mut overrides, "base_node.mining_enabled", "true");
        }
        if let Some(ref addr) = self.grpc_address {
            replace_or_add_override(&mut overrides, "base_node.grpc_enabled", "true");
            replace_or_add_override(&mut overrides, "base_node.grpc_address", addr);
        } else if self.grpc_enabled {
            replace_or_add_override(&mut overrides, "base_node.grpc_enabled", "true");
        } else {
            // clippy
        }
        if self.second_layer_grpc_enabled {
            replace_or_add_override(&mut overrides, "base_node.grpc_enabled", "true");
            replace_or_add_override(&mut overrides, "base_node.second_layer_grpc_enabled", "true");
        }
        overrides
    }
}

fn replace_or_add_override(overrides: &mut Vec<(String, String)>, key: &str, value: &str) {
    if let Some(index) = overrides.iter().position(|(k, _)| k == key) {
        overrides.remove(index);
    }
    overrides.push((key.to_string(), value.to_string()));
}