vibe-action 0.2.2

Command router — execute shell commands and LLM prompts via simple YAML actions.
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

//! Query provider for `{query}` — raw input (positional arg or clipboard text).

use super::query::QueryKey;
use super::query::QueryProvider;
use super::query::read_clipboard_text;
use crate::models::context::ContextModel;
use anyhow::Result;

pub struct QueryRawProvider {
    raw_value: Option<String>,
}

impl QueryRawProvider {
    pub fn new(raw_value: Option<String>) -> Self {
        Self { raw_value }
    }
}

impl QueryProvider for QueryRawProvider {
    fn key(&self) -> QueryKey {
        QueryKey::Raw
    }

    fn resolve(&self) -> Result<ContextModel> {
        if let Some(value) = &self.raw_value {
            if !value.is_empty() {
                return Ok(ContextModel::String(value.clone()));
            }
        }

        match read_clipboard_text() {
            Ok(text) => Ok(ContextModel::String(text)),
            Err(_) => Ok(ContextModel::String(String::new())),
        }
    }
}