zhenyi
103 lines
3.3 KiB
Rust
103 lines
3.3 KiB
Rust
//! SQL query methods for the `Workspace` entity.
|
|
|
|
use sqlx::PgPool;
|
|
use uuid::Uuid;
|
|
|
|
use super::workspace::Workspace;
|
|
use crate::models::common::{Role, Visibility};
|
|
|
|
impl Workspace {
|
|
/// Find a non-deleted workspace by primary key.
|
|
pub async fn find_by_name(pool: &PgPool, name: &str) -> Result<Option<Self>, sqlx::Error> {
|
|
sqlx::query_as::<_, Workspace>(
|
|
"SELECT id, owner_id, name, description, avatar_url, visibility, plan, status, \
|
|
default_role, is_personal, archived_at, created_at, updated_at, deleted_at \
|
|
FROM workspace WHERE name = $1 AND deleted_at IS NULL",
|
|
)
|
|
.bind(name)
|
|
.fetch_optional(pool)
|
|
.await
|
|
}
|
|
|
|
pub async fn find_by_id(pool: &PgPool, id: Uuid) -> Result<Option<Self>, sqlx::Error> {
|
|
sqlx::query_as::<_, Workspace>(
|
|
r#"SELECT id, owner_id, name, description, avatar_url, visibility, plan, status,
|
|
default_role, is_personal, archived_at, created_at, updated_at, deleted_at
|
|
FROM workspace WHERE id = $1 AND deleted_at IS NULL"#,
|
|
)
|
|
.bind(id)
|
|
.fetch_optional(pool)
|
|
.await
|
|
}
|
|
|
|
/// Count non-deleted workspaces owned by a user.
|
|
pub async fn count_owned(pool: &PgPool, owner_id: Uuid) -> Result<i64, sqlx::Error> {
|
|
sqlx::query_scalar(
|
|
r#"SELECT COUNT(*) FROM workspace WHERE owner_id = $1 AND deleted_at IS NULL"#,
|
|
)
|
|
.bind(owner_id)
|
|
.fetch_one(pool)
|
|
.await
|
|
}
|
|
|
|
/// Check if a user is an active member of a workspace.
|
|
pub async fn is_member(
|
|
pool: &PgPool,
|
|
workspace_id: Uuid,
|
|
user_id: Uuid,
|
|
) -> Result<bool, sqlx::Error> {
|
|
sqlx::query_scalar::<_, bool>(
|
|
r#"SELECT EXISTS(
|
|
SELECT 1 FROM workspace_member
|
|
WHERE workspace_id = $1 AND user_id = $2 AND status = 'active'
|
|
)"#,
|
|
)
|
|
.bind(workspace_id)
|
|
.bind(user_id)
|
|
.fetch_one(pool)
|
|
.await
|
|
}
|
|
|
|
/// Get the role of a user in a workspace.
|
|
/// Returns `Role::Owner` if the user is the workspace owner but has no explicit member row.
|
|
pub async fn user_role(
|
|
pool: &PgPool,
|
|
workspace_id: Uuid,
|
|
user_id: Uuid,
|
|
owner_id: Uuid,
|
|
) -> Result<Option<Role>, sqlx::Error> {
|
|
if owner_id == user_id {
|
|
return Ok(Some(Role::Owner));
|
|
}
|
|
let role_str: Option<String> = sqlx::query_scalar(
|
|
r#"SELECT role FROM workspace_member
|
|
WHERE workspace_id = $1 AND user_id = $2 AND status = 'active'"#,
|
|
)
|
|
.bind(workspace_id)
|
|
.bind(user_id)
|
|
.fetch_optional(pool)
|
|
.await?;
|
|
|
|
Ok(role_str.and_then(|r| r.parse::<Role>().ok()))
|
|
}
|
|
|
|
/// Check if the user can read the workspace.
|
|
/// Readable if: owner, active member, or workspace is public/internal.
|
|
pub async fn is_readable(
|
|
pool: &PgPool,
|
|
ws: &Workspace,
|
|
user_id: Uuid,
|
|
) -> Result<bool, sqlx::Error> {
|
|
if ws.owner_id == user_id {
|
|
return Ok(true);
|
|
}
|
|
if Self::is_member(pool, ws.id, user_id).await? {
|
|
return Ok(true);
|
|
}
|
|
Ok(matches!(
|
|
ws.visibility,
|
|
Visibility::Public | Visibility::Internal
|
|
))
|
|
}
|
|
}
|