Add tracing for cg3

Author: bbqsrc

Cargo.lock

@@ -198,6 +198,7 @@ impl Display for Command {
 #[serde(untagged)]
 pub enum Value {
     Int(isize),
+    Bool(bool),
     String(String),
     Array(Vec<Value>),
     Map(IndexMap<String, Value>),
@@ -215,6 +216,13 @@ impl Value {
                     format!("{}", x)
                 }
             }
+            Value::Bool(x) => {
+                if ansi {
+                    format!("\x1b[1;34m{}\x1b[0m", x)
+                } else {
+                    format!("{}", x)
+                }
+            }
             Value::String(x) => {
                 if ansi {
                     format!("\x1b[1;31m{:?}\x1b[0m", x)
@@ -295,6 +303,13 @@ impl Value {
         }
     }
 
+    pub fn try_as_bool(&self) -> Option<bool> {
+        match self {
+            Value::Bool(x) => Some(*x),
+            _ => None,
+        }
+    }
+
     pub fn try_as_string(&self) -> Option<String> {
         match self {
             Value::String(x) => Some(x.clone()),
@@ -355,6 +370,7 @@ impl Value {
 
     pub fn try_as_json(&self) -> Result<serde_json::Value, serde_json::Error> {
         match self {
+            Value::Bool(x) => Ok(serde_json::Value::Bool(*x)),
             Value::Int(x) => Ok(serde_json::Value::Number(serde_json::Number::from(*x))),
             Value::String(x) => Ok(serde_json::Value::String(x.clone())),
             Value::Array(x) => Ok(serde_json::Value::Array(

src/ast/mod.rs

@@ -1,10 +1,10 @@
 use std::{collections::HashMap, str::FromStr, sync::Arc, thread::JoinHandle};
 
 use async_trait::async_trait;
-use divvun_runtime_macros::rt_command;
-use indexmap::IndexMap;
+use divvun_runtime_macros::{rt_command, rt_struct};
 use once_cell::sync::Lazy;
 use regex::Regex;
+use serde::{Deserialize, Serialize};
 use tokio::sync::{
     Mutex,
     mpsc::{self, Receiver, Sender},
@@ -460,13 +460,22 @@ pub struct Vislcg3 {
     _thread: JoinHandle<()>,
 }
 
+#[rt_struct(module = "cg3")]
+#[derive(Clone, Debug, Default, Serialize, Deserialize)]
+struct Vislcg3Config {
+    trace: bool,
+}
+
 #[rt_command(
     module = "cg3",
     name = "vislcg3",
     input = [String],
     output = "String",
     kind = "cg3",
-    args = [model_path = "Path"]
+    args = [
+        model_path = "Path",
+        config? = "Vislcg3Config",
+    ]
 )]
 impl Vislcg3 {
     pub fn new(
@@ -482,11 +491,27 @@ impl Vislcg3 {
             .ok_or_else(|| Error("model_path missing".to_string()))?;
         let model_path = context.extract_to_temp_dir(model_path)?;
 
+        let config = match kwargs
+            .remove("config")
+            .and_then(|x| x.value)
+            .map(|x| x.try_as_json())
+        {
+            Some(Ok(c)) => {
+                let config: Vislcg3Config = serde_json::from_value(c)
+                    .map_err(|e| Error(format!("config arg is not valid SpellerConfig: {}", e)))?;
+                Some(config)
+            }
+            Some(Err(e)) => return Err(Error(format!("config arg is not valid JSON: {}", e))),
+            None => None,
+        };
+        let config = config.unwrap_or_default();
+
         let (input_tx, mut input_rx) = mpsc::channel(1);
         let (output_tx, output_rx) = mpsc::channel(1);
 
         let thread = std::thread::spawn(move || {
             let applicator = cg3::Applicator::new(&model_path);
+            applicator.set_trace(config.trace);
 
             loop {
                 let Some(Some(input)): Option<Option<String>> = input_rx.blocking_recv() else {