Find unknown mergeable PRs on periodic refresh

Author: Sakib25800

.sqlx/query-1e5d97de9e3b8c34ac2cfa4c00a6dd191dc70a3db77def03371d8bdfde2ae5e6.json renamed to .sqlx/query-81feaa8d94edd26607d6415207f549cf5a104f426f71e3394ef0713d25020da6.json

@@ -16,10 +16,11 @@ pub async fn refresh_repository(
     team_api_client: &TeamApiClient,
 ) -> anyhow::Result<()> {
     let repo = repo.as_ref();
-    if let (Ok(_), _, Ok(_)) = tokio::join!(
+    if let (Ok(_), _, Ok(_), Ok(_)) = tokio::join!(
         cancel_timed_out_builds(repo, db.as_ref()),
         reload_permission(repo, team_api_client),
-        reload_config(repo)
+        reload_config(repo),
+        reload_unknown_mergeable_prs(repo, db.as_ref())
     ) {
         Ok(())
     } else {
@@ -79,6 +80,31 @@ async fn reload_permission(
     Ok(())
 }
 
+async fn reload_unknown_mergeable_prs(
+    repo: &RepositoryState,
+    db: &PgDbClient,
+) -> anyhow::Result<()> {
+    let prs = db
+        .get_prs_with_unknown_mergeable_state(repo.repository())
+        .await?;
+
+    tracing::info!(
+        "Refreshing {} PR(s) with unknown mergeable state",
+        prs.len()
+    );
+
+    if prs.is_empty() {
+        return Ok(());
+    }
+
+    for pr in prs {
+        tracing::info!("PR #{} has unknown mergeable state", pr.number);
+        // Add to 'queue'
+    }
+
+    Ok(())
+}
+
 async fn reload_config(repo: &RepositoryState) -> anyhow::Result<()> {
     let config = repo.client.load_config().await?;
     repo.config.store(Arc::new(config));

.sqlx/query-8b97d98a060fcd7bb38cf072b33432ed024bc81b5af28e24202ddd23b3a03eaa.json

@@ -10,11 +10,11 @@ use crate::github::{CommitSha, GithubRepoName};
 
 use super::operations::{
     approve_pull_request, create_build, create_pull_request, create_workflow,
-    delegate_pull_request, find_build, find_pr_by_build, get_pull_request, get_repository,
-    get_running_builds, get_workflow_urls_for_build, get_workflows_for_build, set_pr_priority,
-    set_pr_rollup, set_pr_status, unapprove_pull_request, undelegate_pull_request,
-    update_build_status, update_mergeable_states_by_base_branch, update_pr_build_id,
-    update_workflow_status, upsert_pull_request, upsert_repository,
+    delegate_pull_request, find_build, find_pr_by_build, get_prs_with_unknown_mergeable_state,
+    get_pull_request, get_repository, get_running_builds, get_workflow_urls_for_build,
+    get_workflows_for_build, set_pr_priority, set_pr_rollup, set_pr_status, unapprove_pull_request,
+    undelegate_pull_request, update_build_status, update_mergeable_states_by_base_branch,
+    update_pr_build_id, update_workflow_status, upsert_pull_request, upsert_repository,
 };
 use super::{ApprovalInfo, MergeableState, RunId};
 
@@ -64,6 +64,13 @@ impl PgDbClient {
         update_mergeable_states_by_base_branch(&self.pool, repo, base_branch, mergeable_state).await
     }
 
+    pub async fn get_prs_with_unknown_mergeable_state(
+        &self,
+        repo: &GithubRepoName,
+    ) -> anyhow::Result<Vec<PullRequestModel>> {
+        get_prs_with_unknown_mergeable_state(&self.pool, repo).await
+    }
+
     pub async fn set_rollup(
         &self,
         pr: &PullRequestModel,

.sqlx/query-9e8a84ab79912f65d2e2db8060bd5a0f2b98c91675d42fa0c767c13f9338b4ff.json renamed to .sqlx/query-94cb702b5a80ade4531a82ea6134b44e2d5d0cce600a11ceb9c55dfd5a8a59c4.json

@@ -39,7 +39,7 @@ pub(crate) async fn get_pull_request(
             pr.approved_by,
             pr.approved_sha
         ) AS "approval_status!: ApprovalStatus",
-        pr.status as "pr_status: PullRequestStatus", 
+        pr.status as "pr_status: PullRequestStatus",
         pr.priority,
         pr.rollup as "rollup: RollupMode",
         pr.delegated,
@@ -137,7 +137,7 @@ pub(crate) async fn upsert_pull_request(
                     pr.approved_by,
                     pr.approved_sha
                 ) AS "approval_status!: ApprovalStatus",
-                pr.status as "pr_status: PullRequestStatus", 
+                pr.status as "pr_status: PullRequestStatus",
                 pr.priority,
                 pr.rollup as "rollup: RollupMode",
                 pr.delegated,
@@ -189,6 +189,45 @@ pub(crate) async fn update_mergeable_states_by_base_branch(
     .await
 }
 
+pub(crate) async fn get_prs_with_unknown_mergeable_state(
+    executor: impl PgExecutor<'_>,
+    repo: &GithubRepoName,
+) -> anyhow::Result<Vec<PullRequestModel>> {
+    measure_db_query("get_prs_with_unknown_mergeable_state", || async {
+        let prs = sqlx::query_as!(
+            PullRequestModel,
+            r#"
+SELECT
+    pr.id,
+    pr.repository as "repository: GithubRepoName",
+    pr.number as "number!: i64",
+    (
+        pr.approved_by,
+        pr.approved_sha
+    ) AS "approval_status!: ApprovalStatus",
+    pr.status as "pr_status: PullRequestStatus",
+    pr.priority,
+    pr.rollup as "rollup: RollupMode",
+    pr.delegated,
+    pr.base_branch,
+    pr.mergeable_state as "mergeable_state: MergeableState",
+    pr.created_at as "created_at: DateTime<Utc>",
+    build AS "try_build: BuildModel"
+FROM pull_request as pr
+LEFT JOIN build ON pr.build_id = build.id
+WHERE pr.repository = $1
+    AND pr.mergeable_state = 'unknown'
+    AND pr.status = 'open'
+"#,
+            repo as &GithubRepoName,
+        )
+        .fetch_all(executor)
+        .await?;
+        Ok(prs)
+    })
+    .await
+}
+
 pub(crate) async fn approve_pull_request(
     executor: impl PgExecutor<'_>,
     pr_id: i32,
@@ -285,7 +324,7 @@ SELECT
         pr.approved_by,
         pr.approved_sha
     ) AS "approval_status!: ApprovalStatus",
-    pr.status as "pr_status: PullRequestStatus",  
+    pr.status as "pr_status: PullRequestStatus",
     pr.delegated,
     pr.priority,
     pr.base_branch,