@rsvelte/vite-plugin-svelte 0.5.0 → 0.5.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rsvelte/vite-plugin-svelte",
-  "version": "0.5.0",
+  "version": "0.5.1",
   "license": "MIT",
   "author": "dominikg",
   "files": [
@@ -34,12 +34,15 @@
   },
   "homepage": "https://github.com/baseballyama/rsvelte/tree/main/apps/npm/vite-plugin-svelte#readme",
   "dependencies": {
+    "@jridgewell/remapping": "^2.3.5",
+    "@jridgewell/trace-mapping": "^0.3.31",
     "@sveltejs/vite-plugin-svelte-inspector": "^5.0.0",
+    "acorn": "^8.16.0",
     "deepmerge": "^4.3.1",
     "magic-string": "^1.0.0",
     "obug": "^2.1.0",
     "vitefu": "^1.1.1",
-    "@rsvelte/vite-plugin-svelte-native": "^0.3.0"
+    "@rsvelte/vite-plugin-svelte-native": "^0.3.7"
   },
   "peerDependencies": {
     "svelte": "^5.0.0",

package/src/utils/compile.js

@@ -1,6 +1,7 @@
 import * as svelte from '@rsvelte/vite-plugin-svelte-native';
 import { safeBase64Hash } from './hash.js';
 import { log } from './log.js';
+import { addPartialAcceptExports, applyMapEdits } from './map-edits.js';
 
 import { mapToRelative } from './sourcemaps.js';
 import { enhanceCompileError } from './error.js';
@@ -109,12 +110,11 @@ export function createCompileSvelte() {
 			// TODO remove later
 			if (
 				options.server?.config.experimental.hmrPartialAccept &&
-				compiled.js.code.includes('import.meta.hot.accept(')
+				compiled.js.code.includes('import.meta.hot')
 			) {
-				compiled.js.code = compiled.js.code.replaceAll(
-					'import.meta.hot.accept(',
-					'import.meta.hot.acceptExports(["default"],'
-				);
+				applyMapEdits(compiled, filename, (editor, generated) => {
+					addPartialAcceptExports(editor, generated);
+				});
 			}
 		} catch (e) {
 			enhanceCompileError(e, code, options.preprocess);
@@ -137,8 +137,9 @@ export function createCompileSvelte() {
 			const hasCss = compiled.css?.code?.trim()?.length ?? 0 > 0;
 			// compiler might not emit css with mode none or it may be empty
 			if (emitCss && hasCss) {
-				// TODO properly update sourcemap?
-				compiled.js.code += `\nimport ${JSON.stringify(cssId)};\n`;
+				applyMapEdits(compiled, filename, (editor) => {
+					editor.append(`\nimport ${JSON.stringify(cssId)};\n`);
+				});
 			}
 		}
 

package/src/utils/map-edits.js

@@ -0,0 +1,76 @@
+import remapping from '@jridgewell/remapping';
+import { parse } from 'acorn';
+import MagicString from 'magic-string';
+
+/** @param {any} callee */
+function isHotAcceptCallee(callee) {
+	return (
+		callee.type === 'MemberExpression' &&
+		!callee.computed &&
+		callee.property.type === 'Identifier' &&
+		callee.property.name === 'accept' &&
+		callee.object.type === 'MemberExpression' &&
+		!callee.object.computed &&
+		callee.object.property.type === 'Identifier' &&
+		callee.object.property.name === 'hot' &&
+		callee.object.object.type === 'MetaProperty' &&
+		callee.object.object.meta.name === 'import' &&
+		callee.object.object.property.name === 'meta'
+	);
+}
+
+/**
+ * @param {any} node
+ * @param {any[]} calls
+ */
+function collectHotAcceptCalls(node, calls) {
+	if (!node || typeof node !== 'object') return;
+	if (node.type === 'CallExpression' && isHotAcceptCallee(node.callee)) {
+		calls.push(node);
+	}
+	for (const value of Object.values(node)) {
+		if (Array.isArray(value)) {
+			for (const child of value) collectHotAcceptCalls(child, calls);
+		} else if (value && typeof value === 'object' && typeof value.type === 'string') {
+			collectHotAcceptCalls(value, calls);
+		}
+	}
+}
+
+/**
+ * @param {{ js: { code: string, map: any } }} compiled
+ * @param {string} filename
+ * @param {(editor: MagicString, code: string) => void} edit
+ */
+export function applyMapEdits(compiled, filename, edit) {
+	const editor = new MagicString(compiled.js.code);
+	edit(editor, compiled.js.code);
+	const previousMap = compiled.js.map;
+	compiled.js.code = editor.toString();
+	if (!previousMap) return;
+	const editMap = editor.generateMap({ hires: true, source: filename });
+	let loadedPreviousMap = false;
+	compiled.js.map = remapping(/** @type {any} */ (editMap), (source) => {
+		if (source !== filename || loadedPreviousMap) return null;
+		loadedPreviousMap = true;
+		return /** @type {any} */ (previousMap);
+	});
+}
+
+/**
+ * @param {MagicString} editor
+ * @param {string} code
+ */
+export function addPartialAcceptExports(editor, code) {
+	const calls = [];
+	collectHotAcceptCalls(parse(code, { ecmaVersion: 'latest', sourceType: 'module' }), calls);
+	for (const call of calls) {
+		const firstArgument = call.arguments[0];
+		if (!firstArgument || firstArgument.type === 'SpreadElement') continue;
+		editor.overwrite(
+			call.callee.start,
+			firstArgument.start,
+			`${code.slice(call.callee.start, call.callee.end)}Exports([\"default\"],`
+		);
+	}
+}