You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

64 lines
2.4 KiB

"""A module defining a repository rule for vendoring the dependencies
of a crate in the current workspace.
"""
def _impl(repository_ctx):
# Link cxx repository into @third-party.
lockfile = repository_ctx.path(repository_ctx.attr.lockfile)
workspace = lockfile.dirname.dirname
repository_ctx.symlink(workspace, "workspace")
# Copy third-party/Cargo.lock since those are the crate versions that the
# BUILD file is written against.
vendor_lockfile = repository_ctx.path("workspace/third-party/Cargo.lock")
root_lockfile = repository_ctx.path("workspace/Cargo.lock")
_copy_file(repository_ctx, src = vendor_lockfile, dst = root_lockfile)
# Execute cargo vendor.
cmd = ["cargo", "vendor", "--versioned-dirs", "third-party/vendor"]
result = repository_ctx.execute(
cmd,
quiet = True,
working_directory = "workspace",
)
_log_cargo_vendor(repository_ctx, result)
if result.return_code != 0:
fail("failed to execute `{}`".format(" ".join(cmd)))
# Copy lockfile back to third-party/Cargo.lock to reflect any modification
# performed by Cargo.
_copy_file(repository_ctx, src = root_lockfile, dst = vendor_lockfile)
# Produce a token for third_party_glob to depend on so that the necessary
# sequencing is visible to Bazel.
repository_ctx.file("BUILD", executable = False)
repository_ctx.file("vendor.bzl", "vendored = True", executable = False)
def _copy_file(repository_ctx, *, src, dst):
content = repository_ctx.read(src)
if not dst.exists or content != repository_ctx.read(dst):
repository_ctx.file(dst, content = content, executable = False)
def _log_cargo_vendor(repository_ctx, result):
relevant = ""
for line in result.stderr.splitlines(True):
if line.strip() and not line.startswith("To use vendored sources,"):
relevant += line
if relevant:
# Render it as command output.
# If we just use print(), Bazel will cache and repeat the output even
# when not rerunning the command.
print = ["echo", relevant]
repository_ctx.execute(print, quiet = False)
vendor = repository_rule(
doc = "A rule used to vendor the dependencies of a crate in the current workspace",
attrs = {
"lockfile": attr.label(
doc = "A lockfile providing the set of crates to vendor",
),
},
local = True,
implementation = _impl,
)