sabiql 1.15.0

A fast, driver-less TUI for browsing and editing PostgreSQL databases
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124

use sabiql_domain::{
    Column, ColumnAttributes, DatabaseMetadata, FkAction, ForeignKey, Index, IndexAttributes,
    IndexType, QueryResult, QuerySource, Table, TableSummary, Trigger, TriggerEvent, TriggerTiming,
};

pub fn sample_metadata() -> DatabaseMetadata {
    DatabaseMetadata {
        database_name: "test_db".to_string(),
        schemas: vec![],
        table_summaries: vec![
            TableSummary::new("public".to_string(), "users".to_string(), Some(100), false),
            TableSummary::new("public".to_string(), "posts".to_string(), Some(50), false),
            TableSummary::new(
                "public".to_string(),
                "comments".to_string(),
                Some(200),
                false,
            ),
        ],
    }
}

pub fn sample_table_detail() -> Table {
    Table {
        schema: "public".to_string(),
        name: "users".to_string(),
        owner: Some("postgres".to_string()),
        columns: vec![
            Column {
                name: "id".to_string(),
                data_type: "integer".to_string(),
                attributes: ColumnAttributes::PRIMARY_KEY | ColumnAttributes::UNIQUE,
                default: None,
                comment: Some("Primary key".to_string()),
                ordinal_position: 1,
            },
            Column {
                name: "name".to_string(),
                data_type: "varchar(255)".to_string(),
                attributes: ColumnAttributes::empty(),
                default: None,
                comment: None,
                ordinal_position: 2,
            },
            Column {
                name: "email".to_string(),
                data_type: "varchar(255)".to_string(),
                attributes: ColumnAttributes::NULLABLE | ColumnAttributes::UNIQUE,
                default: None,
                comment: None,
                ordinal_position: 3,
            },
        ],
        primary_key: Some(vec!["id".to_string()]),
        indexes: vec![
            Index {
                name: "users_pkey".to_string(),
                columns: vec!["id".to_string()],
                attributes: IndexAttributes::UNIQUE | IndexAttributes::PRIMARY,
                index_type: IndexType::BTree,
                definition: None,
            },
            Index {
                name: "idx_users_email".to_string(),
                columns: vec!["email".to_string()],
                attributes: IndexAttributes::UNIQUE,
                index_type: IndexType::BTree,
                definition: None,
            },
        ],
        foreign_keys: vec![ForeignKey {
            name: "fk_users_department".to_string(),
            from_schema: "public".to_string(),
            from_table: "users".to_string(),
            from_columns: vec!["department_id".to_string()],
            to_schema: "public".to_string(),
            to_table: "departments".to_string(),
            to_columns: vec!["id".to_string()],
            on_delete: FkAction::Cascade,
            on_update: FkAction::NoAction,
        }],
        rls: None,
        triggers: vec![Trigger {
            name: "audit_users".to_string(),
            timing: TriggerTiming::After,
            events: vec![TriggerEvent::Insert, TriggerEvent::Update],
            function_name: "audit_func".to_string(),
            security_definer: false,
        }],
        row_count_estimate: Some(100),
        comment: Some("User accounts".to_string()),
    }
}

pub fn sample_query_result() -> QueryResult {
    QueryResult::success(
        "SELECT * FROM users LIMIT 100".to_string(),
        vec!["id".to_string(), "name".to_string(), "email".to_string()],
        vec![
            vec![
                "1".to_string(),
                "Alice".to_string(),
                "alice@example.com".to_string(),
            ],
            vec![
                "2".to_string(),
                "Bob".to_string(),
                "bob@example.com".to_string(),
            ],
        ],
        15,
        QuerySource::Preview,
    )
}

pub fn empty_query_result() -> QueryResult {
    QueryResult::success(
        "SELECT * FROM users WHERE 1=0".to_string(),
        vec!["id".to_string(), "name".to_string(), "email".to_string()],
        vec![],
        5,
        QuerySource::Preview,
    )
}