rustploy 0.1.14

Rustploy PaaS daemon — lightweight self-hosted deployment platform
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

//! Troca de uma env var de conexão (`DATABASE_URL`…) num projeto, com o valor
//! antigo guardado para rollback.
//!
//! Uma chave pode estar no **projeto** e/ou em **serviços** (o serviço
//! sobrescreve o projeto no deploy). Trocar só um dos lugares deixaria o outro
//! apontando para o banco antigo — a falha silenciosa que o doc de desenho
//! descreve. Então a troca vai em todo lugar onde a chave já existe; se não
//! existe em lugar nenhum, entra no projeto.

use crate::store::Db;
use anyhow::Result;
use shared::{EnvChange, EnvVar, EnvVarValue};

/// A chave já está definida no projeto ou em algum serviço dele?
pub async fn is_defined(db: &Db, project_id: &str, key: &str) -> Result<bool> {
    if let Some(p) = crate::store::projects::get(db, project_id).await?
        && p.env_vars.iter().any(|e| e.key == key)
    {
        return Ok(true);
    }
    Ok(crate::store::services::list(db, project_id)
        .await?
        .iter()
        .any(|s| s.spec.env_vars.iter().any(|e| e.key == key)))
}

/// Grava `value` (Plain) em todo lugar onde `key` existe; se em nenhum, no projeto.
pub async fn switch(db: &Db, project_id: &str, key: &str, value: &str) -> Result<Vec<EnvChange>> {
    let new = EnvVarValue::Plain(value.to_string());
    let mut changes = vec![];

    for svc in crate::store::services::list(db, project_id).await? {
        let mut spec = svc.spec.clone();
        let Some(e) = spec.env_vars.iter_mut().find(|e| e.key == key) else {
            continue;
        };
        changes.push(EnvChange {
            service_id: Some(svc.id.clone()),
            key: key.to_string(),
            previous: Some(std::mem::replace(&mut e.value, new.clone())),
        });
        crate::store::services::update_spec(db, &svc.id, spec).await?;
    }

    let Some(project) = crate::store::projects::get(db, project_id).await? else {
        anyhow::bail!("projeto não encontrado");
    };
    let mut vars = project.env_vars.clone();
    let project_has = vars.iter().any(|e| e.key == key);
    if project_has || changes.is_empty() {
        let previous = match vars.iter_mut().find(|e| e.key == key) {
            Some(e) => Some(std::mem::replace(&mut e.value, new.clone())),
            None => {
                vars.push(EnvVar {
                    key: key.to_string(),
                    value: new.clone(),
                });
                None
            }
        };
        crate::store::projects::update_env_vars(db, project_id, vars, project.env_comments).await?;
        changes.push(EnvChange {
            service_id: None,
            key: key.to_string(),
            previous,
        });
    }
    Ok(changes)
}

/// Desfaz [`switch`]: devolve o valor antigo (ou remove a chave criada).
pub async fn restore(db: &Db, project_id: &str, changes: &[EnvChange]) -> Result<()> {
    for c in changes {
        match &c.service_id {
            Some(sid) => {
                let Some(svc) = crate::store::services::get(db, sid).await? else {
                    continue;
                };
                let mut spec = svc.spec.clone();
                match &c.previous {
                    Some(v) => {
                        if let Some(e) = spec.env_vars.iter_mut().find(|e| e.key == c.key) {
                            e.value = v.clone();
                        }
                    }
                    None => spec.env_vars.retain(|e| e.key != c.key),
                }
                crate::store::services::update_spec(db, sid, spec).await?;
            }
            None => {
                let Some(project) = crate::store::projects::get(db, project_id).await? else {
                    continue;
                };
                let mut vars = project.env_vars.clone();
                match &c.previous {
                    Some(v) => {
                        if let Some(e) = vars.iter_mut().find(|e| e.key == c.key) {
                            e.value = v.clone();
                        }
                    }
                    None => vars.retain(|e| e.key != c.key),
                }
                crate::store::projects::update_env_vars(db, project_id, vars, project.env_comments)
                    .await?;
            }
        }
    }
    Ok(())
}