SRS/cli.py

#! /usr/bin/env python

import argparse
import subprocess
import sys
import os

def updateIssue(issuePath: str, verbose = False):
    if verbose:
        print("get issues")
    p = subprocess.run(["deno", "run", "-A","tools/getIssue.ts", "--path",issuePath])
    p.check_returncode()

def printDocument(issuePath:str, outDir:str, watch = False, verbose = False):
    if verbose:
        print("build document")
    cmd = ["deno", "run", "-A","tools/printDocument.ts", "--issue_path", issuePath, "--outDir", outDir]
    if watch:
        cmd.append("--watch")
    p = subprocess.run(cmd)
    p.check_returncode()

def build(args):
    parser = argparse.ArgumentParser(description='Compiling the documentation', prog="cli.py build")
    parser.add_argument('-v', '--verbose', action='store_true', help='verbose mode')
    parser.add_argument('--update_issues', action='store_true', help='update issues')
    parser.add_argument('--outDir', default="build", help='output directory')
    parser.add_argument('-w', '--watch', action='store_true', help='watch for changes')

    args = parser.parse_args(args)
    if args.verbose:
        print("build start")

    issuePath = os.path.join(args.outDir,"issues.json")
    if args.update_issues:
        updateIssue(issuePath, args.verbose)
    printDocument(issuePath, args.outDir, args.watch, args.verbose)

def serve(args):
    """serve the documentation and reload on changes"""
    parser = argparse.ArgumentParser(description='Serve the documentation and reload changes', prog="cli.py serve")
    parser.add_argument('-v', '--verbose', action='store_true', help='verbose mode')
    parser.add_argument('-p', '--port', default=8000, help='port')
    parser.add_argument('--update_issues', action='store_true', help='update issues')
    outDir = "build"
    issuePath = os.path.join(outDir,"issues.json")
    args = parser.parse_args(args)
    if args.verbose:
        print("serve start")
    cmd = ["mdbook", "serve", "--port", str(args.port)]
    p = subprocess.Popen(cmd)
    printDocument(issuePath, outDir, True, args.verbose)
    p.wait()
    if p.returncode != 0:
        sys.exit(p.returncode)

def help(_args):
    global commandList
    print("subcommands:")
    for command in commandList.keys():
        print("\t", command)

def issueUpdate(args):
    parser = argparse.ArgumentParser(description='Update issues', prog="cli.py issueUpdate")
    parser.add_argument('-v', '--verbose', action='store_true', help='verbose mode')
    parser.add_argument('--outDir', default="build", help='output directory')
    args = parser.parse_args(args)
    issuePath = os.path.join(args.outDir,"issues.json")
    updateIssue(issuePath, args.verbose)

commandList = {
    'build': build,
    'help': help,
    'issueUpdate': issueUpdate,
    'serve': serve
}

def main():
    args = sys.argv
    if len(args) < 2:
        print("no subcommand", file=sys.stderr)
        help(args)
        return
    _command = args[0]
    subCommand = args[1]
    if subCommand not in commandList:
        print("Command not found", file=sys.stderr)
        help(args)
        sys.exit(1)
    args = args[2:]
    os.makedirs("build", exist_ok=True)
    commandList[subCommand](args)

if __name__ == '__main__':
    main()