0
0
mirror of https://github.com/mongodb/mongo.git synced 2024-11-29 16:47:28 +01:00
mongodb/site_scons/site_tools/idl_tool.py

144 lines
4.4 KiB
Python
Executable File

# Copyright 2020 MongoDB Inc.
#
# Permission is hereby granted, free of charge, to any person obtaining
# a copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to
# the following conditions:
#
# The above copyright notice and this permission notice shall be included
# in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY
# KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
# WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
#
"""IDL Compiler Scons Tool."""
import os.path
import subprocess
import sys
import SCons
# We lazily import this at generate time.
idlc = None
IDL_GLOBAL_DEPS = []
def idlc_emitter(target, source, env):
"""For each input IDL file, the tool produces a .cpp and .h file."""
first_source = str(source[0])
if not first_source.endswith(".idl"):
raise ValueError(
"Bad idl file name '%s', it must end with '.idl' " % (first_source)
)
base_file_name, _ = SCons.Util.splitext(str(target[0]))
target_source = env.File(base_file_name + "_gen.cpp")
target_header = env.File(base_file_name + "_gen.h")
env.Alias("generated-sources", [target_source, target_header])
return [target_source, target_header], source
IDLCAction = SCons.Action.Action("$IDLCCOM", "$IDLCCOMSTR")
def idl_scanner(node, env, path):
# When generating ninja we only need to add the IDL_GLOBAL_DEPS
# because the implicit dependencies will be picked up using the
# deps=msvc method.
if env.get("GENERATING_NINJA", False):
return IDL_GLOBAL_DEPS
nodes_deps_list = getattr(node.attributes, "IDL_NODE_DEPS", None)
if nodes_deps_list is not None:
return nodes_deps_list
nodes_deps_list = IDL_GLOBAL_DEPS[:]
# Compute the include paths to use based on the include flags in IDLCFLAGS
flags = env["IDLCFLAGS"]
include_paths = []
for i in range(len(flags)):
if flags[i] == "--include":
include_paths.append(flags[i + 1])
resolver = idlc.CompilerImportResolver(include_paths)
try:
with open(str(node), encoding="utf-8") as file_stream:
parsed_doc = idlc.parser.parse(
file_stream, str(node), resolver
)
except OSError:
return nodes_deps_list
if not parsed_doc.errors and parsed_doc.spec.imports is not None:
nodes_deps_list.extend(
[env.File(d) for d in sorted(parsed_doc.spec.imports.dependencies)]
)
setattr(node.attributes, "IDL_NODE_DEPS", nodes_deps_list)
return nodes_deps_list
idl_scanner = SCons.Scanner.Scanner(function=idl_scanner, skeys=[".idl"])
# TODO: create a scanner for imports when imports are implemented
IDLCBuilder = SCons.Builder.Builder(
action=IDLCAction,
emitter=idlc_emitter,
src_suffix=".idl",
suffix=".cpp",
source_scanner=idl_scanner,
)
def generate(env):
bld = IDLCBuilder
env.Append(SCANNERS=idl_scanner)
env["BUILDERS"]["Idlc"] = bld
sys.path.append(env.Dir("#buildscripts").get_abspath())
import buildscripts.idl.idl.compiler as idlc_mod
global idlc
idlc = idlc_mod
env["IDLC"] = "$PYTHON buildscripts/idl/idlc.py"
base_dir = env.Dir("$BUILD_DIR").path
env["IDLCFLAGS"] = [
"--include", "src",
"--base_dir", base_dir,
"--target_arch", "$TARGET_ARCH",
]
env["IDLCCOM"] = "$IDLC $IDLCFLAGS --header ${TARGETS[1]} --output ${TARGETS[0]} $SOURCES"
env["IDLCCOMSTR"] = ("Generating ${TARGETS[0]}"
if not env.get("VERBOSE", "").lower() in ['true', '1']
else None)
env["IDLCSUFFIX"] = ".idl"
global IDL_GLOBAL_DEPS
IDL_GLOBAL_DEPS = env.Glob("#buildscripts/idl/*.py") + env.Glob(
"#buildscripts/idl/idl/*.py"
)
env["IDL_HAS_INLINE_DEPENDENCIES"] = True
def exists(env):
return True