sabiql 2.0.0

A fast, driver-less TUI for browsing and editing PostgreSQL and SQLite 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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146

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

pub fn minimal_table(schema: &str, name: &str) -> Table {
    Table {
        schema: schema.to_string(),
        name: name.to_string(),
        owner: None,
        columns: Vec::new(),
        primary_key: None,
        foreign_keys: Vec::new(),
        indexes: Vec::new(),
        rls: None,
        triggers: Vec::new(),
        row_count_estimate: None,
        comment: None,
        source_ddl: None,
        kind_info: TableKindInfo::default(),
    }
}

pub fn view_kind_info() -> TableKindInfo {
    TableKindInfo {
        kind: TableKind::View,
        ..TableKindInfo::default()
    }
}

pub fn sample_metadata() -> DatabaseMetadata {
    let mut metadata = DatabaseMetadata::new("test_db".to_string());
    metadata.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,
        ),
    ];
    metadata
}

pub fn sample_table_detail() -> Table {
    let mut table = minimal_table("public", "users");
    table.owner = Some("postgres".to_string());
    table.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,
        },
    ];
    table.primary_key = Some(vec!["id".to_string()]);
    table.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,
        },
    ];
    table.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,
        reference_resolved: true,
    }];
    table.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,
    }];
    table.row_count_estimate = Some(100);
    table.comment = Some("User accounts".to_string());
    table
}

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,
    )
}