feat(config-migrate): couple to infra-agent type, simplify

Author: DavSanchez

agent-control/src/bin/main_config_migrate.rs

@@ -1,38 +1,32 @@
 use newrelic_agent_control::agent_control::config_repository::store::AgentControlConfigStore;
-use newrelic_agent_control::agent_control::defaults::{
-    AGENT_CONTROL_DATA_DIR, AGENT_CONTROL_LOCAL_DATA_DIR, AGENT_CONTROL_LOG_DIR, SUB_AGENT_DIR,
-};
 use newrelic_agent_control::config_migrate::cli::Cli;
 use newrelic_agent_control::config_migrate::migration::agent_config_getter::AgentConfigGetter;
 use newrelic_agent_control::config_migrate::migration::config::MigrationConfig;
 use newrelic_agent_control::config_migrate::migration::converter::ConfigConverter;
-use newrelic_agent_control::config_migrate::migration::defaults::NEWRELIC_INFRA_AGENT_TYPE_CONFIG_MAPPING;
 use newrelic_agent_control::config_migrate::migration::migrator::{ConfigMigrator, MigratorError};
 use newrelic_agent_control::config_migrate::migration::persister::legacy_config_renamer::LegacyConfigRenamer;
 use newrelic_agent_control::config_migrate::migration::persister::values_persister_file::ValuesPersisterFile;
 use newrelic_agent_control::instrumentation::tracing::{TracingConfig, try_init_tracing};
 use newrelic_agent_control::values::file::ConfigRepositoryFile;
 use std::error::Error;
-use std::path::PathBuf;
 use std::sync::Arc;
 use tracing::{debug, info, warn};
 
 fn main() -> Result<(), Box<dyn Error>> {
-    let tracing_config = TracingConfig::from_logging_path(PathBuf::from(AGENT_CONTROL_LOG_DIR));
+    let cli = Cli::load();
+    let tracing_config = TracingConfig::from_logging_path(cli.log_dir());
     let _tracer = try_init_tracing(tracing_config);
 
     info!("Starting config conversion tool...");
 
-    let config: MigrationConfig = MigrationConfig::parse(NEWRELIC_INFRA_AGENT_TYPE_CONFIG_MAPPING)?;
+    let config = MigrationConfig::parse(&cli.get_migration_config_str()?)?;
 
-    let cli = Cli::init_config_migrate_cli();
-    let remote_dir = PathBuf::from(AGENT_CONTROL_DATA_DIR);
-    let vr = ConfigRepositoryFile::new(cli.local_data_dir(), remote_dir);
+    let vr = ConfigRepositoryFile::new(cli.local_data_dir(), cli.remote_data_dir());
     let sa_local_config_loader = AgentControlConfigStore::new(Arc::new(vr));
     let config_migrator = ConfigMigrator::new(
         ConfigConverter::default(),
         AgentConfigGetter::new(sa_local_config_loader),
-        ValuesPersisterFile::new(PathBuf::from(AGENT_CONTROL_LOCAL_DATA_DIR).join(SUB_AGENT_DIR)),
+        ValuesPersisterFile::new(cli.local_sub_agent_data_dir()),
     );
 
     let legacy_config_renamer = LegacyConfigRenamer::default();

agent-control/src/config_migrate/cli.rs

@@ -1,29 +1,67 @@
-use crate::agent_control::defaults::AGENT_CONTROL_LOCAL_DATA_DIR;
+use crate::{
+    agent_control::defaults::{
+        AGENT_CONTROL_DATA_DIR, AGENT_CONTROL_LOCAL_DATA_DIR, AGENT_CONTROL_LOG_DIR, SUB_AGENT_DIR,
+    },
+    config_migrate::migration::defaults::NEWRELIC_INFRA_AGENT_TYPE_CONFIG_MAPPING,
+};
 use clap::Parser;
-use std::path::PathBuf;
+use std::{error::Error, fs, path::PathBuf};
 
 #[derive(Parser, Debug)]
 #[command(author, version, about, long_about = None)] // Read from `Cargo.toml`
 pub struct Cli {
-    /// Overrides the default local configuration path `/etc/newrelic-agent-control/`.
-    #[cfg(debug_assertions)]
+    /// Local data path used by Agent Control.
+    #[arg(long, default_value_os_t = PathBuf::from(AGENT_CONTROL_LOCAL_DATA_DIR))]
+    local_dir: PathBuf,
+
+    /// Remote data path used by Agent Control.
+    #[arg(long, default_value_os_t = PathBuf::from(AGENT_CONTROL_DATA_DIR))]
+    remote_dir: PathBuf,
+
+    /// Logs path used by Agent Control.
+    #[arg(long, default_value_os_t = PathBuf::from(AGENT_CONTROL_LOG_DIR))]
+    logs_dir: PathBuf,
+
+    /// Provides an external configuration mapping for the migration of agents to Agent Control.
     #[arg(long)]
-    local_dir: Option<PathBuf>,
+    migration_config_file: Option<PathBuf>,
 }
 
 impl Cli {
     /// Parses command line arguments
-    pub fn init_config_migrate_cli() -> Self {
+    pub fn load() -> Self {
         // Get command line args
         Self::parse()
     }
 
     pub fn local_data_dir(&self) -> PathBuf {
-        #[cfg(debug_assertions)]
-        if let Some(path) = &self.local_dir {
-            return path.clone();
-        }
+        self.local_dir.to_path_buf()
+    }
+
+    pub fn local_sub_agent_data_dir(&self) -> PathBuf {
+        self.local_dir.join(SUB_AGENT_DIR)
+    }
+
+    pub fn remote_data_dir(&self) -> PathBuf {
+        self.remote_dir.to_path_buf()
+    }
 
-        PathBuf::from(AGENT_CONTROL_LOCAL_DATA_DIR)
+    pub fn log_dir(&self) -> PathBuf {
+        self.logs_dir.to_path_buf()
+    }
+
+    pub fn get_migration_config_str(&self) -> Result<String, Box<dyn Error>> {
+        if let Some(path) = &self.migration_config_file {
+            fs::read_to_string(path).map_err(|e| {
+                format!(
+                    "Could not read provided migration config file ({}): {}",
+                    path.display(),
+                    e
+                )
+                .into()
+            })
+        } else {
+            Ok(NEWRELIC_INFRA_AGENT_TYPE_CONFIG_MAPPING.to_owned())
+        }
     }
 }

agent-control/src/config_migrate/migration.rs

@@ -1,4 +1,5 @@
 pub mod agent_config_getter;
+pub mod agent_configs;
 pub mod agent_value_spec;
 pub mod config;
 pub mod converter;