Update the main branch of a repo

1 files changed, 6 insertions, 1 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 196078c..134ba97 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -33,12 +33,17 @@ pub fn run_command(user_input: String) -> Result<ControlFlow<(), ()>, ShackleErr
             git::set_description(&directory, &description)?;
             println!("Successfully updated description");
         }
+        Ok(ShackleCommand::SetBranch(SetBranchArgs { directory, branch })) => {
+            git::set_branch(&directory, &branch)?;
+            println!("Successfully updated branch");
+        }
         Ok(ShackleCommand::Init(InitArgs {
             repo_name,
             group,
             description,
+            branch,
         })) => {
-            let init_result = git::init(&repo_name, &group, &description)?;
+            let init_result = git::init(&repo_name, &group, &description, &branch)?;
             println!("Successfully created \"{}\"", init_result.path.display());
         }
         Ok(ShackleCommand::GitUploadPack(upload_pack_args)) => {