#!/usr/bin/python3

from git import Repo
import re
import sys

version = sys.argv[1]

repo = Repo()

tag = repo.git.describe(tags = True, abbrev=0)
m = re.match('llvmorg-([0-9]+)\.([0-9]+)\.([0-9]+)', tag)
if not m:
    print("error: Tag is not valid: ", tag)
    sys.exit(1)

expected_major = m.group(1)
expected_minor = m.group(2)
expected_patch = int(m.group(3)) + 1
expected_version = f"{expected_major}.{expected_minor}.{expected_patch}"

m = re.match("[0-9]+\.[0-9]+\.[0-9]+", version)
if not m:
    print("error: Version is not valid: ", version)
    sys.exit(1)

if version != expected_version:
    print("error: Expected version", expected_version, "but found version", version)
    sys.exit(1)

print("Versions match:", version, expected_version)
sys.exit(0)