diff --git a/packages/moon-mvl/lib/addCSS.js b/packages/moon-mvl/lib/addCSS.js index 7cb31c11..8b137891 100644 --- a/packages/moon-mvl/lib/addCSS.js +++ b/packages/moon-mvl/lib/addCSS.js @@ -1,10 +1 @@ -module.exports = (code) => { - const head = document.querySelector("head"); - const style = document.createElement("style"); - style.appendChild(document.createTextNode(code)); - head.appendChild(style); - return () => { - head.removeChild(style); - }; -}; diff --git a/packages/moon-mvl/lib/hot.js b/packages/moon-mvl/lib/hot.js index d3a4f90c..c90bb6b6 100644 --- a/packages/moon-mvl/lib/hot.js +++ b/packages/moon-mvl/lib/hot.js @@ -1,13 +1,16 @@ -let components = []; - -module.exports.register = (component) => { - components.push(component); +module.exports.registerJS = (component) => { + return () => { + component.destroy(); + }; }; -module.exports.remove = () => { - for (let i = 0; i < components.length; i++) { - components[i].destroy(); - } +module.exports.registerCSS = (code) => { + const head = document.querySelector("head"); + const style = document.createElement("style"); + style.appendChild(document.createTextNode(code)); + head.appendChild(style); - components = []; + return () => { + head.removeChild(style); + }; };