Synchronize GitHub app installation

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);
@@ -138,7 +138,7 @@ impl HttpClient {
                 }
             }
 
-            f(resp.json().with_context(|| {
+            f(resp.json_annotated().with_context(|| {
                 format!("Failed to deserialize response body for {method} request to '{next_url}'")
             })?)?;
         }
@@ -248,10 +248,23 @@ 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,
+    pub(crate) node_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/api/write.rs

@@ -230,7 +230,8 @@ impl GitHubWrite {
         debug!("Creating the repo {org}/{name} with {req:?}");
         if self.dry_run {
             Ok(Repo {
-                id: String::from("ID"),
+                node_id: String::from("ID"),
+                repo_id: 0,
                 name: name.to_string(),
                 org: org.to_string(),
                 description: settings.description.clone(),
@@ -273,6 +274,42 @@ impl GitHubWrite {
         Ok(())
     }
 
+    pub(crate) fn add_repo_to_app_installation(
+        &self,
+        installation_id: u64,
+        repository_id: u64,
+    ) -> anyhow::Result<()> {
+        debug!("Adding repository {repository_id} to installation {installation_id}");
+        if !self.dry_run {
+            self.client
+                .req(
+                    Method::PUT,
+                    &format!("user/installations/{installation_id}/repositories/{repository_id}"),
+                )?
+                .send()?
+                .custom_error_for_status()?;
+        }
+        Ok(())
+    }
+
+    pub(crate) fn remove_repo_from_app_installation(
+        &self,
+        installation_id: u64,
+        repository_id: u64,
+    ) -> anyhow::Result<()> {
+        debug!("Removing repository {repository_id} from installation {installation_id}");
+        if !self.dry_run {
+            self.client
+                .req(
+                    Method::DELETE,
+                    &format!("user/installations/{installation_id}/repositories/{repository_id}"),
+                )?
+                .send()?
+                .custom_error_for_status()?;
+        }
+        Ok(())
+    }
+
     /// Update a team's permissions to a repo
     pub(crate) fn update_team_repo_permissions(
         &self,