diff --git a/mojo/extensions.bzl b/mojo/extensions.bzl index 4379ded..589694a 100644 --- a/mojo/extensions.bzl +++ b/mojo/extensions.bzl @@ -17,7 +17,7 @@ _PLATFORM_MAPPINGS = { def _mojo_toolchain_impl(rctx): rctx.download_and_extract( - url = "https://dl.modular.com/public/nightly/python/max-{}-py3-none-{}.whl".format( + url = rctx.attr.url_override or "https://dl.modular.com/public/nightly/python/max-{}-py3-none-{}.whl".format( rctx.attr.version, _PLATFORM_MAPPINGS[rctx.attr.platform], ), @@ -47,6 +47,10 @@ _mojo_toolchain_repository = repository_rule( values = _PLATFORMS, mandatory = True, ), + "url_override": attr.string( + doc = "Override the download URL for the prebuilt package.", + default = "", + ), "use_prebuilt_packages": attr.bool( doc = "Whether to automatically add prebuilt mojopkgs to every mojo target.", mandatory = True, @@ -106,6 +110,7 @@ def _mojo_impl(mctx): name = name, version = tags.version, platform = platform, + url_override = tags.url_override, use_prebuilt_packages = tags.use_prebuilt_packages, ) @@ -124,6 +129,10 @@ _toolchain_tag = tag_class( doc = "The version of the Mojo toolchain to download.", default = _DEFAULT_VERSION, ), + "url_override": attr.string( + doc = "Override the download URL for the prebuilt package.", + default = "", + ), "use_prebuilt_packages": attr.bool( doc = "Whether to automatically add prebuilt mojopkgs to every mojo target.", default = True,