import { NextResponse } from 'next/server';
import Imap from 'imap';
import nodemailer from 'nodemailer';
import { parseEmailHeaders, decodeEmailBody } from '@/lib/email-parser';
import { cookies } from 'next/headers';

interface StoredCredentials {
  email: string;
  password: string;
  host: string;
  port: number;
}

interface Email {
  id: string;
  from: string;
  subject: string;
  date: Date;
  read: boolean;
  starred: boolean;
  body: string;
  to?: string;
  folder: string;
}

interface ImapBox {
  messages: {
    total: number;
  };
}

interface ImapMessage {
  on: (event: string, callback: (data: any) => void) => void;
  once: (event: string, callback: (data: any) => void) => void;
  attributes: {
    uid: number;
    flags: string[];
    size: number;
  };
  body: {
    [key: string]: {
      on: (event: string, callback: (data: any) => void) => void;
    };
  };
}

interface ImapConfig {
  user: string;
  password: string;
  host: string;
  port: number;
  tls: boolean;
  authTimeout: number;
  connTimeout: number;
  debug?: (info: string) => void;
}

function getStoredCredentials(): StoredCredentials | null {
  const cookieStore = cookies();
  
  const credentialsCookie = cookieStore.get('imap_credentials');
  console.log('Retrieved credentials cookie:', credentialsCookie ? 'Found' : 'Not found');

  if (!credentialsCookie?.value) {
    console.log('No credentials cookie found');
    return null;
  }

  try {
    const credentials = JSON.parse(credentialsCookie.value);
    console.log('Parsed credentials:', {
      ...credentials,
      password: '***'
    });

    // Validate required fields
    if (!credentials.email || !credentials.password || !credentials.host || !credentials.port) {
      console.error('Missing required credentials fields');
      return null;
    }

    return {
      email: credentials.email,
      password: credentials.password,
      host: credentials.host,
      port: credentials.port
    };
  } catch (error) {
    console.error('Error parsing credentials cookie:', error);
    return null;
  }
}

export async function GET() {
  try {
    const credentials = getStoredCredentials();
    if (!credentials) {
      return NextResponse.json(
        { error: 'No stored credentials found' },
        { status: 401 }
      );
    }

    const availableMailboxes: string[] = [];
    const emailsByFolder: { [key: string]: any[] } = {};

    return new Promise((resolve) => {
      const imap = new Imap({
        user: credentials.email,
        password: credentials.password,
        host: credentials.host,
        port: credentials.port,
        tls: true,
        tlsOptions: { rejectUnauthorized: false },
        authTimeout: 30000,
        connTimeout: 30000
      });

      // Add a timeout to prevent hanging
      const timeout = setTimeout(() => {
        console.error('IMAP connection timeout');
        imap.end();
        resolve(NextResponse.json({ 
          emails: [],
          error: 'Connection timeout'
        }));
      }, 60000);

      imap.once('error', (err: Error) => {
        console.error('IMAP error:', err);
        clearTimeout(timeout);
        resolve(NextResponse.json({ 
          emails: [],
          error: 'IMAP connection error'
        }));
      });

      imap.once('ready', () => {
        imap.getBoxes((err, boxes) => {
          if (err) {
            console.error('Error getting mailboxes:', err);
            clearTimeout(timeout);
            imap.end();
            resolve(NextResponse.json({ emails: [], error: 'Failed to get mailboxes' }));
            return;
          }

          // Process mailboxes
          Object.keys(boxes).forEach((box) => {
            availableMailboxes.push(box);
          });

          console.log('Available mailboxes:', availableMailboxes);

          // Process each mailbox
          const foldersToProcess = availableMailboxes;
          let processedFolders = 0;
          let activeFetches = 0;

          function checkCompletion() {
            processedFolders++;
            if (processedFolders === foldersToProcess.length && activeFetches === 0) {
              clearTimeout(timeout);
              finishProcessing();
            }
          }

          function finishProcessing() {
            // Combine all emails from all folders
            const allEmails = Object.entries(emailsByFolder).flatMap(([folder, emails]) => emails);
            
            console.log('Emails by folder:', Object.fromEntries(
              Object.entries(emailsByFolder).map(([folder, emails]) => [folder, emails.length])
            ));
            console.log('All folders processed, total emails:', allEmails.length);

            const response = {
              emails: allEmails,
              folders: availableMailboxes,
              mailUrl: process.env.NEXTCLOUD_URL ? `${process.env.NEXTCLOUD_URL}/apps/mail/` : null
            };
            imap.end();
            resolve(NextResponse.json(response));
          }

          foldersToProcess.forEach((folderName) => {
            // Initialize array for this folder
            emailsByFolder[folderName] = [];

            imap.openBox(folderName, false, (err, box) => {
              if (err) {
                console.error(`Error opening box ${folderName}:`, err);
                checkCompletion();
                return;
              }

              // Search for all emails in this folder
              imap.search(['ALL'], (err, results) => {
                if (err) {
                  console.error(`Error searching in ${folderName}:`, err);
                  checkCompletion();
                  return;
                }

                if (!results || results.length === 0) {
                  checkCompletion();
                  return;
                }

                activeFetches++;
                // Fetch emails
                const fetch = imap.fetch(results, {
                  bodies: ['HEADER', 'TEXT'],
                  struct: true
                });

                fetch.on('message', (msg) => {
                  let header = '';
                  let text = '';
                  let messageId: number | null = null;
                  let messageFlags: string[] = [];

                  // Handle attributes first
                  msg.once('attributes', (attrs) => {
                    messageId = attrs.uid;
                    messageFlags = attrs.flags || [];
                  });

                  msg.on('body', (stream, info) => {
                    let buffer = '';
                    stream.on('data', (chunk) => {
                      buffer += chunk.toString('utf8');
                    });
                    stream.on('end', () => {
                      if (info.which === 'HEADER') {
                        header = buffer;
                      } else if (info.which === 'TEXT') {
                        text = buffer;
                      }
                    });
                  });

                  msg.on('end', () => {
                    if (!messageId) {
                      console.error('No message ID found for email');
                      return;
                    }

                    const parsedHeader = Imap.parseHeader(header);
                    const email = {
                      id: messageId,
                      from: parsedHeader.from?.[0] || '',
                      to: parsedHeader.to?.[0] || '',
                      subject: parsedHeader.subject?.[0] || '(No subject)',
                      date: parsedHeader.date?.[0] || new Date().toISOString(),
                      body: text,
                      folder: folderName,
                      flags: messageFlags
                    };
                    emailsByFolder[folderName].push(email);
                  });
                });

                fetch.on('error', (err) => {
                  console.error(`Error fetching emails from ${folderName}:`, err);
                  activeFetches--;
                  checkCompletion();
                });

                fetch.on('end', () => {
                  activeFetches--;
                  checkCompletion();
                });
              });
            });
          });
        });
      });

      imap.connect();
    });
  } catch (error) {
    console.error('Error in mail API:', error);
    return NextResponse.json({ 
      emails: [],
      error: error instanceof Error ? error.message : 'Unknown error'
    });
  }
}

export async function POST(request: Request) {
  try {
    const credentials = getStoredCredentials();
    if (!credentials) {
      return NextResponse.json(
        { error: 'No stored credentials found' },
        { status: 401 }
      );
    }

    let body;
    try {
      body = await request.json();
    } catch (error) {
      return NextResponse.json(
        { error: 'Invalid JSON in request body' },
        { status: 400 }
      );
    }

    const { to, subject, body: emailBody, attachments } = body;

    if (!to || !subject || !emailBody) {
      return NextResponse.json(
        { error: 'Missing required fields: to, subject, or body' },
        { status: 400 }
      );
    }

    const transporter = nodemailer.createTransport({
      host: credentials.host,
      port: credentials.port,
      secure: true,
      auth: {
        user: credentials.email,
        pass: credentials.password,
      },
    });

    const mailOptions = {
      from: credentials.email,
      to,
      subject,
      text: emailBody,
      attachments: attachments || [],
    };

    const info = await transporter.sendMail(mailOptions);
    console.log('Email sent:', info.messageId);

    return NextResponse.json({ 
      success: true, 
      messageId: info.messageId,
      message: 'Email sent successfully'
    });
  } catch (error) {
    console.error('Error sending email:', error);
    return NextResponse.json(
      { 
        error: error instanceof Error ? error.message : 'Failed to send email',
        details: error instanceof Error ? error.stack : undefined
      },
      { status: 500 }
    );
  }
}