From df03c385d61b1f2f3fc85462da06ea502e9142fe Mon Sep 17 00:00:00 2001 From: Nathan Flurry Date: Thu, 10 Jul 2025 14:17:42 -0700 Subject: [PATCH] chore(pegboard): add workaround fetching image size when not using ats --- docker/dev-full/docker-compose.yml | 1 + .../edge/infra/client/manager/src/image_download_handler.rs | 5 +++++ 2 files changed, 6 insertions(+) diff --git a/docker/dev-full/docker-compose.yml b/docker/dev-full/docker-compose.yml index 9602796578..ac1ef2676b 100644 --- a/docker/dev-full/docker-compose.yml +++ b/docker/dev-full/docker-compose.yml @@ -211,6 +211,7 @@ services: - RIVET_SERVICE_NAME=rivet-client - RIVET_OTEL_ENDPOINT=http://otel-collector:4317 - RUST_LOG=debug,hyper=info + - __HACK__DISABLE_FETCH_IMAGE_SIZE=1 stop_grace_period: 0s depends_on: # HACK: rivet-server dependency since we need to do a DNS lookup in entrypoint.sh for the iptables chain diff --git a/packages/edge/infra/client/manager/src/image_download_handler.rs b/packages/edge/infra/client/manager/src/image_download_handler.rs index d12d0c5648..60e0606565 100644 --- a/packages/edge/infra/client/manager/src/image_download_handler.rs +++ b/packages/edge/infra/client/manager/src/image_download_handler.rs @@ -573,6 +573,11 @@ impl ImageDownloadHandler { ctx: &Ctx, image_config: &protocol::Image, ) -> Result { + // HEAD does not work if not using ATS + if std::env::var("__HACK__DISABLE_FETCH_IMAGE_SIZE").map_or(false, |x| x == "1") { + return Ok(0); + } + let addresses = self.get_image_addresses(ctx, image_config).await?; let mut iter = addresses.into_iter();