from enum import Enum

class Color(Enum):
    Black = 0
    Red = 1
    Green = 2
    Yellow = 3
    Blue = 4
    Magenta = 5
    Cyan = 6
    White = 7

    # escape sequence format string
    __fescseq = '\033[3{0}m'
    # ansi reset code
    __ansi_rst = '\033[0m'
    
    __ansi_black = '\033[30m'
    __ansi_red = '\033[31m'
    __ansi_green = '\033[32m'
    __ansi_yellow = '\033[33m'
    __ansi_blue = '\033[34m'
    __ansi_magenta = '\033[35m'
    __ansi_cyan= '\033[36m'
    __ansi_white = '\033[37m'

    @staticmethod
    def _ansi_ret(ansi):
        return ansi, Color.__ansi_white

    @staticmethod
    def code(color: 'Color'):
        return Color.__fescseq.format(color.value)
    
    def ansi(self):
        code = Color.code(self)
        if self == Color.White:
            return code, ''
        return code, Color.__ansi_rst

# !! ULTRA PRINT !!
def uprint(text: str, 
           color: Color = Color.White, 
           end: str = '\n', 
           flush: bool = True):
    c = color.ansi()
    text = f'{c[0]}{text}{c[1]}'
    print(text, end=end, flush=flush)