objectiveai-cli 2.1.1

ObjectiveAI command-line interface and embeddable library
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

use jaq_core::{Compiler, Ctx, FilterT, RcIter, load};
use jaq_json::Val;
use serde::Serialize;

pub fn run_jq(
    value: &impl Serialize,
    filter: &str,
) -> Result<Vec<serde_json::Value>, super::Error> {
    let program = load::File {
        code: filter,
        path: (),
    };
    let loader = load::Loader::new(jaq_std::defs().chain(jaq_json::defs()));
    let arena = load::Arena::default();

    let modules = loader.load(&arena, program).map_err(|errs| {
        super::Error::JqParse(
            errs.into_iter()
                .map(|e| format!("{e:?}"))
                .collect::<Vec<_>>()
                .join(", "),
        )
    })?;

    let filter = Compiler::default()
        .with_funs(jaq_std::funs().chain(jaq_json::funs()))
        .compile(modules)
        .map_err(|errs| {
            super::Error::JqCompile(
                errs.into_iter()
                    .map(|e| format!("{e:?}"))
                    .collect::<Vec<_>>()
                    .join(", "),
            )
        })?;

    let input_value =
        serde_json::to_value(value).map_err(super::Error::Serialize)?;

    let inputs = RcIter::new(core::iter::empty());
    let out = filter.run((Ctx::new([], &inputs), Val::from(input_value)));

    let mut results = Vec::new();
    for item in out {
        match item {
            Ok(val) => results.push(serde_json::Value::from(val)),
            Err(err) => {
                return Err(super::Error::JqRuntime(format!("{err:?}")));
            }
        }
    }
    Ok(results)
}