""" Flowtool console output wrappers.  """

import click

from collections import namedtuple
from functools import partial

DEBUG_OUTPUT = None

color_names = [
    'black', 'red', 'green', 'yellow', 'blue', 'magenta', 'cyan', 'white',
]

term_effects = ['dim', 'bold', 'blink', 'underline', 'reverse']

[docs]def color_function(name):
    """ Creates a function that decorates a string with
        the named color code.
    """
    return partial(click.style, fg=name)

[docs]def effect_function(name):
    """ Creates a function that decorates a string with the named effect
        (bold, underline, blink, etc).
    """
    return partial(click.style, **{name: True})

[docs]def echo_function(style_func):
    """ Creates a function that echos (a list of) arguments
        using `style_func`.
    """
    global DEBUG_OUTPUT
    def echo(*message, **kwd):
        click.secho(' '.join(map(style_func, map(str, message))), **kwd)
    return echo

[docs]def debug_function(style_func):
    """ Creates a function that echos (a list of) arguments using `style_func`
        only if bool(DEBUG_OUTPUT) evals to True.
    """
    global DEBUG_OUTPUT
    def echo(*message, **kwd):
        if not DEBUG_OUTPUT:
            return
        click.secho(style_func(' '.join(map(str, message))), **kwd)
    return echo

color_functions = {name: color_function(name) for name in color_names}
effect_functions = {name: effect_function(name) for name in term_effects}

functions = {}
functions.update(color_functions, **effect_functions)
ConvenienceFunctions = namedtuple('ConvenienceFunctions', color_names + term_effects)
colors = ConvenienceFunctions(**functions)
echo = ConvenienceFunctions(**{name: echo_function(f) for name, f in functions.items()})
debug = ConvenienceFunctions(**{name: debug_function(f) for name, f in functions.items()})