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() {
  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 }
      );
    }

    // 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();
      const mailbox = await client.mailboxOpen('INBOX');
      
      // Fetch only essential message data
      const messages = await client.fetch('1:20', { 
        envelope: true,
        flags: true
      });
      
      const result = [];
      for await (const message of messages) {
        result.push({
          id: message.uid.toString(),
          from: message.envelope.from[0].address,
          subject: message.envelope.subject || '(No subject)',
          date: message.envelope.date.toISOString(),
          read: message.flags.has('\\Seen'),
          starred: message.flags.has('\\Flagged'),
          folder: mailbox.path
        });
      }

      return NextResponse.json({
        emails: result,
        folders: ['INBOX', 'Sent', 'Drafts', 'Trash', 'Spam']
      });
    } finally {
      try {
        await client.logout();
      } catch (e) {
        console.error('Error during logout:', e);
      }
    }
  } catch (error) {
    console.error('Error in mail route:', error);
    return NextResponse.json(
      { error: 'An unexpected error occurred' },
      { status: 500 }
    );
  }
}