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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

use super::{
    output::{emit, failure},
    state,
};
use crate::render::{RenderFormat, TerminalEvent};
use saya_store::{AuditOperation, AuditStatus, SchemaStore, SqliteStateStore};
use std::time::Instant;

pub(super) async fn fallback(
    store: &SqliteStateStore,
    identity: &str,
    started: Instant,
    error: saya_types::ConnectionError,
    format: RenderFormat,
    persistence_failed: &mut bool,
) -> Result<i32, Box<dyn std::error::Error>> {
    match store.get_schema(identity).await {
        Ok(Some(cached)) => {
            *persistence_failed |= state::audit_silent(
                store,
                identity,
                AuditOperation::SchemaRefresh,
                AuditStatus::Cached,
                started.elapsed(),
                None,
                None,
            )
            .await
            .is_err();
            warn(*persistence_failed, format);
            emit(
                TerminalEvent::Diagnostic {
                    message: "Using cached schema metadata; it may be stale.".into(),
                },
                format,
            );
            emit(
                TerminalEvent::Schema {
                    schema: cached.schema,
                },
                format,
            );
            Ok(0)
        }
        Err(_) => {
            *persistence_failed = true;
            fail(store, identity, started, error, format, persistence_failed).await
        }
        Ok(None) => fail(store, identity, started, error, format, persistence_failed).await,
    }
}

async fn fail(
    store: &SqliteStateStore,
    identity: &str,
    started: Instant,
    error: saya_types::ConnectionError,
    format: RenderFormat,
    persistence_failed: &mut bool,
) -> Result<i32, Box<dyn std::error::Error>> {
    *persistence_failed |= state::audit_silent(
        store,
        identity,
        AuditOperation::SchemaRefresh,
        AuditStatus::Failure,
        started.elapsed(),
        None,
        None,
    )
    .await
    .is_err();
    warn(*persistence_failed, format);
    failure(3, error, format)
}

fn warn(persistence_failed: bool, format: RenderFormat) {
    if persistence_failed {
        state::diagnostic(format);
    }
}