turbovault-sql 1.4.0

SQL query engine for Obsidian vault frontmatter — powered by GlueSQL
Documentation 
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104

//! Conversion between GlueSQL values and serde_json values

use gluesql::prelude::{Payload, Value as GlueValue};
use serde_json::{Value, json};

/// Convert a GlueSQL `Payload` to a JSON value for MCP responses
pub fn payload_to_json(payload: Payload) -> Value {
    match payload {
        Payload::Select { labels, rows } => {
            let results: Vec<Value> = rows
                .into_iter()
                .map(|row| {
                    let obj: serde_json::Map<String, Value> = labels
                        .iter()
                        .zip(row)
                        .map(|(label, val)| (label.clone(), glue_to_json(val)))
                        .collect();
                    Value::Object(obj)
                })
                .collect();
            json!({
                "rows": results,
                "count": results.len()
            })
        }
        Payload::Insert(n) => json!({"type": "insert", "affected": n}),
        Payload::Update(n) => json!({"type": "update", "affected": n}),
        Payload::Delete(n) => json!({"type": "delete", "affected": n}),
        Payload::Create => json!({"type": "create", "status": "ok"}),
        _ => json!({"type": "other", "status": "ok"}),
    }
}

/// Convert a GlueSQL `Value` to a serde_json `Value`
pub fn glue_to_json(val: GlueValue) -> Value {
    match val {
        GlueValue::Bool(b) => json!(b),
        GlueValue::I8(n) => json!(n),
        GlueValue::I16(n) => json!(n),
        GlueValue::I32(n) => json!(n),
        GlueValue::I64(n) => json!(n),
        GlueValue::I128(n) => json!(n),
        GlueValue::U8(n) => json!(n),
        GlueValue::U16(n) => json!(n),
        GlueValue::U32(n) => json!(n),
        GlueValue::U64(n) => json!(n),
        GlueValue::U128(n) => json!(n),
        GlueValue::F32(n) => json!(n),
        GlueValue::F64(n) => json!(n),
        GlueValue::Str(s) => Value::String(s),
        GlueValue::Null => Value::Null,
        GlueValue::List(items) => Value::Array(items.into_iter().map(glue_to_json).collect()),
        GlueValue::Map(map) => {
            let obj: serde_json::Map<String, Value> = map
                .into_iter()
                .map(|(k, v)| (format!("{k:?}"), glue_to_json(v)))
                .collect();
            Value::Object(obj)
        }
        other => Value::String(format!("{other:?}")),
    }
}

/// Determine the JSON type name for schema inspection
pub fn json_type_name(value: &Value) -> &'static str {
    match value {
        Value::String(_) => "string",
        Value::Number(_) => "number",
        Value::Bool(_) => "boolean",
        Value::Array(_) => "array",
        Value::Object(_) => "object",
        Value::Null => "null",
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_json_type_name() {
        assert_eq!(json_type_name(&json!("hello")), "string");
        assert_eq!(json_type_name(&json!(42)), "number");
        assert_eq!(json_type_name(&json!(true)), "boolean");
        assert_eq!(json_type_name(&json!([1, 2])), "array");
        assert_eq!(json_type_name(&json!({"a": 1})), "object");
        assert_eq!(json_type_name(&json!(null)), "null");
    }

    #[test]
    fn test_glue_to_json_primitives() {
        assert_eq!(glue_to_json(GlueValue::Bool(true)), json!(true));
        assert_eq!(glue_to_json(GlueValue::I64(42)), json!(42));
        assert_eq!(glue_to_json(GlueValue::F64(2.72)), json!(2.72));
        assert_eq!(glue_to_json(GlueValue::Str("hi".into())), json!("hi"));
        assert_eq!(glue_to_json(GlueValue::Null), Value::Null);
    }

    #[test]
    fn test_glue_to_json_list() {
        let list = GlueValue::List(vec![GlueValue::Str("a".into()), GlueValue::Str("b".into())]);
        assert_eq!(glue_to_json(list), json!(["a", "b"]));
    }
}