first commit

src/renderer/hooks/use-download.ts

@@ -0,0 +1,134 @@
+import { addMilliseconds } from "date-fns";
+import prettyBytes from "pretty-bytes";
+
+import { formatDownloadProgress } from "@renderer/helpers";
+import { useLibrary } from "./use-library";
+import { useAppDispatch, useAppSelector } from "./redux";
+import {
+  addPacket,
+  clearDownload,
+  setGameDeleting,
+  removeGameFromDeleting,
+} from "@renderer/features";
+import type { GameShop, TorrentProgress } from "@types";
+import { useDate } from "./use-date";
+
+export function useDownload() {
+  const { updateLibrary } = useLibrary();
+  const { formatDistance } = useDate();
+
+  const { packets, gamesWithDeletionInProgress } = useAppSelector(
+    (state) => state.download
+  );
+  const dispatch = useAppDispatch();
+
+  const lastPacket = packets.at(-1);
+
+  const startDownload = (
+    repackId: number,
+    objectID: string,
+    title: string,
+    shop: GameShop
+  ) =>
+    window.electron
+      .startGameDownload(repackId, objectID, title, shop)
+      .then((game) => {
+        dispatch(clearDownload());
+        updateLibrary();
+
+        return game;
+      });
+
+  const pauseDownload = (gameId: number) =>
+    window.electron.pauseGameDownload(gameId).then(() => {
+      dispatch(clearDownload());
+      updateLibrary();
+    });
+
+  const resumeDownload = (gameId: number) =>
+    window.electron.resumeGameDownload(gameId).then(() => {
+      updateLibrary();
+    });
+
+  const cancelDownload = (gameId: number) =>
+    window.electron.cancelGameDownload(gameId).then(() => {
+      dispatch(clearDownload());
+      updateLibrary();
+      deleteGame(gameId);
+    });
+
+  const removeGame = (gameId: number) =>
+    window.electron.removeGame(gameId).then(() => {
+      updateLibrary();
+    });
+
+  const isVerifying = ["downloading_metadata", "checking_files"].includes(
+    lastPacket?.game.status
+  );
+
+  const getETA = () => {
+    if (isVerifying || !isFinite(lastPacket?.timeRemaining)) {
+      return "";
+    }
+
+    try {
+      return formatDistance(
+        addMilliseconds(new Date(), lastPacket?.timeRemaining ?? 1),
+        new Date(),
+        { addSuffix: true }
+      );
+    } catch (err) {
+      return "";
+    }
+  };
+
+  const getProgress = () => {
+    if (lastPacket?.game.status === "checking_files") {
+      return formatDownloadProgress(lastPacket?.game.fileVerificationProgress);
+    }
+
+    return formatDownloadProgress(lastPacket?.game.progress);
+  };
+
+  const deleteGame = (gameId: number) =>
+    window.electron
+      .cancelGameDownload(gameId)
+      .then(() => {
+        dispatch(setGameDeleting(gameId));
+        return window.electron.deleteGameFolder(gameId);
+      })
+      .catch(() => {})
+      .finally(() => {
+        updateLibrary();
+        dispatch(removeGameFromDeleting(gameId));
+      });
+
+  const isGameDeleting = (gameId: number) => {
+    return gamesWithDeletionInProgress.includes(gameId);
+  };
+
+  return {
+    game: lastPacket?.game,
+    bytesDownloaded: lastPacket?.game.bytesDownloaded,
+    fileSize: lastPacket?.game.fileSize,
+    isVerifying,
+    gameId: lastPacket?.game.id,
+    downloadSpeed: `${prettyBytes(lastPacket?.downloadSpeed ?? 0, {
+      bits: true,
+    })}/s`,
+    isDownloading: Boolean(lastPacket),
+    progress: getProgress(),
+    numPeers: lastPacket?.numPeers,
+    numSeeds: lastPacket?.numSeeds,
+    eta: getETA(),
+    startDownload,
+    pauseDownload,
+    resumeDownload,
+    cancelDownload,
+    removeGame,
+    deleteGame,
+    isGameDeleting,
+    clearDownload: () => dispatch(clearDownload()),
+    addPacket: (packet: TorrentProgress) => dispatch(addPacket(packet)),
+  };
+}