From 945549acbcff41982de37cfe6d807152be37adc0 Mon Sep 17 00:00:00 2001
From: surafelabebe <surafel@clumio.com>
Date: Mon, 21 Dec 2020 15:16:11 -0800
Subject: [PATCH] The remote branch reference was exposed as an environment
 variable for a pre-push hook, which makes decisions on git pushes to a
 particular remote branch name pattern.

---
 pre_commit/commands/hook_impl.py | 7 ++++++-
 pre_commit/commands/run.py       | 5 +++++
 testing/util.py                  | 2 ++
 3 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/pre_commit/commands/hook_impl.py b/pre_commit/commands/hook_impl.py
index d0e226f8..25c5fdff 100644
--- a/pre_commit/commands/hook_impl.py
+++ b/pre_commit/commands/hook_impl.py
@@ -69,6 +69,7 @@ def _ns(
         color: bool,
         *,
         all_files: bool = False,
+        remote_branch: Optional[str] = None,
         from_ref: Optional[str] = None,
         to_ref: Optional[str] = None,
         remote_name: Optional[str] = None,
@@ -79,6 +80,7 @@ def _ns(
     return argparse.Namespace(
         color=color,
         hook_stage=hook_type.replace('pre-', ''),
+        remote_branch=remote_branch,
         from_ref=from_ref,
         to_ref=to_ref,
         remote_name=remote_name,
@@ -106,13 +108,14 @@ def _pre_push_ns(
     remote_url = args[1]
 
     for line in stdin.decode().splitlines():
-        _, local_sha, _, remote_sha = line.split()
+        _, local_sha, remote_branch, remote_sha = line.split()
         if local_sha == Z40:
             continue
         elif remote_sha != Z40 and _rev_exists(remote_sha):
             return _ns(
                 'pre-push', color,
                 from_ref=remote_sha, to_ref=local_sha,
+                remote_branch=remote_branch,
                 remote_name=remote_name, remote_url=remote_url,
             )
         else:
@@ -133,6 +136,7 @@ def _pre_push_ns(
                         'pre-push', color,
                         all_files=True,
                         remote_name=remote_name, remote_url=remote_url,
+                        remote_branch=remote_branch,
                     )
                 else:
                     rev_cmd = ('git', 'rev-parse', f'{first_ancestor}^')
@@ -141,6 +145,7 @@ def _pre_push_ns(
                         'pre-push', color,
                         from_ref=source, to_ref=local_sha,
                         remote_name=remote_name, remote_url=remote_url,
+                        remote_branch=remote_branch,
                     )
 
     # nothing to push
diff --git a/pre_commit/commands/run.py b/pre_commit/commands/run.py
index 1e8fad23..8523b29f 100644
--- a/pre_commit/commands/run.py
+++ b/pre_commit/commands/run.py
@@ -371,6 +371,11 @@ def run(
         environ['PRE_COMMIT_FROM_REF'] = args.from_ref
         environ['PRE_COMMIT_TO_REF'] = args.to_ref
 
+    # Exposes remote branch ref as an environment variable
+    # for pre-push hooks consumption
+    if args.remote_branch:
+        environ['PRE_COMMIT_REMOTE_REF'] = args.remote_branch
+
     if args.remote_name and args.remote_url:
         environ['PRE_COMMIT_REMOTE_NAME'] = args.remote_name
         environ['PRE_COMMIT_REMOTE_URL'] = args.remote_url
diff --git a/testing/util.py b/testing/util.py
index 18cd7342..1f8cb35d 100644
--- a/testing/util.py
+++ b/testing/util.py
@@ -61,6 +61,7 @@ def run_opts(
         color=False,
         verbose=False,
         hook=None,
+        remote_branch='',
         from_ref='',
         to_ref='',
         remote_name='',
@@ -78,6 +79,7 @@ def run_opts(
         color=color,
         verbose=verbose,
         hook=hook,
+        remote_branch=remote_branch,
         from_ref=from_ref,
         to_ref=to_ref,
         remote_name=remote_name,