refactor: restore hmr for lazy styles after refactor

webpack-contrib · Aug 5, 2019 · adb9c31 · adb9c31
1 parent 3824f83
commit adb9c31
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/src/index.js b/src/index.js
@@ -66,7 +66,8 @@ if (module.hot) {
   var lastRefs = module.hot.data && module.hot.data.refs || 0;
   
   if (lastRefs) {
-    exports.ref();
+    exports.use();
+
     if (!content.locals) {
       refs = lastRefs;
     }
@@ -94,8 +95,13 @@ var options = ${JSON.stringify(options)};
 options.insert = ${insert};
 options.singleton = ${isSingleton};
 
-if (typeof content === 'string') content = [[module.id, content, '']];
-if (content.locals) exports.locals = content.locals;
+if (typeof content === 'string') {
+  content = [[module.id, content, '']];
+}
+
+if (content.locals) {
+  exports.locals = content.locals;
+}
 
 exports.use = function() {
   if (!(refs++)) {
@@ -171,8 +177,6 @@ if (module.hot) {
 
 if (typeof content === 'string') content = [[module.id, content, '']];
 
-var insertInto;
-
 var options = ${JSON.stringify(options)}
 
 options.insert = ${insert};