From bf2e03e76836ef2618bcb00ab27c3624ae35b789 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Somhairle=20MacLe=C3=B2id?= <smacleod@cloudflare.com>
Date: Thu, 29 Aug 2024 19:58:22 +0100
Subject: [PATCH] Support `.vsix` artifacts in pre-release registry (#6602)

---
 .changeset/cool-crews-happen.md                        |  5 +++++
 .../functions/utils/getArtifactForWorkflowRun.ts       | 10 ++++++----
 2 files changed, 11 insertions(+), 4 deletions(-)
 create mode 100644 .changeset/cool-crews-happen.md

diff --git a/.changeset/cool-crews-happen.md b/.changeset/cool-crews-happen.md
new file mode 100644
index 000000000000..b55b3d5fc361
--- /dev/null
+++ b/.changeset/cool-crews-happen.md
@@ -0,0 +1,5 @@
+---
+"@cloudflare/prerelease-registry": minor
+---
+
+feat: Support `.vsix` artifacts in pre-release registry
diff --git a/packages/prerelease-registry/functions/utils/getArtifactForWorkflowRun.ts b/packages/prerelease-registry/functions/utils/getArtifactForWorkflowRun.ts
index 3a34bfc0dd31..4c4124e9446b 100644
--- a/packages/prerelease-registry/functions/utils/getArtifactForWorkflowRun.ts
+++ b/packages/prerelease-registry/functions/utils/getArtifactForWorkflowRun.ts
@@ -130,13 +130,15 @@ export const getArtifactForWorkflowRun = async ({
 
 		const files = zip.files;
 		const fileNames = Object.keys(files);
-		const tgzFileName = fileNames.find((fileName) => fileName.endsWith(".tgz"));
-		if (tgzFileName === undefined) {
+		const downloadableFileName = fileNames.find(
+			(fileName) => fileName.endsWith(".tgz") || fileName.endsWith(".vsix")
+		);
+		if (downloadableFileName === undefined) {
 			return Response.json({ fileNames }, { status: 404 });
 		}
 
-		const tgzBlob = await files[tgzFileName].async("blob");
-		const response = new Response(tgzBlob, {
+		const downloadableBlob = await files[downloadableFileName].async("blob");
+		const response = new Response(downloadableBlob, {
 			headers: { "Cache-Control": `public, s-maxage=${ONE_WEEK}` },
 		});