# https://stackoverflow.com/questions/4842424/list-of-ansi-color-escape-sequences
# https://en.wikipedia.org/wiki/ANSI_escape_code#3-bit_and_4-bit
COLORS_FG = {
  'BLACK': '\33[30m',
  'RED': '\33[31m',
  'GREEN': '\33[32m',
  'YELLOW': '\33[33m',
  'BLUE': '\33[34m',
  'MAGENTA': '\33[35m',
  'CYAN': '\33[36m',
  'WHITE': '\33[37m',
  'GREY': '\33[90m',
  'BRIGHT_RED': '\33[91m',
  'BRIGHT_GREEN': '\33[92m',
  'BRIGHT_YELLOW': '\33[93m',
  'BRIGHT_BLUE': '\33[94m',
  'BRIGHT_MAGENTA': '\33[95m',
  'BRIGHT_CYAN': '\33[96m',
  'BRIGHT_WHITE': '\33[97m',
}
COLORS_STYLE = {
  'RESET': '\33[0m',
  'BOLD': '\33[1m',
  'NORMAL': '\33[22m',
  'ITALIC': '\33[3m',
  'UNDERLINE': '\33[4m',
  'BLINK': '\33[5m',
  'BLINK2': '\33[6m',
  'SELECTED': '\33[7m',
}
COLORS_BG = {
  'BLACK': '\33[40m',
  'RED': '\33[41m',
  'GREEN': '\33[42m',
  'YELLOW': '\33[43m',
  'BLUE': '\33[44m',
  'MAGENTA': '\33[45m',
  'CYAN': '\33[46m',
  'WHITE': '\33[47m',
  'GREY': '\33[100m',
  'BRIGHT_RED': '\33[101m',
  'BRIGHT_GREEN': '\33[102m',
  'BRIGHT_YELLOW': '\33[103m',
  'BRIGHT_BLUE': '\33[104m',
  'BRIGHT_MAGENTA': '\33[105m',
  'BRIGHT_CYAN': '\33[106m',
  'BRIGHT_WHITE': '\33[107m',
}


def log_welcome(num_nodes=None):
  """Logs the welcome message."""
  msg = f"{COLORS_FG['GREEN']}{COLORS_STYLE['BOLD']}rgthree\'s comfy nodes:"
  msg += f"{COLORS_STYLE['RESET']}{COLORS_STYLE['BOLD']} Loaded"
  if num_nodes:
    print(f"{msg} {num_nodes} exciting nodes.{COLORS_STYLE['RESET']}")
  else:
    print(f"{msg}.{COLORS_STYLE['RESET']}")


def log_node_success(node_name, message):
  """Logs a success message."""
  _log_node(COLORS_FG["GREEN"], node_name, message, prefix='✓  ')


def log_node_info(node_name, message):
  """Logs an info message."""
  _log_node(COLORS_FG["CYAN"], node_name, message, prefix='🛈  ')


def log_node_warn(node_name, message):
  """Logs an warn message."""
  _log_node(COLORS_FG["YELLOW"], node_name, message, prefix='⚠  ')


def log_node(node_name, message):
  """Logs a message."""
  _log_node(COLORS_FG["CYAN"], node_name, message, prefix='   ')


def _log_node(color, node_name, message, prefix=''):
  print(_get_log_msg(color, node_name, message, prefix=''))


def _get_log_msg(color, node_name, message, prefix=''):
  return f'{COLORS_STYLE["BOLD"]}{color}{prefix}rgthree {node_name.replace(" (rgthree)", "")}" +
        f':{COLORS_STYLE["RESET"]} {message}'