Add custom properties

docs/toml-schema.md

@@ -353,6 +353,7 @@ private-non-synced = false
 #### Bots
 
 Here are the available bots that can be added to the `bots` array:
+
 - `datadog`: installs a GitHub App used by DataDog. Used for [CI Visibility](https://www.datadoghq.com/product/ci-cd-monitoring/). It has read access to organization members, and repository
   actions, administration, checks, code scanning alerts, commit statuses, contents,
   deployments, issues, pull requests, secret scanning alerts and secrets.
@@ -527,7 +528,25 @@ branches = ["develop", "staging"]
 # No branch or tag patterns specified - any branch or tag can deploy
 ```
 
+### Repository custom properties
+
+[Repository custom properties] are values set on a repository to opt it into org-wide tooling. The property must first be defined at the organization level.
+
+Only boolean values are supported.
+
+[Repository custom properties]: https://docs.github.com/en/organizations/managing-organization-settings/managing-custom-properties-for-repositories-in-your-organization
+
+```toml
+# Repository custom properties (optional)
+[custom-properties]
+# Set a property name to a boolean value
+crabwatch = true
+```
+
+Properties set on GitHub but not declared here are left unchanged.
+
 ### Crates.io crate management
+
 Configure properties of crates.io crates that are deployed using Trusted Publishing from the given repository.
 
 ```toml

repos/rust-lang/crabwatch.toml

@@ -11,3 +11,6 @@ pattern = "main"
 ci-checks = ["CI"]
 required-approvals = 0
 
+[custom-properties]
+crabwatch = "true"
+

rust_team_data/src/v1.rs

@@ -195,6 +195,7 @@ pub struct Repo {
     // Is the GitHub "Auto-merge" option enabled?
     // https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/incorporating-changes-from-a-pull-request/automatically-merging-a-pull-request
     pub auto_merge_enabled: bool,
+    pub custom_properties: IndexMap<String, String>,
 }
 
 #[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]

src/schema.rs

@@ -844,6 +844,8 @@ pub(crate) struct Repo {
     pub crates_io: Vec<CratesIoConfiguration>,
     #[serde(default)]
     pub environments: BTreeMap<String, Environment>,
+    #[serde(default)]
+    pub custom_properties: BTreeMap<String, String>,
 }
 
 #[derive(serde::Deserialize, Debug, Clone, PartialEq)]

src/static_api.rs

@@ -233,6 +233,11 @@ impl<'a> Generator<'a> {
                 },
                 archived,
                 auto_merge_enabled: !managed_by_bors,
+                custom_properties: r
+                    .custom_properties
+                    .iter()
+                    .map(|(k, v)| (k.clone(), v.clone()))
+                    .collect(),
             };
 
             self.add(&format!("v1/repos/{}.json", r.name), &repo)?;
@@ -491,7 +496,7 @@ impl<'a> Generator<'a> {
         T: serde::Serialize + serde::de::DeserializeOwned + PartialEq,
     {
         info!("writing API object {path}...");
-        let json = serde_json::to_string_pretty(obj)?;
+        let json = serde_json::to_string_pretty(obj)? + "\n";
         self.write(path, json.as_bytes())?;
 
         let obj2: T =

src/sync/github/api/mod.rs

@@ -746,3 +746,15 @@ pub(crate) struct BranchPolicy {
 fn default_branch_policy_type() -> String {
     "branch".to_string()
 }
+
+/// A GitHub repository custom property. Values are strings even for booleans.
+#[derive(Clone, Debug, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
+pub(crate) struct CustomPropertyValue {
+    pub(crate) property_name: String,
+    pub(crate) value: Option<String>,
+}
+
+#[derive(Debug, serde::Serialize)]
+pub(crate) struct SetCustomPropertiesRequest {
+    pub(crate) properties: Vec<CustomPropertyValue>,
+}

src/sync/github/api/read.rs

@@ -1,6 +1,6 @@
 use crate::sync::github::api;
 use crate::sync::github::api::url::TokenType;
-use crate::sync::github::api::{BranchPolicy, Ruleset};
+use crate::sync::github::api::{BranchPolicy, CustomPropertyValue, Ruleset};
 use crate::sync::github::api::{
     BranchProtection, GraphNode, GraphNodes, GraphPageInfo, HttpClient, Login, OrgAppInstallation,
     Repo, RepoAppInstallation, RepoTeam, RepoUser, RestPaginatedError, Team, TeamMember, TeamRole,
@@ -88,6 +88,13 @@ pub(crate) trait GithubRead {
     /// Returns a vector of rulesets
     async fn repo_rulesets(&self, org: &str, repo: &str) -> anyhow::Result<Vec<Ruleset>>;
 
+    /// Get custom property values for a repository
+    async fn repo_custom_properties(
+        &self,
+        org: &str,
+        repo: &str,
+    ) -> anyhow::Result<Vec<CustomPropertyValue>>;
+
     async fn environment_branch_policies(
         &self,
         org: &str,
@@ -712,6 +719,26 @@ impl GithubRead for GitHubApiRead {
         Ok(rulesets)
     }
 
+    async fn repo_custom_properties(
+        &self,
+        org: &str,
+        repo: &str,
+    ) -> anyhow::Result<Vec<CustomPropertyValue>> {
+        // REST API endpoint for repository custom property values
+        // https://docs.github.com/en/rest/repos/custom-properties#get-all-custom-property-values-for-a-repository
+        let values: Vec<CustomPropertyValue> = self
+            .client
+            .req(
+                Method::GET,
+                &GitHubUrl::repos(org, repo, "properties/values")?,
+            )?
+            .send()
+            .await?
+            .json_annotated()
+            .await?;
+        Ok(values)
+    }
+
     async fn environment_branch_policies(
         &self,
         org: &str,

src/sync/github/api/write.rs

@@ -4,8 +4,8 @@ use std::collections::HashSet;
 
 use crate::sync::github::api::url::GitHubUrl;
 use crate::sync::github::api::{
-    GitHubApiRead, GithubRead, HttpClient, Repo, RepoPermission, RepoSettings, Ruleset, RulesetOp,
-    Team, TeamPrivacy, TeamRole, allow_not_found,
+    CustomPropertyValue, GitHubApiRead, GithubRead, HttpClient, Repo, RepoPermission, RepoSettings,
+    Ruleset, RulesetOp, SetCustomPropertiesRequest, Team, TeamPrivacy, TeamRole, allow_not_found,
 };
 use crate::sync::utils::ResponseExt;
 
@@ -696,4 +696,27 @@ impl GitHubWrite {
         }
         Ok(())
     }
+
+    /// Set a custom property value on a repository
+    pub(crate) async fn set_custom_property(
+        &self,
+        org: &str,
+        repo: &str,
+        property: &CustomPropertyValue,
+    ) -> anyhow::Result<()> {
+        debug!(
+            "Setting custom property '{}' on '{}/{}'",
+            property.property_name, org, repo
+        );
+        if !self.dry_run {
+            // REST API: PATCH /repos/{owner}/{repo}/properties/values
+            // https://docs.github.com/en/rest/repos/custom-properties#create-or-update-custom-property-values-for-a-repository
+            let url = GitHubUrl::repos(org, repo, "properties/values")?;
+            let body = SetCustomPropertiesRequest {
+                properties: vec![property.clone()],
+            };
+            self.client.send(Method::PATCH, &url, &body).await?;
+        }
+        Ok(())
+    }
 }