fix: set telegram as source with /pr

AOSC-Dev · Mar 11, 2024 · c7a0bb0 · c7a0bb0
1 parent c10d9da
commit c7a0bb0
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 4 deletions.
diff --git a/server/src/api.rs b/server/src/api.rs
@@ -148,6 +148,7 @@ pub async fn pipeline_new_pr(
     pool: DbPool,
     pr: u64,
     archs: Option<&str>,
+    source: &JobSource
 ) -> anyhow::Result<Pipeline> {
     match octocrab::instance()
         .pulls("AOSC-Dev", "aosc-os-abbs")
@@ -194,7 +195,7 @@ pub async fn pipeline_new_pr(
                     Some(pr.number),
                     &packages.join(","),
                     &archs,
-                    &JobSource::Github(pr.number),
+                    &source,
                 )
                 .await
             } else {

diff --git a/server/src/bot.rs b/server/src/bot.rs
@@ -134,7 +134,14 @@ pub async fn answer(bot: Bot, msg: Message, cmd: Command, pool: DbPool) -> Respo
                     Some(parts[1])
                 };
                 for pr_number in pr_numbers {
-                    match pipeline_new_pr(pool.clone(), pr_number, archs).await {
+                    match pipeline_new_pr(
+                        pool.clone(),
+                        pr_number,
+                        archs,
+                        &JobSource::Telegram(msg.chat.id.0 as i64),
+                    )
+                    .await
+                    {
                         Ok(pipeline) => {
                             bot.send_message(
                                 msg.chat.id,

diff --git a/server/src/routes.rs b/server/src/routes.rs
@@ -94,8 +94,13 @@ pub async fn pipeline_new_pr(
     State(AppState { pool, .. }): State<AppState>,
     Json(payload): Json<PipelineNewPRRequest>,
 ) -> Result<Json<PipelineNewResponse>, AnyhowError> {
-    let pipeline =
-        api::pipeline_new_pr(pool, payload.pr, payload.archs.as_ref().map(|s| s.as_str())).await?;
+    let pipeline = api::pipeline_new_pr(
+        pool,
+        payload.pr,
+        payload.archs.as_ref().map(|s| s.as_str()),
+        &JobSource::Manual,
+    )
+    .await?;
     Ok(Json(PipelineNewResponse { id: pipeline.id }))
 }