Download information about organization app installations

Author: Kobzol

src/github/api/mod.rs

@@ -113,7 +113,7 @@ impl HttpClient {
 
     fn rest_paginated<F, T>(&self, method: &Method, url: String, mut f: F) -> anyhow::Result<()>
     where
-        F: FnMut(Vec<T>) -> anyhow::Result<()>,
+        F: FnMut(T) -> anyhow::Result<()>,
         T: DeserializeOwned,
     {
         let mut next = Some(url);
@@ -248,10 +248,24 @@ impl fmt::Display for RepoPermission {
     }
 }
 
+#[derive(serde::Deserialize, Debug)]
+pub(crate) struct OrgAppInstallation {
+    #[serde(rename = "id")]
+    pub(crate) installation_id: u64,
+    pub(crate) app_id: u64,
+}
+
+#[derive(serde::Deserialize, Debug)]
+pub(crate) struct RepoAppInstallation {
+    pub(crate) name: String,
+}
+
 #[derive(serde::Deserialize, Debug)]
 pub(crate) struct Repo {
     #[serde(rename = "node_id")]
     pub(crate) id: String,
+    #[serde(rename = "id")]
+    pub(crate) repo_id: u64,
     pub(crate) name: String,
     #[serde(alias = "owner", deserialize_with = "repo_owner")]
     pub(crate) org: String,

src/github/api/read.rs

@@ -1,6 +1,7 @@
 use crate::github::api::{
     team_node_id, user_node_id, BranchProtection, GraphNode, GraphNodes, GraphPageInfo, HttpClient,
-    Login, Repo, RepoTeam, RepoUser, Team, TeamMember, TeamRole,
+    Login, OrgAppInstallation, Repo, RepoAppInstallation, RepoTeam, RepoUser, Team, TeamMember,
+    TeamRole,
 };
 use reqwest::Method;
 use std::collections::{HashMap, HashSet};
@@ -12,6 +13,15 @@ pub(crate) trait GithubRead {
     /// Get the owners of an org
     fn org_owners(&self, org: &str) -> anyhow::Result<HashSet<u64>>;
 
+    /// Get the app installations of an org
+    fn org_app_installations(&self, org: &str) -> anyhow::Result<Vec<OrgAppInstallation>>;
+
+    /// Get the repositories enabled for an app installation.
+    fn app_installation_repos(
+        &self,
+        installation_id: u64,
+    ) -> anyhow::Result<Vec<RepoAppInstallation>>;
+
     /// Get all teams associated with a org
     ///
     /// Returns a list of tuples of team name and slug
@@ -110,6 +120,45 @@ impl GithubRead for GitHubApiRead {
         Ok(owners)
     }
 
+    fn org_app_installations(&self, org: &str) -> anyhow::Result<Vec<OrgAppInstallation>> {
+        #[derive(serde::Deserialize, Debug)]
+        struct InstallationPage {
+            installations: Vec<OrgAppInstallation>,
+        }
+
+        let mut installations = Vec::new();
+        self.client.rest_paginated(
+            &Method::GET,
+            format!("orgs/{org}/installations"),
+            |response: InstallationPage| {
+                installations.extend(response.installations);
+                Ok(())
+            },
+        )?;
+        Ok(installations)
+    }
+
+    fn app_installation_repos(
+        &self,
+        installation_id: u64,
+    ) -> anyhow::Result<Vec<RepoAppInstallation>> {
+        #[derive(serde::Deserialize, Debug)]
+        struct InstallationPage {
+            repositories: Vec<RepoAppInstallation>,
+        }
+
+        let mut installations = Vec::new();
+        self.client.rest_paginated(
+            &Method::GET,
+            format!("user/installations/{installation_id}/repositories"),
+            |response: InstallationPage| {
+                installations.extend(response.repositories);
+                Ok(())
+            },
+        )?;
+        Ok(installations)
+    }
+
     fn org_teams(&self, org: &str) -> anyhow::Result<Vec<(String, String)>> {
         let mut teams = Vec::new();
 

src/github/mod.rs

@@ -23,12 +23,37 @@ pub(crate) fn create_diff(
     github.diff_all()
 }
 
+type OrgName = String;
+type RepoName = String;
+
+#[derive(Clone, Debug)]
+enum GithubApp {
+    RenovateBot,
+}
+
+impl GithubApp {
+    fn from_id(app_id: u64) -> Option<Self> {
+        match app_id {
+            2740 => Some(GithubApp::RenovateBot),
+            _ => None,
+        }
+    }
+}
+
+#[derive(Clone, Debug)]
+struct GithubAppInstallation {
+    app: GithubApp,
+    installation_id: u64,
+    repositories: HashSet<RepoName>,
+}
+
 struct SyncGitHub {
     github: Box<dyn GithubRead>,
     teams: Vec<rust_team_data::v1::Team>,
     repos: Vec<rust_team_data::v1::Repo>,
     usernames_cache: HashMap<u64, String>,
-    org_owners: HashMap<String, HashSet<u64>>,
+    org_owners: HashMap<OrgName, HashSet<u64>>,
+    org_apps: HashMap<OrgName, Vec<GithubAppInstallation>>,
 }
 
 impl SyncGitHub {
@@ -50,15 +75,37 @@ impl SyncGitHub {
         let usernames_cache = github.usernames(&users)?;
 
         debug!("caching organization owners");
-        let orgs = teams
+        let mut orgs = teams
             .iter()
             .filter_map(|t| t.github.as_ref())
             .flat_map(|gh| &gh.teams)
             .map(|gh_team| &gh_team.org)
             .collect::<HashSet<_>>();
+
         let mut org_owners = HashMap::new();
+        let mut org_apps = HashMap::new();
+
         for org in &orgs {
             org_owners.insert((*org).to_string(), github.org_owners(org)?);
+
+            let mut installations: Vec<GithubAppInstallation> = vec![];
+
+            for installation in github.org_app_installations(org)? {
+                if let Some(app) = GithubApp::from_id(installation.app_id) {
+                    let mut repositories = HashSet::new();
+                    for repo_installation in
+                        github.app_installation_repos(installation.installation_id)?
+                    {
+                        repositories.insert(repo_installation.name);
+                    }
+                    installations.push(GithubAppInstallation {
+                        app,
+                        installation_id: installation.installation_id,
+                        repositories,
+                    });
+                }
+            }
+            org_apps.insert(org.to_string(), installations);
         }
 
         Ok(SyncGitHub {
@@ -67,6 +114,7 @@ impl SyncGitHub {
             repos,
             usernames_cache,
             org_owners,
+            org_apps,
         })
     }