tools/build_defs/pkg/pkg.bzl - bazel - Git at Google

 # Copyright 2015 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.
 """Rules for manipulation of various packaging."""

 load(":path.bzl", "compute_data_path", "dest_path")

 # Filetype to restrict inputs
 tar_filetype = [".tar", ".tar.gz", ".tgz", ".tar.xz", ".tar.bz2"]
 deb_filetype = [".deb", ".udeb"]

 def _remap(remap_paths, path):
     """If path starts with a key in remap_paths, rewrite it."""
     for prefix, replacement in remap_paths.items():
         if path.startswith(prefix):
             return replacement + path[len(prefix):]
     return path

 def _quote(filename, protect = "="):
     """Quote the filename, by escaping = by \= and \ by \\"""
     return filename.replace("\\", "\\\\").replace(protect, "\\" + protect)

 def _pkg_tar_impl(ctx):
     """Implementation of the pkg_tar rule."""

     # Compute the relative path
     data_path = compute_data_path(ctx.outputs.out, ctx.attr.strip_prefix)

     # Find a list of path remappings to apply.
     remap_paths = ctx.attr.remap_paths

     # Start building the arguments.
     args = [
         "--output=" + ctx.outputs.out.path,
         "--directory=" + ctx.attr.package_dir,
         "--mode=" + ctx.attr.mode,
         "--owner=" + ctx.attr.owner,
         "--owner_name=" + ctx.attr.ownername,
     ]
     if ctx.attr.mtime != -1:  # Note: Must match default in rule def.
         if ctx.attr.portable_mtime:
             fail("You may not set both mtime and portable_mtime")
         args.append("--mtime=%d" % ctx.attr.mtime)
     if ctx.attr.portable_mtime:
         args.append("--mtime=portable")

     # Add runfiles if requested
     file_inputs = []
     if ctx.attr.include_runfiles:
         runfiles_depsets = []
         for f in ctx.attr.srcs:
             default_runfiles = f[DefaultInfo].default_runfiles
             if default_runfiles != None:
                 runfiles_depsets.append(default_runfiles.files)

         # deduplicates files in srcs attribute and their runfiles
         file_inputs = depset(ctx.files.srcs, transitive = runfiles_depsets).to_list()
     else:
         file_inputs = ctx.files.srcs[:]

     args += [
         "--file=%s=%s" % (_quote(f.path), _remap(remap_paths, dest_path(f, data_path)))
         for f in file_inputs
     ]
     for target, f_dest_path in ctx.attr.files.items():
         target_files = target.files.to_list()
         if len(target_files) != 1:
             fail("Each input must describe exactly one file.", attr = "files")
         file_inputs += target_files
         args += ["--file=%s=%s" % (_quote(target_files[0].path), f_dest_path)]
     if ctx.attr.modes:
         args += [
             "--modes=%s=%s" % (_quote(key), ctx.attr.modes[key])
             for key in ctx.attr.modes
         ]
     if ctx.attr.owners:
         args += [
             "--owners=%s=%s" % (_quote(key), ctx.attr.owners[key])
             for key in ctx.attr.owners
         ]
     if ctx.attr.ownernames:
         args += [
             "--owner_names=%s=%s" % (_quote(key), ctx.attr.ownernames[key])
             for key in ctx.attr.ownernames
         ]
     if ctx.attr.empty_files:
         args += ["--empty_file=%s" % empty_file for empty_file in ctx.attr.empty_files]
     if ctx.attr.empty_dirs:
         args += ["--empty_dir=%s" % empty_dir for empty_dir in ctx.attr.empty_dirs]
     if ctx.attr.extension:
         dotPos = ctx.attr.extension.find(".")
         if dotPos > 0:
             dotPos += 1
             args += ["--compression=%s" % ctx.attr.extension[dotPos:]]
         elif ctx.attr.extension == "tgz":
             args += ["--compression=gz"]
     args += ["--tar=" + f.path for f in ctx.files.deps]
     args += [
         "--link=%s:%s" % (_quote(k, protect = ":"), ctx.attr.symlinks[k])
         for k in ctx.attr.symlinks
     ]
     arg_file = ctx.actions.declare_file(ctx.label.name + ".args")
     ctx.actions.write(arg_file, "\n".join(args))

     ctx.actions.run(
         inputs = file_inputs + ctx.files.deps + [arg_file],
         executable = ctx.executable.build_tar,
         arguments = ["--flagfile", arg_file.path],
         outputs = [ctx.outputs.out],
         mnemonic = "PackageTar",
         use_default_shell_env = True,
     )

 def _pkg_deb_impl(ctx):
     """The implementation for the pkg_deb rule."""
     files = [ctx.file.data]
     args = [
         "--output=" + ctx.outputs.deb.path,
         "--changes=" + ctx.outputs.changes.path,
         "--data=" + ctx.file.data.path,
         "--package=" + ctx.attr.package,
         "--architecture=" + ctx.attr.architecture,
         "--maintainer=" + ctx.attr.maintainer,
     ]
     if ctx.attr.preinst:
         args += ["--preinst=@" + ctx.file.preinst.path]
         files += [ctx.file.preinst]
     if ctx.attr.postinst:
         args += ["--postinst=@" + ctx.file.postinst.path]
         files += [ctx.file.postinst]
     if ctx.attr.prerm:
         args += ["--prerm=@" + ctx.file.prerm.path]
         files += [ctx.file.prerm]
     if ctx.attr.postrm:
         args += ["--postrm=@" + ctx.file.postrm.path]
         files += [ctx.file.postrm]

     # Conffiles can be specified by a file or a string list
     if ctx.attr.conffiles_file:
         if ctx.attr.conffiles:
             fail("Both conffiles and conffiles_file attributes were specified")
         args += ["--conffile=@" + ctx.file.conffiles_file.path]
         files += [ctx.file.conffiles_file]
     elif ctx.attr.conffiles:
         args += ["--conffile=%s" % cf for cf in ctx.attr.conffiles]

     # Version and description can be specified by a file or inlined
     if ctx.attr.version_file:
         if ctx.attr.version:
             fail("Both version and version_file attributes were specified")
         args += ["--version=@" + ctx.file.version_file.path]
         files += [ctx.file.version_file]
     elif ctx.attr.version:
         args += ["--version=" + ctx.attr.version]
     else:
         fail("Neither version_file nor version attribute was specified")

     if ctx.attr.description_file:
         if ctx.attr.description:
             fail("Both description and description_file attributes were specified")
         args += ["--description=@" + ctx.file.description_file.path]
         files += [ctx.file.description_file]
     elif ctx.attr.description:
         args += ["--description=" + ctx.attr.description]
     else:
         fail("Neither description_file nor description attribute was specified")

     # Built using can also be specified by a file or inlined (but is not mandatory)
     if ctx.attr.built_using_file:
         if ctx.attr.built_using:
             fail("Both build_using and built_using_file attributes were specified")
         args += ["--built_using=@" + ctx.file.built_using_file.path]
         files += [ctx.file.built_using_file]
     elif ctx.attr.built_using:
         args += ["--built_using=" + ctx.attr.built_using]

     if ctx.attr.priority:
         args += ["--priority=" + ctx.attr.priority]
     if ctx.attr.section:
         args += ["--section=" + ctx.attr.section]
     if ctx.attr.homepage:
         args += ["--homepage=" + ctx.attr.homepage]

     args += ["--distribution=" + ctx.attr.distribution]
     args += ["--urgency=" + ctx.attr.urgency]
     args += ["--depends=" + d for d in ctx.attr.depends]
     args += ["--suggests=" + d for d in ctx.attr.suggests]
     args += ["--enhances=" + d for d in ctx.attr.enhances]
     args += ["--conflicts=" + d for d in ctx.attr.conflicts]
     args += ["--pre_depends=" + d for d in ctx.attr.predepends]
     args += ["--recommends=" + d for d in ctx.attr.recommends]

     ctx.actions.run(
         executable = ctx.executable.make_deb,
         arguments = args,
         inputs = files,
         outputs = [ctx.outputs.deb, ctx.outputs.changes],
         mnemonic = "MakeDeb",
     )
     ctx.actions.run_shell(
         command = "ln -s %s %s" % (ctx.outputs.deb.basename, ctx.outputs.out.path),
         inputs = [ctx.outputs.deb],
         outputs = [ctx.outputs.out],
     )

 # A rule for creating a tar file, see README.md
 _real_pkg_tar = rule(
     implementation = _pkg_tar_impl,
     attrs = {
         "strip_prefix": attr.string(),
         "package_dir": attr.string(default = "/"),
         "deps": attr.label_list(allow_files = tar_filetype),
         "srcs": attr.label_list(allow_files = True),
         "files": attr.label_keyed_string_dict(allow_files = True),
         "mode": attr.string(default = "0555"),
         "modes": attr.string_dict(),
         "mtime": attr.int(default = -1),
         "portable_mtime": attr.bool(default = True),
         "owner": attr.string(default = "0.0"),
         "ownername": attr.string(default = "."),
         "owners": attr.string_dict(),
         "ownernames": attr.string_dict(),
         "extension": attr.string(default = "tar"),
         "symlinks": attr.string_dict(),
         "empty_files": attr.string_list(),
         "include_runfiles": attr.bool(),
         "empty_dirs": attr.string_list(),
         "remap_paths": attr.string_dict(),
         # Implicit dependencies.
         "build_tar": attr.label(
             default = Label("//tools/build_defs/pkg:build_tar"),
             cfg = "host",
             executable = True,
             allow_files = True,
         ),
     },
     outputs = {
         "out": "%{name}.%{extension}",
     },
 )

 def pkg_tar(**kwargs):
     # Compatibility with older versions of pkg_tar that define files as
     # a flat list of labels.
     if "srcs" not in kwargs:
         if "files" in kwargs:
             if not hasattr(kwargs["files"], "items"):
                 label = "%s//%s:%s" % (native.repository_name(), native.package_name(), kwargs["name"])
                 print("%s: you provided a non dictionary to the pkg_tar `files` attribute. " % (label,) +
                       "This attribute was renamed to `srcs`. " +
                       "Consider renaming it in your BUILD file.")
                 kwargs["srcs"] = kwargs.pop("files")
     _real_pkg_tar(**kwargs)

 # A rule for creating a deb file, see README.md
 pkg_deb = rule(
     implementation = _pkg_deb_impl,
     attrs = {
         "data": attr.label(mandatory = True, allow_single_file = tar_filetype),
         "package": attr.string(mandatory = True),
         "architecture": attr.string(default = "all"),
         "distribution": attr.string(default = "unstable"),
         "urgency": attr.string(default = "medium"),
         "maintainer": attr.string(mandatory = True),
         "preinst": attr.label(allow_single_file = True),
         "postinst": attr.label(allow_single_file = True),
         "prerm": attr.label(allow_single_file = True),
         "postrm": attr.label(allow_single_file = True),
         "conffiles_file": attr.label(allow_single_file = True),
         "conffiles": attr.string_list(default = []),
         "version_file": attr.label(allow_single_file = True),
         "version": attr.string(),
         "description_file": attr.label(allow_single_file = True),
         "description": attr.string(),
         "built_using_file": attr.label(allow_single_file = True),
         "built_using": attr.string(),
         "priority": attr.string(),
         "section": attr.string(),
         "homepage": attr.string(),
         "depends": attr.string_list(default = []),
         "suggests": attr.string_list(default = []),
         "enhances": attr.string_list(default = []),
         "conflicts": attr.string_list(default = []),
         "predepends": attr.string_list(default = []),
         "recommends": attr.string_list(default = []),
         # Implicit dependencies.
         "make_deb": attr.label(
             default = Label("//tools/build_defs/pkg:make_deb"),
             cfg = "host",
             executable = True,
             allow_files = True,
         ),
     },
     outputs = {
         "out": "%{name}.deb",
         "deb": "%{package}_%{version}_%{architecture}.deb",
         "changes": "%{package}_%{version}_%{architecture}.changes",
     },
 )
	# Copyright 2015 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.
	"""Rules for manipulation of various packaging."""

	load(":path.bzl", "compute_data_path", "dest_path")

	# Filetype to restrict inputs
	tar_filetype = [".tar", ".tar.gz", ".tgz", ".tar.xz", ".tar.bz2"]
	deb_filetype = [".deb", ".udeb"]

	def _remap(remap_paths, path):
	"""If path starts with a key in remap_paths, rewrite it."""
	for prefix, replacement in remap_paths.items():
	if path.startswith(prefix):
	return replacement + path[len(prefix):]
	return path

	def _quote(filename, protect = "="):
	"""Quote the filename, by escaping = by \= and \ by \\"""
	return filename.replace("\\", "\\\\").replace(protect, "\\" + protect)

	def _pkg_tar_impl(ctx):
	"""Implementation of the pkg_tar rule."""

	# Compute the relative path
	data_path = compute_data_path(ctx.outputs.out, ctx.attr.strip_prefix)

	# Find a list of path remappings to apply.
	remap_paths = ctx.attr.remap_paths

	# Start building the arguments.
	args = [
	"--output=" + ctx.outputs.out.path,
	"--directory=" + ctx.attr.package_dir,
	"--mode=" + ctx.attr.mode,
	"--owner=" + ctx.attr.owner,
	"--owner_name=" + ctx.attr.ownername,
	]
	if ctx.attr.mtime != -1: # Note: Must match default in rule def.
	if ctx.attr.portable_mtime:
	fail("You may not set both mtime and portable_mtime")
	args.append("--mtime=%d" % ctx.attr.mtime)
	if ctx.attr.portable_mtime:
	args.append("--mtime=portable")

	# Add runfiles if requested
	file_inputs = []
	if ctx.attr.include_runfiles:
	runfiles_depsets = []
	for f in ctx.attr.srcs:
	default_runfiles = f[DefaultInfo].default_runfiles
	if default_runfiles != None:
	runfiles_depsets.append(default_runfiles.files)

	# deduplicates files in srcs attribute and their runfiles
	file_inputs = depset(ctx.files.srcs, transitive = runfiles_depsets).to_list()
	else:
	file_inputs = ctx.files.srcs[:]

	args += [
	"--file=%s=%s" % (_quote(f.path), _remap(remap_paths, dest_path(f, data_path)))
	for f in file_inputs
	]
	for target, f_dest_path in ctx.attr.files.items():
	target_files = target.files.to_list()
	if len(target_files) != 1:
	fail("Each input must describe exactly one file.", attr = "files")
	file_inputs += target_files
	args += ["--file=%s=%s" % (_quote(target_files[0].path), f_dest_path)]
	if ctx.attr.modes:
	args += [
	"--modes=%s=%s" % (_quote(key), ctx.attr.modes[key])
	for key in ctx.attr.modes
	]
	if ctx.attr.owners:
	args += [
	"--owners=%s=%s" % (_quote(key), ctx.attr.owners[key])
	for key in ctx.attr.owners
	]
	if ctx.attr.ownernames:
	args += [
	"--owner_names=%s=%s" % (_quote(key), ctx.attr.ownernames[key])
	for key in ctx.attr.ownernames
	]
	if ctx.attr.empty_files:
	args += ["--empty_file=%s" % empty_file for empty_file in ctx.attr.empty_files]
	if ctx.attr.empty_dirs:
	args += ["--empty_dir=%s" % empty_dir for empty_dir in ctx.attr.empty_dirs]
	if ctx.attr.extension:
	dotPos = ctx.attr.extension.find(".")
	if dotPos > 0:
	dotPos += 1
	args += ["--compression=%s" % ctx.attr.extension[dotPos:]]
	elif ctx.attr.extension == "tgz":
	args += ["--compression=gz"]
	args += ["--tar=" + f.path for f in ctx.files.deps]
	args += [
	"--link=%s:%s" % (_quote(k, protect = ":"), ctx.attr.symlinks[k])
	for k in ctx.attr.symlinks
	]
	arg_file = ctx.actions.declare_file(ctx.label.name + ".args")
	ctx.actions.write(arg_file, "\n".join(args))

	ctx.actions.run(
	inputs = file_inputs + ctx.files.deps + [arg_file],
	executable = ctx.executable.build_tar,
	arguments = ["--flagfile", arg_file.path],
	outputs = [ctx.outputs.out],
	mnemonic = "PackageTar",
	use_default_shell_env = True,
	)

	def _pkg_deb_impl(ctx):
	"""The implementation for the pkg_deb rule."""
	files = [ctx.file.data]
	args = [
	"--output=" + ctx.outputs.deb.path,
	"--changes=" + ctx.outputs.changes.path,
	"--data=" + ctx.file.data.path,
	"--package=" + ctx.attr.package,
	"--architecture=" + ctx.attr.architecture,
	"--maintainer=" + ctx.attr.maintainer,
	]
	if ctx.attr.preinst:
	args += ["--preinst=@" + ctx.file.preinst.path]
	files += [ctx.file.preinst]
	if ctx.attr.postinst:
	args += ["--postinst=@" + ctx.file.postinst.path]
	files += [ctx.file.postinst]
	if ctx.attr.prerm:
	args += ["--prerm=@" + ctx.file.prerm.path]
	files += [ctx.file.prerm]
	if ctx.attr.postrm:
	args += ["--postrm=@" + ctx.file.postrm.path]
	files += [ctx.file.postrm]

	# Conffiles can be specified by a file or a string list
	if ctx.attr.conffiles_file:
	if ctx.attr.conffiles:
	fail("Both conffiles and conffiles_file attributes were specified")
	args += ["--conffile=@" + ctx.file.conffiles_file.path]
	files += [ctx.file.conffiles_file]
	elif ctx.attr.conffiles:
	args += ["--conffile=%s" % cf for cf in ctx.attr.conffiles]

	# Version and description can be specified by a file or inlined
	if ctx.attr.version_file:
	if ctx.attr.version:
	fail("Both version and version_file attributes were specified")
	args += ["--version=@" + ctx.file.version_file.path]
	files += [ctx.file.version_file]
	elif ctx.attr.version:
	args += ["--version=" + ctx.attr.version]
	else:
	fail("Neither version_file nor version attribute was specified")

	if ctx.attr.description_file:
	if ctx.attr.description:
	fail("Both description and description_file attributes were specified")
	args += ["--description=@" + ctx.file.description_file.path]
	files += [ctx.file.description_file]
	elif ctx.attr.description:
	args += ["--description=" + ctx.attr.description]
	else:
	fail("Neither description_file nor description attribute was specified")

	# Built using can also be specified by a file or inlined (but is not mandatory)
	if ctx.attr.built_using_file:
	if ctx.attr.built_using:
	fail("Both build_using and built_using_file attributes were specified")
	args += ["--built_using=@" + ctx.file.built_using_file.path]
	files += [ctx.file.built_using_file]
	elif ctx.attr.built_using:
	args += ["--built_using=" + ctx.attr.built_using]

	if ctx.attr.priority:
	args += ["--priority=" + ctx.attr.priority]
	if ctx.attr.section:
	args += ["--section=" + ctx.attr.section]
	if ctx.attr.homepage:
	args += ["--homepage=" + ctx.attr.homepage]

	args += ["--distribution=" + ctx.attr.distribution]
	args += ["--urgency=" + ctx.attr.urgency]
	args += ["--depends=" + d for d in ctx.attr.depends]
	args += ["--suggests=" + d for d in ctx.attr.suggests]
	args += ["--enhances=" + d for d in ctx.attr.enhances]
	args += ["--conflicts=" + d for d in ctx.attr.conflicts]
	args += ["--pre_depends=" + d for d in ctx.attr.predepends]
	args += ["--recommends=" + d for d in ctx.attr.recommends]

	ctx.actions.run(
	executable = ctx.executable.make_deb,
	arguments = args,
	inputs = files,
	outputs = [ctx.outputs.deb, ctx.outputs.changes],
	mnemonic = "MakeDeb",
	)
	ctx.actions.run_shell(
	command = "ln -s %s %s" % (ctx.outputs.deb.basename, ctx.outputs.out.path),
	inputs = [ctx.outputs.deb],
	outputs = [ctx.outputs.out],
	)

	# A rule for creating a tar file, see README.md
	_real_pkg_tar = rule(
	implementation = _pkg_tar_impl,
	attrs = {
	"strip_prefix": attr.string(),
	"package_dir": attr.string(default = "/"),
	"deps": attr.label_list(allow_files = tar_filetype),
	"srcs": attr.label_list(allow_files = True),
	"files": attr.label_keyed_string_dict(allow_files = True),
	"mode": attr.string(default = "0555"),
	"modes": attr.string_dict(),
	"mtime": attr.int(default = -1),
	"portable_mtime": attr.bool(default = True),
	"owner": attr.string(default = "0.0"),
	"ownername": attr.string(default = "."),
	"owners": attr.string_dict(),
	"ownernames": attr.string_dict(),
	"extension": attr.string(default = "tar"),
	"symlinks": attr.string_dict(),
	"empty_files": attr.string_list(),
	"include_runfiles": attr.bool(),
	"empty_dirs": attr.string_list(),
	"remap_paths": attr.string_dict(),
	# Implicit dependencies.
	"build_tar": attr.label(
	default = Label("//tools/build_defs/pkg:build_tar"),
	cfg = "host",
	executable = True,
	allow_files = True,
	),
	},
	outputs = {
	"out": "%{name}.%{extension}",
	},
	)

	def pkg_tar(**kwargs):
	# Compatibility with older versions of pkg_tar that define files as
	# a flat list of labels.
	if "srcs" not in kwargs:
	if "files" in kwargs:
	if not hasattr(kwargs["files"], "items"):
	label = "%s//%s:%s" % (native.repository_name(), native.package_name(), kwargs["name"])
	print("%s: you provided a non dictionary to the pkg_tar `files` attribute. " % (label,) +
	"This attribute was renamed to `srcs`. " +
	"Consider renaming it in your BUILD file.")
	kwargs["srcs"] = kwargs.pop("files")
	_real_pkg_tar(**kwargs)

	# A rule for creating a deb file, see README.md
	pkg_deb = rule(
	implementation = _pkg_deb_impl,
	attrs = {
	"data": attr.label(mandatory = True, allow_single_file = tar_filetype),
	"package": attr.string(mandatory = True),
	"architecture": attr.string(default = "all"),
	"distribution": attr.string(default = "unstable"),
	"urgency": attr.string(default = "medium"),
	"maintainer": attr.string(mandatory = True),
	"preinst": attr.label(allow_single_file = True),
	"postinst": attr.label(allow_single_file = True),
	"prerm": attr.label(allow_single_file = True),
	"postrm": attr.label(allow_single_file = True),
	"conffiles_file": attr.label(allow_single_file = True),
	"conffiles": attr.string_list(default = []),
	"version_file": attr.label(allow_single_file = True),
	"version": attr.string(),
	"description_file": attr.label(allow_single_file = True),
	"description": attr.string(),
	"built_using_file": attr.label(allow_single_file = True),
	"built_using": attr.string(),
	"priority": attr.string(),
	"section": attr.string(),
	"homepage": attr.string(),
	"depends": attr.string_list(default = []),
	"suggests": attr.string_list(default = []),
	"enhances": attr.string_list(default = []),
	"conflicts": attr.string_list(default = []),
	"predepends": attr.string_list(default = []),
	"recommends": attr.string_list(default = []),
	# Implicit dependencies.
	"make_deb": attr.label(
	default = Label("//tools/build_defs/pkg:make_deb"),
	cfg = "host",
	executable = True,
	allow_files = True,
	),
	},
	outputs = {
	"out": "%{name}.deb",
	"deb": "%{package}_%{version}_%{architecture}.deb",
	"changes": "%{package}_%{version}_%{architecture}.changes",
	},
	)