update vite project

Author: wkozyra95

demos/vite-demo/src/components/SmelterCanvas.tsx

@@ -0,0 +1,95 @@
+import React, { useCallback, useEffect, useRef, useState } from 'react';
+import type Smelter from '@swmansion/smelter-web-wasm';
+
+type CanvasProps = React.DetailedHTMLProps<
+  React.CanvasHTMLAttributes<HTMLCanvasElement>,
+  HTMLCanvasElement
+>;
+
+type SmelterCanvasProps = {
+  smelter: Smelter;
+  audio?: boolean;
+  children: React.ReactElement;
+} & CanvasProps;
+
+type UpdatedProps = {
+  smelter: Smelter;
+  width?: number | string;
+  height?: number | string;
+  audio?: boolean;
+};
+
+type RegisterOptions = UpdatedProps & {
+  outputId: string;
+  canvas: HTMLCanvasElement;
+};
+
+const getNewOutputId = (() => {
+  let counter = 1;
+  return () => {
+    const outputId = `output-${counter}`;
+    counter += 1;
+    return outputId;
+  };
+})();
+
+export default function SmelterCanvasOutput(props: SmelterCanvasProps) {
+  const { children, smelter, audio, ...canvasProps } = props;
+  const [updatedProps, setUpdatedProps] = useState<UpdatedProps | undefined>();
+  const [registerOptions, setRegisterOptions] = useState<RegisterOptions | undefined>(undefined);
+  const key = useRef<number>(1);
+
+  const canvasRef = useCallback(
+    async (updatedCanvas: HTMLCanvasElement | null) => {
+      if (!updatedCanvas || !updatedProps || updatedCanvas === registerOptions?.canvas) {
+        return;
+      }
+      const outputId = getNewOutputId();
+      setRegisterOptions({
+        ...updatedProps,
+        outputId,
+        canvas: updatedCanvas,
+      });
+    },
+    [updatedProps, registerOptions]
+  );
+
+  useEffect(() => {
+    // force new canvas
+    key.current += 1;
+
+    setUpdatedProps({
+      smelter,
+      width: canvasProps.width,
+      height: canvasProps.height,
+      audio,
+    });
+  }, [canvasProps.width, canvasProps.height, smelter, audio]);
+
+  useEffect(() => {
+    if (!registerOptions) {
+      return;
+    }
+    const { audio, outputId, width, height, canvas } = registerOptions;
+    const promise = smelter.registerOutput(outputId, children, {
+      type: 'canvas',
+      video: {
+        canvas,
+        resolution: {
+          width: Number(width ?? canvas?.width),
+          height: Number(height ?? canvas?.height),
+        },
+      },
+      audio,
+    });
+
+    return () => {
+      void promise
+        .catch(console.error)
+        .then(() => smelter.unregisterOutput(outputId))
+        .catch(console.error);
+    };
+  }, [registerOptions]);
+
+  return <canvas key={key.current} ref={canvasRef} {...canvasProps} />;
+}

demos/vite-demo/src/components/SmelterCanvasOutput.tsx

@@ -0,0 +1,27 @@
+import Smelter from '@swmansion/smelter-web-wasm';
+import { useEffect, useState } from 'react';
+
+export function useSmelter(): Smelter | undefined {
+  const [smelter, setSmelter] = useState<Smelter>();
+  useEffect(() => {
+    const smelter = new Smelter();
+
+    let cancel = false;
+    const promise = (async () => {
+      await smelter.init();
+      await smelter.start();
+      if (!cancel) {
+        setSmelter(smelter);
+      }
+    })();
+
+    return () => {
+      cancel = true;
+      void (async () => {
+        await promise.catch(() => {});
+        await smelter.terminate();
+      })();
+    };
+  }, []);
+  return smelter;
+}

demos/vite-demo/src/hooks/useSmelter.ts

@@ -1,15 +1,17 @@
 import { View, Text, useInputStreams, InputStream, Tiles } from '@swmansion/smelter';
 import Smelter, { setWasmBundleUrl } from '@swmansion/smelter-web-wasm';
 import { useCallback, useState } from 'react';
-import SmelterCanvas from '../components/SmelterCanvas';
+import SmelterCanvas from '../components/SmelterCanvasOutput';
+import { useSmelter } from '../hooks/useSmelter';
+import SmelterCanvasOutput from '../components/SmelterCanvasOutput';
 
 setWasmBundleUrl('/assets/smelter.wasm');
 
 const CAMERA_ID = 'camera';
 const SCREEN_SHARE_ID = 'screen';
 
 export default function CanvasPage() {
-  const [smelter, setSmelter] = useState<Smelter>();
+  const smelter = useSmelter();
   const [camera, setCamera] = useState<boolean>();
   const [screen, setScreen] = useState<boolean>();
 
@@ -63,9 +65,9 @@ export default function CanvasPage() {
         <p className="">Canvas: </p>
       </div>
       <div>
-        <SmelterCanvas width={1280} height={720} onSmelterCreated={setSmelter}>
+        {smelter && <SmelterCanvasOutput smelter={smelter} width={1280} height={720} audio>
           <SmelterComponent />
-        </SmelterCanvas>
+        </SmelterCanvasOutput>}
       </div>
     </div>
   );