1 files changed, 9 insertions, 0 deletions
diff --git a/src/parser.rs b/src/parser.rs
index 14b1a49..3a7d129 100644
--- a/src/parser.rs
+++ b/src/parser.rs
@@ -14,6 +14,7 @@ pub enum ShackleCommand {
     /// List all repositories available
     List,
     SetDescription(SetDescriptionArgs),
+    SetBranch(SetBranchArgs),
     Init(InitArgs),
     GitUploadPack(GitUploadPackArgs),
     GitReceivePack(GitReceivePackArgs),
@@ -26,11 +27,19 @@ pub struct SetDescriptionArgs {
 }
 
 #[derive(Parser, Clone, Debug, PartialEq, Eq)]
+pub struct SetBranchArgs {
+    pub directory: PathBuf,
+    pub branch: String,
+}
+
+#[derive(Parser, Clone, Debug, PartialEq, Eq)]
 pub struct InitArgs {
     #[arg(long)]
     pub group: Option<String>,
     #[arg(long)]
     pub description: Option<String>,
+    #[arg(long, default_value = "main")]
+    pub branch: String,
     pub repo_name: String,
 }