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

use sea_query::{Condition, Expr, Query, SelectStatement, SimpleExpr};

use super::query::{SqliteMaster, SqliteSchema};
use super::Sqlite;
use crate::probe::{Has, Schema, SchemaProbe};

impl SchemaProbe for Sqlite {
    fn get_current_schema() -> SimpleExpr {
        unimplemented!()
    }

    fn query_tables() -> SelectStatement {
        Query::select()
            .expr_as(Expr::col(SqliteSchema::Name), Schema::TableName)
            .from(SqliteMaster)
            .cond_where(
                Condition::all()
                    .add(Expr::col(SqliteSchema::Type).eq("table"))
                    .add(Expr::col(SqliteSchema::Name).ne("sqlite_sequence")),
            )
            .take()
    }

    fn has_column<T, C>(table: T, column: C) -> SelectStatement
    where
        T: AsRef<str>,
        C: AsRef<str>,
    {
        Query::select()
            .expr(Expr::cust_with_values(
                "COUNT(*) > 0 AS \"has_column\" FROM pragma_table_info(?)",
                [table.as_ref()],
            ))
            .and_where(Expr::col(SqliteSchema::Name).eq(column.as_ref()))
            .take()
    }

    fn has_index<T, C>(table: T, index: C) -> SelectStatement
    where
        T: AsRef<str>,
        C: AsRef<str>,
    {
        Query::select()
            .expr_as(Expr::cust("COUNT(*) > 0"), Has::Index)
            .from(SqliteMaster)
            .cond_where(
                Condition::all()
                    .add(Expr::col(SqliteSchema::Type).eq("index"))
                    .add(Expr::col(SqliteSchema::TblName).eq(table.as_ref()))
                    .add(Expr::col(SqliteSchema::Name).eq(index.as_ref())),
            )
            .take()
    }
}