import os
import os.path
import logging
from logging.handlers import RotatingFileHandler
from memgpt.constants import (
    LOGGER_NAME,
    LOGGER_DEFAULT_LEVEL,
    LOGGER_DIR,
    LOGGER_FILENAME,
    LOGGER_FILE_BACKUP_COUNT,
    LOGGER_MAX_FILE_SIZE,
)

# Checking if log directory exists
if not os.path.exists(LOGGER_DIR):
    os.makedirs(LOGGER_DIR, exist_ok=True)

# Create logger for MemGPT
logger = logging.getLogger(LOGGER_NAME)
logger.setLevel(LOGGER_DEFAULT_LEVEL)

# create console handler and set level to debug
console_handler = logging.StreamHandler()

# create rotatating file handler
file_handler = RotatingFileHandler(
    os.path.join(LOGGER_DIR, LOGGER_FILENAME), maxBytes=LOGGER_MAX_FILE_SIZE, backupCount=LOGGER_FILE_BACKUP_COUNT
)

# create formatters
console_formatter = logging.Formatter("%(name)s - %(levelname)s - %(message)s")  # not datetime
file_formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")

# add formatter to console handler
console_handler.setFormatter(console_formatter)

# add formatter for file handler
file_handler.setFormatter(file_formatter)

# add ch to logger
logger.addHandler(console_handler)
logger.addHandler(file_handler)