/*
 * Portions of this file are based on code from radix-ui-primitives.
 * MIT Licensed, Copyright (c) 2022 WorkOS.
 *
 * Credits to the Radix UI team:
 * https://github.com/radix-ui/primitives/blob/21a7c97dc8efa79fecca36428eec49f187294085/packages/react/avatar/src/Avatar.tsx
 */

import {
	ComponentProps,
	Show,
	type ValidComponent,
	createEffect,
	createSignal,
	on,
	onCleanup,
} from "solid-js";
import {
	type ElementOf,
	Polymorphic,
	type PolymorphicProps,
} from "../polymorphic";

import { useImageContext } from "./image-context";
import type { ImageLoadingStatus } from "./types";

export interface ImageImgOptions {}

export interface ImageImgCommonProps<T extends HTMLElement = HTMLElement> {
	src?: string;
}

export interface ImageImgRenderProps extends ImageImgCommonProps {}

export type ImageImgProps<
	T extends ValidComponent | HTMLElement = HTMLElement,
> = ImageImgOptions & Partial<ImageImgCommonProps<ElementOf<T>>>;

/**
 * The image to render. By default, it will only render when it has loaded.
 */
export function ImageImg<T extends ValidComponent = "img">(
	props: PolymorphicProps<T, ImageImgProps<T>>,
) {
	const context = useImageContext();

	const [loadingStatus, setLoadingStatus] =
		createSignal<ImageLoadingStatus>("idle");

	createEffect(
		on(
			() => props.src,
			(src) => {
				if (!src) {
					setLoadingStatus("error");
					return;
				}

				let isMounted = true;
				const image = new window.Image();

				const updateStatus = (status: ImageLoadingStatus) => () => {
					if (!isMounted) {
						return;
					}

					setLoadingStatus(status);
				};

				setLoadingStatus("loading");
				image.onload = updateStatus("loaded");
				image.onerror = updateStatus("error");
				image.src = src;

				onCleanup(() => {
					isMounted = false;
				});
			},
		),
	);

	createEffect(() => {
		const imageLoadingStatus = loadingStatus();

		if (imageLoadingStatus !== "idle") {
			context.onImageLoadingStatusChange(imageLoadingStatus);
		}
	});

	return (
		<Show when={loadingStatus() === "loaded"}>
			<Polymorphic<ImageImgRenderProps>
				as="img"
				{...(props as ImageImgProps)}
			/>
		</Show>
	);
}