80 lines
2.1 KiB
Rust
80 lines
2.1 KiB
Rust
//! Attune Universal Sensor Agent.
|
|
|
|
use anyhow::Result;
|
|
use attune_common::agent_bootstrap::{bootstrap_runtime_env, print_detect_only_report};
|
|
use attune_common::config::Config;
|
|
use attune_sensor::startup::{
|
|
apply_sensor_name_override, init_tracing, log_config_details, run_sensor_service,
|
|
set_config_path,
|
|
};
|
|
use clap::Parser;
|
|
use tracing::info;
|
|
|
|
#[derive(Parser, Debug)]
|
|
#[command(name = "attune-sensor-agent")]
|
|
#[command(
|
|
version,
|
|
about = "Attune Universal Sensor Agent - Injected into runtime containers to auto-detect sensor runtimes"
|
|
)]
|
|
struct Args {
|
|
/// Path to configuration file (optional)
|
|
#[arg(short, long)]
|
|
config: Option<String>,
|
|
|
|
/// Sensor worker name override
|
|
#[arg(short, long)]
|
|
name: Option<String>,
|
|
|
|
/// Run runtime detection, print results, and exit
|
|
#[arg(long)]
|
|
detect_only: bool,
|
|
}
|
|
|
|
fn main() -> Result<()> {
|
|
attune_common::auth::install_crypto_provider();
|
|
init_tracing(tracing::Level::INFO);
|
|
|
|
let args = Args::parse();
|
|
|
|
info!("Starting Attune Universal Sensor Agent");
|
|
info!(
|
|
"Agent binary: attune-sensor-agent {}",
|
|
env!("CARGO_PKG_VERSION")
|
|
);
|
|
|
|
// Safe: no async runtime or worker threads are running yet.
|
|
std::env::set_var("ATTUNE_SENSOR_AGENT_MODE", "true");
|
|
std::env::set_var("ATTUNE_SENSOR_AGENT_BINARY_NAME", "attune-sensor-agent");
|
|
std::env::set_var(
|
|
"ATTUNE_SENSOR_AGENT_BINARY_VERSION",
|
|
env!("CARGO_PKG_VERSION"),
|
|
);
|
|
|
|
let bootstrap = bootstrap_runtime_env("ATTUNE_SENSOR_RUNTIMES");
|
|
|
|
if args.detect_only {
|
|
print_detect_only_report("ATTUNE_SENSOR_RUNTIMES", &bootstrap);
|
|
return Ok(());
|
|
}
|
|
|
|
set_config_path(args.config.as_deref());
|
|
|
|
let runtime = tokio::runtime::Runtime::new()?;
|
|
runtime.block_on(async_main(args))
|
|
}
|
|
|
|
async fn async_main(args: Args) -> Result<()> {
|
|
let mut config = Config::load()?;
|
|
config.validate()?;
|
|
|
|
if let Some(name) = args.name {
|
|
apply_sensor_name_override(&mut config, name);
|
|
}
|
|
|
|
log_config_details(&config);
|
|
run_sensor_service(config, "Attune Sensor Agent is ready").await?;
|
|
info!("Attune Sensor Agent shutdown complete");
|
|
|
|
Ok(())
|
|
}
|