From 635e1423d311d775b074ed7e2c8a867f88b7cf29 Mon Sep 17 00:00:00 2001
From: Marc Cataford <mcat@riseup.net>
Date: Sun, 3 Nov 2024 20:59:35 -0500
Subject: [PATCH 1/2] refactor: pull git wrapper together

---
 frg/context.py        | 14 ++++----------
 frg/git.py            | 29 +++++++++++++++++++++++++++++
 tests/test_context.py |  3 +++
 3 files changed, 36 insertions(+), 10 deletions(-)
 create mode 100644 frg/git.py

diff --git a/frg/context.py b/frg/context.py
index 530463f..c5a2466 100644
--- a/frg/context.py
+++ b/frg/context.py
@@ -6,10 +6,11 @@ invocation environment (i.e. git context, ...) to make things work.
 """
 
 import logging
-import subprocess
 
 import pydantic
 
+import frg.git as git
+
 logger = logging.getLogger(__name__)
 
 
@@ -56,15 +57,8 @@ def get_git_context(*, domain_aliases: dict[str, str] | None = None) -> GitConte
     if not domain_aliases:
         domain_aliases = dict()
 
-    current_branch_cmd = subprocess.run(
-        ["git", "branch", "--show-current"], capture_output=True
-    )
-    current_branch = current_branch_cmd.stdout.decode("utf8").strip()
-
-    remote_url_cmd = subprocess.run(
-        ["git", "config", "--get", "remote.origin.url"], capture_output=True
-    )
-    remote_url = remote_url_cmd.stdout.decode("utf8").strip()
+    current_branch = git.get_current_branch().stdout
+    remote_url = git.get_current_remote_url().stdout
 
     host, owner, repo = parse_remote_url(remote_url)
 
diff --git a/frg/git.py b/frg/git.py
new file mode 100644
index 0000000..83b05bd
--- /dev/null
+++ b/frg/git.py
@@ -0,0 +1,29 @@
+import subprocess
+
+import pydantic
+
+
+class CommandResult(pydantic.BaseModel):
+    return_code: int
+    stdout: str
+    stderr: str
+
+
+def _git(args: list[str]) -> CommandResult:
+    result = subprocess.run(["git", *args], capture_output=True)
+
+    return CommandResult(
+        stdout=result.stdout.decode("utf8").strip(),
+        stderr=result.stderr.decode("utf8").strip(),
+        return_code=result.returncode,
+    )
+
+
+def get_current_branch() -> CommandResult:
+    """Returns the current checked out branch."""
+    return _git(["branch", "--show-current"])
+
+
+def get_current_remote_url() -> CommandResult:
+    """Returns the remote origin url."""
+    return _git(["config", "--get", "remote.origin.url"])
diff --git a/tests/test_context.py b/tests/test_context.py
index c94afd3..58c0b33 100644
--- a/tests/test_context.py
+++ b/tests/test_context.py
@@ -28,6 +28,9 @@ def mock_subprocess(mock_context):
                 "utf8"
             )
 
+        return_val.stderr = b""
+        return_val.returncode = 0
+
         return return_val
 
     with unittest.mock.patch("subprocess.run") as mock_run:
-- 
2.45.2


From fd11d844667e7ffb80b4a14515128a4d008aff24 Mon Sep 17 00:00:00 2001
From: Marc Cataford <mcat@riseup.net>
Date: Sun, 3 Nov 2024 21:11:12 -0500
Subject: [PATCH 2/2] feat: ensure push on pr create

---
 frg/cli.py | 4 ++++
 frg/git.py | 5 +++++
 2 files changed, 9 insertions(+)

diff --git a/frg/cli.py b/frg/cli.py
index 32a229b..f982d28 100644
--- a/frg/cli.py
+++ b/frg/cli.py
@@ -4,6 +4,7 @@ import click
 import pydantic
 
 import frg.forgejo.browser as forgejo_browser
+import frg.git as git
 from frg.configuration import Config, get_configuration
 from frg.context import GitContext, get_git_context
 
@@ -61,6 +62,9 @@ def pr(ctx):
 @click.pass_obj
 def create_pr(ctx, web: bool):
     """Interacts with pull requests."""
+
+    git.push(branch=ctx.git.current_branch)
+
     if web:
         forgejo_browser.create_pull_request_via_web(
             head=ctx.git.current_branch,
diff --git a/frg/git.py b/frg/git.py
index 83b05bd..50ccffd 100644
--- a/frg/git.py
+++ b/frg/git.py
@@ -27,3 +27,8 @@ def get_current_branch() -> CommandResult:
 def get_current_remote_url() -> CommandResult:
     """Returns the remote origin url."""
     return _git(["config", "--get", "remote.origin.url"])
+
+
+def push(*, branch: str) -> CommandResult:
+    """Pushes the current local commits to remote."""
+    return _git(["push", "--set-upstream", "origin", branch])
-- 
2.45.2