add useImperativeMethods

bitovi · Nov 20, 2023 · 8dfe6d0 · 8dfe6d0
1 parent 6a0cb93
commit 8dfe6d0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/packages/core/src/core.ts b/packages/core/src/core.ts
@@ -179,3 +179,15 @@ export default function r2wc<Props extends R2WCBaseProps, Context>(
 
   return ReactWebComponent
 }
+
+export function useImperativeMethods<Methods extends string>(
+  container: HTMLElement | ShadowRoot,
+  methods: Record<Methods, () => unknown>,
+): void {
+  const element = "host" in container ? container.host : container
+
+  for (const method in methods) {
+    //@ts-ignore
+    element[method] = methods[method]
+  }
+}
diff --git a/packages/react-to-web-component/src/react-to-web-component.ts b/packages/react-to-web-component/src/react-to-web-component.ts
@@ -6,6 +6,8 @@ import { createRoot } from "react-dom/client"
 
 import r2wcCore from "@r2wc/core"
 
+export { useImperativeMethods } from "@r2wc/core"
+
 interface Context<Props extends R2WCBaseProps> {
   root: Root
   ReactComponent: React.ComponentType<Props>