crude 0.1.4

Migration toolkit for databases
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

use anstream::println;
use chrono::Utc;
use clap::Parser;
use eyre::eyre;
use owo_colors::OwoColorize;
use sha2::{Digest, Sha256};
use tracing::instrument;

use crate::{
    Options,
    db::get_db_adapter,
    error::Result,
    migration::{
        dir::get_migrations_dir,
        planner::{MigrationState, Planner},
    },
};

/// Squash all applied migrations into a single baseline migration
#[derive(Debug, Parser)]
pub struct Rollup {}

impl Rollup {
    #[instrument(name = "rollup", skip_all)]
    pub(crate) fn run(&self, opts: &Options) -> Result {
        let migrations_dir = get_migrations_dir(opts);
        let local = migrations_dir.load()?;

        let mut db = get_db_adapter(opts, false)?;

        // Error out if status is not clean
        if Planner::new(opts)?
            .status()?
            .iter()
            .any(|s| s.state != MigrationState::Applied)
        {
            return Err(eyre!(
                "cannot rollup when there are pending, variant, or divergent migrations"
            ));
        }

        let url = opts.get_url()?;

        // Dump current schema excluding the migrations table
        let up_sql = String::from_utf8_lossy(&db.dump_schema(url, true)?).into_owned();

        // Build baseline migration
        let ts = Utc::now();
        let compound_name = format!("{}_rollup", ts.format("%Y%m%d%H%M%S"));

        let mut hasher = Sha256::new();
        hasher.update(up_sql.as_bytes());
        let hash = hex::encode(hasher.finalize());

        // Dump data-only SQL for seed (exclude migrations table)
        let seed_sql = String::from_utf8_lossy(&db.dump_data(url, true)?).into_owned();

        // Create the rollup migration
        migrations_dir.create_migration(&compound_name, Some(&up_sql), Some(&seed_sql))?;

        // Reset migration history and apply baseline record
        db.clear_migrations()?;
        db.record_baseline(&compound_name, &hash)?;

        // Remove all other migrations including previous rollup
        local
            .iter()
            .filter(|m| m.name != "init")
            .try_for_each(|m| {
                println!("{} {}", "Rolled up".cyan(), m.compound_name);

                migrations_dir.remove_migration(&m.compound_name)
            })?;

        Ok(())
    }
}