Refactor

1 files changed, 28 insertions, 25 deletions

diff --git a/dots_manager/kawaii_logger.py b/dots_manager/kawaii_logger.py
index f82da23..f13ea14 100644
--- a/dots_manager/kawaii_logger.py
+++ b/dots_manager/kawaii_logger.py
@@ -2,20 +2,22 @@ import logging
 import sys
 import random
 
-MOOD_EMOTICONS = {
-    "debug": ["(=^･^=)", "(=・ｪ・=)", "(=^-ω-^=)", "(=^‥^=)"],
-    "info": ["(＾• ω •＾)", "(=^･ω･^)y＝", "(≚ᄌ≚)", "(｡♥‿♥｡)"],
-    "warning": ["(；￣Д￣)", "(￢_￢;)", "(･ัω･ั)", "(・_・ヾ"],
-    "error": ["(╥﹏╥)", "(≧Д≦)", "(；′⌒`)", "(Ｔ▽Ｔ)"],
+MOOD_KAOMOJI = {
+    "<_mood.happy>": ["(ﾉ◕ヮ◕)ﾉ*:･ﾟ✧", "(＾▽＾)", "( ˘⌣˘)♡(˘⌣˘ )"],
+    "<_mood.excited>": ["(๑˃ᴗ˂)ﻭ"],
+    "<_mood.sad>": ["(｡•́︿•̀｡)", "(╯︵╰,)", "(ಥ﹏ಥ)", "(︶︹︺)"],
+    "<_mood.anxious>": ["(ノдヽ)", "(◎_◎;)", "(・_・;)", "(ﾟДﾟ;)"],
 }
 
-MOOD_SUFFIXES = {
-    "_happy": ["(ﾉ◕ヮ◕)ﾉ*:･ﾟ✧", "(๑˃ᴗ˂)ﻭ", "(＾▽＾)", "( ˘⌣˘)♡(˘⌣˘ )"],
-    "_sad": ["(｡•́︿•̀｡)", "(╯︵╰,)", "(ಥ﹏ಥ)", "(︶︹︺)"],
-    "_anxious": ["(ノдヽ)", "(◎_◎;)", "(・_・;)", "(ﾟДﾟ;)"],
+LEVEL_KAOMOJI = {
+    logging.DEBUG: ["(=^･^=)", "(=・ｪ・=)", "(=^-ω-^=)", "(=^‥^=)"],
+    logging.INFO: ["(＾• ω •＾)", "(=^･ω･^)y＝", "(≚ᄌ≚)", "(｡♥‿♥｡)"],
+    logging.WARNING: ["(；￣Д￣)", "(￢_￢;)", "(･ัω･ั)", "(・_・ヾ"],
+    logging.ERROR: ["(╥﹏╥)", "(≧Д≦)", "(；′⌒`)", "(Ｔ▽Ｔ)"],
+    logging.CRITICAL: ["(╯°□°)╯︵ ┻━┻"],
 }
 
-LEVEL_COLORS = {
+LEVEL_ANSI_STYLES = {
     logging.DEBUG: "\033[95m",  # light magenta
     logging.INFO: "\033[96m",  # light cyan
     logging.WARNING: "\033[93m",  # light yellow
@@ -23,32 +25,33 @@ LEVEL_COLORS = {
     logging.CRITICAL: "\033[35m",  # magenta
 }
 
-RESET_COLOR = "\033[0m"
+RESET_ANSI = "\033[0m"
 
 
 class KawaiiFormatter(logging.Formatter):
-    def format(self, record):
-        level = record.levelname.lower()
-        color = LEVEL_COLORS.get(record.levelno, "")
-        base_emotes = MOOD_EMOTICONS.get(level, [])
-        mood_emotes = []
-
-        emote_pool = mood_emotes if mood_emotes else base_emotes
-        emote = random.choice(emote_pool) if emote_pool else "(・ω・)"
-
+    def format(self, record: logging.LogRecord):
         message = record.msg.strip()
         ts = self.formatTime(record, "%Y-%m-%d %H:%M:%S")
         lvl = record.levelname
         filename = record.filename
         lineno = str(record.lineno)
 
-        formatted = f"[{ts}] {color}[{lvl} {filename}:{lineno}]{RESET_COLOR} {color}{message}{RESET_COLOR} {emote}"
-        return f"{formatted}{RESET_COLOR}"
+        split_last_word = message.rsplit(" ", 1)
+        last_word = split_last_word[-1] if split_last_word else ""
+        emote_pool = LEVEL_KAOMOJI.get(record.levelno, [])
+        if last_word in MOOD_KAOMOJI:
+            message = split_last_word[0] if len(split_last_word) > 1 else ""
+            emote_pool = MOOD_KAOMOJI.get(last_word, [])
+        emote = random.choice(emote_pool) if emote_pool else "(・ω・)"
+
+        color = LEVEL_ANSI_STYLES.get(record.levelno, "")
+
+        formatted = f"[{ts}] {color}[{lvl} {filename}:{lineno}]{RESET_ANSI} {color}{message}{RESET_ANSI} {emote}"
+        return f"{formatted}{RESET_ANSI}"
 
 
-def setup_logger(verbose: bool = False) -> logging.Logger:
-    """sets up a super cute logger with sparkles and cat faces ✨"""
-    logger = logging.getLogger("dotfiles")
+def setup_logger(verbose: bool = False, logger_name: str = "") -> logging.Logger:
+    logger = logging.getLogger(logger_name)
     logger.setLevel(logging.DEBUG if verbose else logging.INFO)
 
     handler = logging.StreamHandler(sys.stdout)