mamba/releaser.py

# script to release any of the mamba packages
import copy
import datetime
import re

template = {"version": None, "changes": []}

templates = {
    "libmamba": "libmamba/include/mamba/version.hpp.tmpl",
    "micromamba": "micromamba/src/version.hpp.tmpl",
    "libmambapy": "libmambapy/libmambapy/_version.py.tmpl",
    "mamba": "mamba/mamba/_version.py.tmpl",
}


def apply_changelog(name, version, changes):
    res = ""
    today = datetime.date.today()
    fmt_today = today.strftime("%B %d, %Y")

    header_line = f"{name} {version} ({fmt_today})"
    res += f"{header_line}\n{'=' * len(header_line)}\n\n"

    for idx, c in enumerate(changes):
        if c.startswith("-"):
            if idx > 0 and not changes[idx - 1].startswith("- "):
                res += f"\n{c}\n"
            else:
                res += f"{c}\n"
        else:
            res += f"{c}\n"
    res += "\n"

    cl_file = name + "/CHANGELOG.md"
    with open(cl_file, "r") as fi:
        prev_cl = fi.read()
    with open(cl_file, "w") as fo:
        fo.write(res + prev_cl)

    version_major, version_minor, version_patch = version.split(".")

    def template_substitute(contents):
        x = contents.replace("{{ version_major }}", version_major)
        x = x.replace("{{ version_minor }}", version_minor)
        x = x.replace("{{ version_patch }}", version_patch)
        return x

    if name in templates:
        template = templates[name]
        with open(template, "r") as fi:
            final = template_substitute(fi.read())
        with open(template[: -len(".tmpl")], "w") as fo:
            fo.write(final)


def commands(changes):
    print("pre-commit run --all")
    print("git diff")

    commit_msg = ", ".join([f"{x} {changes[x]['version']}" for x in changes])

    today = datetime.date.today()
    date_stamp = today.strftime("%Y.%m.%d")

    files_to_commit = ""
    for c in changes:
        files_to_commit += f"    {c}/CHANGELOG.md \\\n"
        files_to_commit += f"    {templates[c][:-len('.tmpl')]} \\\n"
    files_to_commit = files_to_commit[:-3]

    print(f"git commit -m 'release {commit_msg}' \\\n{files_to_commit}")

    print(f"git tag {date_stamp}")
    for c in changes:
        print(f"git tag {c}_{changes[c]['version']}")


class Section:
    def __init__(self):
        self.items = []
        self.applies_to = ["all"]
        self.text = ""


class Item:
    def __init__(self):
        self.applies_to = ["all"]
        self.text = ""


def populate_changes(name, sections, changes):
    el = changes[name]

    def applies(x):
        return "all" in x or name in x

    for s in sections:
        s_applies = applies(s.applies_to)
        if s_applies and len(s.items):
            s_applies = any(applies(i.applies_to) for i in s.items)

        if s_applies:
            if s != sections[0]:
                el["changes"].append("\n" + s.text.strip())
            else:
                el["changes"].append(s.text.strip())
            for i in s.items:
                if applies(i.applies_to):
                    el["changes"].append(f"- {i.text.strip()}")


def main():
    changes = {}
    with open("CHANGELOG.md", "r") as fi:
        contents = fi.readlines()

    for idx, line in enumerate(contents):
        if line.startswith("====="):
            release_start = idx + 1
            break

    brackets_re = re.compile(r"\[(.*)\]")

    # section with groups, heading + items
    sections = []
    in_section = False

    contents = contents[release_start:]
    for idx, c in enumerate(contents):

        if c.startswith("Releases"):
            releases = [x.strip() for x in c[len("Releases: ") :].split(",")]
            for r in releases:
                rsplit = r.split()
                changes[rsplit[0].strip()] = copy.deepcopy(template)
                changes[rsplit[0].strip()]["version"] = rsplit[1].strip()
            continue

        if contents[idx + 1].startswith("===="):
            break

        if c.strip() == "" or c[0] == "-":
            in_section = False

        if c.strip() == "":
            continue

        if c[0] != "-":
            if not in_section:
                sections.append(Section())
                in_section = True
            sections[-1].text += c

        if m := re.search(brackets_re, c):
            if in_section:
                sections[-1].applies_to = [x.strip() for x in m.groups(1)[0].split(",")]
            else:
                sections[-1].items.append(Item())
                sections[-1].items[-1].text = c[m.end() :].strip()
                sections[-1].items[-1].applies_to = [
                    x.strip() for x in m.groups(1)[0].split(",")
                ]

        else:
            if c.startswith(" "):
                if in_section:
                    sections[-1].text += " " + c.strip()
                else:
                    sections[-1].items[-1].text += c.strip()
            else:
                if not in_section:
                    sections[-1].items.append(Item())
                    sections[-1].items[-1].text = c.strip()
                    sections[-1].items[-1].applies_to = ["all"]

    for c in changes:
        populate_changes(c, sections, changes)

    for el in changes:
        apply_changelog(el, changes[el]["version"], changes[el]["changes"])

    commands(changes)


if __name__ == "__main__":
    main()