iocaine 3.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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

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

#[cfg(all(not(target_env = "musl"), feature = "jemalloc"))]
#[global_allocator]
static GLOBAL: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;

use anyhow::Result;
use clap::{Parser, Subcommand};

use iocaine::morgue::Body;

mod config;
use config::Config;
mod commands;
use commands::show::{self, ShowCommands};
use commands::tests::{self, TestCommands};

#[derive(Debug, Parser, Default)]
#[command(version, about)]
pub struct Args {
    /// Path to load configuration from. Can be repeated.
    #[arg(short = 'c', long = "config-path")]
    pub config_paths: Vec<String>,

    #[command(subcommand)]
    command: Option<Commands>,
}

#[derive(Debug, Subcommand)]
enum Commands {
    /// Start the iocaine server. This is the default, if no command is specified.
    #[command(bin_name = "iocaine start")]
    Start,
    /// Run tests against the request handler.
    #[command(bin_name = "iocaine test")]
    Test {
        #[command(subcommand)]
        command: Option<TestCommands>,
    },
    /// Show something about iocaine (its configuration, for example), then exit.
    #[command(bin_name = "iocaine show")]
    Show {
        #[command(subcommand)]
        command: ShowCommands,
    },
}

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

    #[cfg(all(feature = "tokio-console", tokio_unstable))]
    console_subscriber::init();

    #[cfg(not(all(feature = "tokio-console", tokio_unstable)))]
    tracing_subscriber::fmt()
        .with_env_filter(
            tracing_subscriber::EnvFilter::try_from_default_env()
                .unwrap_or_else(|_| tracing_subscriber::EnvFilter::new("warn")),
        )
        .with_writer(std::io::stderr)
        .init();

    let args = Args::parse();
    let command = args.command.unwrap_or(Commands::Start);

    let config = Config::load(&args.config_paths).unwrap_or_else(|err| panic!("{}", err));

    match command {
        Commands::Start => {
            let app = Body::new(config.try_into()?)?;
            app.run().await?;
        }
        Commands::Test { command } => tests::run(&config, command.unwrap_or_default())?,
        Commands::Show { command } => show::run(config, command)?,
    }

    Ok(())
}