rustbrain 0.1.1

CLI for rustbrain — a lightweight Rust-native second brain for humans and AI agents
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
105
106
107
108
109
110
111
112
113

//! End-to-end CLI smoke tests.

use assert_cmd::cargo::cargo_bin_cmd;
use predicates::prelude::*;
use std::fs;
use tempfile::tempdir;

#[test]
fn init_sync_query_context_export() {
    let dir = tempdir().unwrap();
    let root = dir.path();
    let docs = root.join("docs");
    let src = root.join("src");
    fs::create_dir_all(&docs).unwrap();
    fs::create_dir_all(&src).unwrap();
    fs::write(
        root.join("Cargo.toml"),
        "[package]\nname = \"demo\"\nversion = \"0.1.0\"\nedition = \"2021\"\n",
    )
    .unwrap();
    fs::write(
        docs.join("raft.md"),
        "---\ntags: [raft]\nnode_type: concept\n---\n# Raft\nSee [[logcompaction]] and symbol:StorageEngine.\n",
    )
    .unwrap();
    fs::write(
        docs.join("logcompaction.md"),
        "---\ntags: [log]\nnode_type: concept\n---\n# Log Compaction\nSee [[raft]].\n",
    )
    .unwrap();
    fs::write(
        src.join("lib.rs"),
        "pub struct StorageEngine;\nimpl StorageEngine { pub fn open() {} }\n",
    )
    .unwrap();

    cargo_bin_cmd!("rustbrain")
        .current_dir(root)
        .args(["init", "."])
        .assert()
        .success()
        .stdout(predicate::str::contains("initialized"));

    cargo_bin_cmd!("rustbrain")
        .current_dir(root)
        .args(["sync", "."])
        .assert()
        .success()
        .stdout(predicate::str::contains("sync complete"));

    // Second sync resolves pending symbol anchors (note may index before rust).
    cargo_bin_cmd!("rustbrain")
        .current_dir(root)
        .args(["sync", "."])
        .assert()
        .success();

    cargo_bin_cmd!("rustbrain")
        .current_dir(root)
        .args(["query", "raft", "-w", ".", "--scores"])
        .assert()
        .success()
        .stdout(predicate::str::contains("Raft"));

    // Critical: context must not panic on short flags.
    cargo_bin_cmd!("rustbrain")
        .current_dir(root)
        .args(["context", "-p", "raft", "-F", "markdown", "-w", "."])
        .assert()
        .success()
        .stdout(predicate::str::contains("rustbrain context"))
        .stdout(predicate::str::contains("tokens:"));

    cargo_bin_cmd!("rustbrain")
        .current_dir(root)
        .args([
            "context",
            "--for-prompt",
            "raft",
            "--format",
            "xml",
            "--hops",
            "1",
            "-w",
            ".",
        ])
        .assert()
        .success()
        .stdout(predicate::str::contains("<rustbrain_context"))
        .stdout(predicate::str::contains("tokens_used="));

    let bundle = root.join("out.brainbundle");
    cargo_bin_cmd!("rustbrain")
        .current_dir(root)
        .args([
            "export",
            "--out",
            bundle.to_str().unwrap(),
            "--decouple-ast",
            "-w",
            ".",
        ])
        .assert()
        .success();
    assert!(bundle.exists());

    // Watch command exists (help only — don't block).
    cargo_bin_cmd!("rustbrain")
        .args(["watch", "--help"])
        .assert()
        .success()
        .stdout(predicate::str::contains("debounce"));
}