vibe-action 0.1.1

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
37
38

//! System provider for `{system_clipboard_image}` — image from clipboard as base64 PNG.

use crate::models::context::ContextModel;
use crate::system::system::{SystemKey, SystemProvider};
use anyhow::Result;
use image::ImageEncoder;
use image::codecs::png::PngEncoder;

pub struct SystemClipboardImageProvider;

impl SystemProvider for SystemClipboardImageProvider {
    fn key(&self) -> SystemKey {
        SystemKey::ClipboardImage
    }

    fn resolve(&self) -> Result<ContextModel> {
        let mut clipboard = arboard::Clipboard::new()
            .map_err(|e| anyhow::anyhow!("Failed to access clipboard: {}", e))?;
        let img = clipboard
            .get_image()
            .map_err(|_| anyhow::anyhow!("Clipboard does not contain an image."))?;

        let mut png_bytes = Vec::new();
        let encoder = PngEncoder::new(&mut png_bytes);
        encoder
            .write_image(
                &img.bytes,
                img.width as u32,
                img.height as u32,
                image::ExtendedColorType::Rgba8,
            )
            .map_err(|e| anyhow::anyhow!("Failed to encode PNG: {}", e))?;

        use base64::Engine;
        let base64 = base64::engine::general_purpose::STANDARD.encode(&png_bytes);
        Ok(ContextModel::String(base64))
    }
}