feat: add /report command

AOSC-Dev · Mar 3, 2024 · 0946877 · 0946877
1 parent 5dab702
commit 0946877
Show file tree

Hide file tree

Showing 3 changed files with 214 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/server/Cargo.toml b/server/Cargo.toml
@@ -27,3 +27,4 @@ console = "0.15.8"
 buildit-utils = { path = "../buildit-utils" }
 jsonwebtoken = "9.2.0"
 size = "0.4.1"
+dickens = { git = "https://github.com/AOSC-Dev/dickens.git", version = "0.1.0" }
diff --git a/server/src/bot.rs b/server/src/bot.rs
@@ -46,6 +46,8 @@ pub enum Command {
     Start(String),
     #[command(description = "Queue all ready messages: /queue [archs]")]
     Queue(String),
+    #[command(description = "Let dickens generate report for GitHub PR: /report pr-number")]
+    Report(String),
 }
 
 pub struct BuildRequest<'a> {
@@ -599,6 +601,76 @@ pub async fn answer(
                 }
             }
         }
+        Command::Report(arguments) => match str::parse::<u64>(&arguments) {
+            Ok(pr_number) => {
+                // create octocrab instance
+                let crab = match octocrab::Octocrab::builder()
+                    .user_access_token(ARGS.github_access_token.clone())
+                    .build()
+                {
+                    Ok(v) => v,
+                    Err(err) => {
+                        bot.send_message(
+                            msg.chat.id,
+                            format!("Cannot create octocrab instance: {err}"),
+                        )
+                        .await?;
+                        return Ok(());
+                    }
+                };
+
+                // get topic of pr
+                match crab.pulls("AOSC-Dev", "aosc-os-abbs").get(pr_number).await {
+                    Ok(pr) => match dickens::topic::report(pr.head.ref_field.as_str()).await {
+                        Ok(report) => {
+                            // post report as github comment
+                            match crab
+                                .issues("AOSC-Dev", "aosc-os-abbs")
+                                .create_comment(pr_number, report)
+                                .await
+                            {
+                                Ok(comment) => {
+                                    bot_send_message_handle_length(
+                                        &bot,
+                                        &msg,
+                                        &format!("Report posted as comment: {}", comment.url),
+                                    )
+                                    .await?;
+                                }
+                                Err(err) => {
+                                    bot_send_message_handle_length(
+                                        &bot,
+                                        &msg,
+                                        &format!("Failed to create github comments: {err}."),
+                                    )
+                                    .await?;
+                                }
+                            }
+                        }
+                        Err(err) => {
+                            bot_send_message_handle_length(
+                                &bot,
+                                &msg,
+                                &format!("Failed to generate dickens report: {err}."),
+                            )
+                            .await?;
+                        }
+                    },
+                    Err(err) => {
+                        bot_send_message_handle_length(
+                            &bot,
+                            &msg,
+                            &format!("Failed to get pr info: {err}."),
+                        )
+                        .await?;
+                    }
+                }
+            }
+            Err(err) => {
+                bot_send_message_handle_length(&bot, &msg, &format!("Bad PR number: {err}"))
+                    .await?;
+            }
+        },
     };
 
     Ok(())