release_utils.py

# Copyright (c) Meta Platforms, Inc. and affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import argparse
from typing import Tuple


def get_current_version() -> Tuple[str, str]:
    current_version = open("version.txt", "r").read().strip()
    current_tag = "v" + current_version
    return current_version, current_tag


def get_next_version(release_type) -> Tuple[Tuple[int, int, int], str, str]:
    current_version = open("version.txt", "r").read().strip()
    version_list = [int(x) for x in current_version.strip("'").split(".")]
    major, minor, patch = version_list[0], version_list[1], version_list[2]
    if release_type == "patch":
        patch += 1
    elif release_type == "minor":
        minor += 1
        patch = 0
    elif release_type == "major":
        major += 1
        minor = patch = 0
    else:
        raise ValueError(
            "Incorrect release type specified. Acceptable types are major, minor and patch."
        )

    new_version_tuple = (major, minor, patch)
    new_version_str = ".".join([str(x) for x in new_version_tuple])
    new_tag_str = "v" + new_version_str
    return new_version_tuple, new_version_str, new_tag_str


def update_version(new_version_tuple) -> None:
    """
    given the current version, update the version to the
    next version depending on the type of release.
    """

    with open("version.txt", "w") as writer:
        writer.write(".".join([str(x) for x in new_version_tuple]))


def main(args):
    if args.release_type:
        if args.release_type in ["major", "minor", "patch"]:
            new_version_tuple, new_version, new_tag = get_next_version(
                args.release_type
            )
            print(new_version, new_tag)
        else:
            raise ValueError("Incorrect release type specified")

    if args.update_version:
        update_version(new_version_tuple)

    if args.get_current_version:
        current_version, current_tag = get_current_version()
        print(current_version, current_tag)


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Versioning utils")
    parser.add_argument(
        "--release-type",
        type=str,
        required=False,
        help="type of release = major/minor/patch",
    )
    parser.add_argument(
        "--update-version",
        action="store_true",
        required=False,
        help="updates the version in version.txt",
    )
    parser.add_argument(
        "--get-current-version",
        action="store_true",
        required=False,
        help="get the current version in version.txt",
    )

    args = parser.parse_args()
    main(args)