build_vimball.py

#!/usr/bin/env python
# vi:set ts=8 sts=4 sw=4 et tw=80:
"""Script to build the Vim run time files."""

# Python 2-3 compatibility.
from __future__ import print_function

import sys
import os
import string
import tempfile
import subprocess
import shutil
import importlib

from lib.clewn import __version__

DEBUGGERS = ('simple', 'gdb', 'pdb')
RUNTIME = [
    'autoload/pyclewn/start.vim',
    'autoload/pyclewn/buffers.vim',
    'autoload/pyclewn/version.vim',
    'doc/pyclewn.txt',
    'plugin/pyclewn.vim',
    'syntax/clewn_variables.vim',
    'macros/.pyclewn_keys.gdb',
    'macros/.pyclewn_keys.pdb',
    'macros/.pyclewn_keys.simple',
    ]
VERSION_FUNC = """
function pyclewn#version#RuntimeVersion()
    return "%s"
endfunction
"""

def keymap_files():
    """Update the key map files for each debugger."""
    with open('runtime/macros/.pyclewn_keys.template') as tf:
        print('Updating:')
        template = tf.read()
        for d in DEBUGGERS:
            filename = 'runtime/macros/.pyclewn_keys.%s' % d
            try:
                module = importlib.import_module('.%s' % d, 'lib.clewn')
            except ImportError:
                print('Warning: cannot update %s' % filename, file=sys.stderr)
                continue
            with open(filename, 'w') as f:
                f.write(string.Template(template).substitute(clazz=d))
                mapkeys = getattr(module, 'MAPKEYS')
                for k in sorted(mapkeys):
                    if len(mapkeys[k]) == 2:
                        comment = ' # ' + mapkeys[k][1]
                        f.write('# %s%s\n' % (('%s : %s' %
                                (k, mapkeys[k][0])).ljust(30), comment))
                    else:
                        f.write('# %s : %s\n' % (k, mapkeys[k][0]))
            print('  %s' % filename)

def vimball():
    """Build the vimball."""
    fd, tmpname = tempfile.mkstemp(prefix='vimball', suffix='.clewn')
    args = ['vim', '-u', 'NORC', '-vN',
            '-c', 'edit %s' % tmpname,
            '-c', '%MkVimball! runtime/pyclewn runtime',
            '-c', 'quit',
           ]

    # Create version.vim.
    version = __version__ + '.' + subprocess.check_output(
                    ['hg',  'id',  '-i'], universal_newlines=True)
    with open('runtime/autoload/pyclewn/version.vim', 'w') as f:
        f.write(VERSION_FUNC % version.rstrip('+\n'))

    data_dir = 'lib/clewn/runtime'
    if not os.path.exists(data_dir):
        os.mkdir(data_dir)
    # Remove the existing vimballs.
    for dirpath, dirnames, filenames in os.walk(data_dir):
        if dirpath == data_dir:
            for fname in filenames:
                if fname.startswith('pyclewn-') and fname.endswith('.vmb'):
                    print('Removing', fname)
                    os.unlink(os.path.join(dirpath, fname))

    # Build the vimball.
    try:
        with os.fdopen(fd, 'w') as f:
            f.write('\n'.join(RUNTIME))
            f.close()
            subprocess.call(args)
    finally:
        try:
            os.unlink(tmpname)
        except OSError:
            pass

    vimball = os.path.join(data_dir, 'pyclewn-%s.vmb' % __version__)
    print('Creation of', vimball)
    shutil.move('runtime/pyclewn.vmb', vimball)

def main():
    keymap_files()
    vimball()

if __name__ == '__main__':
        main()