#!/usr/bin/env python3

import argparse
import os
import re


def main():
    filepath = "pyzebra/__init__.py"

    parser = argparse.ArgumentParser()
    parser.add_argument("level", type=str, choices=["patch", "minor", "major"])
    parser.add_argument("tag_msg", type=str, help="tag message")
    args = parser.parse_args()

    with open(filepath) as f:
        file_content = f.read()

    version = re.search(r'__version__ = "(.*?)"', file_content).group(1)
    major, minor, patch = map(int, version.split(sep="."))

    if args.level == "patch":
        patch += 1
    elif args.level == "minor":
        minor += 1
        patch = 0
    elif args.level == "major":
        major += 1
        minor = 0
        patch = 0

    new_version = f"{major}.{minor}.{patch}"

    with open(filepath, "w") as f:
        f.write(re.sub(r'__version__ = "(.*?)"', f'__version__ = "{new_version}"', file_content))

    os.system(f"git commit {filepath} -m 'Updating for version {new_version}'")
    os.system(f"git tag -a {new_version} -m '{args.tag_msg}'")


if __name__ == "__main__":
    main()