import logging

import coloredlogs

from llm_server import opts


class LoggingInfo:
    def __init__(self):
        self._level = logging.INFO
        self._format = opts.LOGGING_FORMAT

    @property
    def level(self):
        return self._level

    @level.setter
    def level(self, value):
        self._level = value

    @property
    def format(self):
        return self._format

    @format.setter
    def format(self, value):
        self._format = value


logging_info = LoggingInfo()


def init_logging():
    """
    Set up the parent logger.
    :return:
    """
    logger = logging.getLogger('llm_server')
    logger.setLevel(logging_info.level)


def create_logger(name):
    logger = logging.getLogger('llm_server').getChild(name)
    logger.setLevel(logging_info.level)
    if not logger.handlers:
        handler = logging.StreamHandler()
        handler.setLevel(logging_info.level)
        formatter = logging.Formatter(logging_info.format)
        handler.setFormatter(formatter)
        logger.addHandler(handler)
        coloredlogs.install(logger=logger, level=logging_info.level)
    return logger