saya-cli 0.3.2

Database-aware AI agent for the terminal: full-screen TUI, schema discovery, and bounded read-only SQL over PostgreSQL, MySQL, SQLite, DuckDB, and Snowflake.
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

use super::output::emit;
use crate::{
    profile_identity::profile_identity,
    render::{RenderFormat, TerminalEvent},
};
use saya_store::{
    AuditEntry, AuditOperation, AuditStatus, AuditStore, SqliteStateStore, StoreError,
};
use saya_types::DatabaseProfile;
use std::time::Duration;

pub(super) fn identity(name: &str, profile: &DatabaseProfile, scope: &std::path::Path) -> String {
    profile_identity(name, profile, scope).as_str().to_owned()
}

#[allow(clippy::too_many_arguments)]
pub(super) async fn audit(
    store: &SqliteStateStore,
    profile_id: &str,
    operation: AuditOperation,
    status: AuditStatus,
    elapsed: Duration,
    rows: Option<usize>,
    truncated: Option<bool>,
    format: RenderFormat,
) {
    if audit_silent(
        store, profile_id, operation, status, elapsed, rows, truncated,
    )
    .await
    .is_err()
    {
        diagnostic(format);
    }
}

#[allow(clippy::too_many_arguments)]
pub(super) async fn audit_silent(
    store: &SqliteStateStore,
    profile_id: &str,
    operation: AuditOperation,
    status: AuditStatus,
    elapsed: Duration,
    rows: Option<usize>,
    truncated: Option<bool>,
) -> Result<(), StoreError> {
    let mut event = AuditEntry::new(profile_id, operation, status, elapsed.as_millis() as u64);
    event.row_count = rows;
    event.truncated = truncated;
    store.record_audit(event).await
}

pub(super) fn diagnostic(format: RenderFormat) {
    emit(
        TerminalEvent::Diagnostic {
            message: "Local state store unavailable; continuing without persistence.".into(),
        },
        format,
    );
}