rustifymyclaw 0.1.1

Run Coding CLI Agents from Messaging Apps - single Rust binary, no server required.
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

use std::sync::Arc;

use anyhow::{Context, Result};
use clap::Parser;
use tokio::sync::{mpsc, RwLock};
use tokio_util::sync::CancellationToken;

mod backend;
mod channel;
mod cli;
mod command;
mod config;
mod config_reload;
mod executor;
mod formatter;
mod rate_limit;
mod router;
mod security;
mod session;
mod startup;
mod types;

use crate::session::SessionStore;
use crate::startup::{
    await_shutdown, build_rate_limiter, build_workspaces, spawn_config_watcher, spawn_providers,
    spawn_router, spawn_signal_handler, WorkspaceSetup,
};

#[tokio::main]
async fn main() -> Result<()> {
    let args = cli::Cli::parse();

    let log_filter = args
        .log_level
        .as_deref()
        .map(tracing_subscriber::EnvFilter::new)
        .or_else(|| tracing_subscriber::EnvFilter::try_from_default_env().ok())
        .unwrap_or_else(|| tracing_subscriber::EnvFilter::new("info"));

    tracing_subscriber::fmt().with_env_filter(log_filter).init();

    if cli::run_command(&args)? {
        return Ok(());
    }

    let config_path = args.config_file.unwrap_or_else(config::dirs_path);
    let app_config =
        config::load_from_path(&config_path).context("failed to load configuration")?;

    let shutdown = CancellationToken::new();
    spawn_signal_handler(shutdown.clone());

    let global_output = Arc::new(app_config.output.clone());
    let (rate_limiter, rate_limiter_shared) = build_rate_limiter(&app_config);
    let session_store = Arc::new(RwLock::new(SessionStore::new()));
    let (tx, rx) = mpsc::channel(64);

    let WorkspaceSetup {
        backends,
        workspaces,
        providers,
    } = build_workspaces(&app_config, &global_output).await?;

    let router_handle = spawn_router(
        session_store,
        backends,
        workspaces,
        rate_limiter,
        rx,
        shutdown.clone(),
    );
    let provider_handles = spawn_providers(providers, tx, shutdown.clone());

    spawn_config_watcher(app_config, shutdown, rate_limiter_shared);

    await_shutdown(provider_handles, router_handle).await;
    Ok(())
}