#coding:utf-8

import logging 
from lib.core.data import now,root_path
from lib.core.enums import CUSTOM_LOGGING

logging.addLevelName(CUSTOM_LOGGING.LOGO, "(∩ᵒ̴̶̷̤⌔ᵒ̴̶̷̤∩)")
logging.addLevelName(CUSTOM_LOGGING.VULN, "~o(〃'▽'〃)o")
logging.addLevelName(CUSTOM_LOGGING.UNVULN, "ε(┬┬﹏┬┬)3")
logging.addLevelName(CUSTOM_LOGGING.NETEERROR, "(*• . •*)?")

logger = logging.getLogger("oFx_l0g")
logger.setLevel(logging.DEBUG)
formatter = logging.Formatter("%(asctime)s - %(levelname)s: %(message)s")


# use filehandler
FILE_HANDLE = logging.FileHandler("%s.log" % (root_path + "/log/" + now))
FILE_HANDLE.setLevel(logging.DEBUG)
FILE_HANDLE.setFormatter(formatter)

# use streamhandler
STREAM_HANDLE = None
try:
    from lib.thirdparty.ansistrm.ansistrm import ColorizingStreamHandler
    disableColor = False

    if disableColor:
        LOGGER_HANDLER = logging.StreamHandler()
    else:
        LOGGER_HANDLER = ColorizingStreamHandler()
        LOGGER_HANDLER.level_map[logging.getLevelName("(∩ᵒ̴̶̷̤⌔ᵒ̴̶̷̤∩)")] = (None, "yellow", False)
        LOGGER_HANDLER.level_map[logging.getLevelName("~o(〃'▽'〃)o")] = (None, "green", False)
        LOGGER_HANDLER.level_map[logging.getLevelName("ε(┬┬﹏┬┬)3")] = (None, "blue", False)
        LOGGER_HANDLER.level_map[logging.getLevelName("(*• . •*)?")] = (None, "cyan", False)
except:
    pass
finally:
    STREAM_HANDLE = logging.StreamHandler()



STREAM_HANDLE.setLevel(logging.DEBUG)
STREAM_HANDLE.setFormatter(formatter)

# add Handler
logger.addHandler(FILE_HANDLE)
logger.addHandler(STREAM_HANDLE)

def loglogo(message):
    print("\033[33m")
    logger.log(CUSTOM_LOGGING.LOGO,message)

def logvuln(message):
    print("\033[32m") 
    logger.log(CUSTOM_LOGGING.VULN,message)

def logunvuln(message):
    print("\033[34m") 
    logger.log(CUSTOM_LOGGING.UNVULN,message)

def logverifyerror(message):
    print("\033[36m") 
    logger.log(CUSTOM_LOGGING.NETEERROR,message)

def logwarning(message):
    # print("\033[35m")
    logger.warning(message)

def logcritical(message):
    # print("\033[31m")
    logger.critical(message)