parsql-cli 0.5.0

Command-line interface for parsql database toolkit
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
125
126
127

//! Configuration handling for parsql CLI

use anyhow::{Context, Result};
use serde::{Deserialize, Serialize};
use std::fs;
use std::path::Path;

#[derive(Debug, Serialize, Deserialize, Default)]
pub struct Config {
    #[serde(default)]
    pub migrations: MigrationConfig,
    
    #[serde(default)]
    pub database: Option<DatabaseConfig>,
    
    #[serde(skip)]
    pub database_url: Option<String>,
}

#[derive(Debug, Serialize, Deserialize)]
pub struct MigrationConfig {
    #[serde(default = "default_migrations_dir")]
    pub directory: String,
    
    #[serde(default = "default_table_name")]
    pub table_name: String,
    
    #[serde(default = "default_true")]
    pub transaction_per_migration: bool,
    
    #[serde(default)]
    pub allow_out_of_order: bool,
    
    #[serde(default = "default_true")]
    pub verify_checksums: bool,
    
    #[serde(default)]
    pub auto_create_table: Option<bool>,
}

#[derive(Debug, Serialize, Deserialize)]
pub struct DatabaseConfig {
    pub url: String,
}

impl MigrationConfig {
    pub fn to_parsql_migrations_config(&self) -> parsql_migrations::config::MigrationConfig {
        let mut config = parsql_migrations::config::MigrationConfig::default();
        config.table.table_name = self.table_name.clone();
        config.transaction_per_migration = self.transaction_per_migration;
        config.verify_checksums = self.verify_checksums;
        config.allow_out_of_order = self.allow_out_of_order;
        config.auto_create_table = self.auto_create_table.unwrap_or(true);
        config
    }
}

impl Default for MigrationConfig {
    fn default() -> Self {
        Self {
            directory: default_migrations_dir(),
            table_name: default_table_name(),
            transaction_per_migration: true,
            allow_out_of_order: false,
            verify_checksums: true,
            auto_create_table: Some(true),
        }
    }
}

fn default_migrations_dir() -> String {
    "migrations".to_string()
}

fn default_table_name() -> String {
    "parsql_migrations".to_string()
}

fn default_true() -> bool {
    true
}

pub fn load_config(path: &str) -> Result<Config> {
    let config_path = Path::new(path);
    
    if !config_path.exists() {
        // Return default config if file doesn't exist
        return Ok(Config::default());
    }
    
    let contents = fs::read_to_string(config_path)
        .with_context(|| format!("Failed to read config file: {}", path))?;
    
    let mut config: Config = toml::from_str(&contents)
        .with_context(|| format!("Failed to parse config file: {}", path))?;
    
    // If database URL is in config, use it
    if let Some(ref db_config) = config.database {
        config.database_url = Some(db_config.url.clone());
    }
    
    Ok(config)
}

impl Config {
    pub fn to_parsql_migration_config(&self) -> parsql_migrations::MigrationConfig {
        let mut config = parsql_migrations::MigrationConfig::default();
        
        config.table.table_name = self.migrations.table_name.clone();
        config.transaction_per_migration = self.migrations.transaction_per_migration;
        config.allow_out_of_order = self.migrations.allow_out_of_order;
        config.verify_checksums = self.migrations.verify_checksums;
        
        if let Some(auto_create) = self.migrations.auto_create_table {
            config.auto_create_table = auto_create;
        }
        
        config
    }

    /// Create a default config with custom migrations directory (useful for testing)
    pub fn default_with_directory(directory: &str) -> Self {
        let mut config = Config::default();
        config.migrations.directory = directory.to_string();
        config
    }
}