vimr/bin/generate_source.py

#!/usr/bin/env python3

import os
import io
import re
from string import Template

print(os.getcwd())
if 'CONFIGURATION' in os.environ and os.environ['CONFIGURATION'] == 'Debug':
    if os.path.isfile('./NvimView/NvimAutoCommandEvent.generated.swift'):
        print("Files already there, exiting...")
        exit(0)

with io.open('./neovim/build/include/auevents_enum.generated.h', 'r') as auto_cmds_file:
    raw_auto_cmds = [line.strip() for line in auto_cmds_file.readlines() if re.match(r'^EVENT_', line.strip())]


def convert(line):
    result = re.match(r'^EVENT_(.*) = (.*)', line.replace(',', ''))
    return result.group(1), result.group(2)


auto_cmds = [convert(line) for line in raw_auto_cmds]
auto_cmds_template = Template(
'''
enum NvimAutoCommandEvent: Int {
${event_cases}
}
'''
)

header = auto_cmds_template.substitute(
    event_cases='\n'.join(
        ['  case {} = {}'.format(event[0].lower(), event[1]) for event in auto_cmds]
    ),
)
with io.open('./NvimView/NvimAutoCommandEvent.generated.swift', 'w') as auto_cmds_header_file:
    auto_cmds_header_file.write(header)