iocaine 2.0.0

The deadliest poison known to AI
Documentation 
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

// SPDX-FileCopyrightText: 2025 Gergely Nagy
// SPDX-FileContributor: Gergely Nagy
//
// SPDX-License-Identifier: MIT

use anyhow::Result;
use clap::Parser;
use figment::{
    providers::{Env, Format, Toml},
    Figment,
};
use figment_file_provider_adapter::FileAdapter;

use iocaine::{
    app::Iocaine,
    cli,
    config::{Config, MetricsLabel},
};

#[tokio::main]
async fn main() -> Result<()> {
    #[cfg(all(feature = "tokio-console", tokio_unstable))]
    console_subscriber::init();
    #[cfg(all(feature = "tokio-console", not(tokio_unstable)))]
    compile_error!("`tokio-console` requires manually enabling the `--cfg tokio_unstable` rust flag during compilation!");

    #[cfg(not(feature = "tokio-console"))]
    tracing_subscriber::fmt()
        .with_env_filter(tracing_subscriber::EnvFilter::from_default_env())
        .with_max_level(tracing::Level::WARN)
        .init();

    let args = cli::Args::parse();
    tracing::debug!(config_file = &args.config_file, "loading configuration");

    let mut config: Config = Figment::new()
        .merge(FileAdapter::wrap(Env::prefixed("IOCAINE_").split("__")))
        .merge(FileAdapter::wrap(Env::prefixed("IOCAINE__").split("__")))
        .merge(FileAdapter::wrap(Toml::file(&args.config_file)))
        .extract()
        .unwrap_or_else(|err| {
            tracing::error!(err = format!("{}", err), "Failed to load configuration");
            panic!(
                "Failed to load configuration from {} (and environment variables): {}",
                &args.config_file, err
            );
        });
    if config.metrics.enable
        && config
            .metrics
            .labels
            .contains(&MetricsLabel::UserAgentGroup)
        && config.metrics.agent_group.is_empty()
    {
        tracing::warn!("UserAgentGroup metrics labels are enabled, but no grouping has been configured. Disabling the label.");
        config
            .metrics
            .labels
            .retain(|label| label != &MetricsLabel::UserAgentGroup);
    }

    let app = Iocaine::new(config)?;
    app.run().await?;

    Ok(())
}