python = repository_ctx.which("python3") or repository_ctx.which("python")

if not python:

fail("Neither python3 nor python executables found")

script = Label("//misc/bazel/internal:git_lfs_probe.py")

def probe(srcs, hash_only = False):

repository_ctx.report_progress("querying LFS url(s) for: %s" % ", ".join([src.basename for src in srcs]))

cmd = [python, script]

if hash_only:

cmd.append("--hash-only")

cmd.extend(srcs)

res = repository_ctx.execute(cmd, quiet = True)

if res.return_code != 0:

fail("git LFS probing failed while instantiating @%s:\n%s" % (repository_ctx.name, res.stderr))

return res.stdout.splitlines()

for src in srcs:

repository_ctx.watch(src)

infos = probe(srcs, hash_only = True)

remote = []

for src, info in zip(srcs, infos):

if info == "local":

repository_ctx.report_progress("symlinking local %s" % src.basename)

repository_ctx.symlink(src, src.basename)

else:

repository_ctx.report_progress("trying cache for remote %s" % src.basename)

res = repository_ctx.download([], src.basename, sha256 = info, allow_fail = True, executable = executable)

if not res.success:

remote.append(src)

if remote:

infos = probe(remote)

for src, info in zip(remote, infos):

sha256, _, url = info.partition(" ")

repository_ctx.report_progress("downloading remote %s" % src.basename)

repository_ctx.download(url, src.basename, sha256 = sha256, executable = executable)

if extract:

for src in srcs:

repository_ctx.report_progress("extracting %s" % src.basename)

repository_ctx.extract(src.basename, stripPrefix = stripPrefix)

attr = repository_ctx.attr

if attr.build_file_content and attr.build_file:

fail("You should specify only one among build_file_content and build_file for rule @%s" % repository_ctx.name)

if attr.build_file_content:

repository_ctx.file("BUILD.bazel", attr.build_file_content)

elif attr.build_file:

for src in repository_ctx.attr.srcs:

lfs_smudge(

repository_ctx,

[repository_ctx.path(src)],

extract = True,

stripPrefix = repository_ctx.attr.strip_prefix,

)

attr = repository_ctx.attr

if int(bool(attr.srcs)) + int(bool(attr.dir)) != 1:

fail("Exactly one between `srcs` and `dir` must be defined for @%s" % repository_ctx.name)

if attr.srcs:

srcs = [repository_ctx.path(src) for src in attr.srcs]

else:

dir = repository_ctx.path(attr.dir)

if not dir.is_dir:

fail("`dir` not a directory in @%s" % repository_ctx.name)

srcs = [f for f in dir.readdir() if not f.is_dir]

lfs_smudge(repository_ctx, srcs, executable = repository_ctx.attr.executable)

# with bzlmod the name is qualified with `+` separators, and we want the base name here

name = repository_ctx.name.split("+")[-1]

basenames = [src.basename for src in srcs]

build = "exports_files(%s)\n" % repr(basenames)

# add a main `name` filegroup only if it doesn't conflict with existing exported files

if name not in basenames:

build += 'filegroup(name = "%s", srcs = %s, visibility = ["//visibility:public"])\n' % (

name,

basenames,

)

repository_ctx.file("BUILD.bazel", build)

# this is for drop-in compatibility with `http_file`

repository_ctx.file(

"file/BUILD.bazel",

'alias(name = "file", actual = "//:%s", visibility = ["//visibility:public"])\n' % name,

"build_file": attr.label(doc = "The file to use as the BUILD file for this repository. " +

"Either build_file or build_file_content can be specified, but not both."),

"build_file_content": attr.string(doc = "The content for the BUILD file for this repository. " +

"Either build_file or build_file_content can be specified, but not both."),

"strip_prefix": attr.string(default = "", doc = "A directory prefix to strip from the extracted files."),

doc = "Export the contents from an on-demand LFS archive. The corresponding path should be added to be ignored " +

"in `.lfsconfig`.",

implementation = _download_and_extract_lfs_archive,

attrs = {

"src": attr.label(mandatory = True, doc = "Local path to the LFS archive to extract."),

} | _lfs_archive_attrs,

doc = "Overlay the contents from on-demand LFS archives. The corresponding paths should be added to be ignored " +

"in `.lfsconfig`.",

implementation = _download_and_extract_lfs_archives,

attrs = {

"srcs": attr.label_list(doc = "Local paths to the LFS archives to extract in order.", mandatory = True),

} | _lfs_archive_attrs,

doc = "Export LFS files for on-demand download. Exactly one between `srcs` and `dir` must be defined. The " +

"corresponding paths should be added to be ignored in `.lfsconfig`.",

implementation = _download_lfs,

attrs = {

"srcs": attr.label_list(doc = "Local paths to the LFS files to export."),

"dir": attr.label(doc = "Local path to a directory containing LFS files to export. Only the direct contents " +

"of the directory are exported"),

"executable": attr.bool(doc = "Whether files should be marked as executable"),

},