types: improve type definitions

just214 · Mar 9, 2022 · d6893fd · d6893fd
1 parent 7024ccb
commit d6893fd
Showing 1 changed file with 13 additions and 9 deletions.
diff --git a/src/react-hook-youtube.tsx b/src/react-hook-youtube.tsx
@@ -11,16 +11,20 @@ declare global {
 export function useYoutubeIframe(options: YT.PlayerOptions) {
   const playerRef = React.useRef<any>();
 
+  type YoutubePlayerProps = {
+    className?: string;
+    style?: React.CSSProperties;
+  };
+
   // COMPONENT
   const YoutubePlayer = React.useMemo(() => {
-    return (props: { className?: string; style?: any }) =>
-      options.videoId ? (
-        <div
-          className={props.className || ""}
-          style={props.style}
-          id={`youtube-player-${options.videoId}`}
-        />
-      ) : null;
+    return (props: YoutubePlayerProps) => (
+      <div
+        className={props.className || ""}
+        style={props.style}
+        id={`youtube-player-${options.videoId}`}
+      />
+    );
   }, [options.videoId]);
 
   /**
@@ -295,5 +299,5 @@ export function useYoutubeIframe(options: YT.PlayerOptions) {
       getIframe,
       destroy,
     },
-  };
+  } as { YoutubePlayer: React.FC<YoutubePlayerProps>; player: YT.Player };
 }