tools/build_defs/cc/cc_import.bzl - bazel - Git at Google

 # Copyright 2020 The Bazel Authors. All rights reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
 #
 #    http://www.apache.org/licenses/LICENSE-2.0
 #
 # Unless required by applicable law or agreed to in writing, software
 # distributed under the License is distributed on an "AS IS" BASIS,
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.

 """Starlark implementation of cc_import.

 We may change the implementation at any moment or even delete this file. Do not
 rely on this. Pass the flag --experimental_starlark_cc_import
 """

 load("@bazel_tools//tools/cpp:toolchain_utils.bzl", "find_cpp_toolchain")
 load("@bazel_tools//tools/build_defs/cc:action_names.bzl", "CPP_LINK_STATIC_LIBRARY_ACTION_NAME")

 def _to_list(element):
     if element == None:
         return []
     else:
         return [element]

 def _to_depset(element):
     if element == None:
         return depset()
     else:
         return depset([element])

 def _is_shared_library_extension_valid(shared_library_name):
     if (shared_library_name.endswith(".so") or
         shared_library_name.endswith(".dll") or
         shared_library_name.endswith(".dylib")):
         return True

     # validate against the regex "^.+\.so(\.\d\w*)+$" for versioned .so files
     parts = shared_library_name.split(".")
     extension = parts[1]
     if extension != "so":
         return False
     version_parts = parts[2:]
     for part in version_parts:
         if not part[0].isdigit():
             return False
         for c in part[1:].elems():
             if not (c.isalnum() or c == "_"):
                 return False
     return True

 def _perform_error_checks(
         system_provided,
         shared_library_artifact,
         interface_library_artifact):
     # If the shared library will be provided by system during runtime, users are not supposed to
     # specify shared_library.
     if system_provided and shared_library_artifact != None:
         fail("'shared_library' shouldn't be specified when 'system_provided' is true")

     # If a shared library won't be provided by system during runtime and we are linking the shared
     # library through interface library, the shared library must be specified.
     if (not system_provided and shared_library_artifact == None and
         interface_library_artifact != None):
         fail("'shared_library' should be specified when 'system_provided' is false")

     if (shared_library_artifact != None and
         not _is_shared_library_extension_valid(shared_library_artifact.basename)):
         fail("'shared_library' does not produce any cc_import shared_library files (expected .so, .dylib or .dll)")

 def _create_archive_action(
         ctx,
         feature_configuration,
         cc_toolchain,
         output_file,
         object_files):
     archiver_path = cc_common.get_tool_for_action(
         feature_configuration = feature_configuration,
         action_name = CPP_LINK_STATIC_LIBRARY_ACTION_NAME,
     )
     archiver_variables = cc_common.create_link_variables(
         feature_configuration = feature_configuration,
         cc_toolchain = cc_toolchain,
         output_file = output_file.path,
         is_using_linker = False,
     )
     command_line = cc_common.get_memory_inefficient_command_line(
         feature_configuration = feature_configuration,
         action_name = CPP_LINK_STATIC_LIBRARY_ACTION_NAME,
         variables = archiver_variables,
     )
     args = ctx.actions.args()
     args.add_all(command_line)
     args.add_all(object_files)
     args.use_param_file("@%s", use_always = True)

     env = cc_common.get_environment_variables(
         feature_configuration = feature_configuration,
         action_name = CPP_LINK_STATIC_LIBRARY_ACTION_NAME,
         variables = archiver_variables,
     )

     # TODO(bazel-team): PWD=/proc/self/cwd env var is missing, but it is present when an analogous archiving
     # action is created by cc_library
     ctx.actions.run(
         executable = archiver_path,
         arguments = [args],
         env = env,
         inputs = depset(
             direct = object_files,
             transitive = [
                 cc_toolchain.all_files,
             ],
         ),
         use_default_shell_env = True,
         outputs = [output_file],
         mnemonic = "CppArchive",
     )

 def _cc_import_impl(ctx):
     cc_toolchain = find_cpp_toolchain(ctx)
     cc_common.check_experimental_starlark_cc_import(
         actions = ctx.actions,
     )

     feature_configuration = cc_common.configure_features(
         ctx = ctx,
         cc_toolchain = cc_toolchain,
         requested_features = ctx.features,
         unsupported_features = ctx.disabled_features,
     )

     _perform_error_checks(
         ctx.attr.system_provided,
         ctx.file.shared_library,
         ctx.file.interface_library,
     )

     pic_static_library = ctx.file.pic_static_library or None
     static_library = ctx.file.static_library or None

     if ctx.files.pic_objects and not pic_static_library:
         lib_name = "lib" + ctx.label.name + ".pic.a"
         pic_static_library = ctx.actions.declare_file(lib_name)
         _create_archive_action(ctx, feature_configuration, cc_toolchain, pic_static_library, ctx.files.pic_objects)

     if ctx.files.objects and not static_library:
         lib_name = "lib" + ctx.label.name + ".a"
         static_library = ctx.actions.declare_file(lib_name)
         _create_archive_action(ctx, feature_configuration, cc_toolchain, static_library, ctx.files.objects)

     library_to_link = cc_common.create_library_to_link(
         actions = ctx.actions,
         feature_configuration = feature_configuration,
         cc_toolchain = ctx.attr._cc_toolchain[cc_common.CcToolchainInfo],
         static_library = static_library,
         pic_static_library = pic_static_library,
         interface_library = ctx.file.interface_library,
         dynamic_library = ctx.file.shared_library,
         pic_objects = ctx.files.pic_objects,
         objects = ctx.files.objects,
         alwayslink = ctx.attr.alwayslink,
     )

     linker_input = cc_common.create_linker_input(
         libraries = depset([library_to_link]),
         user_link_flags = depset(ctx.attr.linkopts),
         owner = ctx.label,
     )

     linking_context = cc_common.create_linking_context(
         linker_inputs = depset([linker_input]),
     )

     (compilation_context, compilation_outputs) = cc_common.compile(
         actions = ctx.actions,
         feature_configuration = feature_configuration,
         cc_toolchain = cc_toolchain,
         public_hdrs = ctx.files.hdrs,
         includes = ctx.attr.includes,
         name = ctx.label.name,
     )

     this_cc_info = CcInfo(compilation_context = compilation_context, linking_context = linking_context)
     cc_infos = [this_cc_info]

     for dep in ctx.attr.deps:
         cc_infos.append(dep[CcInfo])
     merged_cc_info = cc_common.merge_cc_infos(cc_infos = cc_infos)

     runfiles = ctx.runfiles(files = ctx.files.data)

     transitive_runfiles_list = []
     if ctx.attr.static_library:
         transitive_runfiles_list.append(ctx.attr.static_library[DefaultInfo].default_runfiles)
     if ctx.attr.pic_static_library:
         transitive_runfiles_list.append(ctx.attr.pic_static_library[DefaultInfo].default_runfiles)
     if ctx.attr.shared_library:
         transitive_runfiles_list.append(ctx.attr.shared_library[DefaultInfo].default_runfiles)
     if ctx.attr.interface_library:
         transitive_runfiles_list.append(ctx.attr.interface_library[DefaultInfo].default_runfiles)
     for dep in ctx.attr.deps:
         transitive_runfiles_list.append(dep[DefaultInfo].default_runfiles)

     for maybe_runfiles in transitive_runfiles_list:
         if maybe_runfiles:
             runfiles = runfiles.merge(maybe_runfiles)

     default_info = DefaultInfo(runfiles = runfiles)

     return [merged_cc_info, default_info]

 cc_import = rule(
     implementation = _cc_import_impl,
     attrs = {
         "hdrs": attr.label_list(allow_files = [".h"]),
         "static_library": attr.label(allow_single_file = [".a", ".lib"]),
         "pic_static_library": attr.label(allow_single_file = [".pic.a", ".pic.lib"]),
         "shared_library": attr.label(allow_single_file = [".so", ".dll", ".dylib"]),
         "interface_library": attr.label(
             allow_single_file = [".ifso", ".tbd", ".lib", ".so", ".dylib"],
         ),
         "pic_objects": attr.label_list(
             allow_files = [".o", ".pic.o"],
         ),
         "objects": attr.label_list(
             allow_files = [".o", ".nopic.o"],
         ),
         "system_provided": attr.bool(default = False),
         "alwayslink": attr.bool(default = False),
         "linkopts": attr.string_list(),
         "includes": attr.string_list(),
         "deps": attr.label_list(),
         "data": attr.label_list(allow_files = True),
         "_cc_toolchain": attr.label(default = "@bazel_tools//tools/cpp:current_cc_toolchain"),
     },
     toolchains = ["@rules_cc//cc:toolchain_type"],  # copybara-use-repo-external-label
     fragments = ["cpp"],
 )
	# Copyright 2020 The Bazel Authors. All rights reserved.
	#
	# Licensed under the Apache License, Version 2.0 (the "License");
	# you may not use this file except in compliance with the License.
	# You may obtain a copy of the License at
	#
	# http://www.apache.org/licenses/LICENSE-2.0
	#
	# Unless required by applicable law or agreed to in writing, software
	# distributed under the License is distributed on an "AS IS" BASIS,
	# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	# See the License for the specific language governing permissions and
	# limitations under the License.

	"""Starlark implementation of cc_import.

	We may change the implementation at any moment or even delete this file. Do not
	rely on this. Pass the flag --experimental_starlark_cc_import
	"""

	load("@bazel_tools//tools/cpp:toolchain_utils.bzl", "find_cpp_toolchain")
	load("@bazel_tools//tools/build_defs/cc:action_names.bzl", "CPP_LINK_STATIC_LIBRARY_ACTION_NAME")

	def _to_list(element):
	if element == None:
	return []
	else:
	return [element]

	def _to_depset(element):
	if element == None:
	return depset()
	else:
	return depset([element])

	def _is_shared_library_extension_valid(shared_library_name):
	if (shared_library_name.endswith(".so") or
	shared_library_name.endswith(".dll") or
	shared_library_name.endswith(".dylib")):
	return True

	# validate against the regex "^.+\.so(\.\d\w*)+$" for versioned .so files
	parts = shared_library_name.split(".")
	extension = parts[1]
	if extension != "so":
	return False
	version_parts = parts[2:]
	for part in version_parts:
	if not part[0].isdigit():
	return False
	for c in part[1:].elems():
	if not (c.isalnum() or c == "_"):
	return False
	return True

	def _perform_error_checks(
	system_provided,
	shared_library_artifact,
	interface_library_artifact):
	# If the shared library will be provided by system during runtime, users are not supposed to
	# specify shared_library.
	if system_provided and shared_library_artifact != None:
	fail("'shared_library' shouldn't be specified when 'system_provided' is true")

	# If a shared library won't be provided by system during runtime and we are linking the shared
	# library through interface library, the shared library must be specified.
	if (not system_provided and shared_library_artifact == None and
	interface_library_artifact != None):
	fail("'shared_library' should be specified when 'system_provided' is false")

	if (shared_library_artifact != None and
	not _is_shared_library_extension_valid(shared_library_artifact.basename)):
	fail("'shared_library' does not produce any cc_import shared_library files (expected .so, .dylib or .dll)")

	def _create_archive_action(
	ctx,
	feature_configuration,
	cc_toolchain,
	output_file,
	object_files):
	archiver_path = cc_common.get_tool_for_action(
	feature_configuration = feature_configuration,
	action_name = CPP_LINK_STATIC_LIBRARY_ACTION_NAME,
	)
	archiver_variables = cc_common.create_link_variables(
	feature_configuration = feature_configuration,
	cc_toolchain = cc_toolchain,
	output_file = output_file.path,
	is_using_linker = False,
	)
	command_line = cc_common.get_memory_inefficient_command_line(
	feature_configuration = feature_configuration,
	action_name = CPP_LINK_STATIC_LIBRARY_ACTION_NAME,
	variables = archiver_variables,
	)
	args = ctx.actions.args()
	args.add_all(command_line)
	args.add_all(object_files)
	args.use_param_file("@%s", use_always = True)

	env = cc_common.get_environment_variables(
	feature_configuration = feature_configuration,
	action_name = CPP_LINK_STATIC_LIBRARY_ACTION_NAME,
	variables = archiver_variables,
	)

	# TODO(bazel-team): PWD=/proc/self/cwd env var is missing, but it is present when an analogous archiving
	# action is created by cc_library
	ctx.actions.run(
	executable = archiver_path,
	arguments = [args],
	env = env,
	inputs = depset(
	direct = object_files,
	transitive = [
	cc_toolchain.all_files,
	],
	),
	use_default_shell_env = True,
	outputs = [output_file],
	mnemonic = "CppArchive",
	)

	def _cc_import_impl(ctx):
	cc_toolchain = find_cpp_toolchain(ctx)
	cc_common.check_experimental_starlark_cc_import(
	actions = ctx.actions,
	)

	feature_configuration = cc_common.configure_features(
	ctx = ctx,
	cc_toolchain = cc_toolchain,
	requested_features = ctx.features,
	unsupported_features = ctx.disabled_features,
	)

	_perform_error_checks(
	ctx.attr.system_provided,
	ctx.file.shared_library,
	ctx.file.interface_library,
	)

	pic_static_library = ctx.file.pic_static_library or None
	static_library = ctx.file.static_library or None

	if ctx.files.pic_objects and not pic_static_library:
	lib_name = "lib" + ctx.label.name + ".pic.a"
	pic_static_library = ctx.actions.declare_file(lib_name)
	_create_archive_action(ctx, feature_configuration, cc_toolchain, pic_static_library, ctx.files.pic_objects)

	if ctx.files.objects and not static_library:
	lib_name = "lib" + ctx.label.name + ".a"
	static_library = ctx.actions.declare_file(lib_name)
	_create_archive_action(ctx, feature_configuration, cc_toolchain, static_library, ctx.files.objects)

	library_to_link = cc_common.create_library_to_link(
	actions = ctx.actions,
	feature_configuration = feature_configuration,
	cc_toolchain = ctx.attr._cc_toolchain[cc_common.CcToolchainInfo],
	static_library = static_library,
	pic_static_library = pic_static_library,
	interface_library = ctx.file.interface_library,
	dynamic_library = ctx.file.shared_library,
	pic_objects = ctx.files.pic_objects,
	objects = ctx.files.objects,
	alwayslink = ctx.attr.alwayslink,
	)

	linker_input = cc_common.create_linker_input(
	libraries = depset([library_to_link]),
	user_link_flags = depset(ctx.attr.linkopts),
	owner = ctx.label,
	)

	linking_context = cc_common.create_linking_context(
	linker_inputs = depset([linker_input]),
	)

	(compilation_context, compilation_outputs) = cc_common.compile(
	actions = ctx.actions,
	feature_configuration = feature_configuration,
	cc_toolchain = cc_toolchain,
	public_hdrs = ctx.files.hdrs,
	includes = ctx.attr.includes,
	name = ctx.label.name,
	)

	this_cc_info = CcInfo(compilation_context = compilation_context, linking_context = linking_context)
	cc_infos = [this_cc_info]

	for dep in ctx.attr.deps:
	cc_infos.append(dep[CcInfo])
	merged_cc_info = cc_common.merge_cc_infos(cc_infos = cc_infos)

	runfiles = ctx.runfiles(files = ctx.files.data)

	transitive_runfiles_list = []
	if ctx.attr.static_library:
	transitive_runfiles_list.append(ctx.attr.static_library[DefaultInfo].default_runfiles)
	if ctx.attr.pic_static_library:
	transitive_runfiles_list.append(ctx.attr.pic_static_library[DefaultInfo].default_runfiles)
	if ctx.attr.shared_library:
	transitive_runfiles_list.append(ctx.attr.shared_library[DefaultInfo].default_runfiles)
	if ctx.attr.interface_library:
	transitive_runfiles_list.append(ctx.attr.interface_library[DefaultInfo].default_runfiles)
	for dep in ctx.attr.deps:
	transitive_runfiles_list.append(dep[DefaultInfo].default_runfiles)

	for maybe_runfiles in transitive_runfiles_list:
	if maybe_runfiles:
	runfiles = runfiles.merge(maybe_runfiles)

	default_info = DefaultInfo(runfiles = runfiles)

	return [merged_cc_info, default_info]

	cc_import = rule(
	implementation = _cc_import_impl,
	attrs = {
	"hdrs": attr.label_list(allow_files = [".h"]),
	"static_library": attr.label(allow_single_file = [".a", ".lib"]),
	"pic_static_library": attr.label(allow_single_file = [".pic.a", ".pic.lib"]),
	"shared_library": attr.label(allow_single_file = [".so", ".dll", ".dylib"]),
	"interface_library": attr.label(
	allow_single_file = [".ifso", ".tbd", ".lib", ".so", ".dylib"],
	),
	"pic_objects": attr.label_list(
	allow_files = [".o", ".pic.o"],
	),
	"objects": attr.label_list(
	allow_files = [".o", ".nopic.o"],
	),
	"system_provided": attr.bool(default = False),
	"alwayslink": attr.bool(default = False),
	"linkopts": attr.string_list(),
	"includes": attr.string_list(),
	"deps": attr.label_list(),
	"data": attr.label_list(allow_files = True),
	"_cc_toolchain": attr.label(default = "@bazel_tools//tools/cpp:current_cc_toolchain"),
	},
	toolchains = ["@rules_cc//cc:toolchain_type"], # copybara-use-repo-external-label
	fragments = ["cpp"],
	)