import React, { useEffect, useState, useRef, memo } from "react";
import makeConfig from "./utils/config";

/**
 * @typedef {object} Colour
 * @property {"hsl"|"rgb"} space
 * @property {number[]} values   // 3-tuple [0-1] or [0-360,0-1,0-1]
 */

/**
 * Complete runtime configuration for the Matrix / Digital-Rain component.
 *
 * @typedef {{
 *   /* ------------- core identity ------------- * /
 *   version?: (
 *     "classic" | "megacity" | "neomatrixology" | "operator" |
 *     "nightmare" | "paradise" | "resurrections" | "trinity" |
 *     "morpheus" | "bugs" | "palimpsest" | "twilight" |
 *     "3d" | "throwback" | "updated" |
 *     "1999" | "2003" | "2021" | string            /* custom * /
 *   ),
 *   font?: keyof typeof fonts,                     // "matrixcode", …
 *   effect?: "palette" | "stripe" | string,
 *
 *   /* ------------- texture assets ------------- * /
 *   baseTexture?:  keyof typeof textureURLs | null,
 *   glintTexture?: keyof typeof textureURLs | null,
 *
 *   /* ------------- global toggles ------------- * /
 *   useCamera?: boolean,
 *   volumetric?: boolean,
 *   loops?: boolean,
 *   skipIntro?: boolean,
 *   renderer?: "regl" | "three" | string,
 *   suppressWarnings?: boolean,
 *   useHalfFloat?: boolean,
 *   isometric?: boolean,
 *
 *   /* ------------- glyph appearance ------------- * /
 *   glyphEdgeCrop?: number,
 *   glyphHeightToWidth?: number,
 *   glyphVerticalSpacing?: number,
 *   glyphFlip?: boolean,
 *   glyphRotation?: number,        // radians (multiples of π/2 supported)
 *
 *   /* ------------- cursor & glint ------------- * /
 *   isolateCursor?: boolean,
 *   cursorColor?:   Colour,
 *   cursorIntensity?: number,
 *   isolateGlint?:  boolean,
 *   glintColor?:    Colour,
 *   glintIntensity?: number,
 *
 *   /* ------------- animation & timing ------------- * /
 *   animationSpeed?: number,
 *   fps?: number,
 *   cycleSpeed?: number,
 *   cycleFrameSkip?: number,
 *   fallSpeed?: number,
 *   forwardSpeed?: number,
 *   raindropLength?: number,
 *   slant?: number,                // radians
 *
 *   /* ------------- optical effects ------------- * /
 *   bloomStrength?: number,
 *   bloomSize?: number,
 *   highPassThreshold?: number,
 *   baseBrightness?: number,
 *   baseContrast?: number,
 *   glintBrightness?: number,
 *   glintContrast?: number,
 *   brightnessOverride?: number,
 *   brightnessThreshold?: number,
 *   brightnessDecay?: number,
 *   ditherMagnitude?: number,
 *   hasThunder?: boolean,
 *
 *   /* ------------- geometry ------------- * /
 *   numColumns?: number,
 *   density?: number,
 *   isPolar?: boolean,
 *   rippleTypeName?: ("circle"|"box"|string|null),
 *   rippleThickness?: number,
 *   rippleScale?: number,
 *   rippleSpeed?: number,
 *
 *   /* ------------- colour mapping ------------- * /
 *   palette?: {color: Colour, at: number}[],
 *   stripeColors?: Colour[],
 *   backgroundColor?: Colour,
 *   glyphIntensity?: number,
 *
 *   /* ------------- misc / experimental ------------- * /
 *   resolution?: number,
 *   testFix?: string|null,
 *
 *   /* ------------- React pass-through ------------- * /
 *   style?: React.CSSProperties,
 *   className?: string,
 *
 *   /* ------------- catch-all ------------- * /
 *   [key: string]: unknown
 * }} MatrixProps
 */

/** @param {MatrixProps} props */
export const Matrix = memo((props) => {
	const { style, className, ...rawConfigProps } = props;
	const elProps = { style, className };
	const domElement = useRef(null);
	const [rRenderer, setRenderer] = useState(null);
	const [rSize, setSize] = useState([1, 1]);
	const [rConfig, setConfig] = useState(makeConfig({}));
	const rendererClasses = {};

	const resizeObserver = new ResizeObserver(entries => {
		for (const entry of entries) {
			const contentBoxSize = entry.contentBoxSize[0];
			setSize([contentBoxSize.inlineSize, contentBoxSize.blockSize]);
			return;
		}
	});

	useEffect(() => {
		if (domElement.current == null) return;
		resizeObserver.observe(domElement.current);
	}, [domElement]);

	useEffect(() => {
		setConfig(makeConfig({
			...Object.fromEntries(
				Object.entries(rawConfigProps).filter(([_, value]) => value != null),
			)
		}));
	}, [props]);

	const supportsWebGPU = () => {
		return (
			window.GPUQueue != null &&
			navigator.gpu != null &&
			navigator.gpu.getPreferredCanvasFormat != null
		);
	};

	const cleanup = () => {
		if (rRenderer == null) return;
		rRenderer.canvas.remove();
		rRenderer.destroy();
		setRenderer(null);
	};

	useEffect(() => {
		const useWebGPU = supportsWebGPU() && rConfig.renderer === "webgpu";
		const isWebGPU = rRenderer?.type === "webgpu";

		const loadRain = async () => {
			let renderer;
			if (useWebGPU) {
				rendererClasses.webgpu ??= (await import("./webgpu/renderer.js")).default;
				renderer = new (rendererClasses.webgpu)();
			} else {
				rendererClasses.regl ??= (await import("./regl/renderer.js")).default;
				renderer = new (rendererClasses.regl)();
			}
			setRenderer(renderer);
			await renderer.ready;
			const canvas = renderer.canvas;
			canvas.style.width = "100%";
			canvas.style.height = "100%";
			domElement.current.appendChild(canvas);
		};

		if (rRenderer == null || useWebGPU !== isWebGPU) {
			cleanup();
			loadRain();
		}

		return cleanup;
	}, [rConfig.renderer]);

	useEffect(() => {
		if (rRenderer?.destroyed ?? true) return;
		rRenderer.formulate(rConfig);
	}, [rRenderer, rConfig]);

	useEffect(() => {
		if (rRenderer?.destroyed ?? true) return;
		rRenderer.size = rSize.map(n => n * rConfig.resolution);
	}, [rRenderer, rConfig.resolution, rSize]);

	return <div ref={domElement} {...elProps}></div>;
});