gistfile1.py

#!/usr/bin/python
import sys
import re


def main():
    if len(sys.argv) == 1:
        sys.stderr.write('Usage: {} <file> <file> ...\n'.format(sys.argv[0]))
        sys.exit(1)

    for fname in sys.argv[1:]:
        s = process_file(open(fname).read().decode('utf-8'))
        open(fname, 'w').write(s.encode('utf-8'))


space = ' ' * 4


def process_file(s):
    for regex, result in [
        # strip trailing whitespaces
        (re.compile('\s*\n'), '\n'),
        # "\n" after ,"
        (re.compile('\s*,\s*'), ',\n'),
        # "\n" after {, "\n" around "}"
        (re.compile('\s*\{([^\}]*)\}\s*', re.MULTILINE),
         lambda m: u' {\n' + space + process_brackets(m.group(1)) + '\n}\n\n'),
    ]:
        s = regex.sub(result, s)
    return s


def process_brackets(s):
    for regex, result in [
        # space after ":"
        (re.compile(':\s*'), ': '),
        # "\n" after ;
        (re.compile('\s*;\s*', re.MULTILINE), ';\n' + space),
        # " " after ,"
        (re.compile('\s*,\s*'), ', '),
        # fix comment
        (re.compile('\s*\*/'), '*/'),
    ]:
        s = regex.sub(result, s)
    return s.strip()


if __name__ == '__main__':
    main()