cargo-gears-core 0.0.1

Core functionality library for cargo-gears
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

use super::{ensure_conn_payload, load_config, save_config, validate_name};
use crate::app_config::DbConnConfig;
use crate::common::PathConfigParams;
use anyhow::{Context, bail};

#[derive(Debug, Eq, PartialEq)]
pub struct DbParams {
    pub command: DbCommand,
}

#[derive(Debug, Eq, PartialEq)]
pub enum DbCommand {
    /// Add or update (upsert) a global database server config under `database.servers`
    Add(AddArgs),
    /// Edit a global database server config under `database.servers`
    Edit(EditArgs),
    /// Remove a global database server config from `database.servers`
    Rm(RemoveArgs),
}

impl DbParams {
    pub fn run(&self) -> anyhow::Result<()> {
        match &self.command {
            DbCommand::Add(args) => args.run(),
            DbCommand::Edit(args) => args.run(),
            DbCommand::Rm(args) => args.run(),
        }
    }
}

#[derive(Debug, Eq, PartialEq)]
pub struct AddArgs {
    pub path_config: PathConfigParams,
    /// Server name under `database.servers.<name>`
    pub name: String,
    pub conn: DbConnConfig,
}

impl AddArgs {
    fn run(&self) -> anyhow::Result<()> {
        let config_path = self.path_config.resolve_config()?;
        validate_name(&self.name, "server")?;
        ensure_conn_payload(&self.conn)?;

        let mut config = load_config(&config_path)?;
        let database = config.database.get_or_insert_default();
        if let Some(existing) = database.servers.get_mut(&self.name) {
            existing.apply_patch(self.conn.clone());
        } else {
            database
                .servers
                .insert(self.name.clone(), self.conn.clone());
        }
        save_config(&config_path, &config)
    }
}

#[derive(Debug, Eq, PartialEq)]
pub struct EditArgs {
    pub path_config: PathConfigParams,
    /// Server name under `database.servers.<name>`
    pub name: String,
    pub conn: DbConnConfig,
}

impl EditArgs {
    fn run(&self) -> anyhow::Result<()> {
        let config_path = self.path_config.resolve_config()?;
        validate_name(&self.name, "server")?;
        ensure_conn_payload(&self.conn)?;

        let mut config = load_config(&config_path)?;
        let database = config
            .database
            .as_mut()
            .context("global database config is missing; use `config db add` first")?;
        let existing = database.servers.get_mut(&self.name).with_context(|| {
            format!(
                "database server '{}' not found in {}",
                self.name,
                config_path.display()
            )
        })?;
        existing.apply_patch(self.conn.clone());

        save_config(&config_path, &config)
    }
}

#[derive(Debug, Eq, PartialEq)]
pub struct RemoveArgs {
    pub path_config: PathConfigParams,
    /// Server name under `database.servers.<name>`
    pub name: String,
}

impl RemoveArgs {
    fn run(&self) -> anyhow::Result<()> {
        let config_path = self.path_config.resolve_config()?;
        validate_name(&self.name, "server")?;

        let mut config = load_config(&config_path)?;
        let Some(database) = config.database.as_mut() else {
            bail!("global database config is missing");
        };

        if database.servers.remove(&self.name).is_none() {
            let name = &self.name;
            bail!("database server '{name}' not found");
        }

        if database.servers.is_empty() && database.auto_provision.is_none() {
            config.database = None;
        }

        save_config(&config_path, &config)
    }
}