shuvarie 0.1.0

Blazingly fast AI coding TUI for chivalrous people
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

use clap::Parser;
use tokio::sync::mpsc::channel;

mod cli;
mod trust;
mod tui;

use crate::cli::show_resume_hint;

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

    if let Some(dir) = args.dir.as_deref() {
        std::env::set_current_dir(dir).map_err(|e| {
            color_eyre::eyre::eyre!("failed to set working directory to {}: {e}", dir.display())
        })?;
    }

    let mut store = shuvarie_db::Store::open(&shuvarie_db::Store::default_path()).await?;

    if let Some(path) = args.import_session.as_deref() {
        return cli::import_session(&mut store, path).await;
    }
    if let Some(dest) = args.export_session.as_deref() {
        return cli::export_session(&mut store, dest, args.session).await;
    }

    let Some(trust::Resolved {
        config,
        grants,
        scenes,
        theme_set,
    }) = trust::resolve(args.config.as_ref()).await?
    else {
        return Ok(());
    };
    let (cmd_tx, cmd_rx) = channel::<shuvarie_core::Command>(64);
    let (event_tx, event_rx) = channel::<shuvarie_core::Event>(64);
    let permissions = std::sync::Arc::new(
        shuvarie_core::permissions::Permissions::build(
            &config.permissions,
            &std::env::current_dir().unwrap_or_else(|_| std::path::PathBuf::from(".")),
        )
        .map_err(|e| color_eyre::eyre::eyre!(e))?,
    );
    let startup = if let Some(id) = args.session {
        shuvarie_core::StartupSession::Session(id)
    } else if args.current {
        shuvarie_core::StartupSession::MostRecent
    } else {
        shuvarie_core::StartupSession::None
    };
    let core = tokio::spawn(shuvarie_core::run(
        config.clone(),
        shuvarie_core::Connections::load()?,
        store,
        startup,
        args.config,
        None,
        permissions,
        grants,
        scenes,
        cmd_rx,
        event_tx,
    ));

    let res = tui::run_tui(config, theme_set, cmd_tx.clone(), event_rx).await;
    drop(cmd_tx);
    let core_res = core.await;

    let res = res?;
    core_res?;

    if let Some(res_enum) = res {
        match res_enum {
            tui::TuiResponse::SessionSaved { session_id } => {
                show_resume_hint(session_id);
            }
        }
    }

    Ok(())
}