hydra/src/main/services/notifications.ts

import { Notification, nativeImage } from "electron";
import { t } from "i18next";
import { parseICO } from "icojs";

import { Game } from "@main/entity";
import { gameRepository, userPreferencesRepository } from "@main/repository";

const getGameIconNativeImage = async (gameId: number) => {
  try {
    const game = await gameRepository.findOne({
      where: {
        id: gameId,
      },
    });

    if (!game?.iconUrl) return undefined;

    const images = await parseICO(
      Buffer.from(game.iconUrl.split("base64,")[1], "base64")
    );

    const highResIcon = images.find((image) => image.width >= 128);
    if (!highResIcon) return undefined;

    return nativeImage.createFromBuffer(Buffer.from(highResIcon.buffer));
  } catch (err) {
    return undefined;
  }
};

export const publishDownloadCompleteNotification = async (game: Game) => {
  const userPreferences = await userPreferencesRepository.findOne({
    where: { id: 1 },
  });

  const icon = await getGameIconNativeImage(game.id);

  if (userPreferences?.downloadNotificationsEnabled) {
    new Notification({
      title: t("download_complete", {
        ns: "notifications",
        lng: userPreferences.language,
      }),
      body: t("game_ready_to_install", {
        ns: "notifications",
        lng: userPreferences.language,
        title: game.title,
      }),
      icon,
    }).show();
  }
};

export const publishNewRepacksNotifications = async (count: number) => {
  const userPreferences = await userPreferencesRepository.findOne({
    where: { id: 1 },
  });

  if (userPreferences?.repackUpdatesNotificationsEnabled) {
    new Notification({
      title: t("repack_list_updated", {
        ns: "notifications",
        lng: userPreferences?.language || "en",
      }),
      body: t("repack_count", {
        ns: "notifications",
        lng: userPreferences?.language || "en",
        count: count,
      }),
    }).show();
  }
};