import { app, ipcMain } from "electron";
import { DownloadManager } from "../managers/download";
import { resolve, join } from "path";
import { WindowManager } from "../managers/window";
import request from "request";
import { createWriteStream, unlink } from "fs";
const progress = require("request-progress");

export function handleDownloaderIPCs() {
  /**
   * Handles the "pauseDownload" IPC message by pausing the download associated with the provided fileName.
   * @param _event - The IPC event object.
   * @param fileName - The name of the file being downloaded.
   */
  ipcMain.handle("pauseDownload", async (_event, fileName) => {
    DownloadManager.instance.networkRequests[fileName]?.pause();
  });

  /**
   * Handles the "resumeDownload" IPC message by resuming the download associated with the provided fileName.
   * @param _event - The IPC event object.
   * @param fileName - The name of the file being downloaded.
   */
  ipcMain.handle("resumeDownload", async (_event, fileName) => {
    DownloadManager.instance.networkRequests[fileName]?.resume();
  });

  /**
   * Handles the "abortDownload" IPC message by aborting the download associated with the provided fileName.
   * The network request associated with the fileName is then removed from the networkRequests object.
   * @param _event - The IPC event object.
   * @param fileName - The name of the file being downloaded.
   */
  ipcMain.handle("abortDownload", async (_event, fileName) => {
    const rq = DownloadManager.instance.networkRequests[fileName];
    DownloadManager.instance.networkRequests[fileName] = undefined;
    const userDataPath = app.getPath("userData");
    const fullPath = join(userDataPath, fileName);
    rq?.abort();
    let result = "NULL";
    unlink(fullPath, function (err) {
      if (err && err.code == "ENOENT") {
        result = `File not exist: ${err}`;
      } else if (err) {
        result = `File delete error: ${err}`;
      } else {
        result = "File deleted successfully";
      }
      console.log(`Delete file ${fileName} from ${fullPath} result: ${result}`);
    });
  });

  /**
   * Downloads a file from a given URL.
   * @param _event - The IPC event object.
   * @param url - The URL to download the file from.
   * @param fileName - The name to give the downloaded file.
   */
  ipcMain.handle("downloadFile", async (_event, url, fileName) => {
    const userDataPath = app.getPath("userData");
    const destination = resolve(userDataPath, fileName);
    const rq = request(url);

    progress(rq, {})
      .on("progress", function (state: any) {
        WindowManager?.instance.currentWindow?.webContents.send(
          "FILE_DOWNLOAD_UPDATE",
          {
            ...state,
            fileName,
          }
        );
      })
      .on("error", function (err: Error) {
        WindowManager?.instance.currentWindow?.webContents.send(
          "FILE_DOWNLOAD_ERROR",
          {
            fileName,
            err,
          }
        );
      })
      .on("end", function () {
        if (DownloadManager.instance.networkRequests[fileName]) {
          WindowManager?.instance.currentWindow?.webContents.send(
            "FILE_DOWNLOAD_COMPLETE",
            {
              fileName,
            }
          );
          DownloadManager.instance.setRequest(fileName, undefined);
        } else {
          WindowManager?.instance.currentWindow?.webContents.send(
            "FILE_DOWNLOAD_ERROR",
            {
              fileName,
              err: "Download cancelled",
            }
          );
        }
      })
      .pipe(createWriteStream(destination));

    DownloadManager.instance.setRequest(fileName, rq);
  });
}