import express from 'express';
import multer from 'multer';
import path from 'path';
import fs from 'fs';
import { db, getConfig } from './db';
import { processUploadedImage } from './imageService';

export const galleryRouter = express.Router();

const upload = multer({ dest: path.join(__dirname, '../uploads/tmp') });

galleryRouter.get('/', (req, res) => {
  const catParam = req.query.cat ? Number(req.query.cat) : 0;
  const page = req.query.page ? Math.max(1, Number(req.query.page)) : 1;
  const config = getConfig();
  const perPage = Number(config.albums_per_page) || 12;

  const categories = db.prepare('SELECT * FROM categories WHERE parent = ? ORDER BY pos ASC, name ASC').all(catParam) as any[];

  const albCountRow = db.prepare('SELECT COUNT(*) as count FROM albums WHERE category = ?').get(catParam) as any;
  const totalAlbums = albCountRow ? albCountRow.count : 0;
  const totalPages = Math.ceil(totalAlbums / perPage) || 1;
  const offset = (page - 1) * perPage;

  const albums = db.prepare(`
    SELECT a.*,
      (SELECT COUNT(*) FROM pictures p WHERE p.aid = a.aid) as pic_count,
      (SELECT filename FROM pictures p WHERE p.aid = a.aid ORDER BY pid DESC LIMIT 1) as thumb_filename
    FROM albums a
    WHERE a.category = ?
    ORDER BY a.pos ASC, a.aid DESC
    LIMIT ? OFFSET ?
  `).all(catParam, perPage, offset) as any[];

  const recentPics = db.prepare('SELECT * FROM pictures ORDER BY pid DESC LIMIT 8').all() as any[];

  res.render('index', {
    title: config.gallery_name,
    description: config.gallery_description,
    categories,
    albums,
    recentPics,
    currentCat: catParam,
    page,
    totalPages,
    user: (req.session as any).user || null
  });
});

galleryRouter.get('/album/:aid', (req, res) => {
  const aid = Number(req.params.aid);
  const album = db.prepare('SELECT * FROM albums WHERE aid = ?').get(aid) as any;
  if (!album) {
    return res.status(404).render('message', { title: 'Not Found', message: 'Album does not exist.', backUrl: '/' });
  }

  db.prepare('UPDATE albums SET alb_hits = alb_hits + 1 WHERE aid = ?').run(aid);

  const page = req.query.page ? Math.max(1, Number(req.query.page)) : 1;
  const config = getConfig();
  const perPage = Number(config.pics_per_page) || 12;

  const picCountRow = db.prepare('SELECT COUNT(*) as count FROM pictures WHERE aid = ?').get(aid) as any;
  const totalPics = picCountRow ? picCountRow.count : 0;
  const totalPages = Math.ceil(totalPics / perPage) || 1;
  const offset = (page - 1) * perPage;

  const pictures = db.prepare('SELECT * FROM pictures WHERE aid = ? ORDER BY pid DESC LIMIT ? OFFSET ?').all(aid, perPage, offset) as any[];

  res.render('album', {
    title: album.title,
    album,
    pictures,
    page,
    totalPages,
    user: (req.session as any).user || null
  });
});

galleryRouter.get('/picture/:pid', (req, res) => {
  const pid = Number(req.params.pid);
  const picture = db.prepare('SELECT * FROM pictures WHERE pid = ?').get(pid) as any;
  if (!picture) {
    return res.status(404).render('message', { title: 'Not Found', message: 'Picture does not exist.', backUrl: '/' });
  }

  db.prepare('UPDATE pictures SET hits = hits + 1 WHERE pid = ?').run(pid);

  const album = db.prepare('SELECT * FROM albums WHERE aid = ?').get(picture.aid) as any;
  const comments = db.prepare('SELECT * FROM comments WHERE pid = ? ORDER BY msg_id ASC').all(pid) as any[];

  const prevPic = db.prepare('SELECT pid FROM pictures WHERE aid = ? AND pid < ? ORDER BY pid DESC LIMIT 1').get(picture.aid, pid) as any;
  const nextPic = db.prepare('SELECT pid FROM pictures WHERE aid = ? AND pid > ? ORDER BY pid ASC LIMIT 1').get(picture.aid, pid) as any;

  res.render('displayimage', {
    title: picture.title || picture.filename,
    picture,
    album,
    comments,
    prevPid: prevPic ? prevPic.pid : null,
    nextPid: nextPic ? nextPic.pid : null,
    user: (req.session as any).user || null
  });
});

galleryRouter.post('/picture/:pid/comment', (req, res) => {
  const pid = Number(req.params.pid);
  const { msg_author, msg_body } = req.body;
  const user = (req.session as any).user;

  const authorName = user ? user.name : (msg_author || 'Guest');
  const authorId = user ? user.id : 0;

  if (msg_body && msg_body.trim().length > 0) {
    db.prepare('INSERT INTO comments (pid, msg_author, msg_body, author_id) VALUES (?, ?, ?, ?)').run(pid, authorName, msg_body.trim(), authorId);
  }

  res.redirect(`/picture/${pid}`);
});

galleryRouter.post('/picture/:pid/rate', (req, res) => {
  const pid = Number(req.params.pid);
  const rating = Math.min(5, Math.max(1, Number(req.body.rating) || 0));

  if (rating > 0) {
    const pic = db.prepare('SELECT pic_rating, votes FROM pictures WHERE pid = ?').get(pid) as any;
    if (pic) {
      const newVotes = pic.votes + 1;
      const newRating = Math.round(((pic.pic_rating * pic.votes) + (rating * 2000)) / newVotes);
      db.prepare('UPDATE pictures SET pic_rating = ?, votes = ? WHERE pid = ?').run(newRating, newVotes, pid);
    }
  }

  res.redirect(`/picture/${pid}`);
});

galleryRouter.get('/upload', (req, res) => {
  const user = (req.session as any).user;
  if (!user) {
    return res.redirect('/login');
  }

  const albums = db.prepare('SELECT * FROM albums ORDER BY title ASC').all() as any[];
  res.render('upload', { title: 'Upload File', albums, user });
});

galleryRouter.post('/upload', upload.single('userfile'), async (req, res) => {
  const user = (req.session as any).user;
  if (!user) {
    return res.redirect('/login');
  }

  const { aid, title, caption, keywords } = req.body;
  if (!req.file || !aid) {
    return res.render('message', { title: 'Upload Failed', message: 'Please select an image file and target album.', backUrl: '/upload' });
  }

  try {
    const ext = path.extname(req.file.originalname).toLowerCase();
    const uniqueFilename = `${Date.now()}_${Math.floor(Math.random() * 10000)}${ext}`;

    const processed = await processUploadedImage(req.file.path, uniqueFilename);

    fs.unlinkSync(req.file.path);

    db.prepare(`
      INSERT INTO pictures (aid, filepath, filename, filesize, pwidth, pheight, owner_id, title, caption, keywords)
      VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
    `).run(
      Number(aid),
      processed.filepath,
      processed.filename,
      processed.filesize,
      processed.pwidth,
      processed.pheight,
      user.id,
      title || '',
      caption || '',
      keywords || ''
    );

    res.redirect(`/album/${aid}`);
  } catch (err: any) {
    console.error('Upload Error:', err);
    res.render('message', { title: 'Upload Error', message: err.message || 'Failed to process uploaded file.', backUrl: '/upload' });
  }
});