hydra/src/main/events/torrenting/cancel-game-download.ts

import { gameRepository } from "@main/repository";

import { registerEvent } from "../register-event";
import { WindowManager } from "@main/services";

import { In } from "typeorm";
import { DownloadManager } from "@main/services";
import { GameStatus } from "@shared";

const cancelGameDownload = async (
  _event: Electron.IpcMainInvokeEvent,
  gameId: number
) => {
  const game = await gameRepository.findOne({
    where: {
      id: gameId,
      isDeleted: false,
      status: In([
        GameStatus.Downloading,
        GameStatus.DownloadingMetadata,
        GameStatus.CheckingFiles,
        GameStatus.Paused,
        GameStatus.Seeding,
        GameStatus.Finished,
      ]),
    },
  });

  if (!game) return;
  DownloadManager.cancelDownload();

  await gameRepository
    .update(
      {
        id: game.id,
      },
      {
        status: GameStatus.Cancelled,
        bytesDownloaded: 0,
        progress: 0,
      }
    )
    .then((result) => {
      if (
        game.status !== GameStatus.Paused &&
        game.status !== GameStatus.Seeding
      ) {
        if (result.affected) WindowManager.mainWindow?.setProgressBar(-1);
      }
    });
};

registerEvent(cancelGameDownload, {
  name: "cancelGameDownload",
});