From dcadf5daadbff0f3ee775a166f27b473b27b3824 Mon Sep 17 00:00:00 2001
From: Shu Ding <g@shud.in>
Date: Wed, 15 Sep 2021 15:07:37 +0200
Subject: [PATCH] fix: Make `JSXStyle` return a noop if the registry context is
 not provided (#749)

---
 src/style.js | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/src/style.js b/src/style.js
index 0a4f916d..5896f79a 100644
--- a/src/style.js
+++ b/src/style.js
@@ -1,12 +1,19 @@
-import { useLayoutEffect, useContext } from 'react'
-import { StyleSheetContext } from './stylesheet-registry'
+import { useLayoutEffect } from 'react'
+import { useStyleRegistry } from './stylesheet-registry'
 import { computeId } from './lib/hash'
 export default function JSXStyle(props) {
-  const registry = useContext(StyleSheetContext)
+  const registry = useStyleRegistry()
+
+  // If `registry` does not exist, we do nothing here.
+  if (!registry) {
+    return null
+  }
+
   if (typeof window === 'undefined') {
     registry.add(props)
     return null
   }
+
   useLayoutEffect(() => {
     registry.add(props)
     return () => {
@@ -14,6 +21,7 @@ export default function JSXStyle(props) {
     }
     // props.children can be string[], will be striped since id is identical
   }, [props.id, String(props.dynamic)])
+
   return null
 }