import logging
import subprocess
import sys

import pydantic
from pydantic.fields import Field

from .console_formatter import ColoredConsoleFormatter
from .json_formatter import JSONFormatter


def get_git_commit() -> str:
    try:
        return subprocess.check_output(['git', 'rev-parse', '--short', 'HEAD'], text=True).strip()
    except Exception:
        return 'unknown'


class LoggerConfig(pydantic.BaseModel):
    APP_NAME: str
    APP_VERSION: str = Field(default_factory=get_git_commit)
    LEVEL: int = logging.INFO
    PRETTY_CONSOLE: bool = False


def init(config: LoggerConfig) -> None:
    root_logger = logging.getLogger()
    root_logger.setLevel(config.LEVEL)
    root_logger.handlers.clear()

    handler = logging.StreamHandler(sys.stdout)
    handler.setLevel(config.LEVEL)

    formatter = ColoredConsoleFormatter() if config.PRETTY_CONSOLE else JSONFormatter()
    handler.setFormatter(formatter)
    root_logger.addHandler(handler)

    for logger_name in ['uvicorn', 'uvicorn.access', 'uvicorn.error']:
        uvicorn_logger = logging.getLogger(logger_name)
        uvicorn_logger.handlers = [handler]
        uvicorn_logger.propagate = False

    old_factory = logging.getLogRecordFactory()

    def record_factory(*args, **kwargs) -> logging.LogRecord:
        record = old_factory(*args, **kwargs)
        record.app_name = config.APP_NAME
        record.app_version = config.APP_VERSION
        return record

    logging.setLogRecordFactory(record_factory)

    logging.info(
        'Logger initialized', extra={'app': config.APP_NAME, 'version': config.APP_VERSION}
    )