shellql 0.1.7-beta

A Vim- and tmux-inspired terminal database manager for developers
Documentation 
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

use std::{
    collections::HashMap,
    time::{Duration, Instant},
};

use crate::connection::{Database, DbPool, list_connections};

use super::cmdline::CommandLine;
use super::form::AddConnectionForm;
use super::pane::Overlay;
use super::tab::{LoadedTable, Tab};

#[derive(PartialEq, Eq, Clone, Copy, Debug)]
pub enum AppMode {
    Home,
    Dashboard,
}

pub struct AppState {
    pub mode: AppMode,
    pub overlay: Option<Overlay>,
    pub should_quit: bool,
    pub connections: Vec<Database>,
    pub selected_connection: usize,
    pub pending_key: Option<char>,
    pub cmdline: CommandLine,
    /// Populated when `Overlay::AddConnection` is active.
    pub form: Option<AddConnectionForm>,

    // ── Shared database session state ─────────────────────────────────────
    /// Active connection metadata.
    pub connection: Option<Database>,
    /// Live DB pool — cheap to clone for async ops.
    pub pool: Option<DbPool>,
    /// Schema table list from the current connection.
    pub tables: Vec<String>,
    /// Shared table cache across all tabs.
    pub table_cache: HashMap<String, LoadedTable>,

    // ── Tabs ──────────────────────────────────────────────────────────────
    pub tabs: Vec<Tab>,
    pub active_tab: usize,

    /// Set by the connection picker; handled in the event loop with a spinner.
    pub pending_connection: Option<Database>,

    /// Live-refresh TableView panes from the database.
    pub live_table_refresh_enabled: bool,
    pub live_table_refresh_interval: Duration,
    pub last_live_table_refresh_at: Option<Instant>,
}

impl AppState {
    pub fn new() -> Self {
        Self {
            mode: AppMode::Home,
            overlay: None,
            should_quit: false,
            connections: list_connections(),
            selected_connection: 0,
            pending_key: None,
            cmdline: CommandLine::new(),
            form: None,
            connection: None,
            pool: None,
            tables: Vec::new(),
            table_cache: HashMap::new(),
            tabs: Vec::new(),
            active_tab: 0,
            pending_connection: None,
            live_table_refresh_enabled: true,
            live_table_refresh_interval: Duration::from_millis(1000),
            last_live_table_refresh_at: None,
        }
    }

    /// True if we're in a database session (tabs exist).
    pub fn has_session(&self) -> bool {
        !self.tabs.is_empty()
    }

    /// Mutable reference to the active tab, if any.
    pub fn active_tab_mut(&mut self) -> Option<&mut Tab> {
        self.tabs.get_mut(self.active_tab)
    }

    /// Reference to the active tab, if any.
    pub fn active_tab(&self) -> Option<&Tab> {
        self.tabs.get(self.active_tab)
    }
}