dev

#!/usr/bin/env python3

# This script is an entrypoint for all developer activities in this project.

import argparse
import os
import subprocess
import sys


def cmd(cmds: argparse._SubParsersAction, name: str, help: str):
    """Decorator that registers subcommands as functions to be executed."""

    def wrapper(func):
        p = cmds.add_parser(name, help=help)
        p.set_defaults(func=func)

    return wrapper


def run(cmd: str, args: list = None, capture_output=False, env: dict = None):
    """Helper to run cmd in a shell."""
    if args:
        if args[0] == "--":
            args = args[1:]
        cmd += " -- " + " ".join(args)
    return subprocess.run(
        cmd, check=True, shell=True, capture_output=capture_output, env=env
    )


def main():
    if not os.getenv("DIRENV_DIR"):
        print("Direnv is not enabled. Fix it first! See README.md for instructions.")
        sys.exit(1)

    cli = argparse.ArgumentParser(
        usage="./dev COMMAND [FLAGS...]",
        description="CLI for developing Mintter. Provides commands for most common developer tasks in this project.",
    )

    cmds = cli.add_subparsers(
        title="commands",
        # This is ugly, but otherwise argparse prints the redundant list of subcommands.
        # And if we just use an empty string it messes up help message alignment for some subcommands.
        metavar="                                                     ",
    )

    @cmd(
        cmds,
        "gen",
        "Check all the generated code is up to date. Otherwise run the code generation process to fix it.",
    )
    def gen(args):
        out = run(
            "plz run parallel $(plz query filter -i 'generated:check') "
            + str.join(" ", args),
            capture_output=True,
        )

        targets_to_gen = []
        for line in out.stdout.decode("utf-8").split("\n"):
            idx = line.find("plz run")
            if idx == -1:
                continue
            targets_to_gen.append(line[idx + 7 : -1])  # 7 is length of 'plz run'

        if len(targets_to_gen) == 0:
            return
        return run("plz run parallel " + " ".join(targets_to_gen))
    
    @cmd(cmds, "run-desktop", "Run frontend desktop app for development.")
    def run_desktop(args):
        run("./scripts/cleanup-desktop.sh")
        run("yarn install")
        run("plz build //backend:mintterd //:yarn")
        return run("yarn desktop", args=args)
    
    @cmd(
        cmds, "build-desktop", "Builds the desktop app for the current platform."
    )
    def build_desktop(args):
        run("./scripts/cleanup-frontend.sh")
        run("./scripts/cleanup-desktop.sh")
        run("yarn install")
        run("plz build //backend:mintterd //:yarn")
        run("yarn desktop:make")
    
    
    @cmd(cmds, "run-site", "Run sites app for development.")
    def run_site(args):
        run("./scripts/cleanup-site.sh")
        run("yarn install")
        run("plz build //:yarn")
        return run(
            "yarn site",
            args=args,
            env={**os.environ, "GW_NEXT_HOST": "http://localhost:3000"},
        )
    
    @cmd(cmds, "build-site", "Build site app for production.")
    def build_site(args):
        run("./scripts/cleanup-frontend.sh")
        run("./scripts/cleanup-site.sh")
        run("yarn install")
        run("plz build //:yarn")
        return run(
            "yarn site:prod",
            args=args,
            # env={**os.environ, "GW_NEXT_HOST": "http://localhost:3000"},
        )
    
    @cmd(
        cmds, "validate", "Formats, Validates and test the site app"
    )
    def validate(args):
        run("yarn validate")

    @cmd(cmds, "run-backend", "Build and run mintterd binary for the current platform.")
    def run_backend(args):
        return run("plz run //backend:mintterd", args=args)
    
    @cmd(cmds, "build-backend", "Build mintterd binary for the current platform.")
    def build_backend(args):
        return run("plz build //backend:mintterd")

    @cmd(cmds, "run-gw-backend", "Build and run backend for mintter web gateway.")
    def run_gateway(args):
        return run("plz run //backend:minttergw", args=args)

    @cmd(cmds, "ping-p2p", "Execute ping utility to check visibility.")
    def ping_p2p(args):
        return run("plz run //backend:pingp2p", args=args)

    @cmd(cmds, "release", "Cut a new release.")
    def release(args):
        allowed_components = ["desktop", "relay"]
        want_branch = "master"

        if len(args) != 1:
            raise ValueError(
                "Must specify a component name to release. One of %s."
                % allowed_components
            )

        component = args[0]

        if component not in allowed_components:
            raise ValueError(
                "Unknown component name to release. Must be one of %s."
                % allowed_components
            )

        branch = (
            run("git branch --show-current", capture_output=True)
            .stdout.decode()
            .strip()
        )
        if branch != want_branch:
            raise ValueError("Must only release on the '%s' branch!" % want_branch)

        print("Checking if we're up to date with remote...")

        run("git fetch origin %s" % want_branch)
        diff = (
            run("git diff-index origin/%s" % want_branch, capture_output=True)
            .stdout.decode()
            .strip()
        )
        if diff:
            raise ValueError("Local repository is not up to date with remote!")

        prev_tag = (
            run(
                "git describe --tags --abbrev=0 --match '%s/*'" % component,
                capture_output=True,
            )
            .stdout.decode()
            .strip()
        )

        prev_version = prev_tag.removeprefix(component + "/")

        new_version = input(
            "Last version was %s. Specify the new version: " % prev_version
        )

        if not new_version:
            raise ValueError("Must specify version.")

        parts = new_version.split(".")
        if len(parts) != 3:
            raise ValueError(
                "Invalid version number %s. Version must be 3 numbers separated by dots."
                % new_version
            )

        print("Tagging release...")
        run("git tag -a %s" % component + "/" + new_version)
        run("git push --follow-tags")

    if len(sys.argv) == 1:
        cli.print_help()
        return

    namespace, args = cli.parse_known_args()

    try:
        namespace.func(args)
    except ValueError as err:
        print(str(err))
        sys.exit(1)
    except (subprocess.CalledProcessError, KeyboardInterrupt):
        return


if __name__ == "__main__":
    main()