[Improv]: Add environment variables for use by wrappers

src/backend/launcher.ts

@@ -45,6 +45,7 @@ import {
   GameInfo,
   Runner,
   EnviromentVariable,
+  WrapperEnv,
   WrapperVariable,
   ExecResult,
   GameSettings,
@@ -275,6 +276,7 @@ async function prepareWineLaunch(
 /**
  * Maps general settings to environment variables
  * @param gameSettings The GameSettings to get the environment variables for
+ * @param wrapperEnv The wrapper info to be added into the environment variables
  * @returns A big string of environment variables, structured key=value
  */
 function setupEnvVars(gameSettings: GameSettings) {
@@ -305,6 +307,34 @@ function setupEnvVars(gameSettings: GameSettings) {
   return ret
 }
 
+/**
+ * Maps launcher info to environment variables for consumption by wrappers
+ * @param wrapperEnv The info to be added into the environment variables
+ * @returns Environment variables
+ */
+function setupWrapperEnvVars(wrapperEnv: WrapperEnv) {
+  const ret: Record<string, string> = {}
+
+  ret.HEROIC_APP_NAME = wrapperEnv.appName
+
+  switch (wrapperEnv.appRunner) {
+    case 'gog':
+      ret.HEROIC_APP_SOURCE = 'gog'
+      break
+    case 'legendary':
+      ret.HEROIC_APP_SOURCE = 'epic'
+      break
+    case 'nile':
+      ret.HEROIC_APP_SOURCE = 'amazon'
+      break
+    case 'sideload':
+      ret.HEROIC_APP_SOURCE = 'sideload'
+      break
+  }
+
+  return ret
+}
+
 /**
  * Maps Wine-related settings to environment variables
  * @param gameSettings The GameSettings to get the environment variables for
@@ -982,6 +1012,7 @@ export {
   launchCleanup,
   prepareWineLaunch,
   setupEnvVars,
+  setupWrapperEnvVars,
   setupWineEnvVars,
   setupWrappers,
   runWineCommand,

src/backend/storeManagers/gog/games.ts

@@ -62,6 +62,7 @@ import {
   prepareWineLaunch,
   runWineCommand as runWineCommandUtil,
   setupEnvVars,
+  setupWrapperEnvVars,
   setupWrappers
 } from '../../launcher'
 import {
@@ -464,9 +465,13 @@ export async function launch(
     ? ['--override-exe', gameSettings.targetExe]
     : []
 
-  let commandEnv = isWindows
-    ? process.env
-    : { ...process.env, ...setupEnvVars(gameSettings) }
+  let commandEnv = {
+    ...process.env,
+    ...setupWrapperEnvVars({ appName, appRunner: 'gog' })
+  }
+  if (!isWindows) {
+    commandEnv = { ...commandEnv, ...setupEnvVars(gameSettings) }
+  }
 
   const wrappers = setupWrappers(
     gameSettings,

src/backend/storeManagers/legendary/games.ts

@@ -47,6 +47,7 @@ import {
   prepareLaunch,
   prepareWineLaunch,
   setupEnvVars,
+  setupWrapperEnvVars,
   setupWrappers,
   launchCleanup,
   getRunnerCallWithoutCredentials,
@@ -822,9 +823,13 @@ export async function launch(
 
   const languageCode = gameSettings.language || configStore.get('language', '')
 
-  let commandEnv = isWindows
-    ? process.env
-    : { ...process.env, ...setupEnvVars(gameSettings) }
+  let commandEnv = {
+    ...process.env,
+    ...setupWrapperEnvVars({ appName, appRunner: 'legendary' })
+  }
+  if (!isWindows) {
+    commandEnv = { ...commandEnv, ...setupEnvVars(gameSettings) }
+  }
 
   const wrappers = setupWrappers(
     gameSettings,

src/backend/storeManagers/nile/games.ts

@@ -35,6 +35,7 @@ import {
   prepareLaunch,
   prepareWineLaunch,
   setupEnvVars,
+  setupWrapperEnvVars,
   setupWrappers
 } from 'backend/launcher'
 import { appendFileSync, existsSync } from 'graceful-fs'
@@ -344,9 +345,13 @@ export async function launch(
     ? ['--override-exe', gameSettings.targetExe]
     : []
 
-  let commandEnv = isWindows
-    ? process.env
-    : { ...process.env, ...setupEnvVars(gameSettings) }
+  let commandEnv = {
+    ...process.env,
+    ...setupWrapperEnvVars({ appName, appRunner: 'nile' })
+  }
+  if (!isWindows) {
+    commandEnv = { ...commandEnv, ...setupEnvVars(gameSettings) }
+  }
 
   const wrappers = setupWrappers(
     gameSettings,

src/backend/storeManagers/storeManagerCommon/games.ts

@@ -11,6 +11,7 @@ import {
   prepareLaunch,
   runWineCommand,
   setupEnvVars,
+  setupWrapperEnvVars,
   setupWrappers
 } from '../../launcher'
 import { access, chmod } from 'fs/promises'
@@ -131,7 +132,12 @@ export async function launchGame(
       })
       return false
     }
-    const env = { ...process.env, ...setupEnvVars(gameSettings) }
+
+    const env = {
+      ...process.env,
+      ...setupWrapperEnvVars({ appName, appRunner: runner }),
+      ...setupEnvVars(gameSettings)
+    }
 
     // Native
     if (isNative) {

src/common/types.ts

@@ -351,6 +351,11 @@ export interface WrapperVariable {
   args: string
 }
 
+export interface WrapperEnv {
+  appName: string
+  appRunner: Runner
+}
+
 type AntiCheat =
   | 'Arbiter'
   | 'BattlEye'