vibe-action 0.0.4

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

//! Contains modifier — checks if a string or list contains a substring.
//! For strings: returns Bool. For lists: returns List<Bool> per element.

use anyhow::Result;

use super::modifier::{Modifier, ModifierKey};
use crate::models::context::ContextModel;

pub struct ContainsModifier;

impl Modifier for ContainsModifier {
    fn key(&self) -> ModifierKey {
        ModifierKey::Contains
    }

    fn apply(&self, value: &ContextModel, arg: &str) -> Result<ContextModel> {
        match value {
            ContextModel::String(s) => Ok(ContextModel::Bool(s.contains(arg))),
            ContextModel::List(items) => {
                let results: Vec<ContextModel> = items
                    .iter()
                    .map(|i| ContextModel::Bool(i.to_string().contains(arg)))
                    .collect();
                Ok(ContextModel::List(results))
            }
            _ => anyhow::bail!("Modifier 'contains' expects a string or list"),
        }
    }
}