from __future__ import annotations

import argparse

import json

import sys

from cti.config import load_config

from cti.pipeline import run

def main(argv=None):

    parser = argparse.ArgumentParser(prog="cti", description="CTI detection automation")

    parser.add_argument("command", choices=["run", "serve"])

    parser.add_argument("-c", "--config", default=None)

    parser.add_argument("--host", default="0.0.0.0")

    parser.add_argument("--port", type=int, default=8080)

    args = parser.parse_args(argv)

    config = load_config(args.config)

    if args.command == "run":

        result = run(config)

        print(json.dumps(result, indent=2))

        return 0

    from cti.web import create_app

    create_app(config).run(host=args.host, port=args.port)

    return 0

if __name__ == "__main__":

    sys.exit(main())