fix(types): add types for ref (#12)

jeetiss · Mar 30, 2021 · da02f23 · da02f23
1 parent af3e08a
commit da02f23
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/index.d.ts b/index.d.ts
@@ -1,4 +1,4 @@
-import { FC, CSSProperties } from "react";
+import { FC, ForwardedRef, CSSProperties } from "react";
 
 type State = {
   width?: number;
@@ -47,6 +47,7 @@ declare const Canvas: FC<{
   frames: ImageData[];
   width?: number;
   height?: number;
+  ref?: ForwardedRef<HTMLCanvasElement>;
   fit?: "cover" | "contain" | "fill";
   className?: string;
   style?: CSSProperties;

diff --git a/test-d/index.tsx b/test-d/index.tsx
@@ -94,5 +94,16 @@ expectError(usePlayback(pstate, update));
 
 expectType(<Canvas {...pstate} />);
 
+const ref = React.createRef<HTMLCanvasElement>();
+expectType(<Canvas {...pstate} ref={ref} />);
+expectType(
+  <Canvas
+    {...pstate}
+    ref={(ref) => {
+      expectType<HTMLCanvasElement | null>(ref);
+    }}
+  />
+);
+
 useWorkerParser("str", update);
 useParser("str", update);