Skip to main content

hematite/
lib.rs

1pub mod agent;
2pub mod memory;
3pub mod runtime;
4pub mod telemetry;
5pub mod tools;
6pub mod ui;
7
8pub const HEMATITE_VERSION: &str = env!("CARGO_PKG_VERSION");
9pub const HEMATITE_AUTHOR: &str = "Ocean Bennett";
10pub const HEMATITE_REPOSITORY_URL: &str = "https://github.com/undergroundrap/hematite-cli";
11pub const HEMATITE_SHORT_DESCRIPTION: &str =
12    "Local-first AI coding harness and workstation assistant for real developer workflows.";
13const HEMATITE_GIT_COMMIT_SHORT_RAW: &str = env!("HEMATITE_GIT_COMMIT_SHORT");
14const HEMATITE_GIT_EXACT_TAG_RAW: &str = env!("HEMATITE_GIT_EXACT_TAG");
15const HEMATITE_GIT_DIRTY_RAW: &str = env!("HEMATITE_GIT_DIRTY");
16
17pub fn hematite_git_commit_short() -> Option<&'static str> {
18    (!HEMATITE_GIT_COMMIT_SHORT_RAW.is_empty()).then_some(HEMATITE_GIT_COMMIT_SHORT_RAW)
19}
20
21pub fn hematite_git_exact_tag() -> Option<&'static str> {
22    (!HEMATITE_GIT_EXACT_TAG_RAW.is_empty()).then_some(HEMATITE_GIT_EXACT_TAG_RAW)
23}
24
25pub fn hematite_git_dirty() -> bool {
26    HEMATITE_GIT_DIRTY_RAW.eq_ignore_ascii_case("true")
27}
28
29pub fn hematite_build_descriptor() -> String {
30    let release_tag = format!("v{}", HEMATITE_VERSION);
31    let exact_release = matches!(hematite_git_exact_tag(), Some(tag) if tag == release_tag);
32
33    if exact_release && !hematite_git_dirty() {
34        "release".to_string()
35    } else {
36        match (hematite_git_commit_short(), hematite_git_dirty()) {
37            (Some(commit), true) => format!("dev+{}-dirty", commit),
38            (Some(commit), false) => format!("dev+{}", commit),
39            (None, true) => "dev-dirty".to_string(),
40            (None, false) => "dev".to_string(),
41        }
42    }
43}
44
45pub fn hematite_version_display() -> String {
46    format!("v{} [{}]", HEMATITE_VERSION, hematite_build_descriptor())
47}
48
49pub fn hematite_version_report() -> String {
50    let mut lines = vec![
51        format!("Hematite v{}", HEMATITE_VERSION),
52        format!("Build: {}", hematite_build_descriptor()),
53    ];
54    if let Some(commit) = hematite_git_commit_short() {
55        lines.push(format!("Commit: {}", commit));
56    }
57    lines.push(format!(
58        "Built from a dirty worktree: {}",
59        if hematite_git_dirty() { "yes" } else { "no" }
60    ));
61    lines.push(format!(
62        "Exact release tag at build time: {}",
63        hematite_git_exact_tag().unwrap_or("none")
64    ));
65    lines.join("\n")
66}
67
68pub fn hematite_about_report() -> String {
69    [
70        format!("Hematite v{}", HEMATITE_VERSION),
71        format!("Build: {}", hematite_build_descriptor()),
72        format!("Created and maintained by {}", HEMATITE_AUTHOR),
73        HEMATITE_SHORT_DESCRIPTION.to_string(),
74        format!("Repo: {}", HEMATITE_REPOSITORY_URL),
75    ]
76    .join("\n")
77}
78
79pub fn hematite_identity_answer() -> String {
80    format!(
81        "Hematite was created and is maintained by {}.\n\n{}\n\nThe running assistant uses a local model runtime, but Hematite itself is the local coding harness: the TUI, tool use, file editing, workflow control, voice integration, and workstation-assistant architecture.\n\nRepo: {}",
82        HEMATITE_AUTHOR, HEMATITE_SHORT_DESCRIPTION, HEMATITE_REPOSITORY_URL
83    )
84}
85
86// Standard imports for library users
87pub use agent::config::HematiteConfig;
88pub use agent::conversation::ConversationManager;
89pub use agent::inference::InferenceEngine;
90
91use clap::Parser;
92
93#[derive(Parser, Debug, Clone)]
94#[command(
95    author,
96    version,
97    about = "Hematite CLI - Local-first AI coding harness and workstation assistant",
98    long_about = None
99)]
100pub struct CliCockpit {
101    #[arg(long, help = "Bypasses the high-risk modal (Danger mode)")]
102    pub yolo: bool,
103
104    #[arg(
105        long,
106        default_value_t = 3,
107        help = "Sets max parallel workers (default 3)"
108    )]
109    pub swarm_size: usize,
110
111    #[arg(
112        long,
113        help = "Forces the Vigil Brief Mode for concise, high-speed output"
114    )]
115    pub brief: bool,
116
117    #[arg(
118        long,
119        help = "Pass a custom salt to reroll the deterministic species hash"
120    )]
121    pub reroll: Option<String>,
122
123    #[arg(
124        long,
125        help = "Rusty Mode: Enables the Rusty personality system, snark, and companion features"
126    )]
127    pub rusty: bool,
128
129    #[arg(long, help = "Show Rusty stats and exit")]
130    pub stats: bool,
131
132    #[arg(
133        long,
134        help = "Skip the blocking splash screen and enter the TUI immediately"
135    )]
136    pub no_splash: bool,
137
138    #[arg(
139        long,
140        help = "Optional model ID for simple tasks (overrides auto-detect)"
141    )]
142    pub fast_model: Option<String>,
143
144    #[arg(
145        long,
146        help = "Optional model ID for complex tasks (overrides auto-detect)"
147    )]
148    pub think_model: Option<String>,
149
150    #[arg(
151        long,
152        default_value = "http://localhost:1234/v1",
153        help = "The base URL for the OpenAI-compatible API"
154    )]
155    pub url: String,
156
157    #[arg(long, hide = true)]
158    pub pdf_extract_helper: Option<String>,
159
160    #[arg(long, hide = true)]
161    pub teleported_from: Option<String>,
162}
163
164#[cfg(test)]
165mod tests {
166    #[test]
167    fn version_report_contains_release_version() {
168        let report = crate::hematite_version_report();
169        assert!(report.contains(crate::HEMATITE_VERSION));
170        assert!(report.contains("Build:"));
171    }
172
173    #[test]
174    fn about_report_contains_author_and_repo() {
175        let report = crate::hematite_about_report();
176        assert!(report.contains(crate::HEMATITE_AUTHOR));
177        assert!(report.contains(crate::HEMATITE_REPOSITORY_URL));
178    }
179}