feat: add baby object match edutainment flow

server-rs/crates/api-server/src/state.rs

@@ -1,8 +1,9 @@
 use std::{
     collections::HashMap,
     error::Error,
-    fmt,
+    fmt, fs,
     sync::{Arc, Mutex},
+    time::{SystemTime, UNIX_EPOCH},
 };
 
 use module_ai::{AiTaskService, InMemoryAiTaskStore};
@@ -369,18 +370,18 @@ impl AppState {
             pool_size: config.spacetime_pool_size,
             procedure_timeout: config.spacetime_procedure_timeout,
         });
+        let mut candidates = Vec::new();
+
         match spacetime_client
             .export_auth_store_snapshot_from_tables()
             .await
         {
             Ok(snapshot) => {
-                if let Some(snapshot_json) = snapshot.snapshot_json {
-                    if !snapshot_json.trim().is_empty() {
-                        let auth_store = InMemoryAuthStore::from_snapshot_json(&snapshot_json)
-                            .map_err(AppStateInitError::AuthStore)?;
-                        info!("已从 SpacetimeDB 表恢复认证快照");
-                        return Self::new_with_auth_store(config, auth_store);
-                    }
+                if let Some(candidate) = auth_store_candidate_from_snapshot_record(
+                    snapshot,
+                    AuthStoreRestoreSource::SpacetimeTables,
+                )? {
+                    candidates.push(candidate);
                 }
             }
             Err(error) => {
@@ -390,13 +391,11 @@ impl AppState {
 
         match spacetime_client.get_auth_store_snapshot().await {
             Ok(snapshot) => {
-                if let Some(snapshot_json) = snapshot.snapshot_json {
-                    if !snapshot_json.trim().is_empty() {
-                        let auth_store = InMemoryAuthStore::from_snapshot_json(&snapshot_json)
-                            .map_err(AppStateInitError::AuthStore)?;
-                        info!("已从 SpacetimeDB 快照记录恢复认证快照");
-                        return Self::new_with_auth_store(config, auth_store);
-                    }
+                if let Some(candidate) = auth_store_candidate_from_snapshot_record(
+                    snapshot,
+                    AuthStoreRestoreSource::SpacetimeSnapshot,
+                )? {
+                    candidates.push(candidate);
                 }
             }
             Err(error) => {
@@ -404,6 +403,30 @@ impl AppState {
             }
         }
 
+        if let Some(candidate) = auth_store_candidate_from_local_file(&config)? {
+            candidates.push(candidate);
+        }
+
+        if let Some(candidate) = select_auth_store_restore_candidate(candidates) {
+            let source = candidate.source;
+            let should_sync_to_spacetime = source == AuthStoreRestoreSource::LocalFile;
+            let state = Self::new_with_auth_store(config, candidate.auth_store)?;
+            info!(
+                source = source.as_str(),
+                updated_at_micros = candidate.updated_at_micros,
+                "已恢复认证快照"
+            );
+            if should_sync_to_spacetime {
+                if let Err(error) = state.sync_auth_store_snapshot_to_spacetime().await {
+                    warn!(
+                        error = %error,
+                        "本地认证快照回写 SpacetimeDB 失败，当前启动继续"
+                    );
+                }
+            }
+            return Ok(state);
+        }
+
         Self::new(config)
     }
 
@@ -695,6 +718,95 @@ impl AppState {
     }
 }
 
+#[derive(Clone, Copy, Debug, PartialEq, Eq)]
+enum AuthStoreRestoreSource {
+    SpacetimeTables,
+    SpacetimeSnapshot,
+    LocalFile,
+}
+
+impl AuthStoreRestoreSource {
+    fn as_str(self) -> &'static str {
+        match self {
+            Self::SpacetimeTables => "spacetime_tables",
+            Self::SpacetimeSnapshot => "spacetime_snapshot",
+            Self::LocalFile => "local_file",
+        }
+    }
+}
+
+#[derive(Debug)]
+struct AuthStoreRestoreCandidate {
+    source: AuthStoreRestoreSource,
+    updated_at_micros: Option<i64>,
+    auth_store: InMemoryAuthStore,
+}
+
+fn auth_store_candidate_from_snapshot_record(
+    snapshot: spacetime_client::AuthStoreSnapshotRecord,
+    source: AuthStoreRestoreSource,
+) -> Result<Option<AuthStoreRestoreCandidate>, AppStateInitError> {
+    let Some(snapshot_json) = snapshot
+        .snapshot_json
+        .filter(|value| !value.trim().is_empty())
+    else {
+        return Ok(None);
+    };
+    let auth_store = InMemoryAuthStore::from_snapshot_json(&snapshot_json)
+        .map_err(AppStateInitError::AuthStore)?;
+
+    Ok(Some(AuthStoreRestoreCandidate {
+        source,
+        updated_at_micros: snapshot.updated_at_micros,
+        auth_store,
+    }))
+}
+
+fn auth_store_candidate_from_local_file(
+    config: &AppConfig,
+) -> Result<Option<AuthStoreRestoreCandidate>, AppStateInitError> {
+    if !config.auth_store_path.is_file() {
+        return Ok(None);
+    }
+
+    let updated_at_micros = fs::metadata(&config.auth_store_path)
+        .ok()
+        .and_then(|metadata| metadata.modified().ok())
+        .and_then(system_time_to_unix_micros);
+    let auth_store = InMemoryAuthStore::from_persistence_path(config.auth_store_path.clone())
+        .map_err(AppStateInitError::AuthStore)?;
+
+    Ok(Some(AuthStoreRestoreCandidate {
+        source: AuthStoreRestoreSource::LocalFile,
+        updated_at_micros,
+        auth_store,
+    }))
+}
+
+fn system_time_to_unix_micros(system_time: SystemTime) -> Option<i64> {
+    let duration = system_time.duration_since(UNIX_EPOCH).ok()?;
+    i64::try_from(duration.as_micros()).ok()
+}
+
+fn select_auth_store_restore_candidate(
+    candidates: Vec<AuthStoreRestoreCandidate>,
+) -> Option<AuthStoreRestoreCandidate> {
+    candidates.into_iter().max_by_key(|candidate| {
+        (
+            candidate.updated_at_micros.unwrap_or(i64::MIN),
+            auth_store_restore_source_priority(candidate.source),
+        )
+    })
+}
+
+fn auth_store_restore_source_priority(source: AuthStoreRestoreSource) -> u8 {
+    match source {
+        AuthStoreRestoreSource::SpacetimeSnapshot => 3,
+        AuthStoreRestoreSource::SpacetimeTables => 2,
+        AuthStoreRestoreSource::LocalFile => 1,
+    }
+}
+
 impl fmt::Display for AppStateInitError {
     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
         match self {