import {FillBlanksExercise} from "@/interfaces/exam";
import clsx from "clsx";
import {Fragment, useEffect, useState} from "react";
import reactStringReplace from "react-string-replace";
import {CommonProps} from ".";
import Button from "../Low/Button";

interface WordsDrawerProps {
	words: {word: string; isDisabled: boolean}[];
	isOpen: boolean;
	blankId?: string;
	previouslySelectedWord?: string;
	onCancel: () => void;
	onAnswer: (answer: string) => void;
}

function WordsDrawer({words, isOpen, blankId, previouslySelectedWord, onCancel, onAnswer}: WordsDrawerProps) {
	const [selectedWord, setSelectedWord] = useState<string | undefined>(previouslySelectedWord);

	useEffect(() => setSelectedWord(previouslySelectedWord), [previouslySelectedWord]);

	return (
		<>
			<div
				className={clsx(
					"w-full h-full absolute top-0 left-0 bg-gradient-to-t from-mti-black to-transparent z-10",
					isOpen ? "visible opacity-10" : "invisible opacity-0",
				)}
			/>
			<div
				className={clsx(
					"absolute w-full bg-white px-7 py-8 bottom-0 left-0 shadow-2xl rounded-2xl z-20 flex flex-col gap-8 transition-opacity duration-300 ease-in-out",
					isOpen ? "visible opacity-100" : "invisible opacity-0",
				)}>
				<div className="w-full flex gap-2">
					<div className="rounded-full w-6 h-6 flex items-center justify-center text-white bg-mti-green-light">{blankId}</div>
					<span> Choose the correct word:</span>
				</div>
				<div className="grid grid-cols-6 gap-6">
					{words.map(({word, isDisabled}) => (
						<button
							key={word}
							onClick={() => setSelectedWord((prev) => (prev === word ? undefined : word))}
							className={clsx(
								"rounded-full py-3 text-center transition duration-300 ease-in-out",
								selectedWord === word ? "text-white bg-mti-green-light" : "bg-mti-green-ultralight",
								!isDisabled && "hover:text-white hover:bg-mti-green",
								"disabled:cursor-not-allowed disabled:text-mti-gray-dim",
							)}
							disabled={isDisabled}>
							{word}
						</button>
					))}
				</div>
				<div className="flex justify-between w-full">
					<Button color="green" variant="outline" className="max-w-[200px] w-full" onClick={onCancel}>
						Back
					</Button>
					<Button color="green" className="max-w-[200px] w-full" onClick={() => onAnswer(selectedWord!)} disabled={!selectedWord}>
						Confirm
					</Button>
				</div>
			</div>
		</>
	);
}

export default function FillBlanks({id, allowRepetition, type, prompt, solutions, text, words, onNext, onBack}: FillBlanksExercise & CommonProps) {
	const [userSolutions, setUserSolutions] = useState<{id: string; solution: string}[]>([]);
	const [currentBlankId, setCurrentBlankId] = useState<string>();
	const [blankSelectedWord, setBlankSelectedWord] = useState<string>();

	useEffect(() => {
		setBlankSelectedWord(currentBlankId ? userSolutions.find((x) => x.id === currentBlankId)?.solution : undefined);
	}, [userSolutions, currentBlankId]);

	const calculateScore = () => {
		const total = text.match(/({{\d+}})/g)?.length || 0;
		const correct = userSolutions.filter((x) => solutions.find((y) => x.id === y.id)?.solution === x.solution.toLowerCase() || false).length;

		return {total, correct};
	};

	const renderLines = (line: string) => {
		return (
			<span className="text-base leading-5">
				{reactStringReplace(line, /({{\d+}})/g, (match) => {
					const id = match.replaceAll(/[\{\}]/g, "");
					const userSolution = userSolutions.find((x) => x.id === id);

					return (
						<button
							className={clsx(
								"rounded-full hover:text-white hover:bg-mti-green transition duration-300 ease-in-out my-1",
								!userSolution && "w-6 h-6 text-center text-mti-green-light bg-mti-green-ultralight",
								currentBlankId === id && "text-white !bg-mti-green-light ",
								userSolution && "px-5 py-2 text-center text-white bg-mti-green-light",
							)}
							onClick={() => setCurrentBlankId(id)}>
							{userSolution ? userSolution.solution : id}
						</button>
					);
				})}
			</span>
		);
	};

	return (
		<>
			<div className="flex flex-col gap-4 mt-4 h-full mb-20">
				<WordsDrawer
					blankId={currentBlankId}
					words={words.map((word) => ({word, isDisabled: allowRepetition ? false : userSolutions.map((x) => x.solution).includes(word)}))}
					previouslySelectedWord={blankSelectedWord}
					isOpen={!!currentBlankId}
					onCancel={() => setCurrentBlankId(undefined)}
					onAnswer={(solution: string) => {
						setUserSolutions((prev) => [...prev.filter((x) => x.id !== currentBlankId), {id: currentBlankId!, solution}]);
						setCurrentBlankId(undefined);
					}}
				/>
				<span className="text-sm w-full leading-6">
					{prompt.split("\\n").map((line, index) => (
						<Fragment key={index}>
							{line}
							<br />
						</Fragment>
					))}
				</span>
				<span className="bg-mti-gray-smoke rounded-xl px-5 py-6">
					{text.split("\\n").map((line, index) => (
						<p key={index}>
							{renderLines(line)}
							<br />
						</p>
					))}
				</span>
			</div>

			<div className="self-end flex justify-between w-full gap-8 absolute bottom-8 left-0 px-8">
				<Button color="green" variant="outline" onClick={onBack} className="max-w-[200px] w-full">
					Back
				</Button>

				<Button
					color="green"
					onClick={() => onNext({exercise: id, solutions: userSolutions, score: calculateScore(), type})}
					className="max-w-[200px] self-end w-full">
					Next
				</Button>
			</div>
		</>
	);
}