重写

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

@@ -1,12 +1,17 @@
 use std::{error::Error, fmt};
 
-use module_auth::{AuthUserService, InMemoryAuthStore, PasswordEntryService, RefreshSessionService};
+use module_auth::{
+    AuthUserService, InMemoryAuthStore, PasswordEntryService, PhoneAuthService,
+    RefreshSessionService, WechatAuthService, WechatAuthStateService,
+};
 use platform_auth::{
     JwtConfig, JwtError, RefreshCookieConfig, RefreshCookieError, RefreshCookieSameSite,
 };
 use platform_oss::{OssClient, OssConfig, OssError};
+use spacetime_client::{SpacetimeClient, SpacetimeClientConfig};
 
 use crate::config::AppConfig;
+use crate::wechat_provider::{WechatProvider, build_wechat_provider};
 
 // 当前阶段先保留最小共享状态壳，后续逐步接入配置、客户端与平台适配。
 #[derive(Clone, Debug)]
@@ -20,6 +25,11 @@ pub struct AppState {
     password_entry_service: PasswordEntryService,
     refresh_session_service: RefreshSessionService,
     auth_user_service: AuthUserService,
+    phone_auth_service: PhoneAuthService,
+    wechat_auth_state_service: WechatAuthStateService,
+    wechat_auth_service: WechatAuthService,
+    wechat_provider: WechatProvider,
+    spacetime_client: SpacetimeClient,
 }
 
 #[derive(Debug)]
@@ -51,8 +61,18 @@ impl AppState {
         let auth_store = InMemoryAuthStore::default();
         let password_entry_service = PasswordEntryService::new(auth_store.clone());
         let auth_user_service = AuthUserService::new(auth_store.clone());
+        let phone_auth_service = PhoneAuthService::new(auth_store.clone());
+        let wechat_auth_state_service =
+            WechatAuthStateService::new(auth_store.clone(), config.wechat_state_ttl_minutes);
+        let wechat_auth_service = WechatAuthService::new(auth_store.clone());
+        let wechat_provider = build_wechat_provider(&config);
         let refresh_session_service =
             RefreshSessionService::new(auth_store, config.refresh_session_ttl_days);
+        let spacetime_client = SpacetimeClient::new(SpacetimeClientConfig {
+            server_url: config.spacetime_server_url.clone(),
+            database: config.spacetime_database.clone(),
+            token: config.spacetime_token.clone(),
+        });
 
         Ok(Self {
             config,
@@ -62,6 +82,11 @@ impl AppState {
             password_entry_service,
             refresh_session_service,
             auth_user_service,
+            phone_auth_service,
+            wechat_auth_state_service,
+            wechat_auth_service,
+            wechat_provider,
+            spacetime_client,
         })
     }
 
@@ -88,6 +113,26 @@ impl AppState {
     pub fn auth_user_service(&self) -> &AuthUserService {
         &self.auth_user_service
     }
+
+    pub fn phone_auth_service(&self) -> &PhoneAuthService {
+        &self.phone_auth_service
+    }
+
+    pub fn wechat_auth_state_service(&self) -> &WechatAuthStateService {
+        &self.wechat_auth_state_service
+    }
+
+    pub fn wechat_auth_service(&self) -> &WechatAuthService {
+        &self.wechat_auth_service
+    }
+
+    pub fn wechat_provider(&self) -> &WechatProvider {
+        &self.wechat_provider
+    }
+
+    pub fn spacetime_client(&self) -> &SpacetimeClient {
+        &self.spacetime_client
+    }
 }
 
 impl fmt::Display for AppStateInitError {