import { NextResponse } from 'next/server';
import { ImapFlow } from 'imapflow';
import { getServerSession } from 'next-auth';
import { authOptions } from '@/app/api/auth/[...nextauth]/route';
import { prisma } from '@/lib/prisma';

export async function GET(request: Request) {
  try {
    const session = await getServerSession(authOptions);
    if (!session?.user?.id) {
      return NextResponse.json(
        { error: 'Unauthorized' },
        { status: 401 }
      );
    }

    // Get credentials from database
    const credentials = await prisma.mailCredentials.findUnique({
      where: {
        userId: session.user.id
      }
    });

    if (!credentials) {
      return NextResponse.json(
        { error: 'No mail credentials found. Please configure your email account.' },
        { status: 401 }
      );
    }

    // Get query parameters
    const url = new URL(request.url);
    const folder = url.searchParams.get('folder') || 'INBOX';
    const page = parseInt(url.searchParams.get('page') || '1');
    const limit = parseInt(url.searchParams.get('limit') || '20');

    // Calculate start and end sequence numbers
    const start = (page - 1) * limit + 1;
    const end = start + limit - 1;

    // Connect to IMAP server
    const client = new ImapFlow({
      host: credentials.host,
      port: credentials.port,
      secure: true,
      auth: {
        user: credentials.email,
        pass: credentials.password,
      },
      logger: false,
      emitLogs: false,
      tls: {
        rejectUnauthorized: false
      }
    });

    try {
      await client.connect();
      
      // Get list of all mailboxes first
      const mailboxes = await client.list();
      const availableFolders = mailboxes.map(box => box.path);
      
      // Open the requested mailbox
      const mailbox = await client.mailboxOpen(folder);
      
      const result = [];
      
      // Only try to fetch if the mailbox has messages
      if (mailbox.exists > 0) {
        // Adjust start and end to be within bounds
        const adjustedStart = Math.min(start, mailbox.exists);
        const adjustedEnd = Math.min(end, mailbox.exists);
        
        // Fetch messages from the current folder
        const messages = await client.fetch(`${adjustedStart}:${adjustedEnd}`, { 
          envelope: true,
          flags: true,
          bodyStructure: true
        });
        
        for await (const message of messages) {
          result.push({
            id: message.uid,
            from: message.envelope.from?.[0]?.address || '',
            fromName: message.envelope.from?.[0]?.name || message.envelope.from?.[0]?.address?.split('@')[0] || '',
            to: message.envelope.to?.map(addr => addr.address).join(', ') || '',
            subject: message.envelope.subject || '(No subject)',
            date: message.envelope.date?.toISOString() || new Date().toISOString(),
            read: message.flags.has('\\Seen'),
            starred: message.flags.has('\\Flagged'),
            folder: mailbox.path,
            hasAttachments: message.bodyStructure?.type === 'multipart',
            flags: Array.from(message.flags)
          });
        }
      }

      return NextResponse.json({
        emails: result,
        folders: availableFolders,
        total: mailbox.exists,
        hasMore: end < mailbox.exists
      });
    } finally {
      try {
        await client.logout();
      } catch (e) {
        console.error('Error during logout:', e);
      }
    }
  } catch (error) {
    console.error('Error in courrier route:', error);
    return NextResponse.json(
      { error: 'An unexpected error occurred' },
      { status: 500 }
    );
  }
}