From 8242b24736a4adf02c048914fd7ac97b2e708200 Mon Sep 17 00:00:00 2001 From: Alex Trotta Date: Fri, 23 May 2025 15:10:24 -0400 Subject: [PATCH] Add ability to override the download URL manually --- mojo/extensions.bzl | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) 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,