From 3ff9ccd28964c1f5bd31741af06894c90d9d7916 Mon Sep 17 00:00:00 2001
From: GQ <tonny008@163.com>
Date: Sat, 25 Mar 2023 17:55:56 +0800
Subject: [PATCH 1/2] add extractStyleNode function to extract style as
 ReactNode.

---
 src/hooks/useStyleRegister.tsx | 19 +++++++++++++++++++
 src/index.ts                   |  3 ++-
 2 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/src/hooks/useStyleRegister.tsx b/src/hooks/useStyleRegister.tsx
index 0d709e11..dc9ec717 100644
--- a/src/hooks/useStyleRegister.tsx
+++ b/src/hooks/useStyleRegister.tsx
@@ -466,3 +466,22 @@ export function extractStyle(cache: Cache) {
 
   return styleText;
 }
+
+export function extractStyleNode(cache: Cache) {
+  const styleKeys = Array.from(cache.cache.keys()).filter((key) =>
+    key.startsWith('style%')
+  );
+
+  return styleKeys.map((key) => {
+    const [styleStr, tokenKey, styleId]: [string, string, string] =
+      cache.cache.get(key)![1];
+    return (
+      <style
+        key={styleId}
+        data-token-hash={tokenKey}
+        data-css-hash={styleId}
+        dangerouslySetInnerHTML={{ __html: styleStr }}
+      />
+    );
+  });
+}
\ No newline at end of file
diff --git a/src/index.ts b/src/index.ts
index 15ca1604..0fc1892c 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -1,6 +1,6 @@
 import useCacheToken from './hooks/useCacheToken';
 import type { CSSInterpolation, CSSObject } from './hooks/useStyleRegister';
-import useStyleRegister, { extractStyle } from './hooks/useStyleRegister';
+import useStyleRegister, { extractStyle, extractStyleNode } from './hooks/useStyleRegister';
 import Keyframes from './Keyframes';
 import type { Linter } from './linters';
 import {
@@ -24,6 +24,7 @@ export {
   StyleProvider,
   Keyframes,
   extractStyle,
+  extractStyleNode,
 
   // Transformer
   legacyLogicalPropertiesTransformer,

From b50a4d8b759d7b55810b481b36989d2702098d16 Mon Sep 17 00:00:00 2001
From: GQ <tonny008@163.com>
Date: Sat, 25 Mar 2023 18:17:27 +0800
Subject: [PATCH 2/2] use costants

---
 src/hooks/useStyleRegister.tsx | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/src/hooks/useStyleRegister.tsx b/src/hooks/useStyleRegister.tsx
index dc9ec717..3f1ffcad 100644
--- a/src/hooks/useStyleRegister.tsx
+++ b/src/hooks/useStyleRegister.tsx
@@ -475,13 +475,11 @@ export function extractStyleNode(cache: Cache) {
   return styleKeys.map((key) => {
     const [styleStr, tokenKey, styleId]: [string, string, string] =
       cache.cache.get(key)![1];
-    return (
-      <style
-        key={styleId}
-        data-token-hash={tokenKey}
-        data-css-hash={styleId}
-        dangerouslySetInnerHTML={{ __html: styleStr }}
-      />
-    );
+    const props = {
+      [ATTR_TOKEN]: tokenKey,
+      [ATTR_MARK]: styleId,
+      dangerouslySetInnerHTML: { __html: styleStr },
+    };
+    return <style key={styleId} {...props} />;
   });
 }
\ No newline at end of file