import re
from os import listdir
from os.path import isfile, join

def source_many(directory, predicate):
    for filename in listdir(directory):
        if not predicate(filename): continue
        source @(join(directory, filename))

is_tako = lambda filename: re.match('^.*\.tako', filename)
source_many('/etc/profile.d', is_tako)
source_many($HOME + '/.config/tako/completers', is_tako)

$TAKO_SETTINGS.auto_pushd = True

$WORKON_HOME = $HOME + '/.virtualenvs'
$XDG_CONFIG_HOME = $HOME + '/.config'
$SXHKD_SHELL = '/bin/sh'
$EDITOR = 'emacs'
$VISUAL = $EDITOR
$PATH.insert(0, $HOME + '/bin')
$PATH.insert(0, $HOME + '/.local/bin')

$LS_COLORS = $(dircolors @($HOME + '/.config/zsh/dircolors.256dark'))[0][len("'LS_COLORS="):-len("';")]

def _workon(args, stdin=None):
    if len(args) < 1:
        return 'Need name of virtualenv\n'

    source_order = [
        'preactivate',
        'activate',
        'postactivate',
    ]

    for f in filter(isfile, [
            join($WORKON_HOME, args[0], 'bin', '{}.tako'.format(name))
            for name in source_order
    ]):
        source @(f)

$TAKO_SETTINGS.aliases.update({
        'startx': 'ssh-agent startx ; vlock',
        'ls': 'ls --color=auto',
        'la': 'ls -A',
        'll': 'ls -l',
        'units': 'units --verbose',
        'trr': 'transmission-remote',
        'vol': 'pulseaudio-ctl set',
        'workon': _workon,
})

if 'TERM' in __tako_env__:
    if '256color' in $TERM:
        colors = {
            'blue': 81,
            'orange': 166,
            'purple': 135,
            'red': 196,
            'green': 118,
            'gray': 241,
        }

        def colorize(color, text):
            prefix = '{COLOR_' + str(colors[color]) + '}'
            suffix = '{NO_COLOR}'
            return prefix + text + suffix
    else:
        colors = {
            'blue': 'BLUE',
            'orange': 'YELLOW',
            'purple': 'PURPLE',
            'red': 'RED',
            'green': 'GREEN',
            'gray': 'WHITE',
        }
        def colorize(color, text):
            prefix = '{%s}' % colors[color]
            suffix = '{NO_COLOR}'
            return prefix + text + suffix

    def git_branch():
        ref=$(git symbolic-ref HEAD err> /dev/null)
        if not ref: return None
        prefix = colorize('gray', ':')
        branch = colorize('orange', re.sub('^refs/heads/', '', ref[0].strip()))
        return prefix + branch

    $TAKO_SETTINGS.prompt_fields['git_branch'] = git_branch

    def make_prompt():
        username = colorize('purple', '{user}')
        host = colorize('blue', '{hostname}')
        directory = colorize('green', '{cwd}')
        lbrkt = colorize('gray', '[')
        rbrkt = colorize('gray', ']')
        colon = colorize('gray', ':')

        return lbrkt + colon.join([
                username,
                host,
                directory,
        ]) + '{git_branch}' + rbrkt + '\n{prompt_end} '

    $TAKO_SETTINGS.prompt = make_prompt()

$TAKO_SETTINGS.show_traceback = True
$TAKO_SETTINGS.traceback_logfile = $HOME + '/tmp/tako.log'