sqldiff 0.5.0

primitives to auto-generate SQL schema migrations
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
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 crate::schema::{Table, Type, Column};
use crate::schema::ToSql;


/// Create table action
#[derive(Debug)]
pub struct CreateTable {
    pub name: String,
    pub columns: Vec<Column>,
}

/// Create index action for a table
#[derive(Debug)]
pub struct CreateIndex {
    pub name: String,
    pub table_name: String,
}


/// Alter table action
#[derive(Debug)]
pub enum AlterAction {
    AddColumn {
        column: Column,
    },
}

#[derive(Debug)]
pub struct AlterTable {
    pub name: String,
    pub alter_action: AlterAction,
}


#[derive(Debug)]
pub enum Statement {
    CreateTable(CreateTable),
    CreateIndex(CreateIndex),
    AlterTable(AlterTable),
}

#[derive(Debug)]
pub enum DebugResults {
    TablesIdentical(String)
}

#[derive(Debug)]
pub struct Migration {
    pub statements: Vec<Statement>,
    pub debug_results: Vec<DebugResults>,
}

impl ToSql for Column {
    fn to_sql(&self) -> String {
        let mut sql = String::new();
        sql.push_str(&self.name);
        sql.push(' ');
        sql.push_str(&self.typ.to_sql());
        if !self.nullable {
            sql.push_str(" NOT NULL");
        }
        if self.primary_key {
            sql.push_str(" PRIMARY KEY");
        }
        sql
    }
}

impl CreateTable {
    pub fn from_table(table: &Table) -> CreateTable {
        CreateTable {
            name: table.name.clone(),
            columns: table.columns.clone(),
        }
    }
}

fn quote(s: &str) -> String {
    if s.contains('"') {
        panic!("Cannot quote string with double quotes");
    }
    format!("\"{}\"", s)
}

fn table_name(schema: &str, table: &str) -> String {
    format!("{}.{}", quote(schema), quote(table))
}

impl Statement {
    pub fn prepare(&self, schema_name: &str) -> String {
        use Statement::*;
        match self {
            CreateTable(create_table) => {
                let mut sql = String::new();
                sql.push_str("CREATE TABLE ");
                sql.push_str(&table_name(schema_name, &create_table.name));
                sql.push_str(" (\n");
                let mut first = true;
                for column in &create_table.columns {
                    if !first {
                        sql.push_str("\n, ");
                    }
                    sql.push_str(&column.to_sql());
                    first = false;
                }
                sql.push_str("\n)");
                sql
            }
            CreateIndex(create_index) => unimplemented!(),
            AlterTable(alter_table) => {
                let mut sql = String::new();
                sql.push_str("ALTER TABLE ");
                sql.push_str(&table_name(schema_name, &alter_table.name));
                match &alter_table.alter_action {
                    AlterAction::AddColumn { column } => {
                        sql.push_str(" ADD COLUMN ");
                        sql.push_str(&column.to_sql());
                    }
                }
                sql
            }
        }
    }
}