From 895d02a19c9ce100076f9ea95c205c8faad1948c Mon Sep 17 00:00:00 2001
From: Jeff Emmett <46964190+Jeff-Emmett@users.noreply.github.com>
Date: Sat, 19 Oct 2024 00:42:23 -0400
Subject: [PATCH] embeds work!

---
 src/components/Board.tsx      |  8 +++--
 src/shapes/EmbedShapeUtil.tsx | 59 +++++++++++++++++++++++++++++++++++
 src/tools/EmbedTool.ts        |  9 ++++++
 src/ui-overrides.tsx          | 12 +++++++
 worker/TldrawDurableObject.ts |  8 ++++-
 5 files changed, 92 insertions(+), 4 deletions(-)
 create mode 100644 src/shapes/EmbedShapeUtil.tsx
 create mode 100644 src/tools/EmbedTool.ts

diff --git a/src/components/Board.tsx b/src/components/Board.tsx
index 42d5427..30009c3 100644
--- a/src/components/Board.tsx
+++ b/src/components/Board.tsx
@@ -12,6 +12,8 @@ import { VideoChatTool } from '@/tools/VideoChatTool'
 import { VideoChatShape } from '@/shapes/VideoChatShapeUtil'
 import { multiplayerAssetStore } from '../client/multiplayerAssetStore'
 import { customSchema } from '../../worker/TldrawDurableObject'
+import { EmbedShape } from '@/shapes/EmbedShapeUtil'
+import { EmbedTool } from '@/tools/EmbedTool'
 
 import React, { useState } from 'react';
 import { ChatBox } from '@/shapes/ChatBoxShapeUtil';
@@ -19,8 +21,8 @@ import { components, uiOverrides } from '@/ui-overrides'
 
 const WORKER_URL = `https://jeffemmett-canvas.jeffemmett.workers.dev`
 
-const shapeUtils = [ChatBoxShape, VideoChatShape]
-const tools = [ChatBoxTool, VideoChatTool]; // Array of tools
+const shapeUtils = [ChatBoxShape, VideoChatShape, EmbedShape]
+const tools = [ChatBoxTool, VideoChatTool, EmbedTool]; // Array of tools
 
 export function Board() {
 	const { slug } = useParams<{ slug: string }>(); // Ensure this is inside the Board component
@@ -107,4 +109,4 @@ async function unfurlBookmarkUrl({ url }: { url: string }): Promise<TLBookmarkAs
 	}
 
 	return asset
-}
\ No newline at end of file
+}
diff --git a/src/shapes/EmbedShapeUtil.tsx b/src/shapes/EmbedShapeUtil.tsx
new file mode 100644
index 0000000..6782bdb
--- /dev/null
+++ b/src/shapes/EmbedShapeUtil.tsx
@@ -0,0 +1,59 @@
+import { BaseBoxShapeUtil, TLBaseShape } from "tldraw";
+import { useCallback, useState } from "react";
+
+export type IEmbedShape = TLBaseShape<
+    'Embed',
+    {
+        w: number;
+        h: number;
+        url: string | null;
+    }
+>;
+
+export class EmbedShape extends BaseBoxShapeUtil<IEmbedShape> {
+    static override type = 'Embed';
+
+    getDefaultProps(): IEmbedShape['props'] {
+        return {
+            url: null,
+            w: 640,
+            h: 480,
+        };
+    }
+
+    indicator(shape: IEmbedShape) {
+        return <rect x={0} y={0} width={shape.props.w} height={shape.props.h} />;
+    }
+
+    component(shape: IEmbedShape) {
+        const [inputUrl, setInputUrl] = useState(shape.props.url || '');
+
+        const handleSubmit = useCallback((e: React.FormEvent) => {
+            e.preventDefault();
+            this.editor.updateShape<IEmbedShape>({ id: shape.id, type: 'Embed', props: { ...shape.props, url: inputUrl } });
+        }, [inputUrl]);
+
+        if (!shape.props.url) {
+            return (
+                <div style={{ pointerEvents: 'all' }}>
+                    <form onSubmit={handleSubmit}>
+                        <input
+                            type="text"
+                            value={inputUrl}
+                            onChange={(e) => setInputUrl(e.target.value)}
+                            placeholder="Enter URL"
+                            style={{ width: shape.props.w, height: shape.props.h }}
+                        />
+                        <button type="submit">Load</button>
+                    </form>
+                </div>
+            );
+        }
+
+        return (
+            <div style={{ pointerEvents: 'all' }}>
+                <iframe src={shape.props.url} width={shape.props.w} height={shape.props.h} />
+            </div>
+        );
+    }
+}
diff --git a/src/tools/EmbedTool.ts b/src/tools/EmbedTool.ts
new file mode 100644
index 0000000..f5b2d69
--- /dev/null
+++ b/src/tools/EmbedTool.ts
@@ -0,0 +1,9 @@
+import { BaseBoxShapeTool } from "tldraw";
+
+export class EmbedTool extends BaseBoxShapeTool {
+    static override id = 'Embed'
+    shapeType = 'Embed';
+    override initial = 'idle';
+
+    // Additional methods for handling video chat functionality can be added here
+}
\ No newline at end of file
diff --git a/src/ui-overrides.tsx b/src/ui-overrides.tsx
index 6986f7d..f00e549 100644
--- a/src/ui-overrides.tsx
+++ b/src/ui-overrides.tsx
@@ -8,6 +8,7 @@ import {
 	useTools,
 } from 'tldraw'
 import { CustomMainMenu } from './components/CustomMainMenu'
+import { EmbedShape } from './shapes/EmbedShapeUtil'
 
 export const uiOverrides: TLUiOverrides = {
 	tools(editor, tools) {
@@ -29,6 +30,15 @@ export const uiOverrides: TLUiOverrides = {
 				editor.setCurrentTool('ChatBox')
 			},
 		}
+		tools.Embed = {
+			id: 'Embed',
+			icon: 'embed',
+			label: 'Embed',
+			kbd: 'e',
+			onSelect: () => {
+				editor.setCurrentTool('Embed')
+			},
+		}
 		return tools
 	},
 }
@@ -38,10 +48,12 @@ export const components: TLComponents = {
 		const tools = useTools()
 		const isChatBoxSelected = useIsToolSelected(tools['ChatBox'])
 		const isVideoSelected = useIsToolSelected(tools['VideoChat'])
+		const isEmbedSelected = useIsToolSelected(tools['Embed'])
 		return (
 			<DefaultToolbar {...props}>
 				<TldrawUiMenuItem {...tools['VideoChat']} isSelected={isVideoSelected} />
 				<TldrawUiMenuItem {...tools['ChatBox']} isSelected={isChatBoxSelected} />
+				<TldrawUiMenuItem {...tools['Embed']} isSelected={isEmbedSelected} />
 				<DefaultToolbarContent />
 			</DefaultToolbar>
 		)
diff --git a/worker/TldrawDurableObject.ts b/worker/TldrawDurableObject.ts
index e76933c..b916883 100644
--- a/worker/TldrawDurableObject.ts
+++ b/worker/TldrawDurableObject.ts
@@ -13,10 +13,16 @@ import throttle from 'lodash.throttle'
 import { Environment } from './types'
 import { ChatBoxShape } from '@/shapes/ChatBoxShapeUtil'
 import { VideoChatShape } from '@/shapes/VideoChatShapeUtil'
+import { EmbedShape } from '@/shapes/EmbedShapeUtil'
 
 // add custom shapes and bindings here if needed:
 export const customSchema = createTLSchema({
-	shapes: { ...defaultShapeSchemas, ChatBox: ChatBoxShape, VideoChat: VideoChatShape },
+	shapes: {
+		...defaultShapeSchemas,
+		ChatBox: ChatBoxShape,
+		VideoChat: VideoChatShape,
+		Embed: EmbedShape
+	},
 	// bindings: { ...defaultBindingSchemas },
 })