import argparse
import json
import logging
import os
from typing import TypedDict
from rich.logging import RichHandler
from rich.console import Console


FORMAT = "%(message)s"
logging.basicConfig(
    level="NOTSET",
    format=FORMAT,
    datefmt="[%X]",
    handlers=[RichHandler(rich_tracebacks=True)],
)
log = logging.getLogger("rich")
console = Console()


class ConfigTools(TypedDict):
    tool: str
    url: str


class ConfigFolders(TypedDict):
    tools: str
    apks: str
    decompiled: str
    patches: str
    dist: str


class ConfigXmlNS(TypedDict):
    android: str
    app: str


class Config(TypedDict):
    log_level: str
    tools: list[ConfigTools]
    folders: ConfigFolders
    xml_ns: ConfigXmlNS


parser = argparse.ArgumentParser(prog="anixart patcher")
parser.add_argument("--config", help="path to config.json file", default="config.json")
parser.add_argument("--no-decompile", action="store_true")
parser.add_argument("--no-compile", action="store_true")
parser.add_argument("--patch", action="store_true", help="only patch decompiled")
parser.add_argument("--sign", action="store_true", help="only sign compiled apk")
parser.add_argument("--init", action="store_true", help="init a new patch")
args = parser.parse_args()


def load_config() -> Config:
    config = None

    if not os.path.exists(args.config):
        log.exception("file `config.json` is not found!")
        exit(1)

    with open(args.config, "r", encoding="utf-8") as file:
        config = json.loads(file.read())

    log.setLevel(config.get("log_level", "NOTSET").upper())

    return config
config = load_config()