New command stacky branch commit which will create a branch and commit (#3)

* New command stacky branch commit which will create a branch and commit

* Updating readme for stacky branch commit

Author: yashNaN

README.md

@@ -38,7 +38,8 @@ Syntax is as follows:
     - `stacky branch up` (`stacky b u`): move down the stack (towards `master`)
     - `stacky branch down` (`stacky b d`): move down the stack (towards `master`)
     - `stacky branch new <name>`: create a new branch on top of the current one
-- `stacky commit [-m <message>] [--amend] [--allow-empty]`: wrapper around `git commit` that syncs everything upstack
+    - `stacky branch commit <name> [-m <message>] [-a]`: create a new branch and commit changes in one command
+- `stacky commit [-m <message>] [--amend] [--allow-empty] [-a]`: wrapper around `git commit` that syncs everything upstack
     - `stacky amend`: will amend currently tracked changes to top commit
 - Based on the first argument (`stack` vs `upstack` vs `downstack`), the following commands operate on the entire current stack, everything upstack from the current PR (inclusive), or everything downstack from the current PR:
     - `stacky stack info [--pr]`

src/stacky/stacky.py

@@ -763,6 +763,34 @@ def cmd_branch_new(stack: StackBranchSet, args):
     run(CmdArgs(["git", "update-ref", "refs/stack-parent/{}".format(name), b.commit, ""]))
 
 
+def cmd_branch_commit(stack: StackBranchSet, args):
+    """Create a new branch and commit all changes with the provided message"""
+    global CURRENT_BRANCH
+    
+    # First create the new branch (same logic as cmd_branch_new)
+    b = stack.stack[CURRENT_BRANCH]
+    assert b.commit
+    name = args.name
+    create_branch(name)
+    run(CmdArgs(["git", "update-ref", "refs/stack-parent/{}".format(name), b.commit, ""]))
+    
+    # Update global CURRENT_BRANCH since we just checked out the new branch
+    CURRENT_BRANCH = BranchName(name)
+    
+    # Reload the stack to include the new branch
+    load_stack_for_given_branch(stack, CURRENT_BRANCH)
+    
+    # Now commit all changes with the provided message (or open editor if no message)
+    do_commit(
+        stack,
+        message=args.message,
+        amend=False,
+        allow_empty=False,
+        edit=True,
+        add_all=args.add_all,
+    )
+
+
 def cmd_branch_checkout(stack: StackBranchSet, args):
     branch_name = args.name
     if branch_name is None:
@@ -1726,6 +1754,12 @@ def main():
         branch_new_parser.add_argument("name", help="Branch name")
         branch_new_parser.set_defaults(func=cmd_branch_new)
 
+        branch_commit_parser = branch_subparsers.add_parser("commit", help="Create a new branch and commit all changes")
+        branch_commit_parser.add_argument("name", help="Branch name")
+        branch_commit_parser.add_argument("-m", help="Commit message", dest="message")
+        branch_commit_parser.add_argument("-a", action="store_true", help="Add all files to commit", dest="add_all")
+        branch_commit_parser.set_defaults(func=cmd_branch_commit)
+
         branch_checkout_parser = branch_subparsers.add_parser("checkout", aliases=["co"], help="Checkout a branch")
         branch_checkout_parser.add_argument("name", help="Branch name", nargs="?")
         branch_checkout_parser.set_defaults(func=cmd_branch_checkout)