Merge pull request #230 from Sakib25800/feat/track-base

Track base branch

Author: Kobzol

.sqlx/query-0f95314b0c0debde970dd27f33ced6c3463e9e31131dbb39b4514f041e95939b.json

@@ -0,0 +1,2 @@
+-- Add down migration script here
+ALTER TABLE pull_request DROP COLUMN base_branch;

.sqlx/query-171a79e16e89f8ad9d2fd897455d87ad5a7ca3eba6f59eabe66a008d1ce864c4.json renamed to .sqlx/query-80d2c1b695162729675c566946a145a78feed990078fbd50a83ada38a225597d.json

@@ -0,0 +1,2 @@
+-- Add up migration script here
+ALTER TABLE pull_request ADD COLUMN base_branch TEXT NOT NULL;

.sqlx/query-8f940c2da5f3cf3f41b39e7c78d9e254da0d1fa8de59af33ade50cb2ac34c917.json

@@ -11,6 +11,8 @@ pub enum BorsRepositoryEvent {
     PullRequestCommitPushed(PullRequestPushed),
     /// When the pull request is edited by its author
     PullRequestEdited(PullRequestEdited),
+    /// When a pull request is opened.
+    PullRequestOpened(PullRequestOpened),
     /// A workflow run on Github Actions or a check run from external CI system has been started.
     WorkflowStarted(WorkflowStarted),
     /// A workflow run on Github Actions or a check run from external CI system has been completed.
@@ -26,6 +28,7 @@ impl BorsRepositoryEvent {
             BorsRepositoryEvent::Comment(comment) => &comment.repository,
             BorsRepositoryEvent::PullRequestCommitPushed(payload) => &payload.repository,
             BorsRepositoryEvent::PullRequestEdited(payload) => &payload.repository,
+            BorsRepositoryEvent::PullRequestOpened(payload) => &payload.repository,
             BorsRepositoryEvent::WorkflowStarted(workflow) => &workflow.repository,
             BorsRepositoryEvent::WorkflowCompleted(workflow) => &workflow.repository,
             BorsRepositoryEvent::CheckSuiteCompleted(payload) => &payload.repository,
@@ -72,6 +75,12 @@ pub struct PullRequestEdited {
     pub from_base_sha: Option<CommitSha>,
 }
 
+#[derive(Debug)]
+pub struct PullRequestOpened {
+    pub repository: GithubRepoName,
+    pub pull_request: PullRequest,
+}
+
 #[derive(Debug)]
 pub struct WorkflowStarted {
     pub repository: GithubRepoName,

.sqlx/query-96805d5e3b9f430d100135ba4dc9face8cca0684d03343d5466deffbdb4fbdac.json

@@ -12,7 +12,7 @@ pub(super) async fn command_info(
 ) -> anyhow::Result<()> {
     // Geting PR info from database
     let pr_model = db
-        .get_or_create_pull_request(repo.client.repository(), pr.number)
+        .get_or_create_pull_request(repo.client.repository(), pr.number, &pr.base.name)
         .await?;
 
     // Building the info message

.sqlx/query-181b548e2a8a50573b30c7d7c703ca6b49c8278bddd103dcb2c84e0b3c95cc3d.json renamed to .sqlx/query-d6aed6aa0d89754e5dbe1fdafd3fbccb522960b2a1b81bba5fa197e9da710126.json

@@ -19,7 +19,9 @@ use crate::permissions::PermissionType;
 use crate::{load_repositories, PgDbClient, TeamApiClient};
 use anyhow::Context;
 use octocrab::Octocrab;
-use pr_events::{handle_pull_request_edited, handle_push_to_pull_request};
+use pr_events::{
+    handle_pull_request_edited, handle_pull_request_opened, handle_push_to_pull_request,
+};
 use review::{command_delegate, command_set_priority, command_set_rollup, command_undelegate};
 use tracing::Instrument;
 
@@ -136,6 +138,14 @@ pub async fn handle_bors_repository_event(
                 .instrument(span.clone())
                 .await?;
         }
+        BorsRepositoryEvent::PullRequestOpened(payload) => {
+            let span =
+                tracing::info_span!("Pull request opened", repo = payload.repository.to_string());
+
+            handle_pull_request_opened(repo, db, payload)
+                .instrument(span.clone())
+                .await?;
+        }
     }
     Ok(())
 }
@@ -425,7 +435,7 @@ async fn has_permission(
     }
 
     let pr_model = db
-        .get_or_create_pull_request(repo_state.repository(), pr.number)
+        .get_or_create_pull_request(repo_state.repository(), pr.number, &pr.base.name)
         .await?;
     let is_delegated = pr_model.delegated && author.id == pr.author.id;