import { useState, useEffect, useRef } from "react";
import { MaterialDialog } from "../core/Dialog";
import { RichTextArea } from "../core/RichTextArea";
import type { Message } from "../../../core/types";
import Quote from "../core/Quote";
import AnimatedHeight from "../core/animations/AnimatedHeight";
import { useImmer } from "use-immer";
import { EmojiMenu } from "./EmojiMenu";

interface ChatInputWrapperProps {
    onSendMessage: (message: string, files: File[]) => void;
    onSaveEdit?: (content: string) => void;
    replyTo?: Message | null;
    replyToVisible: boolean;
    onClearReply?: () => void;
    onCloseReply?: () => void;
    editingMessage?: Message | null;
    editVisible?: boolean;
    onClearEdit?: () => void;
    onCloseEdit?: () => void;
    onProvideFileAdder?: (adder: (files: File[]) => void) => void;
    messagePanelRef?: React.RefObject<HTMLDivElement | null>;
}

export function ChatInputWrapper(
    { 
        onSendMessage, 
        onSaveEdit, 
        replyTo, 
        replyToVisible, 
        onClearReply,
        onCloseReply, 
        editingMessage, 
        editVisible = false, 
        onClearEdit, 
        onCloseEdit,
        onProvideFileAdder,
        messagePanelRef
    }: ChatInputWrapperProps
) {
    const [message, setMessage] = useState("");
    const [selectedFiles, setSelectedFiles] = useImmer<File[]>([]);
    const [attachmentsVisible, setAttachmentsVisible] = useState(false);
    const [errorOpen, setErrorOpen] = useState(false);
    const [emojiMenuOpen, setEmojiMenuOpen] = useState(false);
    const [emojiMenuPosition, setEmojiMenuPosition] = useState({ x: 0, y: 0 });
    const emojiButtonRef = useRef<HTMLButtonElement>(null);
    const chatInputWrapperRef = useRef<HTMLDivElement>(null);

    // Expose a way for parent to programmatically add files
    useEffect(() => {
        if (onProvideFileAdder) {
            const addFiles = (files: File[]) => {
                if (!files || files.length === 0) return;
                setSelectedFiles(draft => { draft.push(...files) });
            };
            onProvideFileAdder(addFiles);
        }
    }, [onProvideFileAdder]);

    // When entering edit mode, preload the message content
    useEffect(() => {
        setMessage(editingMessage ? editingMessage.content || "" : "");
    }, [editingMessage]);

    useEffect(() => {
        setAttachmentsVisible(selectedFiles.length > 0);
    }, [selectedFiles]);

    function handleEmojiButtonClick() {
        if (chatInputWrapperRef.current && messagePanelRef?.current) {
            const inputRect = chatInputWrapperRef.current.getBoundingClientRect();
            const panelRect = messagePanelRef.current.getBoundingClientRect();
            
            // Position menu 10px from message panel edge and 10px above the chat input
            // The animation will start 30px below this position
            setEmojiMenuPosition({
                x: panelRect.left + 10, // 10px from message panel edge
                y: window.innerHeight - inputRect.top + 10 // 10px above the top of chat input
            });
            setEmojiMenuOpen(true);
        }
    };

    function handleEmojiSelect(emoji: string) {
        setMessage(prev => prev + emoji);
    };

    async function handleSubmit(e: React.FormEvent | Event) {
        e.preventDefault();
        const hasText = Boolean(message.trim());
        const hasFiles = selectedFiles.length > 0;
        if (hasText || hasFiles) {
            const totalSize = selectedFiles.reduce((acc, f) => acc + f.size, 0);
            const limit = 4 * 1024 * 1024 * 1024; // 4GB
            if (totalSize > limit) {
                setErrorOpen(true);
                return;
            }
            if (editingMessage && onSaveEdit) {
                onSaveEdit(message);
                setMessage("");
                if (onClearEdit) onClearEdit();
            } else {
                onSendMessage(message, selectedFiles);
                setMessage("");
                setAttachmentsVisible(false);
                if (onClearReply) onClearReply();
            }
        }
    };

    function handleAttachClick() {
        const input = document.createElement("input");
        input.type = "file";
        input.multiple = true;
        input.addEventListener("change", () => {
            setSelectedFiles(draft => { draft.push(...Array.from(input.files || [])) });
        });
        input.click();
    }

    return (
        <div className="chat-input-wrapper" ref={chatInputWrapperRef}>
            <form className="input-group" id="message-form" onSubmit={handleSubmit}>
                <AnimatedHeight visible={editVisible} onFinish={onCloseEdit}>
                    {editingMessage && (
                        <div className="reply-preview contextual-preview">
                            <mdui-icon name="edit" />
                            <Quote className="reply-content contextual-content" background="surfaceContainer">
                                <span className="reply-username">{editingMessage!.username}</span>
                                <span className="reply-text">{editingMessage!.content}</span>
                            </Quote>
                            <mdui-button-icon icon="close" className="reply-cancel" onClick={onClearEdit}></mdui-button-icon>
                        </div>
                    )}
                </AnimatedHeight>
                <AnimatedHeight visible={replyToVisible} onFinish={onCloseReply}>
                    {replyTo && (
                        <div className="reply-preview contextual-preview">
                            <mdui-icon name="reply" />
                            <Quote className="reply-content contextual-content" background="surfaceContainer">
                                <span className="reply-username">{replyTo!.username}</span>
                                <span className="reply-text">{replyTo!.content}</span>
                            </Quote>
                            <mdui-button-icon icon="close" className="reply-cancel" onClick={onClearReply}></mdui-button-icon>
                        </div>
                    )}
                </AnimatedHeight>
                <AnimatedHeight visible={attachmentsVisible} onFinish={() => setSelectedFiles([])}>
                    {selectedFiles.length > 0 && (
                        <div className="attachments-preview contextual-preview">
                            <mdui-icon name="attach_file" />
                            <div className="attachments-chips">
                                {selectedFiles.map((file, i) => (
                                    <mdui-chip
                                        key={i}
                                        variant="input"
                                        end-icon="close"
                                        title={`${file.name} (${Math.round(file.size/1024/1024)} MB)`}
                                        onClick={() => {
                                            if (selectedFiles.length == 1) {
                                                setAttachmentsVisible(false);
                                            } else {
                                                setSelectedFiles(draft => { draft.splice(i) })
                                            }
                                        }}
                                    >
                                        <mdui-icon slot="icon" name="attach_file"></mdui-icon>
                                        <span className="name">{file.name}</span>
                                    </mdui-chip>
                                ))}
                            </div>
                            <mdui-button-icon icon="close" className="reply-cancel" onClick={() => setAttachmentsVisible(false)}></mdui-button-icon>
                        </div>
                    )}
                </AnimatedHeight>
                <div className="chat-input">
                    <mdui-button-icon 
                        ref={emojiButtonRef}
                        icon="mood" 
                        onClick={handleEmojiButtonClick} 
                        className="emoji-btn"
                    ></mdui-button-icon>
                    <RichTextArea
                        className="message-input" 
                        id="message-input" 
                        placeholder="Напишите сообщение..." 
                        autoComplete="off"
                        text={message}
                        rows={1}
                        onTextChange={(value) => setMessage(value)}
                        onEnter={handleSubmit} />
                    <div className="buttons">
                        <mdui-button-icon icon="attach_file" onClick={handleAttachClick} className="attach-btn"></mdui-button-icon>
                        <button type="submit" className="send-btn">
                            <span className="material-symbols filled">{editingMessage ? "check" : "send"}</span>
                        </button>
                    </div>
                </div>
            </form>
            <MaterialDialog open={errorOpen} onOpenChange={setErrorOpen} close-on-overlay-click close-on-esc>
                <div slot="headline">Ошибка</div>
                <div>Общий размер вложений превышает 4 ГБ.</div>
                <mdui-button slot="action" onClick={() => setErrorOpen(false)}>Закрыть</mdui-button>
            </MaterialDialog>
            
            <EmojiMenu
                isOpen={emojiMenuOpen}
                onClose={() => setEmojiMenuOpen(false)}
                onEmojiSelect={handleEmojiSelect}
                position={emojiMenuPosition}
            />
        </div>
    );
}