Fix createUseStyle() sheets not rendered during SSR by calling manageSheet() during useReducer initialisation instead of in useEffect

Author: felthy

packages/react-jss/.size-snapshot.json

@@ -1,23 +1,23 @@
 {
   "dist/react-jss.js": {
-    "bundled": 167552,
-    "minified": 58003,
-    "gzipped": 18880
+    "bundled": 142107,
+    "minified": 51017,
+    "gzipped": 16807
   },
   "dist/react-jss.min.js": {
-    "bundled": 110827,
-    "minified": 41357,
-    "gzipped": 13942
+    "bundled": 108586,
+    "minified": 40704,
+    "gzipped": 13875
   },
   "dist/react-jss.cjs.js": {
-    "bundled": 26174,
-    "minified": 11471,
-    "gzipped": 3751
+    "bundled": 26052,
+    "minified": 11426,
+    "gzipped": 3739
   },
   "dist/react-jss.esm.js": {
-    "bundled": 25212,
-    "minified": 10636,
-    "gzipped": 3624,
+    "bundled": 25090,
+    "minified": 10591,
+    "gzipped": 3612,
     "treeshaked": {
       "rollup": {
         "code": 1946,

packages/react-jss/src/createUseStyles.js

@@ -55,11 +55,14 @@ const createUseStyles = <Theme: {}>(styles: Styles<Theme>, options?: HookOptions
       let dynamicRules
       let classes
       if (sheet) {
-        if (context.registry) {
-          context.registry.add(sheet)
-        }
         dynamicRules = addDynamicRules(sheet, data)
         classes = getSheetClasses(sheet, dynamicRules)
+        manageSheet({
+          index,
+          context,
+          sheet,
+          theme
+        })
       }
 
       return {
@@ -70,31 +73,20 @@ const createUseStyles = <Theme: {}>(styles: Styles<Theme>, options?: HookOptions
     })
 
     useEffectOrLayoutEffect(
-      () => {
-        if (state.sheet) {
-          manageSheet({
-            index,
-            context,
-            sheet: state.sheet,
-            theme
-          })
-        }
-
-        return () => {
-          const {sheet, dynamicRules} = state
+      () => () => {
+        const {sheet, dynamicRules} = state
 
-          if (!sheet) return
+        if (!sheet) return
 
-          unmanageSheet({
-            index,
-            context,
-            sheet,
-            theme
-          })
+        unmanageSheet({
+          index,
+          context,
+          sheet,
+          theme
+        })
 
-          if (dynamicRules) {
-            removeDynamicRules(sheet, dynamicRules)
-          }
+        if (dynamicRules) {
+          removeDynamicRules(sheet, dynamicRules)
         }
       },
       [state.sheet]