vik 0.1.0

Vik is an issue-driven coding workflow automation tool.
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
86
87
88
89
90

//! Provider-adapter seam.
//!
//! Adapters describe what CLI to invoke and how to decode its JSONL
//! output. Subprocess spawn, stdin wiring, stdout streaming, and
//! cancellation all live in [`crate::session`]. Adapter files therefore
//! carry no `cfg(unix)`/`cfg(windows)` gates, no `libc` calls, and no
//! `tokio::process` references — adding a new provider is two pure
//! functions, not a fresh copy of the spawn scaffolding.
mod claude_code;
mod codex;

use serde_json::Value;

use crate::config::AgentProfileSchema;

use super::response::AgentEvent;

pub(super) use claude_code::ClaudeCodeAdapter;
pub(super) use codex::CodexAdapter;

#[derive(Debug, Clone)]
pub struct AgentCommand {
  pub program: String,
  pub args: Vec<String>,
  pub stdin: AgentStdin,
}

/// Stdin wiring requested by the adapter. Adapters never call
/// `Stdio::piped()` themselves; the runner inspects this and chooses.
#[allow(dead_code)]
#[derive(Debug, Clone)]
pub enum AgentStdin {
  /// Closed immediately — for providers that take the prompt on argv.
  None,
  /// Reserved for symmetry; not used by any current adapter.
  Inherit,
  /// Write the string and close the handle so the child sees EOF.
  Pipe(String),
}

/// Implementations are stateless per spawn: `build_command` is called
/// once per run and `map_event` once per JSONL line. Returning an empty
/// `Vec` from `map_event` drops the line; returning multiple events
/// fans one line out (e.g. Claude's `result` line yields both
/// `TokenUsage` and `Completed`).
pub trait AgentAdapter: Send + Sync {
  fn build_command(&self, profile: &AgentProfileSchema, prompt: String) -> AgentCommand;

  /// Unknown shapes return `vec![]`; new provider event types must not
  /// crash the stream.
  fn map_event(&self, value: Value) -> Vec<AgentEvent>;
}

/// Flatten the YAML `args` map into a flat CLI token list. Booleans
/// follow GNU conventions: `true` becomes a no-value flag, `false`
/// drops the flag entirely. Sequences are joined with `,` because every
/// supported provider accepts comma lists in one argument and that
/// keeps logs readable.
fn build_extra_args(map: &serde_yaml::Mapping) -> Vec<String> {
  map
    .iter()
    .flat_map(|(k, v)| {
      let k = k.as_str()?;

      let kv = match v {
        serde_yaml::Value::String(s) => Some((k, s.clone())),
        serde_yaml::Value::Number(n) => Some((k, n.to_string())),
        serde_yaml::Value::Bool(true) => return Some(vec![k.to_string()]),
        serde_yaml::Value::Bool(false) => return Some(Vec::new()),
        serde_yaml::Value::Sequence(seq) => Some((
          k,
          seq
            .iter()
            .filter_map(|item| item.as_str().map(|s| s.to_string()))
            .collect::<Vec<String>>()
            .join(","),
        )),

        _ => return None,
      };

      if let Some((k, v)) = kv {
        Some(vec![k.to_string(), v])
      } else {
        None
      }
    })
    .flatten()
    .collect()
}