frontend/src/lib/PaintFrame.svelte

<script lang="ts">
	import PPButton from '$lib/Buttons/PPButton.svelte';
	import DragButton from '$lib/Buttons/DragButton.svelte';
	import MaskButton from '$lib/Buttons/MaskButton.svelte';
	import UndoButton from '$lib/Buttons/UndoButton.svelte';
	import LoadingIcon from '$lib/Icons/LoadingIcon.svelte';

	import { drag } from 'd3-drag';
	import { select, type Selection } from 'd3-selection';
	import { round } from '$lib/utils';

	import type { ZoomTransform } from 'd3-zoom';
	import { onMount, createEventDispatcher } from 'svelte';

	import { useMyPresence } from '$lib/liveblocks';
	import { loadingState, canvasEl, maskEl } from '$lib/store';

	import { toggle_class } from 'svelte/internal';
	const myPresence = useMyPresence();

	const dispatch = createEventDispatcher();

	export let transform: ZoomTransform;
	export let color = 'black';
	export let interactive = false;

	let maskCtx: CanvasRenderingContext2D;

	let position = {
		x: 768,
		y: 768
	};

	let frameElement: HTMLDivElement;
	let dragEnabled = true;
	let isDragging = false;
	$: prompt = $myPresence?.currentPrompt;
	$: isLoading = $myPresence?.isLoading || false;

	$: coord = {
		x: transform.applyX(position.x),
		y: transform.applyY(position.y)
	};

	let offsetX = 0;
	let offsetY = 0;

	function cropCanvas(cursor: { x: number; y: number }) {
		maskCtx.save();
		maskCtx.clearRect(0, 0, 512, 512);
		maskCtx.globalCompositeOperation = 'source-over';
		maskCtx.drawImage($canvasEl, cursor.x, cursor.y, 512, 512, 0, 0, 512, 512);
		maskCtx.restore();
	}
	function drawCircle(cursor: { x: number; y: number }) {
		maskCtx.save();
		maskCtx.globalCompositeOperation = 'destination-out';
		maskCtx.beginPath();
		maskCtx.arc(cursor.x, cursor.y, 20, 0, 2 * Math.PI);
		maskCtx.fill();
		maskCtx.restore();
	}

	onMount(() => {
		maskCtx = $maskEl.getContext('2d') as CanvasRenderingContext2D;

		select(frameElement)
			.call(dragMoveHandler() as any)
			.call(cursorUpdate);
		select($maskEl).call(maskingHandler() as any);
	});

	function cursorUpdate(selection: Selection) {
		function handlePointerMove(event: PointerEvent) {
			myPresence.update({
				cursor: {
					x: transform.invertX(event.clientX),
					y: transform.invertY(event.clientY)
				}
			});
		}
		function handlePointerLeave() {
			myPresence.update({
				cursor: null
			});
		}
		return selection.on('pointermove', handlePointerMove).on('pointerleave', handlePointerLeave);
	}
	function maskingHandler() {
		function dragstarted(event: Event) {
			const x = event.x / transform.k;
			const y = event.y / transform.k;
		}

		function dragged(event: Event) {
			const x = event.x / transform.k;
			const y = event.y / transform.k;
			drawCircle({ x, y });
		}

		function dragended(event: Event) {
			const x = event.x / transform.k;
			const y = event.y / transform.k;
		}
		return drag().on('start', dragstarted).on('drag', dragged).on('end', dragended);
	}
	function dragMoveHandler() {
		function dragstarted(event: Event) {
			const rect = (event.sourceEvent.target as HTMLElement).getBoundingClientRect();
			if (event.sourceEvent instanceof TouchEvent) {
				offsetX = event.sourceEvent.targetTouches[0].pageX - rect.left;
				offsetY = event.sourceEvent.targetTouches[0].pageY - rect.top;
			} else {
				offsetX = event.sourceEvent.pageX - rect.left;
				offsetY = event.sourceEvent.pageY - rect.top;
			}
			isDragging = true;
		}

		function dragged(event: Event) {
			const x = round(transform.invertX(event.x - offsetX));
			const y = round(transform.invertY(event.y - offsetY));
			position = {
				x,
				y
			};
			myPresence.update({
				cursor: {
					x: transform.invertX(event.x),
					y: transform.invertY(event.y)
				}
			});
			cropCanvas({ x, y });
		}

		function dragended(event: Event) {
			isDragging = false;

			const x = round(transform.invertX(event.x - offsetX));
			const y = round(transform.invertY(event.y - offsetY));
			cropCanvas({ x, y });

			myPresence.update({
				frame: {
					x,
					y
				}
			});
		}
		return drag().on('start', dragstarted).on('drag', dragged).on('end', dragended);
	}
	function toggleDrag() {
		dragEnabled = true;
	}
	function toggleMask() {
		dragEnabled = false;
		cropCanvas(position);
	}
</script>

<div>
	<div
		class="absolute top-0 left-0"
		style={`transform: translateX(${coord.x}px) translateY(${coord.y}px) scale(${transform.k}); border-color: ${color}; transform-origin: 0 0;`}
	>
		<div class="frame">
			<canvas
				class={dragEnabled || isLoading ? '' : 'bg-white'}
				bind:this={$maskEl}
				width="512"
				height="512"
			/>
			<div class="pointer-events-none touch-none">
				{#if $loadingState}
					<div class="col-span-2 row-start-1">
						<span class="text-white drop-shadow-lg">{$loadingState}</span>
					</div>
				{/if}
				{#if isLoading}
					<div class="col-start-2 row-start-2">
						<LoadingIcon />
					</div>
				{/if}

				<h2 class="text-lg">Click to paint</h2>
				<div class="absolute bottom-0 font-bold text-lg">{prompt}</div>
			</div>
			{#if !isDragging}
				<div class="absolute top-full ">
					<div class="py-2">
						<PPButton on:click={() => dispatch('paintMode', { mode: 'paint' })} />
					</div>
				</div>
				<div class="absolute left-full bottom-0">
					<div class="px-2">
						<DragButton isActive={dragEnabled} on:click={toggleDrag} />
						<div class="flex bg-white rounded-full mt-3">
							<MaskButton isActive={!dragEnabled} on:click={toggleMask} />
							{#if !dragEnabled}
								<span class="border-gray-800 border-opacity-50 border-r-2 my-2" />
								<UndoButton on:click={() => {}} />
							{/if}
						</div>
					</div>
				</div>
			{/if}
		</div>
	</div>
	<div
		bind:this={frameElement}
		class="absolute top-0 left-0 w-[512px] h-[512px] ring-2 ring-black
		{isDragging ? 'cursor-grabbing' : 'cursor-grab'}
		{dragEnabled ? 'block' : 'hidden'}"
		style={`transform: translateX(${coord.x}px) translateY(${coord.y}px) scale(${transform.k}); transform-origin: 0 0;`}
	/>
</div>

<style lang="postcss" scoped>
	.frame {
		@apply relative grid grid-cols-3 grid-rows-3 ring-2 ring-blue-500 w-[512px] h-[512px];
	}
</style>