decky-loader/frontend/src/components/modals/filepicker/patches/library.ts

import { Export, Patch, findModuleExport, replacePatch, sleep } from '@decky/ui';

import Logger from '../../../../logger';
import { FileSelectionType } from '..';

const logger = new Logger('LibraryPatch');

let patch: Patch;

function rePatch() {
  // If you patch anything on SteamClient within the first few seconds of the client having loaded it will get redefined for some reason, so repatch any of these changes that occur with History.listen or an interval
  patch = replacePatch(window.SteamClient.Apps, 'PromptToChangeShortcut', async ([appid]: number[]) => {
    try {
      const details = window.appDetailsStore.GetAppDetails(appid);
      logger.debug('game details', details);
      // strShortcutStartDir
      const file = await DeckyPluginLoader.openFilePicker(
        FileSelectionType.FILE,
        details?.strShortcutStartDir.replaceAll('"', '') || '/',
        true,
        true,
      );
      logger.debug('user selected', file);
      window.SteamClient.Apps.SetShortcutExe(appid, JSON.stringify(file.path));
      const pathArr = file.path.split('/');
      pathArr.pop();
      const folder = pathArr.join('/');
      window.SteamClient.Apps.SetShortcutStartDir(appid, JSON.stringify(folder));
    } catch (e) {
      logger.error(e);
    }
  });
}

export default async function libraryPatch() {
  try {
    rePatch();
    // TODO type and add to frontend-lib
    let History: any;

    while (!History) {
      History = findModuleExport((e: Export) => e.m_history)?.m_history;
      if (!History) {
        logger.debug('Waiting 5s for history to become available.');
        await sleep(5000);
      }
    }

    const unlisten = History.listen(() => {
      if (window.SteamClient.Apps.PromptToChangeShortcut !== patch.patchedFunction) {
        rePatch();
      }
    });

    return () => {
      unlisten();
      patch.unpatch();
    };
  } catch (e) {
    logger.error('Error patching library file picker', e);
  }
  return () => {};
}