From 501ef7439d2dc043bcdf5b05e82583a3a46d12d0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ben=20P=C3=BCschel?= <ben.pueschel2004@gmail.com>
Date: Fri, 30 Aug 2024 11:11:03 +0200
Subject: [PATCH] feat: add missing fields in `CreateForkBuilder` (#682)

---
 src/api/repos/forks.rs | 26 ++++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

diff --git a/src/api/repos/forks.rs b/src/api/repos/forks.rs
index 3de1071c..f3d71934 100644
--- a/src/api/repos/forks.rs
+++ b/src/api/repos/forks.rs
@@ -57,12 +57,18 @@ pub struct CreateForkBuilder<'octo, 'r> {
     handler: &'r RepoHandler<'octo>,
     #[serde(skip_serializing_if = "Option::is_none")]
     organization: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    name: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    default_branch_only: Option<bool>,
 }
 impl<'octo, 'r> CreateForkBuilder<'octo, 'r> {
     pub(crate) fn new(handler: &'r RepoHandler<'octo>) -> Self {
         Self {
             handler,
             organization: None,
+            name: None,
+            default_branch_only: None,
         }
     }
 
@@ -72,6 +78,18 @@ impl<'octo, 'r> CreateForkBuilder<'octo, 'r> {
         self
     }
 
+    /// When forking from an existing repository, a new name for the fork.
+    pub fn name(mut self, name: impl Into<String>) -> Self {
+        self.name = Some(name.into());
+        self
+    }
+
+    /// When forking from an existing repository, fork with only the default branch.
+    pub fn default_branch_only(mut self, default_branch_only: impl Into<bool>) -> Self {
+        self.default_branch_only = Some(default_branch_only.into());
+        self
+    }
+
     /// Sends the actual request.
     pub async fn send(self) -> crate::Result<crate::models::Repository> {
         let route = format!(
@@ -108,8 +126,10 @@ impl<'octo> RepoHandler<'octo> {
     }
 
     /// Creates a fork of a repository. Optionally, specify the target
-    /// [organization](CreateForkBuilder::organization()) to
-    /// create the fork in.
+    /// [organization](CreateForkBuilder::organization())
+    /// or [name](CreateForkBuilder::name()) to create the fork in,
+    /// or [default_branch_only](CreateForkBuilder::default_branch_only()) to fork with
+    /// only the default branch.
     /// ```no_run
     /// # async fn run() -> octocrab::Result<()> {
     /// let new_fork = octocrab::instance()
@@ -117,6 +137,8 @@ impl<'octo> RepoHandler<'octo> {
     ///     .create_fork()
     ///     // Optional Parameters
     ///     .organization("weyland-yutani")
+    ///     .name("new-repo-name")
+    ///     .default_branch_only(true)
     ///     .send()
     ///     .await?;
     /// # Ok(())