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

//! Query provider for `{query|line}` — cursor line in IDE, first line in CLI.

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

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

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

impl QueryProvider for LineProvider {
    fn key(&self) -> QueryKey {
        QueryKey::Line
    }

    fn resolve(&self) -> Result<ContextModel> {
        let text = self
            .raw_value
            .as_deref()
            .filter(|v| !v.is_empty())
            .map(|s| s.to_string())
            .unwrap_or_else(|| utils::clipboard::read_text().unwrap_or_default());

        let line = text.lines().next().unwrap_or("").to_string();
        Ok(ContextModel::String(line))
    }
}