"""
SCHAINPY - LOG
    Simple helper for log standarization
    Usage:
        from schainpy.utils import log
        log.error('A kitten died beacuse of you')
        log.warning('You are doing it wrong but what the heck, I'll allow it)
        log.succes('YOU ROCK!')
    To create your own logger inside your class do it like this:
        from schainpy.utils import log
        awesomeLogger = log.makelogger("never gonna", bg="red", fg="white")
        awesomeLogger('give you up')
    which will look like this:
        [NEVER GONNA] - give you up
    with color red as background and white as foreground.
"""
import os
import sys
import click

def warning(message):
    click.echo(click.style('[WARNING] - ' + message, fg='yellow'))


def error(message):
    click.echo(click.style('[ERROR] - ' + message, fg='red', bg='black'))


def success(message):
    click.echo(click.style(message, fg='green'))


def log(message, topic='LOG'):
    click.echo('[{}] - {}'.format(topic, message))

def makelogger(topic, bg='reset', fg='reset'):
    def func(message):
        click.echo(click.style('[{}] - '.format(topic.upper()) + message,
                   bg=bg, fg=fg))
    return func

class LoggerForFile():
    def __init__(self, filename):
        self.old_stdout=sys.stdout
        cwd = os.getcwd()
        self.log_file = open(os.path.join(cwd, filename), 'w+')
    def write(self, text):
        text = text.rstrip()
        if not text: 
            return
        self.log_file.write(text + '\n')
        self.old_stdout.write(text + '\n')
    def flush(self):
        self.old_stdout.flush()

def logToFile(filename='log.log'):
    logger = LoggerForFile(filename)
    sys.stdout = logger