ember-vite-hmr 1.0.0 → 1.1.0

package/dist/lib/babel-plugin.js

@@ -24,6 +24,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.hotAstProcessor = void 0;
+const core_1 = require("@babel/core");
 const glimmer = __importStar(require("@glimmer/syntax"));
 const babel_import_util_1 = require("babel-import-util");
 const builtInComponents = ['LinkTo'];
@@ -209,6 +210,7 @@ function hotReplaceAst({ types: t }) {
         },
         visitor: {
             Program(path, state) {
+                var _a, _b;
                 if (!exports.hotAstProcessor.meta.importVar) {
                     return;
                 }
@@ -226,6 +228,23 @@ function hotReplaceAst({ types: t }) {
                 const varDeclaration = path.node.body.findIndex((e) => e.type === 'VariableDeclaration' && e.declarations[0].id.name === exports.hotAstProcessor.meta.importVar) + 1;
                 const lastImportIndex = path.node.body.findLastIndex((e) => e.type === 'ImportDeclaration') + 1;
                 path.node.body.splice(Math.max(varDeclaration, lastImportIndex), 0, assign);
+                const findImport = function findImport(specifier) {
+                    return path.node.body.find(b => b.type === 'ImportDeclaration' && b.specifiers.some(s => s.local.name === specifier));
+                };
+                const hotAccepts = [];
+                for (const imp of bindings) {
+                    const importDeclaration = findImport(imp);
+                    const source = importDeclaration.source.value;
+                    const specifier = importDeclaration.specifiers.find((s) => s.local.name === imp);
+                    const specifierName = ((_a = specifier.imported) === null || _a === void 0 ? void 0 : _a.name) ||
+                        ((_b = specifier.imported) === null || _b === void 0 ? void 0 : _b.value) ||
+                        'default';
+                    const ast = (0, core_1.parse)(`import.meta.hot.accept('${source}', (module) => (${exports.hotAstProcessor.meta.importVar}.${imp}=module['${specifierName}']))`);
+                    const impHot = ast === null || ast === void 0 ? void 0 : ast.program.body[0];
+                    hotAccepts.push(impHot);
+                }
+                const ifHot = t.ifStatement(t.memberExpression(t.metaProperty(t.identifier('import'), t.identifier('meta')), t.identifier('hot')), t.blockStatement([...hotAccepts]));
+                path.node.body.push(ifHot);
             }
         },
     };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ember-vite-hmr",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "keywords": [
     "ember-addon"
   ],