#!/usr/bin/env python3
""" Driver script to run all code generation """

import argparse
import logging
import os
import sys
import pathlib
import typing

if 'BUILD_WORKSPACE_DIRECTORY' not in os.environ:
    # we are not running with `bazel run`, set up module search path
    _repo_root = pathlib.Path(__file__).resolve().parents[2]
    sys.path.append(str(_repo_root))

from swift.codegen.lib import render, paths
from swift.codegen.generators import generate


def _parse_args() -> argparse.Namespace:
    p = argparse.ArgumentParser(description="Code generation suite")
    p.add_argument("--generate", type=lambda x: x.split(","), default=["dbscheme", "ql"],
                   help="specify what targets to generate as a comma separated list, choosing among dbscheme, ql, trap "
                        "and cpp")
    p.add_argument("--verbose", "-v", action="store_true", help="print more information")
    p.add_argument("--quiet", "-q", action="store_true", help="only print errors")
    p.add_argument("--swift-dir", type=_abspath, default=paths.swift_dir,
                   help="the directory that should be regarded as the root of the swift codebase. Used to compute QL "
                        "imports and in some comments (default %(default)s)")
    p.add_argument("--schema", type=_abspath, default=paths.swift_dir / "schema.py",
                   help="input schema file (default %(default)s)")
    p.add_argument("--dbscheme", type=_abspath, default=paths.swift_dir / "ql/lib/swift.dbscheme",
                   help="output file for dbscheme generation, input file for trap generation (default %(default)s)")
    p.add_argument("--ql-output", type=_abspath, default=paths.swift_dir / "ql/lib/codeql/swift/generated",
                   help="output directory for generated QL files (default %(default)s)")
    p.add_argument("--ql-stub-output", type=_abspath, default=paths.swift_dir / "ql/lib/codeql/swift/elements",
                   help="output directory for QL stub/customization files (default %(default)s). Defines also the "
                        "generated qll file importing every class file")
    p.add_argument("--ql-test-output", type=_abspath, default=paths.swift_dir / "ql/test/extractor-tests/generated",
                   help="output directory for QL generated extractor test files (default %(default)s)")
    p.add_argument("--ql-format", action="store_true", default=True,
                   help="use codeql to autoformat QL files (which is the default)")
    p.add_argument("--no-ql-format", action="store_false", dest="ql_format", help="do not format QL files")
    p.add_argument("--codeql-binary", default="codeql", help="command to use for QL formatting (default %(default)s)")
    p.add_argument("--cpp-output", type=_abspath,
                   help="output directory for generated C++ files, required if trap or cpp is provided to --generate")
    p.add_argument("--generated-registry", type=_abspath, default=paths.swift_dir / "ql/.generated.list",
                   help="registry file containing information about checked-in generated code")
    p.add_argument("--force", "-f", action="store_true",
                   help="generate all files without skipping unchanged files and overwriting modified ones")
    return p.parse_args()


def _abspath(x: str) -> typing.Optional[pathlib.Path]:
    return pathlib.Path(x).resolve() if x else None


def run():
    opts = _parse_args()
    if opts.verbose:
        log_level = logging.DEBUG
    elif opts.quiet:
        log_level = logging.ERROR
    else:
        log_level = logging.INFO
    logging.basicConfig(format="{levelname} {message}", style='{', level=log_level)
    for target in opts.generate:
        generate(target, opts, render.Renderer(opts.swift_dir))


if __name__ == "__main__":
    run()