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

use std::sync::Arc;

use axum::extract::FromRef;
use futures_util::future::BoxFuture;
use platform_oss::OssClient;
use spacetime_client::{
    EditorProjectCreateRecordInput, EditorProjectDeleteRecordInput, EditorProjectGetRecordInput,
    EditorProjectLayoutSaveV2AckRecord, EditorProjectLayoutSaveV2RecordInput, EditorProjectRecord,
    EditorProjectRenameRecordInput, SpacetimeClient, SpacetimeClientError,
};

use super::AppState;

// 元数据入口只依赖现役 typed facade 的项目能力，不暴露其它领域的数据库操作。
pub(crate) trait EditorProjectRepository: Send + Sync {
    fn get_recent_editor_project(
        &self,
        owner_user_id: String,
    ) -> BoxFuture<'_, Result<Option<EditorProjectRecord>, SpacetimeClientError>>;

    fn list_editor_projects(
        &self,
        owner_user_id: String,
    ) -> BoxFuture<'_, Result<Vec<EditorProjectRecord>, SpacetimeClientError>>;

    fn get_editor_project(
        &self,
        input: EditorProjectGetRecordInput,
    ) -> BoxFuture<'_, Result<EditorProjectRecord, SpacetimeClientError>>;

    fn create_editor_project(
        &self,
        input: EditorProjectCreateRecordInput,
    ) -> BoxFuture<'_, Result<EditorProjectRecord, SpacetimeClientError>>;

    fn rename_editor_project(
        &self,
        input: EditorProjectRenameRecordInput,
    ) -> BoxFuture<'_, Result<EditorProjectRecord, SpacetimeClientError>>;

    fn delete_editor_project(
        &self,
        input: EditorProjectDeleteRecordInput,
    ) -> BoxFuture<'_, Result<String, SpacetimeClientError>>;

    fn save_editor_project_layout_v2_ack(
        &self,
        input: EditorProjectLayoutSaveV2RecordInput,
    ) -> BoxFuture<'_, Result<EditorProjectLayoutSaveV2AckRecord, SpacetimeClientError>>;
}

impl EditorProjectRepository for SpacetimeClient {
    fn get_recent_editor_project(
        &self,
        owner_user_id: String,
    ) -> BoxFuture<'_, Result<Option<EditorProjectRecord>, SpacetimeClientError>> {
        Box::pin(SpacetimeClient::get_recent_editor_project(
            self,
            owner_user_id,
        ))
    }

    fn list_editor_projects(
        &self,
        owner_user_id: String,
    ) -> BoxFuture<'_, Result<Vec<EditorProjectRecord>, SpacetimeClientError>> {
        Box::pin(SpacetimeClient::list_editor_projects(self, owner_user_id))
    }

    fn get_editor_project(
        &self,
        input: EditorProjectGetRecordInput,
    ) -> BoxFuture<'_, Result<EditorProjectRecord, SpacetimeClientError>> {
        Box::pin(SpacetimeClient::get_editor_project(self, input))
    }

    fn create_editor_project(
        &self,
        input: EditorProjectCreateRecordInput,
    ) -> BoxFuture<'_, Result<EditorProjectRecord, SpacetimeClientError>> {
        Box::pin(SpacetimeClient::create_editor_project(self, input))
    }

    fn rename_editor_project(
        &self,
        input: EditorProjectRenameRecordInput,
    ) -> BoxFuture<'_, Result<EditorProjectRecord, SpacetimeClientError>> {
        Box::pin(SpacetimeClient::rename_editor_project(self, input))
    }

    fn delete_editor_project(
        &self,
        input: EditorProjectDeleteRecordInput,
    ) -> BoxFuture<'_, Result<String, SpacetimeClientError>> {
        Box::pin(SpacetimeClient::delete_editor_project(self, input))
    }

    fn save_editor_project_layout_v2_ack(
        &self,
        input: EditorProjectLayoutSaveV2RecordInput,
    ) -> BoxFuture<'_, Result<EditorProjectLayoutSaveV2AckRecord, SpacetimeClientError>> {
        Box::pin(SpacetimeClient::save_editor_project_layout_v2_ack(
            self, input,
        ))
    }
}

pub(crate) trait EditorProjectMediaRepair: Send + Sync {
    fn repair_project(&self, project: EditorProjectRecord) -> BoxFuture<'_, EditorProjectRecord>;
}

#[derive(Clone)]
pub struct EditorProjectState {
    projects: Arc<dyn EditorProjectRepository>,
    media_repair: Arc<dyn EditorProjectMediaRepair>,
}

impl EditorProjectState {
    pub(crate) fn new(
        projects: Arc<dyn EditorProjectRepository>,
        media_repair: Arc<dyn EditorProjectMediaRepair>,
    ) -> Self {
        Self {
            projects,
            media_repair,
        }
    }

    pub(crate) fn projects(&self) -> &dyn EditorProjectRepository {
        self.projects.as_ref()
    }

    pub(crate) async fn repair_project_media(
        &self,
        project: EditorProjectRecord,
    ) -> EditorProjectRecord {
        self.media_repair.repair_project(project).await
    }
}

impl FromRef<AppState> for EditorProjectState {
    fn from_ref(state: &AppState) -> Self {
        state.editor_project_state.clone()
    }
}

// 读项目时的内联媒体修复与生成写入共用相同存储依赖，保持原有上传和登记顺序。
#[derive(Clone)]
pub(crate) struct EditorMediaStorageState {
    spacetime_client: SpacetimeClient,
    oss_client: Option<OssClient>,
    http_client: reqwest::Client,
}

impl EditorMediaStorageState {
    pub(super) fn new(
        spacetime_client: SpacetimeClient,
        oss_client: Option<OssClient>,
        http_client: reqwest::Client,
    ) -> Self {
        Self {
            spacetime_client,
            oss_client,
            http_client,
        }
    }

    pub(crate) fn spacetime_client(&self) -> &SpacetimeClient {
        &self.spacetime_client
    }

    pub(crate) fn oss_client(&self) -> Option<&OssClient> {
        self.oss_client.as_ref()
    }

    pub(crate) fn editor_oss_http_client(&self) -> &reqwest::Client {
        &self.http_client
    }
}

impl FromRef<AppState> for EditorMediaStorageState {
    fn from_ref(state: &AppState) -> Self {
        Self::new(
            state.spacetime_client().clone(),
            state.oss_client().cloned(),
            state.editor_oss_http_client().clone(),
        )
    }
}

impl EditorProjectMediaRepair for EditorMediaStorageState {
    fn repair_project(&self, project: EditorProjectRecord) -> BoxFuture<'_, EditorProjectRecord> {
        Box::pin(crate::editor_project::repair_editor_project_record_inline_media(self, project))
    }
}