feat: init

2026-06-07 11:30:56 +08:00
commit 563381c1ca
361 changed files with 41327 additions and 0 deletions
@@ -0,0 +1,123 @@
+use serde::{Deserialize, Serialize};
+use uuid::Uuid;
+
+use crate::error::AppError;
+use crate::models::common::Role;
+use crate::models::workspaces::WorkspaceCustomBranding;
+use crate::service::WorkspaceService;
+use crate::session::Session;
+
+use super::util::merge_optional_text;
+
+#[derive(Deserialize, Serialize, Clone, Debug, utoipa::ToSchema)]
+pub struct UpdateBrandingParams {
+    pub logo_url: Option<String>,
+    pub favicon_url: Option<String>,
+    pub primary_color: Option<String>,
+    pub accent_color: Option<String>,
+    pub custom_css: Option<String>,
+    pub support_url: Option<String>,
+    pub enabled: Option<bool>,
+}
+
+impl WorkspaceService {
+    pub async fn workspace_branding(
+        &self,
+        ctx: &Session,
+        workspace_id: Uuid,
+    ) -> Result<WorkspaceCustomBranding, AppError> {
+        let user_uid = ctx.user().ok_or(AppError::Unauthorized)?;
+        let ws = self.find_workspace_by_id(workspace_id).await?;
+        self.ensure_workspace_role_at_least(user_uid, &ws, Role::Admin)
+            .await?;
+        self.ensure_workspace_branding(workspace_id).await
+    }
+
+    pub async fn workspace_update_branding(
+        &self,
+        ctx: &Session,
+        workspace_id: Uuid,
+        params: UpdateBrandingParams,
+    ) -> Result<WorkspaceCustomBranding, AppError> {
+        let user_uid = ctx.user().ok_or(AppError::Unauthorized)?;
+        let ws = self.find_workspace_by_id(workspace_id).await?;
+        self.ensure_workspace_role_at_least(user_uid, &ws, Role::Admin)
+            .await?;
+
+        let current = self.ensure_workspace_branding(workspace_id).await?;
+        let now = chrono::Utc::now();
+
+        let mut txn = self
+            .ctx
+            .db
+            .writer()
+            .begin()
+            .await
+            .map_err(|_| AppError::TxnError)?;
+        sqlx::query("SET LOCAL app.current_user_id = $1")
+            .bind(user_uid)
+            .execute(&mut *txn)
+            .await
+            .map_err(AppError::Database)?;
+
+        let result = sqlx::query_as::<_, WorkspaceCustomBranding>(
+            "UPDATE workspace_custom_branding SET logo_url = $1, favicon_url = $2, primary_color = $3, \
+             accent_color = $4, custom_css = $5, support_url = $6, enabled = $7, updated_at = $8 \
+             WHERE workspace_id = $9 \
+             RETURNING workspace_id, logo_url, favicon_url, primary_color, accent_color, \
+             custom_css, support_url, enabled, created_at, updated_at",
+        )
+        .bind(merge_optional_text(params.logo_url, current.logo_url))
+        .bind(merge_optional_text(params.favicon_url, current.favicon_url))
+        .bind(merge_optional_text(params.primary_color, current.primary_color))
+        .bind(merge_optional_text(params.accent_color, current.accent_color))
+        .bind(merge_optional_text(params.custom_css, current.custom_css))
+        .bind(merge_optional_text(params.support_url, current.support_url))
+        .bind(params.enabled.unwrap_or(current.enabled))
+        .bind(now)
+        .bind(workspace_id)
+        .fetch_one(&mut *txn)
+        .await
+        .map_err(AppError::Database)?;
+
+        txn.commit().await.map_err(|_| AppError::TxnError)?;
+        Ok(result)
+    }
+
+    async fn ensure_workspace_branding(
+        &self,
+        workspace_id: Uuid,
+    ) -> Result<WorkspaceCustomBranding, AppError> {
+        if let Some(branding) = self.find_workspace_branding(workspace_id).await? {
+            return Ok(branding);
+        }
+        let now = chrono::Utc::now();
+        sqlx::query(
+            "INSERT INTO workspace_custom_branding (workspace_id, enabled, created_at, updated_at) \
+             VALUES ($1, false, $2, $2) ON CONFLICT (workspace_id) DO NOTHING",
+        )
+        .bind(workspace_id)
+        .bind(now)
+        .execute(self.ctx.db.writer())
+        .await
+        .map_err(AppError::Database)?;
+        self.find_workspace_branding(workspace_id)
+            .await?
+            .ok_or(AppError::NotFound("workspace branding not found".into()))
+    }
+
+    async fn find_workspace_branding(
+        &self,
+        workspace_id: Uuid,
+    ) -> Result<Option<WorkspaceCustomBranding>, AppError> {
+        sqlx::query_as::<_, WorkspaceCustomBranding>(
+            "SELECT workspace_id, logo_url, favicon_url, primary_color, accent_color, \
+             custom_css, support_url, enabled, created_at, updated_at \
+             FROM workspace_custom_branding WHERE workspace_id = $1",
+        )
+        .bind(workspace_id)
+        .fetch_optional(self.ctx.db.reader())
+        .await
+        .map_err(AppError::Database)
+    }
+}