Set av-pushed-remote and av-pushed-ref on push

internal/actions/pr.go

@@ -209,6 +209,12 @@ func CreatePullRequest(
 		if _, err := repo.Git(pushFlags...); err != nil {
 			return nil, errors.WrapIf(err, "failed to push")
 		}
+		if err := repo.BranchSetConfig(opts.BranchName, "av-pushed-remote", "origin"); err != nil {
+			return nil, err
+		}
+		if err := repo.BranchSetConfig(opts.BranchName, "av-pushed-ref", fmt.Sprintf("refs/heads/%s", opts.BranchName)); err != nil {
+			return nil, err
+		}
 	} else {
 		_, _ = fmt.Fprint(os.Stderr,
 			"  - skipping push to GitHub",

internal/actions/push.go

@@ -114,6 +114,12 @@ func Push(repo *git.Repo, branchName string, opts PushOpts) error {
 		}
 		return errors.WrapIff(err, "failed to push branch %q", branchName)
 	}
+	if err := repo.BranchSetConfig(branchName, "av-pushed-remote", "origin"); err != nil {
+		return err
+	}
+	if err := repo.BranchSetConfig(branchName, "av-pushed-ref", fmt.Sprintf("refs/heads/%s", branchName)); err != nil {
+		return err
+	}
 	_, _ = fmt.Fprint(os.Stderr,
 		colors.Success("okay"), "\n",
 	)

internal/git/branch.go

@@ -1,5 +1,7 @@
 package git
 
+import "fmt"
+
 // BranchDelete deletes the given branches (equivalent to `git branch -D`).
 func (r *Repo) BranchDelete(names ...string) error {
 	_, err := r.Run(&RunOpts{
@@ -8,3 +10,13 @@ func (r *Repo) BranchDelete(names ...string) error {
 	})
 	return err
 }
+
+// BranchSetConfig sets a config on the given branch (equivalent to `git config
+// branch.<branch>.<key> <value>`).
+func (r *Repo) BranchSetConfig(name, key, value string) error {
+	_, err := r.Run(&RunOpts{
+		Args:      []string{"config", fmt.Sprintf("branch.%s.%s", name, key), value},
+		ExitError: true,
+	})
+	return err
+}