Skip to content

Commit e02c72b

Browse files
committed
fix: Fix ref issue.
1 parent c88a9c0 commit e02c72b

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/index.tsx

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
import React, { useMemo, useRef, useState } from 'react';
1+
import React, { useMemo, useRef, useState, useImperativeHandle } from 'react';
22
import ReactDOM from 'react-dom';
33

44
export interface RunWebProps extends React.IframeHTMLAttributes<HTMLIFrameElement> {
@@ -13,6 +13,9 @@ export default React.forwardRef<HTMLIFrameElement, RunWebProps>((props, ref) =>
1313
const { html = '', css, js, title = 'Demo Title', head, onLoad, ...other } = props;
1414
const [isLoaded, setIsLoaded] = useState(false);
1515
const frameRef = useRef<HTMLIFrameElement>(null);
16+
17+
useImperativeHandle(ref, () => frameRef.current as HTMLIFrameElement, [frameRef]);
18+
1619
const srcDoc = useMemo(() => {
1720
setIsLoaded(false);
1821
const jsString = js ? `<script type="text/javascript">${js}</script>` : '';

0 commit comments

Comments
 (0)