From af147ab0f31b634adaafa4f70316e009329c7f76 Mon Sep 17 00:00:00 2001
From: p-sw <shinwoo.park@psw.kr>
Date: Sun, 2 Jun 2024 08:35:03 +0900
Subject: [PATCH] fix: add forwardRef in overrideComponents

---
 packages/react/src/App.tsx | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/packages/react/src/App.tsx b/packages/react/src/App.tsx
index d039fce..288aed2 100644
--- a/packages/react/src/App.tsx
+++ b/packages/react/src/App.tsx
@@ -27,20 +27,24 @@ import ComponentsCheckbox, {
 import ComponentsDialog, {
   tableOfContents as componentsDialogToc,
 } from "./docs/components/Dialog.mdx";
+import { forwardRef } from "react";
 
 const overrideComponents = {
-  pre: (props: any) => <pre {...props} className={`${props.className} hljs`} />,
-  code: (props: any) => (
+  pre: forwardRef<HTMLPreElement, any>((props: any, ref) => (
+    <pre ref={ref} {...props} className={`${props.className} hljs`} />
+  )),
+  code: forwardRef<HTMLElement, any>((props: any, ref) => (
     <code
+      ref={ref}
       {...props}
       className={`${props.className} rounded-md bg-neutral-800 text-orange-500 font-light p-1 before:content-none after:content-none`}
     />
-  ),
-  table: (props: any) => (
+  )),
+  table: forwardRef<HTMLTableElement, any>((props: any, ref) => (
     <div className="overflow-auto">
-      <table {...props} className={`${props.className}`} />
+      <table ref={ref} {...props} className={`${props.className}`} />
     </div>
-  ),
+  )),
 };
 
 const router = createBrowserRouter(