make.py

#!/usr/bin/env python2
import sys
import os.path
import argparse
import shutil

from mako.template import Template

import tcolor


def make_hi(use_bold=True):
    colors = set()
    def get_256_color(color):
        c256 = tcolor.find_nearest(color)[1]
        colors.add((int(c256), color.upper()))
        return c256

    def hi(fg=None, bg=None, bold=None, italic=None, underline=None, reverse=None):
        parts = []
        if fg:
            parts.append(('guifg', '#' + fg))
            parts.append(('ctermfg', get_256_color(fg)))

        if bg:
            parts.append(('guibg', '#' + bg))
            parts.append(('ctermbg', get_256_color(bg)))

        attrs = []
        if use_bold and bold:
            attrs.append('bold')

        if italic:
            attrs.append('italic')

        if underline:
            attrs.append('underline')

        if reverse:
            attrs.append('reverse')

        if not attrs:
            attrs.append('none')

        if attrs:
            cattrs = ','.join(attrs)
            parts.append(('gui', cattrs))
            parts.append(('cterm', cattrs))

        return ' '.join('{}={}'.format(*r) for r in parts)

    bgfix = lambda bg: r'\033]11;#{}\007'.format(bg)
    cfix = lambda: r'\033]4;{}\007'.format(
        ';'.join('{};#{}'.format(*r) for r in sorted(colors)))

    hi.xterm = lambda bg: bgfix(bg) + cfix()
    hi.urxvt = cfix
    hi.cterm = get_256_color
    return hi


if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Render vim colorscheme file')
    parser.add_argument('--disable-bold', dest='use_bold',
        action='store_false', default=True)
    parser.add_argument('template', nargs='+')

    args = parser.parse_args()

    hi = make_hi(args.use_bold)
    theme_tpl = None

    for fname in args.template:
        tpl = Template(filename=fname)
        outname = os.path.expanduser(tpl.module.out)
        with open(outname, 'w') as f:
            f.write(tpl.render(hi=hi, cterm=hi.cterm, theme_tpl=theme_tpl))

        shutil.copymode(fname, outname)
        if not theme_tpl:
            theme_tpl = tpl.module