allow custom wasm binary name

Author: PgBiel

godot-macros/src/gdextension.rs

@@ -29,6 +29,7 @@ pub fn attribute_gdextension(item: venial::Item) -> ParseResult<TokenStream> {
     let mut parser = KvParser::parse_required(&drained_attributes, "gdextension", &impl_decl)?;
     let entry_point = parser.handle_ident("entry_point")?;
     let entry_symbol = parser.handle_ident("entry_symbol")?;
+    let wasm_binary = parser.handle_expr("wasm_binary")?;
     parser.finish()?;
 
     if entry_point.is_some() && entry_symbol.is_some() {
@@ -48,6 +49,13 @@ pub fn attribute_gdextension(item: venial::Item) -> ParseResult<TokenStream> {
         .or(entry_point)
         .unwrap_or_else(|| ident("gdext_rust_init"));
 
+    let wasm_binary = match wasm_binary {
+        Some(wasm_binary) => {
+            quote! { format!("'{}'", (&#wasm_binary as &str).replace("\\", "\\\\").replace("'", "\\'")) }
+        }
+        None => quote! { "snakePkgName + '.wasm'" },
+    };
+
     let impl_ty = &impl_decl.self_ty;
 
     Ok(quote! {
@@ -69,34 +77,35 @@ pub fn attribute_gdextension(item: venial::Item) -> ParseResult<TokenStream> {
             // involved, but I don't know what guarantees we have here.
             //
             // We should keep an eye out for these sorts of failures!
-            let script = std::ffi::CString::new(concat!(
-                "var pkgName = '", env!("CARGO_PKG_NAME"), "';", r#"
-                var libName = pkgName.replaceAll('-', '_') + '.wasm';
-                if (!(libName in LDSO.loadedLibsByName)) {
+            let script = std::ffi::CString::new(format!(
+                r#"var pkgName = '{}';
+                var snakePkgName = pkgName.replaceAll('-', '_');
+                var libName = {};
+                if (!(libName in LDSO.loadedLibsByName)) {{
                     // Always print to console, even if the error is suppressed.
-                    console.error(`godot-rust could not find the Wasm module '${libName}', needed to load the '${pkgName}' crate. Please ensure a file named '${libName}' exists in the game's web export files. This may require updating Wasm paths in the crate's corresponding '.gdextension' file, or just renaming the Wasm file to the correct name otherwise.`);
-                    throw new Error(`Wasm module '${libName}' not found. Check the console for more information.`);
-                }
+                    console.error(`godot-rust could not find the Wasm module '${{libName}}', needed to load the '${{pkgName}}' crate. Please ensure a file named '${{libName}}' exists in the game's web export files. This may require updating Wasm paths in the crate's corresponding '.gdextension' file, or just renaming the Wasm file to the correct name otherwise.`);
+                    throw new Error(`Wasm module '${{libName}}' not found. Check the console for more information.`);
+                }}
                 var dso = LDSO.loadedLibsByName[libName];
                 // This property was renamed as of emscripten 3.1.34
                 var dso_exports = "module" in dso ? dso["module"] : dso["exports"];
                 var registrants = [];
-                for (sym in dso_exports) {
-                    if (sym.startsWith("dynCall_")) {
-                        if (!(sym in Module)) {
-                            console.log(`Patching Module with ${sym}`);
+                for (sym in dso_exports) {{
+                    if (sym.startsWith("dynCall_")) {{
+                        if (!(sym in Module)) {{
+                            console.log(`Patching Module with ${{sym}}`);
                             Module[sym] = dso_exports[sym];
-                        }
-                    } else if (sym.startsWith("__godot_rust_registrant_")) {
+                        }}
+                    }} else if (sym.startsWith("__godot_rust_registrant_")) {{
                         registrants.push(sym);
-                    }
-                }
-                for (sym of registrants) {
-                    console.log(`Running registrant ${sym}`);
+                    }}
+                }}
+                for (sym of registrants) {{
+                    console.log(`Running registrant ${{sym}}`);
                     dso_exports[sym]();
-                }
+                }}
                 console.log("Added",  registrants.length, "plugins to registry!");
-            "#)).expect("Unable to create CString from script");
+            "#, env!("CARGO_PKG_NAME"), #wasm_binary)).expect("Unable to create CString from script");
 
             extern "C" { fn emscripten_run_script(script: *const std::ffi::c_char); }
             unsafe { emscripten_run_script(script.as_ptr()); }