diff --git a/script/bump-version.py b/script/bump-version.py new file mode 100755 index 0000000..2f05c2e --- /dev/null +++ b/script/bump-version.py @@ -0,0 +1,95 @@ +#!/usr/bin/env python3 + +import argparse +import re +import subprocess +from dataclasses import dataclass +import sys + + +@dataclass +class Version: + major: int + minor: int + patch: int + beta: int = 0 + dev: bool = False + + def __str__(self): + return f'{self.major}.{self.minor}.{self.full_patch}' + + @property + def full_patch(self): + res = f'{self.patch}' + if self.beta > 0: + res += f'b{self.beta}' + if self.dev: + res += '-dev' + return res + + @classmethod + def parse(cls, value): + match = re.match(r'(\d+).(\d+).(\d+)(b\d+)?(-dev)?', value) + assert match is not None + major = int(match[1]) + minor = int(match[2]) + patch = int(match[3]) + beta = int(match[4][1:]) if match[4] else 0 + dev = bool(match[5]) + return Version( + major=major, minor=minor, patch=patch, + beta=beta, dev=dev + ) + + +def sub(path, pattern, repl, expected_count=1): + with open(path) as fh: + content = fh.read() + content, count = re.subn(pattern, repl, content, re.MULTILINE) + if expected_count is not None: + assert count == expected_count + with open(path, "wt") as fh: + fh.write(content) + + +def write_version(target: str, version: Version): + # version: '1.14.5' # BETA + # version: '1.14.5' # STABLE + sub( + 'template/config.yaml', + r" version: '[^']+' # {}".format(target.upper()), + f" version: '{version}' # {target.upper()}" + ) + + +def main(): + parser = argparse.ArgumentParser() + parser.add_argument('new_version', type=str) + parser.add_argument('--beta', action='store_true', help="Update the beta addon") + parser.add_argument('--stable', action='store_true', help="Update the stable addon") + parser.add_argument('--commit', action='store_true') + args = parser.parse_args() + + if not any([args.beta, args.stable]): + print("At least one of beta or stable has to be bumped") + return 1 + + if args.commit and subprocess.call(["git", "diff", "--quiet"]) == 1: + print("Cannot use --commit because git is dirty.") + return 1 + + version = Version.parse(args.new_version) + print(f"Bumping to {version}") + if args.beta: + write_version('beta', version) + if args.stable: + assert not version.beta + write_version('stable', version) + + if args.commit: + subprocess.check_call(["git", "commit", "-nam", f"Bump version to v{version}"]) + return 0 + + +if __name__ == "__main__": + sys.exit(main() or 0)