import type {RequestHandler as Middleware, Request, Response, NextFunction} from "express";
import multer from "multer";
import express from "express";
import ffmpeg from "fluent-ffmpeg";
import imageProbe from "probe-image-size";
import ffmpegpath from "@ffmpeg-installer/ffmpeg";
import ffprobepath from "@ffprobe-installer/ffprobe";

ffmpeg.setFfmpegPath(ffmpegpath.path);
ffmpeg.setFfprobePath(ffprobepath.path);

import fs from "fs";

import {extension, videoExtensions} from "../types/lib";
import {db, MediaRow, getPath, deleteId} from "../types/db";
import {fileStorage} from "../types/multer";
import {checkAuth, checkSharexAuth, convertTo720p, createEmbedData, handleUpload} from "../types/middleware";

const upload = multer({ storage: fileStorage /**, fileFilter: fileFilter**/ }); //maybe make this a env variable?

/**Middleware to grab media from media database */
const fetchMedia: Middleware = (req, res, next) => {
  try {
    const admin: boolean = req.user.username == "admin" ? true : false;
    /**Check if the user is an admin, if so, show all posts from all users */
    const query: string = admin ? "SELECT * FROM media" : `SELECT * FROM media WHERE username = ?`;
    const rows = (admin ? db.prepare(query).all() : db.prepare(query).all(req.user.username)) as MediaRow[];
    const files = rows.map((row: MediaRow) => {
      return {
        id: row.id,
        path: row.path,
        expire: row.expire,
        sername: row.username,
        url: "/" + row.id
      };
    });

    res.locals.files = files.reverse(); //reverse so newest files appear first
    res.locals.Count = files.length;
    next();
  } catch (err) {
    next(err);
  }

  /**db.all(query, (err:Error, rows: []) => {
    if (err) return next(err);
    const files = rows.map((row: MediaRow)=> {
      return {
        id: row.id,
        path: row.path,
        expire: row.expire,
        username: row.username,
        url: "/" + row.id
      };
    });
    res.locals.files = files.reverse(); //reverse so newest files appear first
    res.locals.Count = files.length;
    next();
  });**/
};

const router = express.Router();

router.get("/", (req: Request, res: Response, next: NextFunction) => {
  if (!req.user)
    return res.render("home");
  next();
}, fetchMedia, (req: Request, res: Response) => {
  res.locals.filter = null;
  res.render("index", { user: req.user });
});

router.get("/gifv/:file", async (req: Request, res: Response, next: NextFunction) => {
  const url = `${req.protocol}://${req.get("host")}/uploads/${req.params.file}`;
  let width; let height;

  const nameAndExtension = extension(`uploads/${req.params.file}`);
  if (nameAndExtension[1] == ".mp4" || nameAndExtension[1] == ".mov" || nameAndExtension[1] == ".webm" || nameAndExtension[1] == ".gif") {
    ffmpeg()
      .input(`uploads/${req.params.file}`)
      .inputFormat(nameAndExtension[1].substring(1))
      .ffprobe((err: Error, data: ffmpeg.FfprobeData) => {
        if (err) return next(err);
        width = data.streams[0].width;
        height = data.streams[0].height;
        return res.render("gifv", { url: url, host: `${req.protocol}://${req.get("host")}`, width: width, height: height });
      }); 
  } else {
    const imageData = await imageProbe(fs.createReadStream(`uploads/${req.params.file}`));
    return res.render("gifv", { url: url, host: `${req.protocol}://${req.get("host")}`, width: imageData.width, height: imageData.height });
  } 
});

router.post("/", [checkAuth, upload.array("fileupload"), convertTo720p, createEmbedData, handleUpload], (req: Request, res: Response) => {
  res.redirect("/");
});

router.post("/sharex", [checkSharexAuth, upload.single("fileupload"), createEmbedData, handleUpload], (req: Request, res: Response) => {
  return res.send(`${req.protocol}://${req.get("host")}/uploads/${req.file.filename}`);
});

router.post("/:id(\\d+)/delete", [checkAuth], async (req: Request, res: Response) => {
  const path: any = await getPath(req.params.id);

  const nameAndExtension = extension(path.path);

  const filesToDelete = [path.path, "oembed-" + path.path + ".json"];

  if (videoExtensions.includes(nameAndExtension[1]) || nameAndExtension[1] == ".gif") {
    filesToDelete.push("720p-" + path.path);
  } 

  filesToDelete.forEach(path => {
    fs.unlink(path, async (err) => {
      console.log(`Deleting ${path}`);
      if (err && err.errno == -4058) {
        await deleteId("media", req.params.id);
      }
      await deleteId("media", req.params.id);
    });
  });

  return res.redirect("/");
});

export default router;