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

# This is a very dumb regex based method of translating from
# gn variable declarations and simple if statements to the
# equivalent cmake code. It only supports a few constructs and
# assumes the code is formatted a certain way. ANGLE gn files
# are auto-formatted so this should work most of the time.

# The output may need a bit of manual fixup, but it beats
# translating the whole thing by hand. If new constructs
# are added to the ANGLE gn files, hopefully we can just add
# a few extra regexes here.

if len(sys.argv) != 3:
    sys.stderr.write('Error: wrong number of arguments.\n\n')
    sys.stderr.write('Two arguments are required. The first argument is the path\n')
    sys.stderr.write('of the input .gni file, and the second argument is the path\n')
    sys.stderr.write('of the output .cmake file.\n')
    exit(1)

file = open(sys.argv[1], 'rb').read()

# First translate gn single line list declaration:
file = re.sub(r'\[ ((?:"[^"]*",? )*)\]$', r' \1)', file, flags=re.M)

# Translate gn list declaration:
# variable_name = [
#   "file/name/foo.cpp",
# ]
# to cmake list declaration:
# set(variable_name
#     "file/name/foo.cpp"
# )
file = re.sub(r'^(\s*)(\w+) = ?\[?', r'\1\1set(\2', file, flags=re.M)
file = re.sub(r'^(\s*)("[^"]+"),$', r'\1\1\2', file, flags=re.M)
file = re.sub(r'^(\s*)]$', r'\1\1)', file, flags=re.M)

# Translate list append fom gn to cmake
file = re.sub(r'^(\s*)(\w+) \+= ?\[?', r'\1\1list(APPEND \2', file, flags=re.M)

# Translate if statements from gn to cmake
file = re.sub(r'^(\s*)((?:} else )?)if \((.+)\) {$', r'\1\1\2if(\3)', file, flags=re.M)
file = re.sub(r'^} else if$', r'elseif', file, flags=re.M)
file = re.sub(r'^(\s*)} else {$', r'\1\1else()', file, flags=re.M)
file = re.sub(r'^(\s*)}$', r'\1\1endif()', file, flags=re.M)

# Translate logic ops from gn to cmake
file = re.sub(r' \|\| ', r' OR ', file, flags=re.M)
file = re.sub(r' \&\& ', r' AND ', file, flags=re.M)
file = re.sub(r' == ', r' STREQUAL ', file, flags=re.M)
file = re.sub(r'!', r' NOT ', file, flags=re.M)

out = open(sys.argv[2], 'wb')

out.write('# This file was generated with the command:\n')
out.write('# ' + ' '.join(['"' + arg.replace('"', '\\"') + '"' for arg in sys.argv]))
out.write('\n\n')
out.write(file)