import { promises as fs, watch } from "fs";
import { join } from "path";
import sharp from "sharp";

export let mediaCache: MediaFile[] = [];
export let cacheInitialized = false;
export const mediaDirectory = join(process.cwd(), "public", "uploads");
export const publicDirectory = join(process.cwd(), "public");

interface MediaFile {
  id: string;
  name: string;
  path: string;
  width?: number;
  height?: number;
  dateModified?: Date;
}

export function filterMediaCache(imagePath) {
  mediaCache = mediaCache.filter((file) => file.id !== imagePath);
}

async function updateMediaCache(directory: string) {
  const fileList: MediaFile[] = [];
  async function readMediaFiles(dir: string) {
    const files = await fs.readdir(dir, { withFileTypes: true });
    for (const file of files) {
      const filePath = join(dir, file.name);
      if (file.isDirectory()) {
        await readMediaFiles(filePath);
      } else if (/\.(jpg|jpeg|png|gif|webp)$/i.test(file.name)) {
        // Only read dimensions for image files
        try {
          const { mtime } = await fs.stat(filePath);
          let webPath = filePath
            .substring(mediaDirectory.length)
            .replace(/\\/g, "/");
          if (!webPath.startsWith("/")) webPath = "/" + webPath;

          const image = sharp(filePath);
          const metadata = await image.metadata();

          fileList.push({
            id: "/uploads" + webPath.replace(/\//g, "_"),
            name: file.name,
            path: "/uploads" + webPath,
            width: metadata.width,
            height: metadata.height,
            dateModified: mtime,
          });
        } catch (error) {
          console.error(`Error accessing file: ${filePath}`, error);
        }
      }
    }
  }
  await readMediaFiles(directory);
  mediaCache = fileList;
  cacheInitialized = true;
}

// Initialize cache and set up watcher
export async function initMediaWatcher() {
  await updateMediaCache(mediaDirectory);
  watch(mediaDirectory, { recursive: true }, async (eventType, filename) => {
    await updateMediaCache(mediaDirectory);
  });
}