import express, { Request, Response } from 'express';
import cors from 'cors';
import sqlite3 from 'sqlite3';
import { open } from 'sqlite';
import dotenv from 'dotenv';
import path from 'path';
import multer from 'multer';
import fs from 'fs';
import { sendRSVPEmail, sendRSVPEditLinkEmail } from './email'; // Import the new email function

dotenv.config();

// Function to generate a random alphanumeric string
function generateAlphanumericId(length: number): string {
  const characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
  let result = '';
  const charactersLength = characters.length;
  for (let i = 0; i < length; i++) {
    result += characters.charAt(Math.floor(Math.random() * charactersLength));
  }
  return result;
}

const app = express();
const port = process.env.PORT || 3000;

// Middleware
app.use(cors());
app.use(express.json());

// Serve static files from the frontend build directory
app.use(express.static(path.join(__dirname, '../frontend/build')));

// Serve uploaded files
app.use('/uploads', express.static(path.join(__dirname, '../uploads')));

// Database connection
let db: any;

interface RSVP {
  id: number;
  event_id: number;
  name: string;
  attending: string;
  bringing_guests: string;
  guest_count: number;
  guest_names: string | null;
  items_bringing: string | null;
  created_at?: string;
}

async function connectToDatabase() {
  try {
    // Database file will be in the app directory
    const dbPath = path.join(__dirname, '../database.sqlite');
    
    db = await open({
      filename: dbPath,
      driver: sqlite3.Database
    });

    // Initialize tables immediately after connection
    await initializeDatabase();
  } catch (error) {
    console.error('Error connecting to database:', error);
    process.exit(1); // Exit if we can't connect to the database
  }
}

// Configure multer for file uploads
const uploadDir = path.join(__dirname, '../uploads/wallpapers');
if (!fs.existsSync(uploadDir)) {
  fs.mkdirSync(uploadDir, { recursive: true });
}

// Define multer request interface
interface MulterRequest extends Request {
  file?: Express.Multer.File;
}

const storage = multer.diskStorage({
  destination: function (req: Request, file: Express.Multer.File, cb: (error: Error | null, destination: string) => void) {
    cb(null, uploadDir);
  },
  filename: function (req: Request, file: Express.Multer.File, cb: (error: Error | null, filename: string) => void) {
    const uniqueSuffix = Date.now() + '-' + Math.round(Math.random() * 1E9);
    cb(null, uniqueSuffix + path.extname(file.originalname));
  }
});

const upload = multer({
  storage: storage,
  limits: {
    fileSize: 5 * 1024 * 1024 // 5MB limit
  },
  fileFilter: (req: Request, file: Express.Multer.File, cb: multer.FileFilterCallback) => {
    const allowedTypes = ['image/jpeg', 'image/png', 'image/gif'];
    if (allowedTypes.includes(file.mimetype)) {
      cb(null, true);
    } else {
      cb(new Error('Invalid file type. Only JPEG, PNG and GIF are allowed.'));
    }
  }
});

// Routes
app.get('/api/events', async (req: Request, res: Response) => {
  try {
    const rows = await db.all('SELECT * FROM events');
    
    // Add the full path to wallpapers
    const events = rows.map((event: { 
      id: number;
      title: string;
      description: string;
      date: string;
      location: string;
      slug: string;
      needed_items: string | null;
      wallpaper: string | null;
      created_at: string;
    }) => ({
      ...event,
      wallpaper: event.wallpaper ? `/uploads/wallpapers/${event.wallpaper}` : null,
      needed_items: event.needed_items ? JSON.parse(event.needed_items) : []
    }));
    
    res.json(events);
  } catch (error) {
    console.error('Error fetching events:', error);
    res.status(500).json({ error: 'Internal server error' });
  }
});

app.get('/api/events/:slug', async (req: Request, res: Response) => {
  try {
    const { slug } = req.params;
    const rows = await db.all('SELECT * FROM events WHERE slug = ?', [slug]);
    if (rows.length === 0) {
      return res.status(404).json({ error: 'Event not found' });
    }

    // Parse needed_items from JSON string to array
    const event = rows[0];
    try {
      event.needed_items = event.needed_items ? JSON.parse(event.needed_items) : [];
    } catch (e) {
      console.error('Error parsing needed_items:', e);
      event.needed_items = [];
    }

    // Add the full path to the wallpaper
    if (event.wallpaper) {
      event.wallpaper = `/uploads/wallpapers/${event.wallpaper}`;
    }

    res.json(event);
  } catch (error) {
    console.error('Error fetching event:', error);
    res.status(500).json({ error: 'Internal server error' });
  }
});

app.post('/api/events', upload.single('wallpaper'), async (req: MulterRequest, res: Response) => {
  try {
    const { 
      title, 
      description, 
      date, 
      location, 
      needed_items, 
      rsvp_cutoff_date, 
      max_guests_per_rsvp,
      email_notifications_enabled,
      email_recipients
    } = req.body;
    const wallpaperPath = req.file ? `${req.file.filename}` : null;
    
    // Generate a slug from the title
    const slug = title.toLowerCase().replace(/[^a-z0-9]+/g, '-');
    
    // Ensure needed_items is properly formatted
    let parsedNeededItems: string[] = [];
    try {
      if (typeof needed_items === 'string') {
        parsedNeededItems = JSON.parse(needed_items);
      } else if (Array.isArray(needed_items)) {
        parsedNeededItems = needed_items;
      }
    } catch (e) {
      console.error('Error parsing needed_items:', e);
    }
    
    // Parse max_guests_per_rsvp to ensure it's a number
    const maxGuests = parseInt(max_guests_per_rsvp as string) || 0;
    
    // Parse email_notifications_enabled to ensure it's a boolean
    const emailNotificationsEnabled = email_notifications_enabled === 'true' || email_notifications_enabled === true;
    
    const result = await db.run(
      'INSERT INTO events (title, description, date, location, slug, needed_items, wallpaper, rsvp_cutoff_date, max_guests_per_rsvp, email_notifications_enabled, email_recipients) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)',
      [title, description, date, location, slug, JSON.stringify(parsedNeededItems), wallpaperPath, rsvp_cutoff_date, maxGuests, emailNotificationsEnabled ? 1 : 0, email_recipients || '']
    );
    
    res.status(201).json({ 
      ...result, 
      slug,
      wallpaper: wallpaperPath ? `/uploads/wallpapers/${wallpaperPath}` : null,
      needed_items: parsedNeededItems,
      rsvp_cutoff_date,
      max_guests_per_rsvp: maxGuests,
      email_notifications_enabled: emailNotificationsEnabled,
      email_recipients: email_recipients || ''
    });
  } catch (error) {
    console.error('Error creating event:', error);
    if (req.file) {
      // Clean up uploaded file if there was an error
      fs.unlink(req.file.path, (err) => {
        if (err) console.error('Error deleting file:', err);
      });
    }
    res.status(500).json({ error: 'Internal server error' });
  }
});

app.delete('/api/events/:slug', async (req: Request, res: Response) => {
  try {
    const { slug } = req.params;
    
    // First check if the event exists
    const eventRows = await db.all('SELECT id FROM events WHERE slug = ?', [slug]);
    
    if (eventRows.length === 0) {
      return res.status(404).json({ error: 'Event not found' });
    }
    
    // Delete the event (RSVPs will be automatically deleted due to ON DELETE CASCADE)
    await db.run('DELETE FROM events WHERE slug = ?', [slug]);
    
    res.status(204).send();
  } catch (error) {
    console.error('Error deleting event:', error);
    res.status(500).json({ error: 'Internal server error' });
  }
});

// RSVP routes
app.get('/api/events/:slug/rsvps', async (req: Request, res: Response) => {
  try {
    const { slug } = req.params;
    const eventRows = await db.all('SELECT id FROM events WHERE slug = ?', [slug]);
    
    if (eventRows.length === 0) {
      return res.status(404).json({ error: 'Event not found' });
    }
    
    const eventId = eventRows[0].id;
    const rows = await db.all('SELECT * FROM rsvps WHERE event_id = ?', [eventId]);
    
    // Parse JSON arrays in each RSVP
    const parsedRows = rows.map((rsvp: RSVP) => {
      try {
        return {
          ...rsvp,
          items_bringing: rsvp.items_bringing ? JSON.parse(rsvp.items_bringing) : [],
          guest_names: rsvp.guest_names ? JSON.parse(rsvp.guest_names) : []
        };
      } catch (e) {
        console.error('Error parsing RSVP JSON fields:', e);
        return {
          ...rsvp,
          items_bringing: [],
          guest_names: []
        };
      }
    });
    
    res.json(parsedRows);
  } catch (error) {
    console.error('Error fetching RSVPs:', error);
    res.status(500).json({ error: 'Internal server error' });
  }
});

app.post('/api/events/:slug/rsvp', async (req: Request, res: Response) => {
  try {
    const { slug } = req.params;
    const { 
      name, 
      attending, 
      bringing_guests, 
      guest_count, 
      guest_names, 
      items_bringing, 
      other_items,
      send_email_confirmation, // New field for email opt-in
      email_address // New field for recipient email
    } = req.body;
    
    // Get the event with email notification settings
    const eventRows = await db.all('SELECT id, title, slug, email_notifications_enabled, email_recipients FROM events WHERE slug = ?', [slug]);
    
    if (eventRows.length === 0) {
      return res.status(404).json({ error: 'Event not found' });
    }
    
    const event = eventRows[0];
    const eventId = event.id;
    const eventTitle = event.title;
    const eventSlug = event.slug;
    const emailNotificationsEnabled = event.email_notifications_enabled;
    const eventEmailRecipients = event.email_recipients;
    
    // Parse items_bringing if it's a string
    let parsedItemsBringing: string[] = [];
    try {
      if (typeof items_bringing === 'string') {
        parsedItemsBringing = JSON.parse(items_bringing);
      } else if (Array.isArray(items_bringing)) {
        parsedItemsBringing = items_bringing;
      }
    } catch (e) {
      console.error('Error parsing items_bringing:', e);
    }

    // Parse guest_names if it's a string
    let parsedGuestNames: string[] = [];
    try {
      if (typeof guest_names === 'string') {
        parsedGuestNames = JSON.parse(guest_names);
      } else if (Array.isArray(guest_names)) {
        parsedGuestNames = guest_names;
      }
    } catch (e) {
      console.error('Error parsing guest_names:', e);
    }
    
    // Generate a unique edit ID
    let editId = '';
    let isUnique = false;
    while (!isUnique) {
      editId = generateAlphanumericId(16);
      const existingRsvp = await db.get('SELECT id FROM rsvps WHERE edit_id = ?', [editId]);
      if (!existingRsvp) {
        isUnique = true;
      }
    }

    const result = await db.run(
      'INSERT INTO rsvps (event_id, name, attending, bringing_guests, guest_count, guest_names, items_bringing, other_items, edit_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)',
      [eventId, name, attending, bringing_guests, guest_count, JSON.stringify(parsedGuestNames), JSON.stringify(parsedItemsBringing), other_items || '', editId]
    );

    // Send email notifications to event recipients if enabled for this event
    if (emailNotificationsEnabled) {
      // Get recipients from event settings
      let recipients: string[] = [];
      
      // Use the event's email recipients
      if (eventEmailRecipients) {
        recipients = eventEmailRecipients.split(',').map((addr: string) => addr.trim()).filter(Boolean);
      }
      
      // If no recipients are set for the event, use the sender email as a fallback
      if (recipients.length === 0 && process.env.EMAIL_USER) {
        recipients = [process.env.EMAIL_USER];
      }
      
      if (recipients.length > 0) {
        try {
          for (const to of recipients) {
            await sendRSVPEmail({
              eventTitle,
              eventSlug,
              name,
              attending,
              bringingGuests: bringing_guests,
              guestCount: guest_count,
              guestNames: parsedGuestNames,
              itemsBringing: parsedItemsBringing,
              otherItems: other_items || '',
              to,
            });
          }
        } catch (emailErr) {
          console.error('Error sending RSVP email to event recipients:', emailErr);
        }
      } else {
        console.warn('No event email recipients set. Skipping RSVP email notification to event recipients.');
      }
    } else {
      console.log('Email notifications disabled for this event. Skipping RSVP email notification to event recipients.');
    }

    // Send email confirmation with edit link to the submitter if requested
    const sendEmailConfirmationBool = send_email_confirmation === 'true' || send_email_confirmation === true;
    const submitterEmail = email_address?.trim();

    if (sendEmailConfirmationBool && submitterEmail && process.env.EMAIL_USER) {
      try {
        const editLink = `${process.env.FRONTEND_BASE_URL}/events/${eventSlug}/rsvp/edit/${editId}`;
        await sendRSVPEditLinkEmail({
          eventTitle,
          eventSlug,
          name,
          to: submitterEmail,
          editLink,
        });
        console.log(`Sent RSVP edit link email to ${submitterEmail}`);
      } catch (emailErr) {
        console.error('Error sending RSVP edit link email:', emailErr);
      }
    } else if (sendEmailConfirmationBool && !submitterEmail) {
      console.warn('Email confirmation requested but no email address provided. Skipping edit link email.');
    } else if (sendEmailConfirmationBool && !process.env.EMAIL_USER) {
       console.warn('Email confirmation requested but EMAIL_USER environment variable is not set. Cannot send edit link email.');
    } else {
      console.log('Email confirmation not requested. Skipping edit link email.');
    }


    // Return the complete RSVP data including the parsed arrays and edit_id
    res.status(201).json({
      id: result.lastID,
      event_id: eventId,
      name,
      attending,
      bringing_guests,
      guest_count,
      guest_names: parsedGuestNames,
      items_bringing: parsedItemsBringing,
      other_items: other_items || '',
      edit_id: editId,
      created_at: new Date().toISOString()
    });
  } catch (error) {
    console.error('Error creating RSVP:', error);
    res.status(500).json({ error: 'Internal server error' });
  }
});

// Get RSVP by edit ID
app.get('/api/rsvps/edit/:editId', async (req: Request, res: Response) => {
  try {
    const { editId } = req.params;
    const rsvp = await db.get('SELECT * FROM rsvps WHERE edit_id = ?', [editId]);

    if (!rsvp) {
      return res.status(404).json({ error: 'RSVP not found' });
    }

    // Parse arrays for response
    try {
      rsvp.items_bringing = rsvp.items_bringing ? JSON.parse(rsvp.items_bringing) : [];
      rsvp.guest_names = rsvp.guest_names ? JSON.parse(rsvp.guest_names) : [];
    } catch (e) {
      console.error('Error parsing arrays in response:', e);
      rsvp.items_bringing = [];
      rsvp.guest_names = [];
    }

    res.json(rsvp);
  } catch (error) {
    console.error('Error fetching RSVP by edit ID:', error);
    res.status(500).json({ error: 'Internal server error' });
  }
});


app.delete('/api/events/:slug/rsvps/:id', async (req: Request, res: Response) => {
  try {
    const { slug, id } = req.params;
    
    // Verify the RSVP belongs to the correct event
    const eventRows = await db.all('SELECT id FROM events WHERE slug = ?', [slug]);
    
    if (eventRows.length === 0) {
      return res.status(404).json({ error: 'Event not found' });
    }
    
    const eventId = eventRows[0].id;
    await db.run('DELETE FROM rsvps WHERE id = ? AND event_id = ?', [id, eventId]);
    res.status(204).send();
  } catch (error) {
    console.error('Error deleting RSVP:', error);
    res.status(500).json({ error: 'Internal server error' });
  }
});

// Update RSVP by edit ID
app.put('/api/rsvps/edit/:editId', async (req: Request, res: Response) => {
  try {
    const { editId } = req.params;
    const { name, attending, bringing_guests, guest_count, guest_names, items_bringing, other_items } = req.body;

    // Find the RSVP by edit_id
    const rsvp = await db.get('SELECT id, event_id FROM rsvps WHERE edit_id = ?', [editId]);

    if (!rsvp) {
      return res.status(404).json({ error: 'RSVP not found' });
    }

    const rsvpId = rsvp.id;
    const eventId = rsvp.event_id;

    // Parse items_bringing if it's a string
    let parsedItemsBringing: string[] = [];
    try {
      if (typeof items_bringing === 'string') {
        parsedItemsBringing = JSON.parse(items_bringing);
      } else if (Array.isArray(items_bringing)) {
        parsedItemsBringing = items_bringing;
      }
    } catch (e) {
      console.error('Error parsing items_bringing:', e);
    }

    // Parse guest_names if it's a string
    let parsedGuestNames: string[] = [];
    try {
      if (typeof guest_names === 'string' && guest_names.includes('[')) {
        // If it's a JSON string array
        parsedGuestNames = JSON.parse(guest_names);
      } else if (typeof guest_names === 'string') {
        // If it's a comma-separated string
        parsedGuestNames = guest_names.split(',').map(name => name.trim()).filter(name => name);
      } else if (Array.isArray(guest_names)) {
        // If it's already an array
        parsedGuestNames = guest_names.filter(name => name && name.trim());
      }
    } catch (e) {
      console.error('Error parsing guest_names:', e);
      parsedGuestNames = [];
    }

    // Update the RSVP
    await db.run(
      'UPDATE rsvps SET name = ?, attending = ?, bringing_guests = ?, guest_count = ?, guest_names = ?, items_bringing = ?, other_items = ? WHERE id = ?',
      [name, attending, bringing_guests, guest_count, JSON.stringify(parsedGuestNames), JSON.stringify(parsedItemsBringing), other_items || '', rsvpId]
    );

    // Get the updated RSVP to verify and return
    const updatedRsvp = await db.get('SELECT * FROM rsvps WHERE id = ?', [rsvpId]);

    if (!updatedRsvp) {
      return res.status(404).json({ error: 'RSVP not found after update' });
    }

    // Parse arrays for response
    try {
      updatedRsvp.items_bringing = updatedRsvp.items_bringing ? JSON.parse(updatedRsvp.items_bringing) : [];
      updatedRsvp.guest_names = updatedRsvp.guest_names ? JSON.parse(updatedRsvp.guest_names) : [];
    } catch (e) {
      console.error('Error parsing arrays in response:', e);
      updatedRsvp.items_bringing = [];
      updatedRsvp.guest_names = [];
    }

    res.json(updatedRsvp);
  } catch (error) {
    console.error('Error updating RSVP by edit ID:', error);
    res.status(500).json({ error: 'Internal server error' });
  }
});


// Update RSVP
app.put('/api/events/:slug/rsvps/:id', async (req: Request, res: Response) => {
  try {
    const { slug, id } = req.params;
    const { name, attending, bringing_guests, guest_count, guest_names, items_bringing, other_items } = req.body;
    
    // Verify the RSVP belongs to the correct event
    const eventRows = await db.all('SELECT id FROM events WHERE slug = ?', [slug]);
    
    if (eventRows.length === 0) {
      return res.status(404).json({ error: 'Event not found' });
    }
    
    const eventId = eventRows[0].id;
    
    // Parse items_bringing if it's a string
    let parsedItemsBringing: string[] = [];
    try {
      if (typeof items_bringing === 'string') {
        parsedItemsBringing = JSON.parse(items_bringing);
      } else if (Array.isArray(items_bringing)) {
        parsedItemsBringing = items_bringing;
      }
    } catch (e) {
      console.error('Error parsing items_bringing:', e);
    }

    // Parse guest_names if it's a string
    let parsedGuestNames: string[] = [];
    try {
      if (typeof guest_names === 'string' && guest_names.includes('[')) {
        // If it's a JSON string array
        parsedGuestNames = JSON.parse(guest_names);
      } else if (typeof guest_names === 'string') {
        // If it's a comma-separated string
        parsedGuestNames = guest_names.split(',').map(name => name.trim()).filter(name => name);
      } else if (Array.isArray(guest_names)) {
        // If it's already an array
        parsedGuestNames = guest_names.filter(name => name && name.trim());
      }
    } catch (e) {
      console.error('Error parsing guest_names:', e);
      parsedGuestNames = [];
    }

    // Update the RSVP
    await db.run(
      'UPDATE rsvps SET name = ?, attending = ?, bringing_guests = ?, guest_count = ?, guest_names = ?, items_bringing = ?, other_items = ? WHERE id = ? AND event_id = ?',
      [name, attending, bringing_guests, guest_count, JSON.stringify(parsedGuestNames), JSON.stringify(parsedItemsBringing), other_items || '', id, eventId]
    );

    // Get the updated RSVP to verify and return
    const updatedRsvp = await db.get('SELECT * FROM rsvps WHERE id = ? AND event_id = ?', [id, eventId]);
    
    if (!updatedRsvp) {
      return res.status(404).json({ error: 'RSVP not found after update' });
    }

    // Parse arrays for response
    try {
      updatedRsvp.items_bringing = updatedRsvp.items_bringing ? JSON.parse(updatedRsvp.items_bringing) : [];
      updatedRsvp.guest_names = updatedRsvp.guest_names ? JSON.parse(updatedRsvp.guest_names) : [];
    } catch (e) {
      console.error('Error parsing arrays in response:', e);
      updatedRsvp.items_bringing = [];
      updatedRsvp.guest_names = [];
    }

    res.json(updatedRsvp);
  } catch (error) {
    console.error('Error updating RSVP:', error);
    res.status(500).json({ error: 'Internal server error' });
  }
});

// Update event
app.put('/api/events/:slug', upload.single('wallpaper'), async (req: MulterRequest, res: Response) => {
  try {
    const { slug } = req.params;
    const { 
      title, 
      description, 
      date, 
      location, 
      needed_items, 
      rsvp_cutoff_date, 
      max_guests_per_rsvp,
      email_notifications_enabled,
      email_recipients
    } = req.body;
    
    // Verify the event exists
    const eventRows = await db.all('SELECT * FROM events WHERE slug = ?', [slug]);
    
    if (eventRows.length === 0) {
      return res.status(404).json({ error: 'Event not found' });
    }
    
    // Ensure needed_items is properly formatted
    let parsedNeededItems: string[] = [];
    try {
      if (typeof needed_items === 'string') {
        parsedNeededItems = JSON.parse(needed_items);
      } else if (Array.isArray(needed_items)) {
        parsedNeededItems = needed_items;
      }
    } catch (e) {
      console.error('Error parsing needed_items:', e);
    }

    // Parse max_guests_per_rsvp to ensure it's a number
    const maxGuests = max_guests_per_rsvp !== undefined ? 
      (parseInt(max_guests_per_rsvp as string) || 0) : 
      eventRows[0].max_guests_per_rsvp || 0;
      
    // Parse email_notifications_enabled to ensure it's a boolean
    const emailNotificationsEnabled = email_notifications_enabled !== undefined ?
      (email_notifications_enabled === 'true' || email_notifications_enabled === true) :
      eventRows[0].email_notifications_enabled;
      
    // Get email recipients
    const emailRecipients = email_recipients !== undefined ?
      email_recipients :
      eventRows[0].email_recipients || '';

    // Handle wallpaper update
    let wallpaperPath = eventRows[0].wallpaper;
    if (req.file) {
      // If there's an existing wallpaper, delete it
      if (eventRows[0].wallpaper) {
        const oldWallpaperPath = path.join(uploadDir, eventRows[0].wallpaper);
        try {
          await fs.promises.unlink(oldWallpaperPath);
        } catch (e) {
          console.error('Error deleting old wallpaper:', e);
        }
      }
      wallpaperPath = req.file.filename;
    }
    
    // Update the event
    await db.run(
      'UPDATE events SET title = ?, description = ?, date = ?, location = ?, needed_items = ?, rsvp_cutoff_date = ?, wallpaper = ?, max_guests_per_rsvp = ?, email_notifications_enabled = ?, email_recipients = ? WHERE slug = ?',
      [
        title ?? eventRows[0].title,
        description === undefined ? eventRows[0].description : description,
        date ?? eventRows[0].date,
        location ?? eventRows[0].location,
        JSON.stringify(parsedNeededItems),
        rsvp_cutoff_date !== undefined ? rsvp_cutoff_date : eventRows[0].rsvp_cutoff_date,
        wallpaperPath,
        maxGuests,
        emailNotificationsEnabled ? 1 : 0,
        emailRecipients,
        slug
      ]
    );

    // Get the updated event
    const updatedEvent = await db.get('SELECT * FROM events WHERE slug = ?', [slug]);
    
    // Add the full path to the wallpaper
    if (updatedEvent.wallpaper) {
      updatedEvent.wallpaper = `/uploads/wallpapers/${updatedEvent.wallpaper}`;
    }
    
    // Parse needed_items for response
    try {
      updatedEvent.needed_items = updatedEvent.needed_items ? JSON.parse(updatedEvent.needed_items) : [];
    } catch (e) {
      console.error('Error parsing needed_items in response:', e);
      updatedEvent.needed_items = [];
    }
    
    res.json(updatedEvent);
  } catch (error) {
    console.error('Error updating event:', error);
    // Clean up uploaded file if there was an error
    if (req.file) {
      fs.unlink(req.file.path, (err) => {
        if (err) console.error('Error deleting file:', err);
      });
    }
    res.status(500).json({ error: 'Internal server error' });
  }
});

// Initialize database tables
async function initializeDatabase() {
  try {
    // Create events table
    await db.exec(`
      CREATE TABLE IF NOT EXISTS events (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        title TEXT NOT NULL,
        description TEXT,
        date TEXT NOT NULL,
        location TEXT,
        slug TEXT NOT NULL UNIQUE,
        needed_items TEXT,
        wallpaper TEXT,
        rsvp_cutoff_date TEXT,
        max_guests_per_rsvp INTEGER DEFAULT 0,
        email_notifications_enabled BOOLEAN DEFAULT 0,
        email_recipients TEXT,
        created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
      )
    `);

    // Create RSVPs table
    await db.exec(`
      CREATE TABLE IF NOT EXISTS rsvps (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        event_id INTEGER NOT NULL,
        name TEXT NOT NULL,
        attending TEXT NOT NULL,
        bringing_guests TEXT NOT NULL,
        guest_count INTEGER DEFAULT 0,
        guest_names TEXT,
        items_bringing TEXT,
        other_items TEXT,
        edit_id TEXT UNIQUE, -- Add a column for the unique edit ID
        created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
        FOREIGN KEY (event_id) REFERENCES events(id) ON DELETE CASCADE
      )
    `);
    console.log('Database initialized successfully');
  } catch (error) {
    console.error('Error initializing database:', error);
    throw error; // Re-throw to handle in the connection function
  }
}

// Handle client-side routing
app.get('*', (req: Request, res: Response) => {
  res.sendFile(path.join(__dirname, '../frontend/build/index.html'));
});

// Start server
app.listen(port, async () => {
  console.log(`Server running on port ${port}`);
  await connectToDatabase();
});