0be354a920
Single-commit clean baseline after security scrub of niche-tells, project codenames, internal jargon, and contributor-email leaks. Contents: - 100 Rust crates (_primitives/_rust/) - 37 agent manifests (_manifests/) + generated specs (_generated/) - 67 user-invocable skills (skills/) - 33 hooks (hooks/) - Composition blocks (_blocks/) - Documentation (docs/, README.md) - TS adapter packages (_ts_packages/) - Assembler (_assembler/) - Roles (_roles/) - Templates (_templates/) - Forgejo CI (.forgejo/) Author: Denis Parfionovich <info@greendragon.info> License: see LICENSE.
77 lines
2.5 KiB
Rust
77 lines
2.5 KiB
Rust
//! kei-chat-store CLI.
|
|
|
|
use clap::{Parser, Subcommand};
|
|
use kei_chat_store::search::search;
|
|
use kei_chat_store::sessions::{archive_session, save_message, start_session, ChatMessage};
|
|
use kei_chat_store::stats::stats;
|
|
use kei_chat_store::Store;
|
|
use std::path::PathBuf;
|
|
use std::process::ExitCode;
|
|
|
|
#[derive(Parser)]
|
|
#[command(name = "kei-chat-store", version)]
|
|
struct Cli {
|
|
#[arg(long)] db: Option<PathBuf>,
|
|
#[command(subcommand)] cmd: Cmd,
|
|
}
|
|
|
|
#[derive(Subcommand)]
|
|
enum Cmd {
|
|
Start { #[arg(long)] project: String,
|
|
#[arg(long, default_value = "")] title: String,
|
|
#[arg(long, default_value = "")] model: String },
|
|
Save { #[arg(long)] session_id: String,
|
|
#[arg(long)] role: String,
|
|
content: String,
|
|
#[arg(long, default_value_t = 0)] tokens_in: i64,
|
|
#[arg(long, default_value_t = 0)] tokens_out: i64,
|
|
#[arg(long, default_value_t = 0.0)] cost: f64 },
|
|
Search { query: String, #[arg(long, default_value_t = 20)] limit: i64 },
|
|
Archive { session_id: String },
|
|
Stats,
|
|
}
|
|
|
|
fn db_path(o: Option<PathBuf>) -> PathBuf {
|
|
if let Some(p) = o { return p; }
|
|
if let Ok(e) = std::env::var("KEI_CHAT_DB") { return PathBuf::from(e); }
|
|
let home = std::env::var("HOME").unwrap_or_else(|_| ".".into());
|
|
PathBuf::from(home).join(".claude/chat/chat.sqlite")
|
|
}
|
|
|
|
fn run() -> anyhow::Result<()> {
|
|
let cli = Cli::parse();
|
|
let s = Store::open(&db_path(cli.db))?;
|
|
match cli.cmd {
|
|
Cmd::Start { project, title, model } => {
|
|
println!("{}", start_session(&s, &project, &title, &model)?);
|
|
}
|
|
Cmd::Save { session_id, role, content, tokens_in, tokens_out, cost } => {
|
|
let id = save_message(&s, &ChatMessage {
|
|
session_id, role, content, tokens_in, tokens_out, cost,
|
|
..Default::default()
|
|
})?;
|
|
println!("{}", id);
|
|
}
|
|
Cmd::Search { query, limit } => {
|
|
for m in search(&s, &query, limit)? {
|
|
println!("{}\t{}\t{}", m.id, m.role, m.content);
|
|
}
|
|
}
|
|
Cmd::Archive { session_id } => {
|
|
archive_session(&s, &session_id)?;
|
|
println!("archived {}", session_id);
|
|
}
|
|
Cmd::Stats => {
|
|
let st = stats(&s)?;
|
|
println!("{}", serde_json::to_string_pretty(&st)?);
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
fn main() -> ExitCode {
|
|
match run() {
|
|
Ok(()) => ExitCode::SUCCESS,
|
|
Err(e) => { eprintln!("kei-chat-store: {e:#}"); ExitCode::from(1) }
|
|
}
|
|
}
|