blob: b06735fde99a8bd7462587244b6c45f21e7df9d6 [file] [log] [blame]
# Copyright 2020 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import argparse
import errno
import io
import json
import os
import sys
# For Node, EvaluateExpression
import grit.node.base
# For CheckConditionalElements
import grit.format.html_inline
_CWD = os.getcwd()
class PreprocessIfExprNode(grit.node.base.Node):
def __init__(self):
super(PreprocessIfExprNode, self).__init__()
def PreprocessIfExpr(self, content):
return grit.format.html_inline.CheckConditionalElements(self, content)
def EvaluateCondition(self, expr):
return grit.node.base.Node.EvaluateExpression(expr, self.defines,
self.target_platform, {})
def SetDefines(self, defines):
self.defines = defines
def SetTargetPlatform(self, target_platform):
self.target_platform = target_platform
def Construct(defines, target_platform):
node = PreprocessIfExprNode()
node.SetTargetPlatform(target_platform or sys.platform)
return node
def ParseDefinesArg(definesArg):
defines = {}
for define_arg in definesArg:
define, = define_arg
parts = [part.strip() for part in define.split('=', 1)]
name = parts[0]
val = True if len(parts) == 1 else parts[1]
if (val == "1" or val == "true"):
val = True
elif (val == "0" or val == "false"):
val = False
defines[name] = val
return defines
def main(argv):
parser = argparse.ArgumentParser()
parser.add_argument('--in-folder', required=True)
parser.add_argument('--out-folder', required=True)
parser.add_argument('--in-files', required=True, nargs="*")
parser.add_argument('-D', '--defines', nargs="*", action='append')
parser.add_argument('-E', '--environment')
parser.add_argument('-t', '--target')
args = parser.parse_args(argv)
in_folder = os.path.normpath(os.path.join(_CWD, args.in_folder))
out_folder = os.path.normpath(os.path.join(_CWD, args.out_folder))
defines = ParseDefinesArg(args.defines)
node = PreprocessIfExprNode.Construct(defines,
for input_file in args.in_files:
content = ""
with, input_file),
mode='r') as f:
content =
preprocessed = node.PreprocessIfExpr(content)
out_path = os.path.join(out_folder, input_file)
out_dir = os.path.dirname(out_path)
assert out_dir.startswith(out_folder), \
'Cannot preprocess files to locations not under %s.' % out_dir
except OSError as e:
# Ignore directory exists errors. This can happen if two build rules
# for overlapping directories hit the makedirs line at the same time.
if e.errno != errno.EEXIST:
with, mode='wb') as f:
if args.out_manifest:
manifest_data = {}
manifest_data['base_dir'] = '%s' % args.out_folder
manifest_data['files'] = args.in_files
manifest_file = open(
os.path.normpath(os.path.join(_CWD, args.out_manifest)), 'wb')
json.dump(manifest_data, manifest_file)
if __name__ == '__main__':