zhenyi
123 lines
4.3 KiB
Rust
123 lines
4.3 KiB
Rust
use uuid::Uuid;
|
|
|
|
use crate::error::AppError;
|
|
use crate::models::common::RelationType;
|
|
use crate::models::issues::IssuePrRelation;
|
|
use crate::service::IssueService;
|
|
use crate::session::Session;
|
|
|
|
use super::util::{clamp_limit_offset, ensure_affected, parse_enum};
|
|
|
|
#[derive(serde::Deserialize, serde::Serialize, Clone, Debug, utoipa::ToSchema)]
|
|
pub struct LinkPrParams {
|
|
pub pull_request_id: Uuid,
|
|
pub relation_type: Option<String>,
|
|
}
|
|
|
|
impl IssueService {
|
|
pub async fn issue_pr_relations(
|
|
&self,
|
|
ctx: &Session,
|
|
wk_name: &str,
|
|
number: i64,
|
|
limit: i64,
|
|
offset: i64,
|
|
) -> Result<Vec<IssuePrRelation>, AppError> {
|
|
let user_uid = ctx.user().ok_or(AppError::Unauthorized)?;
|
|
let issue = self.resolve_issue(wk_name, number).await?;
|
|
let issue_id = issue.id;
|
|
self.ensure_issue_readable(user_uid, &issue).await?;
|
|
let (limit, offset) = clamp_limit_offset(limit, offset);
|
|
sqlx::query_as::<_, IssuePrRelation>(
|
|
"SELECT id, issue_id, pull_request_id, relation_type, created_by, created_at \
|
|
FROM issue_pr_relation WHERE issue_id = $1 ORDER BY created_at ASC LIMIT $2 OFFSET $3",
|
|
)
|
|
.bind(issue_id)
|
|
.bind(limit)
|
|
.bind(offset)
|
|
.fetch_all(self.ctx.db.reader())
|
|
.await
|
|
.map_err(AppError::Database)
|
|
}
|
|
|
|
pub async fn issue_link_pr(
|
|
&self,
|
|
ctx: &Session,
|
|
wk_name: &str,
|
|
number: i64,
|
|
params: LinkPrParams,
|
|
) -> Result<IssuePrRelation, AppError> {
|
|
let user_uid = ctx.user().ok_or(AppError::Unauthorized)?;
|
|
let issue = self.resolve_issue(wk_name, number).await?;
|
|
let issue_id = issue.id;
|
|
self.ensure_issue_editable(user_uid, &issue).await?;
|
|
let relation_type = match params.relation_type {
|
|
Some(ref v) => parse_enum(
|
|
Some(v.clone()),
|
|
RelationType::References,
|
|
RelationType::Unknown,
|
|
"relation_type",
|
|
)?,
|
|
None => RelationType::References,
|
|
};
|
|
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 rel = sqlx::query_as::<_, IssuePrRelation>(
|
|
"INSERT INTO issue_pr_relation (id, issue_id, pull_request_id, relation_type, created_by, created_at) \
|
|
VALUES ($1, $2, $3, $4, $5, $6) ON CONFLICT (issue_id, pull_request_id) DO NOTHING \
|
|
RETURNING id, issue_id, pull_request_id, relation_type, created_by, created_at",
|
|
)
|
|
.bind(Uuid::now_v7()).bind(issue_id).bind(params.pull_request_id)
|
|
.bind(relation_type).bind(user_uid).bind(now)
|
|
.fetch_optional(&mut *txn).await.map_err(AppError::Database)?
|
|
.ok_or(AppError::Conflict("PR already linked".into()))?;
|
|
txn.commit().await.map_err(|_| AppError::TxnError)?;
|
|
Ok(rel)
|
|
}
|
|
|
|
pub async fn issue_unlink_pr(
|
|
&self,
|
|
ctx: &Session,
|
|
wk_name: &str,
|
|
number: i64,
|
|
relation_id: Uuid,
|
|
) -> Result<(), AppError> {
|
|
let user_uid = ctx.user().ok_or(AppError::Unauthorized)?;
|
|
let issue = self.resolve_issue(wk_name, number).await?;
|
|
let issue_id = issue.id;
|
|
self.ensure_issue_editable(user_uid, &issue).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 issue_pr_relation WHERE id = $1 AND issue_id = $2")
|
|
.bind(relation_id)
|
|
.bind(issue_id)
|
|
.execute(&mut *txn)
|
|
.await
|
|
.map_err(AppError::Database)?;
|
|
ensure_affected(result.rows_affected(), "PR relation not found")?;
|
|
txn.commit().await.map_err(|_| AppError::TxnError)?;
|
|
Ok(())
|
|
}
|
|
}
|