native-document 1.0.55 → 1.0.56

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "native-document",
-  "version": "1.0.55",
+  "version": "1.0.56",
   "main": "index.js",
   "type": "module",
   "scripts": {
@@ -12,10 +12,13 @@
   "license": "ISC",
   "description": "",
   "devDependencies": {
+    "@babel/parser": "^7.28.5",
+    "@babel/traverse": "^7.28.5",
     "@rollup/plugin-alias": "^5.1.1",
     "@rollup/plugin-replace": "^6.0.2",
     "@rollup/plugin-terser": "^0.4.4",
     "eslint": "^9.33.0",
+    "magic-string": "^0.30.21",
     "rollup": "^4.53.3"
   }
-}
+}

package/src/hrm/transformComponent.js

@@ -4,14 +4,55 @@ import path from 'node:path';
 import MagicString from 'magic-string';
 
 import { fileURLToPath } from 'node:url';
+import { parse } from '@babel/parser';
+import { default as traverse } from '@babel/traverse';
 
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
 
+
 export default function transformComponent(id, code) {
+    let hasDefaultExport = false;
+    let componentName = null;
+    let exportStart = 0;
+    let exportEnd = 0;
     // TODO: move this line outside the function
     const hrmHookTemplate = fs.readFileSync(__dirname + '/hrm.hook.template.js', 'utf8');
     const s = new MagicString('');
+    const codeParsed = parse(code, {
+        sourceType: 'module',
+        plugins: []
+    });
+
+    console.log(traverse)
+    traverse(codeParsed, {
+        ExportDefaultDeclaration(path) {
+            hasDefaultExport = true;
+            const declaration = path.node.declaration;
+
+            if (declaration.id) {
+                componentName = declaration.id.name;
+            } else if (declaration.type === 'Identifier') {
+                componentName = declaration.name;
+            } else {
+                componentName = 'AnonymousComponent';
+            }
+
+            exportStart = path.node.start;
+            exportEnd = path.node.end;
+        }
+    });
+    if (!hasDefaultExport) {
+        return null;
+    }
+
+    const originalExport = code.slice(exportStart, exportEnd);
+    const hrmComponentName = `__HRM_${componentName}__`;
+    const newExport = originalExport.replace(
+        'export default',
+        `const ${hrmComponentName} =`
+    );
+    s.overwrite(exportStart, exportEnd, newExport);
 
     const data = {
         id: id,
@@ -23,6 +64,7 @@ export default function transformComponent(id, code) {
 
     s.append('import ComponentRegistry from "native-document/src/hrm/ComponentRegistry";');
     s.append(code);
+    s.append(`export default ComponentRegistry.register('${id}', ${hrmComponentName}, { preserveState: ${options.preserveState} });`);
     s.append(hrmHookTemplateFormatted);
 
     return {