feat: init
This commit is contained in:
zhenyi
@@ -0,0 +1,220 @@
|
||||
use serde::{Deserialize, Serialize};
|
||||
use uuid::Uuid;
|
||||
|
||||
use crate::error::AppError;
|
||||
use crate::models::common::Provider;
|
||||
use crate::models::workspaces::WorkspaceIntegration;
|
||||
use crate::service::WorkspaceService;
|
||||
use crate::session::Session;
|
||||
|
||||
use super::util::{clamp_limit_offset, ensure_affected, required_text};
|
||||
|
||||
#[derive(Deserialize, Serialize, Clone, Debug, utoipa::ToSchema)]
|
||||
pub struct CreateIntegrationParams {
|
||||
pub provider: String,
|
||||
pub name: String,
|
||||
pub config: Option<crate::models::json_types::WorkspaceIntegrationConfig>,
|
||||
pub secret_ciphertext: Option<String>,
|
||||
pub enabled: Option<bool>,
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Serialize, Clone, Debug, utoipa::ToSchema)]
|
||||
pub struct UpdateIntegrationParams {
|
||||
pub name: Option<String>,
|
||||
pub config: Option<crate::models::json_types::WorkspaceIntegrationConfig>,
|
||||
pub secret_ciphertext: Option<String>,
|
||||
pub enabled: Option<bool>,
|
||||
}
|
||||
|
||||
impl WorkspaceService {
|
||||
pub async fn workspace_integrations(
|
||||
&self,
|
||||
ctx: &Session,
|
||||
workspace_id: Uuid,
|
||||
limit: i64,
|
||||
offset: i64,
|
||||
) -> Result<Vec<WorkspaceIntegration>, 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, crate::models::common::Role::Admin)
|
||||
.await?;
|
||||
let (limit, offset) = clamp_limit_offset(limit, offset);
|
||||
sqlx::query_as::<_, WorkspaceIntegration>(
|
||||
"SELECT id, workspace_id, provider, name, config, secret_ciphertext, enabled, \
|
||||
installed_by, last_used_at, created_at, updated_at FROM workspace_integration \
|
||||
WHERE workspace_id = $1 ORDER BY created_at DESC LIMIT $2 OFFSET $3",
|
||||
)
|
||||
.bind(workspace_id)
|
||||
.bind(limit)
|
||||
.bind(offset)
|
||||
.fetch_all(self.ctx.db.reader())
|
||||
.await
|
||||
.map_err(AppError::Database)
|
||||
}
|
||||
|
||||
pub async fn workspace_create_integration(
|
||||
&self,
|
||||
ctx: &Session,
|
||||
workspace_id: Uuid,
|
||||
params: CreateIntegrationParams,
|
||||
) -> Result<WorkspaceIntegration, 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, crate::models::common::Role::Admin)
|
||||
.await?;
|
||||
|
||||
let provider = params
|
||||
.provider
|
||||
.trim()
|
||||
.parse::<Provider>()
|
||||
.map_err(|_| AppError::BadRequest("invalid provider".into()))?;
|
||||
if provider == Provider::Unknown {
|
||||
return Err(AppError::BadRequest("invalid provider".into()));
|
||||
}
|
||||
let name = required_text(params.name, "name")?;
|
||||
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::<_, WorkspaceIntegration>(
|
||||
"INSERT INTO workspace_integration (id, workspace_id, provider, name, config, secret_ciphertext, \
|
||||
enabled, installed_by, created_at, updated_at) \
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $9) \
|
||||
RETURNING id, workspace_id, provider, name, config, secret_ciphertext, enabled, \
|
||||
installed_by, last_used_at, created_at, updated_at",
|
||||
)
|
||||
.bind(Uuid::now_v7())
|
||||
.bind(workspace_id)
|
||||
.bind(provider)
|
||||
.bind(&name)
|
||||
.bind(params.config.map(sqlx::types::Json))
|
||||
.bind(¶ms.secret_ciphertext)
|
||||
.bind(params.enabled.unwrap_or(true))
|
||||
.bind(user_uid)
|
||||
.bind(now)
|
||||
.fetch_one(&mut *txn)
|
||||
.await
|
||||
.map_err(AppError::Database)?;
|
||||
|
||||
txn.commit().await.map_err(|_| AppError::TxnError)?;
|
||||
Ok(result)
|
||||
}
|
||||
|
||||
pub async fn workspace_update_integration(
|
||||
&self,
|
||||
ctx: &Session,
|
||||
workspace_id: Uuid,
|
||||
integration_id: Uuid,
|
||||
params: UpdateIntegrationParams,
|
||||
) -> Result<WorkspaceIntegration, 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, crate::models::common::Role::Admin)
|
||||
.await?;
|
||||
|
||||
let current = sqlx::query_as::<_, WorkspaceIntegration>(
|
||||
"SELECT id, workspace_id, provider, name, config, secret_ciphertext, enabled, \
|
||||
installed_by, last_used_at, created_at, updated_at FROM workspace_integration \
|
||||
WHERE id = $1 AND workspace_id = $2",
|
||||
)
|
||||
.bind(integration_id)
|
||||
.bind(workspace_id)
|
||||
.fetch_optional(self.ctx.db.reader())
|
||||
.await
|
||||
.map_err(AppError::Database)?
|
||||
.ok_or(AppError::NotFound("integration not found".into()))?;
|
||||
|
||||
let name = params
|
||||
.name
|
||||
.map(|n| n.trim().to_string())
|
||||
.unwrap_or(current.name);
|
||||
let enabled = params.enabled.unwrap_or(current.enabled);
|
||||
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::<_, WorkspaceIntegration>(
|
||||
"UPDATE workspace_integration SET name = $1, config = $2, secret_ciphertext = $3, \
|
||||
enabled = $4, updated_at = $5 WHERE id = $6 AND workspace_id = $7 \
|
||||
RETURNING id, workspace_id, provider, name, config, secret_ciphertext, enabled, \
|
||||
installed_by, last_used_at, created_at, updated_at",
|
||||
)
|
||||
.bind(&name)
|
||||
.bind(
|
||||
params
|
||||
.config
|
||||
.map(sqlx::types::Json)
|
||||
.or_else(|| current.config.clone()),
|
||||
)
|
||||
.bind(params.secret_ciphertext.or(current.secret_ciphertext))
|
||||
.bind(enabled)
|
||||
.bind(now)
|
||||
.bind(integration_id)
|
||||
.bind(workspace_id)
|
||||
.fetch_one(&mut *txn)
|
||||
.await
|
||||
.map_err(AppError::Database)?;
|
||||
|
||||
txn.commit().await.map_err(|_| AppError::TxnError)?;
|
||||
Ok(result)
|
||||
}
|
||||
|
||||
pub async fn workspace_delete_integration(
|
||||
&self,
|
||||
ctx: &Session,
|
||||
workspace_id: Uuid,
|
||||
integration_id: Uuid,
|
||||
) -> Result<(), 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, crate::models::common::Role::Admin)
|
||||
.await?;
|
||||
|
||||
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("DELETE FROM workspace_integration WHERE id = $1 AND workspace_id = $2")
|
||||
.bind(integration_id)
|
||||
.bind(workspace_id)
|
||||
.execute(&mut *txn)
|
||||
.await
|
||||
.map_err(AppError::Database)?;
|
||||
ensure_affected(result.rows_affected(), "integration not found")?;
|
||||
|
||||
txn.commit().await.map_err(|_| AppError::TxnError)?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user