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

use std::{
    fmt::{
        Display,
        Debug,
    },
    collections::HashSet,
};
use crate::{
    utils::Tokens,
    sqlite::types::to_sql_type,
    graphmigrate::Comparison,
};
use super::{
    field::{
        FieldId,
        FieldDef,
    },
    utils::{
        SqliteNodeData,
        SqliteMigrateCtx,
        SqliteNodeDataDispatch,
    },
    node::{
        Id,
        Node,
    },
};

#[derive(Clone, Eq, PartialEq, Hash, Debug)]
pub struct TableId(pub String);

impl Display for TableId {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        std::fmt::Display::fmt(&self.0, f)
    }
}

#[derive(Clone)]
pub struct NodeTable_ {
    pub id: TableId,
    pub fields: Vec<(FieldId, FieldDef)>,
}

impl NodeTable_ {
    pub fn compare(&self, _old: &Self, _created: &HashSet<Id>) -> Comparison {
        Comparison::DoNothing
    }
}

impl SqliteNodeData for NodeTable_ {
    fn update(&self, _ctx: &mut SqliteMigrateCtx, _old: &Self) {
        unreachable!();
    }
}

impl SqliteNodeDataDispatch for NodeTable_ {
    fn create_coalesce(&mut self, other: Node) -> Option<Node> {
        match other {
            Node::Field(f) if f.id.0 == self.id => {
                self.fields.push((f.id.clone(), f.def.clone()));
                None
            },
            other => Some(other),
        }
    }

    fn delete_coalesce(&mut self, other: Node) -> Option<Node> {
        match other {
            Node::Field(f) if f.id.0 == self.id => None,
            Node::Constraint(e) if e.id.0 == self.id => None,
            Node::Index(e) if e.id.0 == self.id => None,
            other => Some(other),
        }
    }

    fn create(&self, ctx: &mut SqliteMigrateCtx) {
        let mut stmt = Tokens::new();
        stmt.s("create table").id(&self.id.0).s("(");
        for (i, f) in self.fields.iter().filter(|f| &f.0.1 != "rowid").enumerate() {
            if i > 0 {
                stmt.s(",");
            }
            stmt.id(&f.0.1).s(to_sql_type(&f.1.type_.type_.type_.type_));
            if !f.1.type_.type_.opt {
                stmt.s("not null");
            }
        }
        stmt.s(")");
        ctx.statements.push(stmt.to_string());
    }

    fn delete(&self, ctx: &mut SqliteMigrateCtx) {
        ctx.statements.push(Tokens::new().s("drop table").id(&self.id.0).to_string());
    }
}