Handle more module declaration cases (#6792)

.changeset/popular-horses-hide.md

@@ -0,0 +1,5 @@
+---
+"wrangler": patch
+---
+
+fix: Handle more module declaration cases

packages/wrangler/src/__tests__/type-generation.test.ts

@@ -1,6 +1,7 @@
 import * as fs from "fs";
 import * as TOML from "@iarna/toml";
 import {
+	constructTSModuleGlob,
 	constructType,
 	constructTypeKey,
 	generateImportSpecifier,
@@ -48,6 +49,22 @@ describe("constructTypeKey", () => {
 	});
 });
 
+describe("constructTSModuleGlob() should return a valid TS glob ", () => {
+	it.each([
+		["**/*.wasm", "*.wasm"],
+		["**/*.txt", "*.txt"],
+		["**/foo", "*/foo"],
+		["**/*foo", "*foo"],
+		["file.foo", "file.foo"],
+		["folder/file.foo", "folder/file.foo"],
+		["folder/*", "folder/*"],
+		["folder/**", "folder/*"],
+		["folder/**/*", "folder/*"],
+	])("$1 -> $2", (from, to) => {
+		expect(constructTSModuleGlob(from)).toBe(to);
+	});
+});
+
 describe("constructType", () => {
 	it("should return a valid type", () => {
 		expect(constructType("valid", "string")).toBe("valid: string;");

packages/wrangler/src/type-generation/index.ts

@@ -165,6 +165,19 @@ export function constructTypeKey(key: string) {
 	return `"${key}"`;
 }
 
+export function constructTSModuleGlob(glob: string) {
+	// Exact module reference, don't transform
+	if (!glob.includes("*")) {
+		return glob;
+		// Usually something like **/*.wasm. Turn into *.wasm
+	} else if (glob.includes(".")) {
+		return `*.${glob.split(".").at(-1)}`;
+	} else {
+		// Replace common patterns
+		return glob.replace("**/*", "*").replace("**/", "*/").replace("/**", "/*");
+	}
+}
+
 /**
  * Generate a import specifier from one module to another
  */
@@ -426,9 +439,7 @@ async function generateTypes(
 				moduleTypeMap[ruleObject.type as keyof typeof moduleTypeMap];
 			if (typeScriptType !== undefined) {
 				ruleObject.globs.forEach((glob) => {
-					modulesTypeStructure.push(`declare module "*.${glob
-						.split(".")
-						.at(-1)}" {
+					modulesTypeStructure.push(`declare module "${constructTSModuleGlob(glob)}" {
 	const value: ${typeScriptType};
 	export default value;
 }`);