qql-cli 0.4.2

Command-line interface, REPL, converter, and migration tools for QQL
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

//! `qql version`.

use crate::output;

pub const VERSION: &str = env!("CARGO_PKG_VERSION");

pub fn build_version_response() -> output::VersionResponse {
    let features = [
        ("grpc", cfg!(feature = "grpc")),
        ("rest", cfg!(feature = "rest")),
        ("edge", cfg!(feature = "edge")),
        ("fastembed", cfg!(feature = "fastembed")),
    ]
    .into_iter()
    .filter(|(_, on)| *on)
    .map(|(name, _)| name.to_string())
    .collect::<Vec<_>>();
    let is_full = cfg!(feature = "edge") && cfg!(feature = "fastembed");
    let is_partial = cfg!(feature = "edge") || cfg!(feature = "fastembed");
    let edition = if is_full {
        "full"
    } else if is_partial {
        "custom"
    } else {
        "standard"
    };
    output::VersionResponse {
        ok: true,
        command: "version".to_string(),
        version: VERSION.to_string(),
        edition: edition.to_string(),
        features,
        message: format!("qql version {} ({})", VERSION, edition),
    }
}

pub fn handle_version() -> Result<(), Box<dyn std::error::Error>> {
    let resp = build_version_response();
    let s = serde_json::to_string_pretty(&resp)?;
    println!("{}", s);
    Ok(())
}

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

    #[test]
    fn test_version_response_edition_and_serialization() {
        let resp = build_version_response();
        assert!(resp.ok);
        assert_eq!(resp.command, "version");
        assert_eq!(resp.version, VERSION);

        let is_full = cfg!(feature = "edge") && cfg!(feature = "fastembed");
        let is_partial = cfg!(feature = "edge") || cfg!(feature = "fastembed");
        let expected_edition = if is_full {
            "full"
        } else if is_partial {
            "custom"
        } else {
            "standard"
        };
        assert_eq!(resp.edition, expected_edition);
        assert_eq!(
            resp.message,
            format!("qql version {} ({})", VERSION, expected_edition)
        );

        // Verify JSON roundtrip
        let json_str = serde_json::to_string_pretty(&resp).expect("must serialize");
        let deserialized: output::VersionResponse =
            serde_json::from_str(&json_str).expect("must deserialize VersionResponse JSON");
        assert_eq!(deserialized, resp);
    }
}