yggdrasil/api/settings/
backup.rs

1// 与 posts 模块一致：Dioxus `#[server]` 宏触发 deprecated/unit/too_many_arguments
2// 提示，按项目惯例放行（限流/运行器等配置项天然参数多）。
3#![allow(clippy::unused_unit, deprecated, clippy::too_many_arguments)]
4
5use dioxus::prelude::*;
6
7#[cfg(feature = "server")]
8use crate::api::auth::get_current_admin_user;
9#[cfg(feature = "server")]
10use crate::api::error::AppError;
11#[cfg(feature = "server")]
12use crate::db::pool::get_conn;
13use crate::models::settings::BackupSettingsView;
14// 仅 server 构建的函数体引用（WASM 端 server fn 体被宏剥离）。
15#[cfg(feature = "server")]
16use crate::models::settings::{BackupSettings, LastBackupRun};
17
18// ============================================================================
19// 自动备份配置
20// ============================================================================
21
22/// 从 settings 表读取自动备份配置（缺键回退默认值）。
23///
24/// 供 `get_backup_settings`、备份核心（include_uploads / retention_count）
25/// 与定时调度任务三处共用。
26#[cfg(feature = "server")]
27pub(crate) async fn load_backup_settings(
28    client: &tokio_postgres::Client,
29) -> Result<BackupSettings, AppError> {
30    let values = super::load_setting_values(
31        client,
32        &[
33            "backup_auto_enabled",
34            "backup_time_utc",
35            "backup_retention_count",
36            "backup_include_uploads",
37        ],
38    )
39    .await?;
40
41    let auto_enabled = values
42        .get("backup_auto_enabled")
43        .and_then(|v| v.parse().ok())
44        .unwrap_or(crate::models::settings::DEFAULT_BACKUP_AUTO_ENABLED);
45    let time_utc = values
46        .get("backup_time_utc")
47        .map(|v| BackupSettings::normalize_time_utc(v))
48        .unwrap_or_else(|| crate::models::settings::DEFAULT_BACKUP_TIME_UTC.to_string());
49    let retention_count = values
50        .get("backup_retention_count")
51        .and_then(|v| v.parse().ok())
52        .map(BackupSettings::clamp_retention)
53        .unwrap_or(crate::models::settings::DEFAULT_BACKUP_RETENTION_COUNT);
54    let include_uploads = values
55        .get("backup_include_uploads")
56        .and_then(|v| v.parse().ok())
57        .unwrap_or(crate::models::settings::DEFAULT_BACKUP_INCLUDE_UPLOADS);
58
59    Ok(BackupSettings {
60        auto_enabled,
61        time_utc,
62        retention_count,
63        include_uploads,
64    })
65}
66
67/// 读取最近一次自动备份结果（未执行过 → None）。
68#[cfg(feature = "server")]
69async fn load_last_backup_run(
70    client: &tokio_postgres::Client,
71) -> Result<Option<LastBackupRun>, AppError> {
72    let rows = client
73        .query(
74            "SELECT key, value FROM settings WHERE key = ANY(ARRAY[
75                'backup_last_run_at', 'backup_last_run_ok',
76                'backup_last_run_file', 'backup_last_run_error'])",
77            &[],
78        )
79        .await
80        .map_err(AppError::query)?;
81    let get = |k: &str| {
82        rows.iter()
83            .find(|r| r.get::<_, String>("key") == k)
84            .map(|r| r.get::<_, String>("value"))
85            .filter(|v| !v.is_empty())
86    };
87    let (Some(at), Some(ok)) = (get("backup_last_run_at"), get("backup_last_run_ok")) else {
88        return Ok(None);
89    };
90    Ok(Some(LastBackupRun {
91        at,
92        ok: ok.parse().unwrap_or(false),
93        file: get("backup_last_run_file"),
94        error: get("backup_last_run_error"),
95    }))
96}
97
98/// 落库最近一次自动备份结果（调度任务执行后调用；空串表示 None）。
99#[cfg(feature = "server")]
100pub(crate) async fn save_last_backup_run(
101    client: &tokio_postgres::Client,
102    run: &LastBackupRun,
103) -> Result<(), AppError> {
104    let pairs: [(&str, String); 4] = [
105        ("backup_last_run_at", run.at.clone()),
106        ("backup_last_run_ok", run.ok.to_string()),
107        ("backup_last_run_file", run.file.clone().unwrap_or_default()),
108        (
109            "backup_last_run_error",
110            run.error.clone().unwrap_or_default(),
111        ),
112    ];
113    for (key, value) in pairs {
114        client
115            .execute(
116                "INSERT INTO settings (key, value, updated_at) VALUES ($1, $2, NOW())
117                 ON CONFLICT (key) DO UPDATE SET value = EXCLUDED.value, updated_at = NOW()",
118                &[&key, &value],
119            )
120            .await
121            .map_err(AppError::query)?;
122    }
123    Ok(())
124}
125
126/// 启动时用 `BACKUP_*` 环境变量播种自动备份配置。
127///
128/// 仅当对应 settings 键**不存在**时插入（`ON CONFLICT DO NOTHING`）——
129/// 即 env 只决定首次部署的初始值，之后以后台面板写入的 DB 值为准，
130/// 重启不会被 env 覆盖。单个变量非法只告警跳过，不影响其他变量与启动。
131#[cfg(feature = "server")]
132pub(crate) async fn seed_backup_settings_from_env(
133    client: &tokio_postgres::Client,
134) -> Result<(), AppError> {
135    let mut seeds: Vec<(&'static str, String)> = Vec::new();
136
137    if let Ok(v) = std::env::var("BACKUP_AUTO_ENABLED") {
138        match v.trim().parse::<bool>() {
139            Ok(b) => seeds.push(("backup_auto_enabled", b.to_string())),
140            Err(_) => tracing::warn!("BACKUP_AUTO_ENABLED={v:?} 非法（期望 true/false），跳过"),
141        }
142    }
143    if let Ok(v) = std::env::var("BACKUP_TIME_UTC") {
144        if BackupSettings::is_valid_time_utc(&v) {
145            seeds.push(("backup_time_utc", BackupSettings::normalize_time_utc(&v)));
146        } else {
147            tracing::warn!("BACKUP_TIME_UTC={v:?} 非法（期望 HH:MM，UTC），跳过");
148        }
149    }
150    if let Ok(v) = std::env::var("BACKUP_RETENTION_COUNT") {
151        match v.trim().parse::<i32>() {
152            Ok(n) => seeds.push((
153                "backup_retention_count",
154                BackupSettings::clamp_retention(n).to_string(),
155            )),
156            Err(_) => tracing::warn!("BACKUP_RETENTION_COUNT={v:?} 非法（期望整数），跳过"),
157        }
158    }
159    if let Ok(v) = std::env::var("BACKUP_INCLUDE_UPLOADS") {
160        match v.trim().parse::<bool>() {
161            Ok(b) => seeds.push(("backup_include_uploads", b.to_string())),
162            Err(_) => tracing::warn!("BACKUP_INCLUDE_UPLOADS={v:?} 非法（期望 true/false），跳过"),
163        }
164    }
165
166    super::insert_env_seeds(client, seeds).await
167}
168
169/// 组装面板视图：设置 + 上次结果 + 下次执行时间。
170#[cfg(feature = "server")]
171async fn load_backup_settings_view(
172    client: &tokio_postgres::Client,
173) -> Result<BackupSettingsView, AppError> {
174    let settings = load_backup_settings(client).await?;
175    let last_run = load_last_backup_run(client).await?;
176    let next_run_at = if settings.auto_enabled {
177        settings
178            .next_run_after(chrono::Utc::now())
179            .map(|dt| dt.to_rfc3339())
180    } else {
181        None
182    };
183    Ok(BackupSettingsView {
184        settings,
185        last_run,
186        next_run_at,
187    })
188}
189
190/// 读取自动备份配置（面板用）。
191#[server(GetBackupSettings, "/api")]
192pub async fn get_backup_settings() -> Result<BackupSettingsView, ServerFnError> {
193    let _user = get_current_admin_user().await?;
194
195    #[cfg(feature = "server")]
196    {
197        let client = get_conn().await.map_err(AppError::db_conn)?;
198        Ok(load_backup_settings_view(&client).await?)
199    }
200
201    #[cfg(not(feature = "server"))]
202    {
203        Ok(BackupSettingsView {
204            settings: BackupSettings::default(),
205            last_run: None,
206            next_run_at: None,
207        })
208    }
209}
210
211/// 更新自动备份配置。
212///
213/// time_utc 会被规范化（非法回退默认），retention_count 会被 clamp。
214/// 写入后唤醒调度任务立即重排，无需等待下一次 tick。
215#[server(UpdateBackupSettings, "/api")]
216pub async fn update_backup_settings(
217    auto_enabled: bool,
218    time_utc: String,
219    retention_count: i32,
220    include_uploads: bool,
221) -> Result<BackupSettingsView, ServerFnError> {
222    let _user = get_current_admin_user().await?;
223
224    let time_utc = BackupSettings::normalize_time_utc(&time_utc);
225    let retention_count = BackupSettings::clamp_retention(retention_count);
226
227    #[cfg(feature = "server")]
228    {
229        let client = get_conn().await.map_err(AppError::db_conn)?;
230        let pairs = [
231            ("backup_auto_enabled", auto_enabled.to_string()),
232            ("backup_time_utc", time_utc.clone()),
233            ("backup_retention_count", retention_count.to_string()),
234            ("backup_include_uploads", include_uploads.to_string()),
235        ];
236        for (key, value) in pairs {
237            client
238                .execute(
239                    "INSERT INTO settings (key, value, updated_at) VALUES ($1, $2, NOW())
240                     ON CONFLICT (key) DO UPDATE SET value = EXCLUDED.value, updated_at = NOW()",
241                    &[&key, &value],
242                )
243                .await
244                .map_err(AppError::query)?;
245        }
246        tracing::info!(
247            "Backup settings updated: auto_enabled={}, time_utc={}, retention={}, include_uploads={}",
248            auto_enabled,
249            time_utc,
250            retention_count,
251            include_uploads
252        );
253        // 唤醒调度器立即重排（设置变更不应等到原定的下次触发才生效）。
254        crate::tasks::backup::notify_settings_changed();
255        Ok(load_backup_settings_view(&client).await?)
256    }
257
258    #[cfg(not(feature = "server"))]
259    {
260        Ok(BackupSettingsView {
261            settings: BackupSettings {
262                auto_enabled,
263                time_utc,
264                retention_count,
265                include_uploads,
266            },
267            last_run: None,
268            next_run_at: None,
269        })
270    }
271}
yggdrasil/api/settings/backup.rs

yggdrasil/api/settings/
backup.rs
