# vim: tabstop=4 shiftwidth=4 softtabstop=4

#
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
#    not use this file except in compliance with the License. You may obtain
#    a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#    License for the specific language governing permissions and limitations
#    under the License.

import logging
import logging.config
import os
import sys

#requires http://pypi.python.org/pypi/termcolor
#but the colors make it worth it :-)
from termcolor import colored

LOG_FN_ENV = 'LOG_FILE'


class TermFormatter(logging.Formatter):
    def __init__(self, reg_fmt, date_format):
        logging.Formatter.__init__(self, reg_fmt, date_format)

    def format(self, record):
        lvl = record.levelno
        lvlname = record.levelname
        if(lvl == logging.DEBUG):
            lvlname = colored(lvlname, 'blue')
        elif(lvl == logging.INFO):
            lvlname = colored(lvlname, 'cyan')
        elif(lvl == logging.WARNING):
            lvlname = colored(lvlname, 'yellow')
        elif(lvl == logging.ERROR):
            lvlname = colored(lvlname, 'red')
        elif(lvl == logging.CRITICAL):
            lvlname = colored(lvlname, 'red')
            record.msg = colored(record.msg, attrs=['bold', 'blink'])
        record.levelname = lvlname
        return logging.Formatter.format(self, record)


class TermHandler(logging.Handler):
    STREAM = sys.stdout
    DO_FLUSH = True
    NL = os.linesep

    def __init__(self):
        logging.Handler.__init__(self)

    def emit(self, record):
        lvl = record.levelno
        msg = self.format(record)
        if(len(msg)):
            TermHandler.STREAM.write(msg + TermHandler.NL)
            if(TermHandler.DO_FLUSH):
                TermHandler.STREAM.flush()


def setupLogging():
    logfn = os.getenv(LOG_FN_ENV)
    if(logfn == None):
        logfn = 'conf/logging.ini'
    logging.config.fileConfig(logfn)


def getLogger(name):
    return logging.getLogger(name)


#this should happen first (and once)
INIT_LOGGING = False
if(not INIT_LOGGING):
    setupLogging()
    INIT_LOGGING = True