"use client"

import React, { useState, useEffect, useRef } from "react";
import { useRouter } from "next/navigation";
import { Card, CardContent, CardHeader, CardTitle } from "@/components/ui/card";
import { Button } from "@/components/ui/button";
import { Input } from "@/components/ui/input";
import { Avatar, AvatarFallback } from "@/components/ui/avatar";
import { cn } from "@/lib/utils";
import { formatDistanceToNow } from "date-fns";
import axios from "axios";
import { toast } from "sonner";
import { ArrowLeft, Send, RefreshCw, File, FileText, Image as ImageIcon, Download } from "lucide-react";
import { getCookie, clientFetch } from "@/lib/client-utils";
import { ImageViewerModal } from "@/components/modals/image-viewer-modal";
import BuyerOrderInfo from "./BuyerOrderInfo";

interface Message {
  _id: string;
  sender: "buyer" | "vendor";
  content: string;
  attachments: string[];
  read: boolean;
  createdAt: string;
  buyerId: string;
  vendorId: string;
}

interface Chat {
  _id: string;
  buyerId: string;
  vendorId: string;
  storeId: string;
  messages: Message[];
  lastUpdated: string;
  orderId?: string;
  telegramUsername?: string | null;
}

// Helper function to extract filename from URL
const getFileNameFromUrl = (url: string): string => {
  // Try to extract filename from the URL path
  const pathParts = url.split('/');
  const lastPart = pathParts[pathParts.length - 1];
  
  // Remove query parameters if any
  const fileNameParts = lastPart.split('?');
  let fileName = fileNameParts[0];
  
  // If filename is too long or not found, create a generic name
  if (!fileName || fileName.length > 30) {
    return 'attachment';
  }
  
  // URL decode the filename (handle spaces and special characters)
  try {
    fileName = decodeURIComponent(fileName);
  } catch (e) {
    // If decoding fails, use the original
  }
  
  return fileName;
};

// Helper function to get file icon based on extension or URL pattern
const getFileIcon = (url: string): React.ReactNode => {
  const fileName = url.toLowerCase();
  
  // Image files
  if (/\.(jpg|jpeg|png|gif|webp|svg|bmp|tiff)($|\?)/i.test(fileName) || 
      url.includes('/photos/') || url.includes('/photo/')) {
    return <ImageIcon className="h-5 w-5" />;
  }
  
  // Document files
  if (/\.(pdf|doc|docx|xls|xlsx|ppt|pptx|txt|rtf|csv)($|\?)/i.test(fileName)) {
    return <FileText className="h-5 w-5" />;
  }
  
  // Default file icon
  return <File className="h-5 w-5" />;
};

export default function ChatDetail({ chatId }: { chatId: string }) {
  const router = useRouter();
  const [chat, setChat] = useState<Chat | null>(null);
  const [loading, setLoading] = useState(true);
  const [message, setMessage] = useState("");
  const [sending, setSending] = useState(false);
  const [messages, setMessages] = useState<Message[]>([]);
  const [chatData, setChatData] = useState<any>(null);
  const messagesEndRef = useRef<HTMLDivElement>(null);
  const [previousMessageCount, setPreviousMessageCount] = useState<number>(0);
  const audioRef = useRef<HTMLAudioElement | null>(null);
  const markReadTimeoutRef = useRef<NodeJS.Timeout | null>(null);
  const isPollingRef = useRef<boolean>(false);
  const [selectedImage, setSelectedImage] = useState<string | null>(null);
  const [selectedMessageIndex, setSelectedMessageIndex] = useState<number | null>(null);
  const [selectedAttachmentIndex, setSelectedAttachmentIndex] = useState<number | null>(null);
  
  // Scroll to bottom utility functions
  const scrollToBottom = () => {
    messagesEndRef.current?.scrollIntoView({ behavior: "smooth" });
  };
  
  const isNearBottom = () => {
    if (!messagesEndRef.current) return true;
    
    const container = messagesEndRef.current.parentElement;
    if (!container) return true;
    
    const { scrollTop, scrollHeight, clientHeight } = container;
    // Consider "near bottom" if within 100px of the bottom
    return scrollHeight - (scrollTop + clientHeight) < 100;
  };
  
  // Initialize audio element
  useEffect(() => {
    // Create audio element for notification sound
    audioRef.current = new Audio('/notification.mp3');
    
    // Fallback if notification.mp3 doesn't exist - use browser API for a simple beep
    audioRef.current.addEventListener('error', () => {
      audioRef.current = null;
    });
    
    return () => {
      if (audioRef.current) {
        audioRef.current = null;
      }
      
      // Clear any pending timeouts when component unmounts
      if (markReadTimeoutRef.current) {
        clearTimeout(markReadTimeoutRef.current);
      }
    };
  }, []);
  
  // Function to play notification sound
  const playNotificationSound = () => {
    if (audioRef.current) {
      audioRef.current.currentTime = 0;
      audioRef.current.play().catch(err => {
        console.log('Error playing sound:', err);
        // Fallback to simple beep if audio file fails
        try {
          const context = new (window.AudioContext || (window as any).webkitAudioContext)();
          const oscillator = context.createOscillator();
          oscillator.type = 'sine';
          oscillator.frequency.setValueAtTime(800, context.currentTime);
          oscillator.connect(context.destination);
          oscillator.start();
          oscillator.stop(context.currentTime + 0.2);
        } catch (e) {
          console.error('Could not play fallback audio', e);
        }
      });
    } else {
      // Fallback to simple beep if audio element is not available
      try {
        const context = new (window.AudioContext || (window as any).webkitAudioContext)();
        const oscillator = context.createOscillator();
        oscillator.type = 'sine';
        oscillator.frequency.setValueAtTime(800, context.currentTime);
        oscillator.connect(context.destination);
        oscillator.start();
        oscillator.stop(context.currentTime + 0.2);
      } catch (e) {
        console.error('Could not play fallback audio', e);
      }
    }
  };
  
  // Function to mark messages as read
  const markMessagesAsRead = async () => {
    try {
      const authToken = getCookie("Authorization");
      
      if (!authToken) return;
      
      const authAxios = axios.create({
        baseURL: process.env.NEXT_PUBLIC_API_URL,
        headers: {
          Authorization: `Bearer ${authToken}`
        }
      });
      
      // Use dedicated endpoint to mark messages as read
      await authAxios.post(`/chats/${chatId}/mark-read`);
      console.log("Marked messages as read");
    } catch (error) {
      console.error("Error marking messages as read:", error);
    }
  };
  
  // Loading effect
  useEffect(() => {
    if (chatId) {
      fetchChatData();
    }
  }, [chatId]);

  // Get a fresh auth axios instance
  const getAuthAxios = () => {
    const authToken = getCookie("Authorization");
    if (!authToken) return null;
    
    return axios.create({
      baseURL: process.env.NEXT_PUBLIC_API_URL,
      headers: {
        Authorization: `Bearer ${authToken}`,
        "Content-Type": "application/json",
      }
    });
  };
  
  // Fetch chat information and messages
  const fetchChatData = async () => {
    if (!chatId) return;
    
    try {
      setLoading(true);
      
      // Use clientFetch instead of direct axios calls
      const response = await clientFetch(`/chats/${chatId}`);
      
      setChatData(response);
      setChat(response); // Set chat data to maintain compatibility
      setMessages(Array.isArray(response.messages) ? response.messages : []);
      
      // Scroll to bottom on initial load
      setTimeout(() => {
        scrollToBottom();
      }, 100);
    } catch (error) {
      console.error("Error fetching chat data:", error);
      toast.error("Failed to load chat");
    } finally {
      setLoading(false);
    }
  };
  
  // Fetch new messages periodically
  useEffect(() => {
    if (!chatId || !chatData) return;
    
    const checkForNewMessages = async () => {
      if (isPollingRef.current) return;
      
      isPollingRef.current = true;
      
      try {
        // Use clientFetch instead of direct axios calls
        const response = await clientFetch(`/chats/${chatId}?markAsRead=true`);
        
        if (
          response && 
          Array.isArray(response.messages) && 
          response.messages.length !== messages.length
        ) {
          setMessages(response.messages);
          // Only auto-scroll if we're already near the bottom
          if (isNearBottom()) {
            setTimeout(() => {
              scrollToBottom();
            }, 100);
          }
        }
      } catch (error) {
        console.error("Error checking for new messages:", error);
      } finally {
        isPollingRef.current = false;
      }
    };
    
    // Check for new messages every 3 seconds
    const intervalId = setInterval(checkForNewMessages, 3000);
    
    return () => {
      clearInterval(intervalId);
    };
  }, [chatId, chatData, messages.length]);

  // Send a message
  const sendMessage = async () => {
    if (!chatId || !message.trim()) return;
    
    // Optimistically add message to UI
    const tempId = `temp-${Date.now()}`;
    const tempMessage = {
      _id: tempId,
      chatId,
      sender: "vendor" as const,
      content: message.trim(),
      attachments: [],
      read: false,
      createdAt: new Date().toISOString(),
      buyerId: chatData?.buyerId || "",
      vendorId: chatData?.vendorId || "",
    };
    
    setMessages(prev => [...prev, tempMessage]);
    scrollToBottom();
    
    // Clear input
    setMessage("");
    
    try {
      // Use clientFetch instead of direct axios calls
      const response = await clientFetch(`/chats/${chatId}/message`, {
        method: 'POST',
        body: JSON.stringify({ message: message.trim() }),
      });
      
      // Replace temp message with real one from server
      setMessages(prev => 
        prev.filter(m => m._id !== tempId)
          .concat(response)
      );
    } catch (error) {
      console.error("Error sending message:", error);
      toast.error("Failed to send message");
      
      // Remove temp message on error
      setMessages(prev => prev.filter(m => m._id !== tempId));
    }
  };
  
  // Handle form submit for sending messages
  const handleSendMessage = (e: React.FormEvent) => {
    e.preventDefault();
    sendMessage();
  };
  
  const handleBackClick = () => {
    router.push("/dashboard/chats");
  };
  
  // Add function to handle image navigation
  const handleImageNavigation = (direction: 'prev' | 'next') => {
    if (!chat || selectedMessageIndex === null || selectedAttachmentIndex === null) return;

    // Get all images from all messages
    const allImages: { messageIndex: number; attachmentIndex: number; url: string }[] = [];
    
    chat.messages.forEach((msg, msgIndex) => {
      msg.attachments.forEach((att, attIndex) => {
        if (/\.(jpg|jpeg|png|gif|webp|svg|bmp|tiff)($|\?)/i.test(att) || 
            att.includes('/photos/') || 
            att.includes('/photo/')) {
          allImages.push({ messageIndex: msgIndex, attachmentIndex: attIndex, url: att });
        }
      });
    });

    if (allImages.length === 0) return;

    // Find current image index
    const currentIndex = allImages.findIndex(img => 
      img.messageIndex === selectedMessageIndex && 
      img.attachmentIndex === selectedAttachmentIndex
    );

    if (currentIndex === -1) return;

    // Calculate new index
    let newIndex;
    if (direction === 'next') {
      newIndex = (currentIndex + 1) % allImages.length;
    } else {
      newIndex = currentIndex === 0 ? allImages.length - 1 : currentIndex - 1;
    }

    // Update state with new image
    const newImage = allImages[newIndex];
    setSelectedMessageIndex(newImage.messageIndex);
    setSelectedAttachmentIndex(newImage.attachmentIndex);
    setSelectedImage(newImage.url);
  };

  // Update the image click handler
  const handleImageClick = (imageUrl: string, messageIndex: number, attachmentIndex: number) => {
    setSelectedImage(imageUrl);
    setSelectedMessageIndex(messageIndex);
    setSelectedAttachmentIndex(attachmentIndex);
  };
  
  if (loading) {
    return (
      <div className="flex flex-col h-screen w-full relative">
        <div className="border-b py-2 px-4 flex items-center space-x-2 bg-card z-10">
          <Button variant="ghost" size="icon" onClick={handleBackClick}>
            <ArrowLeft className="h-5 w-5" />
          </Button>
          <span className="text-lg font-semibold">Loading conversation...</span>
        </div>
        <div className="flex-1 flex items-center justify-center">
          <RefreshCw className="h-8 w-8 animate-spin text-muted-foreground" />
        </div>
      </div>
    );
  }
  
  if (!chat) {
    return (
      <div className="flex flex-col h-screen w-full relative">
        <div className="border-b py-2 px-4 flex items-center space-x-2 bg-card z-10">
          <Button variant="ghost" size="icon" onClick={handleBackClick}>
            <ArrowLeft className="h-5 w-5" />
          </Button>
          <span className="text-lg font-semibold">Chat not found</span>
        </div>
        <div className="flex-1 flex items-center justify-center">
          <div className="text-center">
            <p className="text-muted-foreground mb-4">This conversation doesn't exist or you don't have access to it.</p>
            <Button onClick={handleBackClick}>Back to Chats</Button>
          </div>
        </div>
      </div>
    );
  }
  
  return (
    <div className="flex flex-col h-screen w-full relative">
      <div className="border-b h-16 px-4 flex items-center justify-between bg-card z-10">
        <div className="flex items-center space-x-2">
          <Button variant="ghost" size="icon" onClick={handleBackClick}>
            <ArrowLeft className="h-5 w-5" />
          </Button>
          <div className="flex flex-col">
            <h3 className="text-lg font-semibold">
              Chat with Customer {chat.buyerId.slice(-4)}
            </h3>
            {chat.telegramUsername && (
              <span className="text-sm text-muted-foreground">
                @{chat.telegramUsername}
              </span>
            )}
          </div>
        </div>
        
        <BuyerOrderInfo buyerId={chat.buyerId} chatId={chatId} />
      </div>
      
      <div className="flex-1 overflow-y-auto p-2 space-y-2 pb-[80px]">
        {chat.messages.length === 0 ? (
          <div className="h-full flex items-center justify-center">
            <p className="text-muted-foreground">No messages yet. Send one to start the conversation.</p>
          </div>
        ) : (
          chat.messages.map((msg, messageIndex) => (
            <div
              key={msg._id || messageIndex}
              className={cn(
                "flex mb-4",
                msg.sender === "vendor" ? "justify-end" : "justify-start"
              )}
            >
              <div
                className={cn(
                  "max-w-[90%] rounded-lg p-3",
                  msg.sender === "vendor"
                    ? "bg-primary text-primary-foreground"
                    : "bg-muted"
                )}
              >
                <div className="flex items-center space-x-2 mb-1">
                  {msg.sender === "buyer" && (
                    <Avatar className="h-6 w-6">
                      <AvatarFallback className="text-xs">
                        {chat.buyerId.slice(0, 2).toUpperCase()}
                      </AvatarFallback>
                    </Avatar>
                  )}
                  <span className="text-xs opacity-70">
                    {formatDistanceToNow(new Date(msg.createdAt), { addSuffix: true })}
                  </span>
                </div>
                <p className="whitespace-pre-wrap break-words">{msg.content}</p>
                {/* Show attachments if any */}
                {msg.attachments && msg.attachments.length > 0 && (
                  <div className="mt-2 space-y-2">
                    {msg.attachments.map((attachment, attachmentIndex) => {
                      const isImage = /\.(jpg|jpeg|png|gif|webp|svg|bmp|tiff)($|\?)/i.test(attachment) || 
                                    attachment.includes('/photos/') || 
                                    attachment.includes('/photo/');
                      
                      const fileName = getFileNameFromUrl(attachment);
                      
                      return isImage ? (
                        <div 
                          key={`attachment-${attachmentIndex}`} 
                          className="rounded-md overflow-hidden bg-background/20 p-1"
                          onClick={() => handleImageClick(attachment, messageIndex, attachmentIndex)}
                        >
                          <div className="flex justify-between items-center mb-1">
                            <span className="text-xs opacity-70 flex items-center">
                              <ImageIcon className="h-3 w-3 mr-1" />
                              {fileName}
                            </span>
                            <a 
                              href={attachment} 
                              download={fileName}
                              target="_blank" 
                              rel="noopener noreferrer"
                              className="text-xs opacity-70 hover:opacity-100"
                              onClick={(e) => e.stopPropagation()}
                            >
                              <Download className="h-3 w-3" />
                            </a>
                          </div>
                          <img 
                            src={attachment} 
                            alt={fileName} 
                            className="max-w-full max-h-60 object-contain cursor-pointer hover:opacity-90 transition-opacity rounded"
                            onError={(e) => {
                              (e.target as HTMLImageElement).src = "/placeholder-image.svg";
                            }}
                          />
                        </div>
                      ) : (
                        // Render file attachment
                        <div key={`attachment-${attachmentIndex}`} className="flex items-center bg-background/20 rounded-md p-2 hover:bg-background/30 transition-colors">
                          <div className="mr-2">
                            {getFileIcon(attachment)}
                          </div>
                          <div className="flex-1 min-w-0">
                            <div className="text-sm truncate font-medium">
                              {fileName}
                            </div>
                          </div>
                          <a 
                            href={attachment} 
                            download={fileName}
                            target="_blank"
                            rel="noopener noreferrer"
                            className="ml-2 p-1 rounded-sm hover:bg-background/50"
                          >
                            <Download className="h-4 w-4" />
                          </a>
                        </div>
                      );
                    })}
                  </div>
                )}
              </div>
            </div>
          ))
        )}
        <div ref={messagesEndRef} />
      </div>
      
      <div className="absolute bottom-0 left-0 right-0 border-t border-border bg-background p-4">
        <form onSubmit={handleSendMessage} className="flex space-x-2">
          <Input
            value={message}
            onChange={(e) => setMessage(e.target.value)}
            placeholder="Type your message..."
            disabled={sending}
            className="flex-1"
          />
          <Button type="submit" disabled={sending || !message.trim()}>
            {sending ? <RefreshCw className="h-4 w-4 animate-spin" /> : <Send className="h-4 w-4" />}
          </Button>
        </form>
      </div>

      {/* Update the image viewer modal */}
      <ImageViewerModal
        isOpen={!!selectedImage}
        onClose={() => {
          setSelectedImage(null);
          setSelectedMessageIndex(null);
          setSelectedAttachmentIndex(null);
        }}
        imageUrl={selectedImage || ""}
        onNavigate={handleImageNavigation}
      />
    </div>
  );
}