Merge branch 'main' into feat/HYD-822

src/main/constants.ts

@@ -40,4 +40,6 @@ export const backupsPath = path.join(SystemPath.getPath("userData"), "Backups");
 
 export const appVersion = app.getVersion() + (isStaging ? "-staging" : "");
 
+export const ASSETS_PATH = path.join(SystemPath.getPath("userData"), "Assets");
+
 export const MAIN_LOOP_INTERVAL = 1500;

src/main/events/index.ts

@@ -35,6 +35,7 @@ import "./library/select-game-wine-prefix";
 import "./library/reset-game-achievements";
 import "./library/toggle-automatic-cloud-sync";
 import "./library/get-default-wine-prefix-selection-path";
+import "./library/create-steam-shortcut";
 import "./misc/open-checkout";
 import "./misc/open-external";
 import "./misc/show-open-dialog";

src/main/events/library/create-steam-shortcut.ts

@@ -0,0 +1,181 @@
+import { registerEvent } from "../register-event";
+import type { GameShop, GameStats } from "@types";
+import { gamesSublevel, levelKeys } from "@main/level";
+import {
+  composeSteamShortcut,
+  getSteamLocation,
+  getSteamShortcuts,
+  getSteamUsersIds,
+  HydraApi,
+  logger,
+  SystemPath,
+  writeSteamShortcuts,
+} from "@main/services";
+import fs from "node:fs";
+import axios from "axios";
+import path from "node:path";
+import { ASSETS_PATH } from "@main/constants";
+
+const downloadAsset = async (downloadPath: string, url?: string | null) => {
+  try {
+    if (fs.existsSync(downloadPath)) {
+      return downloadPath;
+    }
+
+    if (!url) {
+      return null;
+    }
+
+    fs.mkdirSync(path.dirname(downloadPath), { recursive: true });
+
+    const response = await axios.get(url, { responseType: "arraybuffer" });
+    fs.writeFileSync(downloadPath, response.data);
+
+    return downloadPath;
+  } catch (error) {
+    logger.error("Failed to download asset", error);
+    return null;
+  }
+};
+
+const downloadAssetsFromSteam = async (
+  shop: GameShop,
+  objectId: string,
+  assets: GameStats["assets"]
+) => {
+  const gameAssetsPath = path.join(ASSETS_PATH, `${shop}-${objectId}`);
+
+  return await Promise.all([
+    downloadAsset(path.join(gameAssetsPath, "icon.ico"), assets?.iconUrl),
+    downloadAsset(
+      path.join(gameAssetsPath, "hero.jpg"),
+      assets?.libraryHeroImageUrl
+    ),
+    downloadAsset(path.join(gameAssetsPath, "logo.png"), assets?.logoImageUrl),
+    downloadAsset(
+      path.join(gameAssetsPath, "cover.jpg"),
+      assets?.coverImageUrl
+    ),
+    downloadAsset(
+      path.join(gameAssetsPath, "library.jpg"),
+      assets?.libraryImageUrl
+    ),
+  ]);
+};
+
+const copyAssetIfExists = async (
+  sourcePath: string | null,
+  destinationPath: string
+) => {
+  if (sourcePath && fs.existsSync(sourcePath)) {
+    logger.info("Copying Steam asset", sourcePath, destinationPath);
+    await fs.promises.cp(sourcePath, destinationPath);
+  }
+};
+
+const createSteamShortcut = async (
+  _event: Electron.IpcMainInvokeEvent,
+  shop: GameShop,
+  objectId: string
+) => {
+  const gameKey = levelKeys.game(shop, objectId);
+  const game = await gamesSublevel.get(gameKey);
+
+  if (game) {
+    if (!game.executablePath) {
+      throw new Error("No executable path found for game");
+    }
+
+    const { assets } = await HydraApi.get<GameStats>(
+      `/games/stats?objectId=${objectId}&shop=${shop}`
+    );
+
+    const steamUserIds = await getSteamUsersIds();
+
+    if (!steamUserIds.length) {
+      logger.error("No Steam user ID found");
+      return;
+    }
+
+    const [iconImage, heroImage, logoImage, coverImage, libraryImage] =
+      await downloadAssetsFromSteam(game.shop, game.objectId, assets);
+
+    const newShortcut = composeSteamShortcut(
+      game.title,
+      game.executablePath,
+      iconImage
+    );
+
+    for (const steamUserId of steamUserIds) {
+      logger.info("Adding shortcut for Steam user", steamUserId);
+
+      const steamShortcuts = await getSteamShortcuts(steamUserId);
+
+      if (steamShortcuts.some((shortcut) => shortcut.appname === game.title)) {
+        continue;
+      }
+
+      const gridPath = path.join(
+        await getSteamLocation(),
+        "userdata",
+        steamUserId.toString(),
+        "config",
+        "grid"
+      );
+
+      await fs.promises.mkdir(gridPath, { recursive: true });
+
+      await Promise.all([
+        copyAssetIfExists(
+          heroImage,
+          path.join(gridPath, `${newShortcut.appid}_hero.jpg`)
+        ),
+        copyAssetIfExists(
+          logoImage,
+          path.join(gridPath, `${newShortcut.appid}_logo.png`)
+        ),
+        copyAssetIfExists(
+          coverImage,
+          path.join(gridPath, `${newShortcut.appid}p.jpg`)
+        ),
+        copyAssetIfExists(
+          libraryImage,
+          path.join(gridPath, `${newShortcut.appid}.jpg`)
+        ),
+      ]);
+
+      steamShortcuts.push(newShortcut);
+
+      logger.info(newShortcut);
+      logger.info("Writing Steam shortcuts", steamShortcuts);
+
+      await writeSteamShortcuts(steamUserId, steamShortcuts);
+    }
+
+    if (process.platform === "linux" && !game.winePrefixPath) {
+      const steamWinePrefixes = path.join(
+        SystemPath.getPath("home"),
+        ".local",
+        "share",
+        "Steam",
+        "steamapps",
+        "compatdata"
+      );
+
+      const winePrefixPath = path.join(
+        steamWinePrefixes,
+        newShortcut.appid.toString(),
+        "pfx"
+      );
+
+      await fs.promises.mkdir(winePrefixPath, { recursive: true });
+
+      await gamesSublevel.put(gameKey, {
+        ...game,
+        winePrefixPath,
+      });
+    }
+  }
+};
+
+registerEvent("createSteamShortcut", createSteamShortcut);

src/main/events/library/get-default-wine-prefix-selection-path.ts

@@ -16,11 +16,7 @@ const getDefaultWinePrefixSelectionPath = async (
       "compatdata"
     );
 
-    if (fs.existsSync(steamWinePrefixes)) {
-      return fs.promises.realpath(steamWinePrefixes);
-    }
-
-    return null;
+    return await fs.promises.realpath(steamWinePrefixes);
   } catch (err) {
     logger.error("Failed to get default wine prefix selection path", err);
 

src/main/events/library/verify-executable-path.ts

@@ -7,11 +7,11 @@ const verifyExecutablePathInUse = async (
 ) => {
   for await (const game of gamesSublevel.values()) {
     if (game.executablePath === executablePath) {
-      return true;
+      return game;
     }
   }
 
-  return false;
+  return null;
 };
 
 registerEvent("verifyExecutablePathInUse", verifyExecutablePathInUse);

src/main/main.ts

@@ -27,7 +27,9 @@ export const loadState = async () => {
 
   await import("./events");
 
-  Aria2.spawn();
+  if (process.platform !== "darwin") {
+    Aria2.spawn();
+  }
 
   if (userPreferences?.realDebridApiToken) {
     RealDebridClient.authorize(userPreferences.realDebridApiToken);

src/main/services/python-rpc.ts

@@ -22,12 +22,6 @@ const binaryNameByPlatform: Partial<Record<NodeJS.Platform, string>> = {
   win32: "hydra-python-rpc.exe",
 };
 
-const rustBinaryNameByPlatform: Partial<Record<NodeJS.Platform, string>> = {
-  darwin: "hydra-httpdl",
-  linux: "hydra-httpdl",
-  win32: "hydra-httpdl.exe",
-};
-
 export class PythonRPC {
   public static readonly BITTORRENT_PORT = "5881";
   public static readonly RPC_PORT = "8084";
@@ -72,20 +66,6 @@ export class PythonRPC {
       rpcPassword,
       initialDownload ? JSON.stringify(initialDownload) : "",
       initialSeeding ? JSON.stringify(initialSeeding) : "",
-      app.isPackaged
-        ? path.join(
-            process.resourcesPath,
-            rustBinaryNameByPlatform[process.platform]!
-          )
-        : path.join(
-            __dirname,
-            "..",
-            "..",
-            "rust_rpc",
-            "target",
-            "debug",
-            rustBinaryNameByPlatform[process.platform]!
-          ),
     ];
 
     if (app.isPackaged) {

src/main/services/steam.ts

@@ -1,8 +1,14 @@
 import axios from "axios";
+import path from "node:path";
+import fs from "node:fs";
+import { crc32 } from "crc";
+import WinReg from "winreg";
+import { parseBuffer, writeBuffer } from "steam-shortcut-editor";
 
-import type { SteamAppDetails } from "@types";
+import type { SteamAppDetails, SteamShortcut } from "@types";
 
 import { logger } from "./logger";
+import { SystemPath } from "./system-path";
 
 export interface SteamAppDetailsResponse {
   [key: string]: {
@@ -11,6 +17,36 @@ export interface SteamAppDetailsResponse {
   };
 }
 
+export const getSteamLocation = async () => {
+  if (process.platform === "linux") {
+    return path.join(SystemPath.getPath("home"), ".local", "share", "Steam");
+  }
+
+  if (process.platform === "darwin") {
+    return path.join(
+      SystemPath.getPath("home"),
+      "Library",
+      "Application Support",
+      "Steam"
+    );
+  }
+
+  const regKey = new WinReg({
+    hive: WinReg.HKCU,
+    key: "\\Software\\Valve\\Steam",
+  });
+
+  return new Promise<string>((resolve, reject) => {
+    regKey.get("SteamPath", (err, value) => {
+      if (err) {
+        reject(err);
+      }
+
+      resolve(value.value);
+    });
+  });
+};
+
 export const getSteamAppDetails = async (
   objectId: string,
   language: string
@@ -40,3 +76,86 @@ export const getSteamAppDetails = async (
       return null;
     });
 };
+
+export const getSteamUsersIds = async () => {
+  const userDataPath = await getSteamLocation();
+
+  const userIds = fs.readdirSync(path.join(userDataPath, "userdata"), {
+    withFileTypes: true,
+  });
+
+  return userIds
+    .filter((dir) => dir.isDirectory())
+    .map((dir) => Number(dir.name));
+};
+
+export const getSteamShortcuts = async (steamUserId: number) => {
+  const shortcutsPath = path.join(
+    await getSteamLocation(),
+    "userdata",
+    steamUserId.toString(),
+    "config",
+    "shortcuts.vdf"
+  );
+
+  if (!fs.existsSync(shortcutsPath)) {
+    return [];
+  }
+
+  const shortcuts = parseBuffer(fs.readFileSync(shortcutsPath));
+
+  return shortcuts.shortcuts as SteamShortcut[];
+};
+
+export const generateSteamShortcutAppId = (
+  exePath: string,
+  gameName: string
+) => {
+  const input = exePath + gameName;
+  const crcValue = crc32(input) >>> 0;
+  const steamAppId = (crcValue | 0x80000000) >>> 0;
+  return steamAppId;
+};
+
+export const composeSteamShortcut = (
+  title: string,
+  executablePath: string,
+  iconPath: string | null
+): SteamShortcut => {
+  return {
+    appid: generateSteamShortcutAppId(executablePath, title),
+    appname: title,
+    Exe: `"${executablePath}"`,
+    StartDir: `"${path.dirname(executablePath)}"`,
+    icon: iconPath ?? "",
+    ShortcutPath: "",
+    LaunchOptions: "",
+    IsHidden: false,
+    AllowDesktopConfig: true,
+    AllowOverlay: true,
+    OpenVR: false,
+    Devkit: false,
+    DevkitGameID: "",
+    DevkitOverrideAppID: false,
+    LastPlayTime: 0,
+    FlatpakAppID: "",
+  };
+};
+
+export const writeSteamShortcuts = async (
+  steamUserId: number,
+  shortcuts: SteamShortcut[]
+) => {
+  const buffer = writeBuffer({ shortcuts });
+
+  return fs.promises.writeFile(
+    path.join(
+      await getSteamLocation(),
+      "userdata",
+      steamUserId.toString(),
+      "config",
+      "shortcuts.vdf"
+    ),
+    buffer
+  );
+};