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

OK = 0
ERROR = 1


def branch_follows_pattern(patterns: AbstractSet[str]) -> 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(
        "-p",
        "--pattern",
        action="append",
        help=("regex pattern that the name of the branch must comply with"),
    )
    args = parser.parse_args(argv)
    return OK if branch_follows_pattern(frozenset(args.pattern)) else ERROR


if __name__ == "__main__":
    raise SystemExit(main())