From 7abaf07fb52bdd4d62f8bcac5b09859cf59803ec Mon Sep 17 00:00:00 2001
From: jared-rimmer <jared.rimmer@brooklyndata.co>
Date: Fri, 4 Nov 2022 16:18:17 +0000
Subject: [PATCH] Add check branch meets naming convention hook

---
 .../check_branch_meets_naming_convention.py   | 47 +++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 pre_commit_hooks/check_branch_meets_naming_convention.py

diff --git a/pre_commit_hooks/check_branch_meets_naming_convention.py b/pre_commit_hooks/check_branch_meets_naming_convention.py
new file mode 100644
index 0000000..db9b77e
--- /dev/null
+++ b/pre_commit_hooks/check_branch_meets_naming_convention.py
@@ -0,0 +1,47 @@
+from __future__ import annotations
+
+import argparse
+import re
+from typing import AbstractSet
+from typing import Sequence
+
+from pre_commit_hooks.util import CalledProcessError
+from pre_commit_hooks.util import cmd_output
+
+def has_naming_convention(
+        patterns: AbstractSet[str] = frozenset(),
+) -> bool:
+    try:
+        ref_name = cmd_output('git', 'symbolic-ref', 'HEAD')
+    except CalledProcessError:
+        return False
+    chunks = ref_name.strip().split('/')
+    branch_name = '/'.join(chunks[2:])
+    
+    return any(
+        re.match(p, branch_name) for p in patterns
+    )
+
+
+def main(argv: Sequence[str] | None = None) -> int:
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        '-b', '--branch', action='append',
+        help='branch to disallow commits to, may be specified multiple times',
+    )
+    parser.add_argument(
+        '-p', '--pattern', action='append',
+        help=(
+            'regex pattern for branch name to allow commits to, '
+            'may be specified multiple times'
+        ),
+    )
+    args = parser.parse_args(argv)
+
+    patterns = frozenset(args.pattern or ())
+    
+    return int(has_naming_convention(patterns))
+
+if __name__ == '__main__':
+    raise SystemExit(main())
+