rigmode 0.1.0

Attach work modes to AI coding agent prompts
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

#![deny(clippy::unwrap_used, clippy::expect_used, clippy::panic)]

use std::io::{self, Read};
use std::path::PathBuf;
use std::process::ExitCode;

use crate::adapters::claude_code;
use crate::cli::Agent;
use crate::config;
use crate::gate;
use crate::log;
use crate::mode;

/// Always returns ExitCode::SUCCESS. Errors become empty stdout.
pub fn execute(agent: Agent, modes_dirs: Vec<PathBuf>) -> ExitCode {
    let output = run(agent, modes_dirs).unwrap_or_default();
    print!("{output}");
    ExitCode::SUCCESS
}

fn run(agent: Agent, modes_dirs: Vec<PathBuf>) -> anyhow::Result<String> {
    let mut stdin = String::new();
    io::stdin().read_to_string(&mut stdin)?;

    match agent {
        Agent::ClaudeCode => {
            let meta = claude_code::decode(&stdin)?;
            // Best-effort: an unreadable config still attaches with defaults.
            let config = config::default_config_path()
                .and_then(|p| config::load_config(&p))
                .unwrap_or_default();

            if let Ok(data_dir) = config::default_data_dir() {
                gate::maybe_record_from_prompt(
                    &data_dir,
                    &meta.prompt,
                    meta.session_id.as_deref(),
                    &config.gate,
                );
            }

            let modes = mode::load_modes(&config.resolve_modes_dirs(modes_dirs)?)?;
            let matched = mode::matching(&meta.prompt, &modes);
            if matched.is_empty() {
                return Ok(String::new());
            }
            if let Ok(data_dir) = config::default_data_dir() {
                log::append_attach(
                    &data_dir.join(log::ATTACH_LOG),
                    agent.as_str(),
                    &meta,
                    &matched,
                );
            }
            Ok(claude_code::encode(&matched))
        }
    }
}