From 40c7c1b515abf4abcecd6480afb28f27fb71d0f0 Mon Sep 17 00:00:00 2001
From: AAGaming <aagaming@riseup.net>
Date: Sat, 30 Dec 2023 21:32:23 -0500
Subject: port updater to ws, also small refactoring

---
 .../modals/MultiplePluginsInstallModal.tsx         |  2 +-
 .../src/components/modals/PluginInstallModal.tsx   |  2 +-
 .../src/components/modals/PluginUninstallModal.tsx |  2 +-
 .../src/components/modals/filepicker/index.tsx     |  2 +-
 .../modals/filepicker/patches/library.ts           |  4 +--
 .../components/settings/pages/developer/index.tsx  | 26 +++++---------
 .../settings/pages/general/BranchSelect.tsx        |  4 +--
 .../pages/general/NotificationSettings.tsx         |  2 +-
 .../settings/pages/general/RemoteDebugging.tsx     |  4 +--
 .../components/settings/pages/general/Updater.tsx  | 10 +++---
 .../settings/pages/plugin_list/index.tsx           | 10 +++---
 frontend/src/developer.tsx                         |  4 +--
 frontend/src/index.ts                              |  2 +-
 frontend/src/plugin-loader.tsx                     | 40 ++++++++++------------
 frontend/src/plugin.ts                             |  6 ++--
 frontend/src/start.tsx                             | 26 +++++++-------
 frontend/src/store.tsx                             |  4 +--
 frontend/src/updater.ts                            | 21 +++---------
 frontend/src/utils/settings.ts                     |  4 +--
 frontend/src/wsrouter.ts                           | 16 ++++-----
 20 files changed, 78 insertions(+), 113 deletions(-)

(limited to 'frontend')

diff --git a/frontend/src/components/modals/MultiplePluginsInstallModal.tsx b/frontend/src/components/modals/MultiplePluginsInstallModal.tsx
index febdc38d..bfbff08c 100644
--- a/frontend/src/components/modals/MultiplePluginsInstallModal.tsx
+++ b/frontend/src/components/modals/MultiplePluginsInstallModal.tsx
@@ -46,7 +46,7 @@ const MultiplePluginsInstallModal: FC<MultiplePluginsInstallModalProps> = ({
         setLoading(true);
         await onOK();
         setTimeout(() => Navigation.OpenQuickAccessMenu(QuickAccessTab.Decky), 250);
-        setTimeout(() => window.DeckyPluginLoader.checkPluginUpdates(), 1000);
+        setTimeout(() => DeckyPluginLoader.checkPluginUpdates(), 1000);
       }}
       onCancel={async () => {
         await onCancel();
diff --git a/frontend/src/components/modals/PluginInstallModal.tsx b/frontend/src/components/modals/PluginInstallModal.tsx
index b37dbc65..bf263b6d 100644
--- a/frontend/src/components/modals/PluginInstallModal.tsx
+++ b/frontend/src/components/modals/PluginInstallModal.tsx
@@ -34,7 +34,7 @@ const PluginInstallModal: FC<PluginInstallModalProps> = ({
         setLoading(true);
         await onOK();
         setTimeout(() => Navigation.OpenQuickAccessMenu(QuickAccessTab.Decky), 250);
-        setTimeout(() => window.DeckyPluginLoader.checkPluginUpdates(), 1000);
+        setTimeout(() => DeckyPluginLoader.checkPluginUpdates(), 1000);
       }}
       onCancel={async () => {
         await onCancel();
diff --git a/frontend/src/components/modals/PluginUninstallModal.tsx b/frontend/src/components/modals/PluginUninstallModal.tsx
index e9c243b6..3fc0a71f 100644
--- a/frontend/src/components/modals/PluginUninstallModal.tsx
+++ b/frontend/src/components/modals/PluginUninstallModal.tsx
@@ -19,7 +19,7 @@ const PluginUninstallModal: FC<PluginUninstallModalProps> = ({ name, title, butt
         await uninstallPlugin(name);
         // uninstalling a plugin resets the hidden setting for it server-side
         // we invalidate here so if you re-install it, you won't have an out-of-date hidden filter
-        await window.DeckyPluginLoader.hiddenPluginsService.invalidate();
+        await DeckyPluginLoader.hiddenPluginsService.invalidate();
       }}
       strTitle={title}
       strOKButtonText={buttonText}
diff --git a/frontend/src/components/modals/filepicker/index.tsx b/frontend/src/components/modals/filepicker/index.tsx
index 2dfa3ccd..2fbc800b 100644
--- a/frontend/src/components/modals/filepicker/index.tsx
+++ b/frontend/src/components/modals/filepicker/index.tsx
@@ -95,7 +95,7 @@ const sortOptions = [
   },
 ];
 
-const getList = window.DeckyBackend.callable<
+const getList = DeckyBackend.callable<
   [
     path: string,
     includeFiles?: boolean,
diff --git a/frontend/src/components/modals/filepicker/patches/library.ts b/frontend/src/components/modals/filepicker/patches/library.ts
index d398b83d..076e78f6 100644
--- a/frontend/src/components/modals/filepicker/patches/library.ts
+++ b/frontend/src/components/modals/filepicker/patches/library.ts
@@ -13,9 +13,7 @@ function rePatch() {
       const details = window.appDetailsStore.GetAppDetails(appid);
       logger.debug('game details', details);
       // strShortcutStartDir
-      const file = await window.DeckyPluginLoader.openFilePicker(
-        details?.strShortcutStartDir.replaceAll('"', '') || '/',
-      );
+      const file = await DeckyPluginLoader.openFilePicker(details?.strShortcutStartDir.replaceAll('"', '') || '/');
       logger.debug('user selected', file);
       window.SteamClient.Apps.SetShortcutExe(appid, JSON.stringify(file.path));
       const pathArr = file.path.split('/');
diff --git a/frontend/src/components/settings/pages/developer/index.tsx b/frontend/src/components/settings/pages/developer/index.tsx
index 36d3b5c0..9c8504e7 100644
--- a/frontend/src/components/settings/pages/developer/index.tsx
+++ b/frontend/src/components/settings/pages/developer/index.tsx
@@ -28,20 +28,13 @@ const installFromZip = async () => {
     logger.error('The default path has not been found!');
     return;
   }
-  window.DeckyPluginLoader.openFilePickerV2(
-    FileSelectionType.FILE,
-    path,
-    true,
-    true,
-    undefined,
-    ['zip'],
-    false,
-    false,
-  ).then((val) => {
-    const url = `file://${val.path}`;
-    console.log(`Installing plugin locally from ${url}`);
-    installFromURL(url);
-  });
+  DeckyPluginLoader.openFilePickerV2(FileSelectionType.FILE, path, true, true, undefined, ['zip'], false, false).then(
+    (val) => {
+      const url = `file://${val.path}`;
+      console.log(`Installing plugin locally from ${url}`);
+      installFromURL(url);
+    },
+  );
 };
 
 export default function DeveloperSettings() {
@@ -92,10 +85,7 @@ export default function DeveloperSettings() {
           <DialogButton
             onClick={async () => {
               try {
-                let tabId = await window.DeckyBackend.call<[name: string], string>(
-                  'utilities/get_tab_id',
-                  'SharedJSContext',
-                );
+                let tabId = await DeckyBackend.call<[name: string], string>('utilities/get_tab_id', 'SharedJSContext');
                 Navigation.NavigateToExternalWeb(
                   'localhost:8080/devtools/inspector.html?ws=localhost:8080/devtools/page/' + tabId,
                 );
diff --git a/frontend/src/components/settings/pages/general/BranchSelect.tsx b/frontend/src/components/settings/pages/general/BranchSelect.tsx
index 1af06823..6e69208a 100644
--- a/frontend/src/components/settings/pages/general/BranchSelect.tsx
+++ b/frontend/src/components/settings/pages/general/BranchSelect.tsx
@@ -3,7 +3,7 @@ import { FunctionComponent } from 'react';
 import { useTranslation } from 'react-i18next';
 
 import Logger from '../../../../logger';
-import { callUpdaterMethod } from '../../../../updater';
+import { checkForUpdates } from '../../../../updater';
 import { useSetting } from '../../../../utils/hooks/useSetting';
 
 const logger = new Logger('BranchSelect');
@@ -37,7 +37,7 @@ const BranchSelect: FunctionComponent<{}> = () => {
         selectedOption={selectedBranch}
         onChange={async (newVal) => {
           await setSelectedBranch(newVal.data);
-          callUpdaterMethod('check_for_updates');
+          checkForUpdates();
           logger.log('switching branches!');
         }}
       />
diff --git a/frontend/src/components/settings/pages/general/NotificationSettings.tsx b/frontend/src/components/settings/pages/general/NotificationSettings.tsx
index 21c2fd82..82f42e3b 100644
--- a/frontend/src/components/settings/pages/general/NotificationSettings.tsx
+++ b/frontend/src/components/settings/pages/general/NotificationSettings.tsx
@@ -6,7 +6,7 @@ import { useDeckyState } from '../../../DeckyState';
 
 const NotificationSettings: FC = () => {
   const { notificationSettings } = useDeckyState();
-  const notificationService = window.DeckyPluginLoader.notificationService;
+  const notificationService = DeckyPluginLoader.notificationService;
 
   const { t } = useTranslation();
 
diff --git a/frontend/src/components/settings/pages/general/RemoteDebugging.tsx b/frontend/src/components/settings/pages/general/RemoteDebugging.tsx
index 60e0e3c1..187f1cf4 100644
--- a/frontend/src/components/settings/pages/general/RemoteDebugging.tsx
+++ b/frontend/src/components/settings/pages/general/RemoteDebugging.tsx
@@ -18,8 +18,8 @@ export default function RemoteDebuggingSettings() {
         value={allowRemoteDebugging || false}
         onChange={(toggleValue) => {
           setAllowRemoteDebugging(toggleValue);
-          if (toggleValue) window.DeckyBackend.call('allow_remote_debugging');
-          else window.DeckyBackend.call('disallow_remote_debugging');
+          if (toggleValue) DeckyBackend.call('allow_remote_debugging');
+          else DeckyBackend.call('disallow_remote_debugging');
         }}
       />
     </Field>
diff --git a/frontend/src/components/settings/pages/general/Updater.tsx b/frontend/src/components/settings/pages/general/Updater.tsx
index 927a99b0..26537b58 100644
--- a/frontend/src/components/settings/pages/general/Updater.tsx
+++ b/frontend/src/components/settings/pages/general/Updater.tsx
@@ -15,7 +15,7 @@ import { useEffect, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 import { FaExclamation } from 'react-icons/fa';
 
-import { VerInfo, callUpdaterMethod, finishUpdate } from '../../../../updater';
+import { VerInfo, checkForUpdates, doRestart, doUpdate } from '../../../../updater';
 import { useDeckyState } from '../../../DeckyState';
 import InlinePatchNotes from '../../../patchnotes/InlinePatchNotes';
 import WithSuspense from '../../../WithSuspense';
@@ -85,7 +85,7 @@ export default function UpdaterSettings() {
       finish: async () => {
         setUpdateProgress(0);
         setReloading(true);
-        await finishUpdate();
+        await doRestart();
       },
     };
   }, []);
@@ -122,13 +122,13 @@ export default function UpdaterSettings() {
               !versionInfo?.remote || versionInfo?.remote?.tag_name == versionInfo?.current
                 ? async () => {
                     setCheckingForUpdates(true);
-                    const res = (await callUpdaterMethod('check_for_updates')) as { result: VerInfo };
-                    setVersionInfo(res.result);
+                    const verInfo = await checkForUpdates();
+                    setVersionInfo(verInfo);
                     setCheckingForUpdates(false);
                   }
                 : async () => {
                     setUpdateProgress(0);
-                    callUpdaterMethod('do_update');
+                    doUpdate();
                   }
             }
           >
diff --git a/frontend/src/components/settings/pages/plugin_list/index.tsx b/frontend/src/components/settings/pages/plugin_list/index.tsx
index 6475b40a..0728b12d 100644
--- a/frontend/src/components/settings/pages/plugin_list/index.tsx
+++ b/frontend/src/components/settings/pages/plugin_list/index.tsx
@@ -35,7 +35,7 @@ async function reinstallPlugin(pluginName: string, currentVersion?: string) {
 
 type PluginTableData = PluginData & { name: string; hidden: boolean; onHide(): void; onShow(): void };
 
-const reloadPluginBackend = window.DeckyBackend.callable<[pluginName: string], void>('loader/reload_plugin');
+const reloadPluginBackend = DeckyBackend.callable<[pluginName: string], void>('loader/reload_plugin');
 
 function PluginInteractables(props: { entry: ReorderableEntry<PluginTableData> }) {
   const { t } = useTranslation();
@@ -58,14 +58,14 @@ function PluginInteractables(props: { entry: ReorderableEntry<PluginTableData> }
               console.error('Error Reloading Plugin Backend', err);
             }
 
-            window.DeckyPluginLoader.importPlugin(name, version);
+            DeckyPluginLoader.importPlugin(name, version);
           }}
         >
           {t('PluginListIndex.reload')}
         </MenuItem>
         <MenuItem
           onSelected={() =>
-            window.DeckyPluginLoader.uninstallPlugin(
+            DeckyPluginLoader.uninstallPlugin(
               name,
               t('PluginLoader.plugin_uninstall.title', { name }),
               t('PluginLoader.plugin_uninstall.button'),
@@ -143,11 +143,11 @@ export default function PluginList() {
   const { t } = useTranslation();
 
   useEffect(() => {
-    window.DeckyPluginLoader.checkPluginUpdates();
+    DeckyPluginLoader.checkPluginUpdates();
   }, []);
 
   const [pluginEntries, setPluginEntries] = useState<ReorderableEntry<PluginTableData>[]>([]);
-  const hiddenPluginsService = window.DeckyPluginLoader.hiddenPluginsService;
+  const hiddenPluginsService = DeckyPluginLoader.hiddenPluginsService;
 
   useEffect(() => {
     setPluginEntries(
diff --git a/frontend/src/developer.tsx b/frontend/src/developer.tsx
index ad5d8a40..99e9890b 100644
--- a/frontend/src/developer.tsx
+++ b/frontend/src/developer.tsx
@@ -38,7 +38,7 @@ export async function setShowValveInternal(show: boolean) {
 }
 
 export async function setShouldConnectToReactDevTools(enable: boolean) {
-  window.DeckyPluginLoader.toaster.toast({
+  DeckyPluginLoader.toaster.toast({
     title: enable ? (
       <TranslationHelper trans_class={TranslationClass.DEVELOPER} trans_text={'enabling'} />
     ) : (
@@ -48,7 +48,7 @@ export async function setShouldConnectToReactDevTools(enable: boolean) {
     icon: <FaReact />,
   });
   await sleep(5000);
-  return enable ? window.DeckyBackend.call('utilities/enable_rdt') : window.DeckyBackend.call('utilities/disable_rdt');
+  return enable ? DeckyBackend.call('utilities/enable_rdt') : DeckyBackend.call('utilities/disable_rdt');
 }
 
 export async function startup() {
diff --git a/frontend/src/index.ts b/frontend/src/index.ts
index 3a0c4880..1ca06b0e 100644
--- a/frontend/src/index.ts
+++ b/frontend/src/index.ts
@@ -6,7 +6,7 @@
   window.deckyAuthToken = await fetch('http://127.0.0.1:1337/auth/token').then((r) => r.text());
   console.debug('Connecting to Decky backend...');
   window.DeckyBackend = new (await import('./wsrouter')).WSRouter();
-  await window.DeckyBackend.connect();
+  await DeckyBackend.connect();
   console.debug('Starting Decky!');
   await import('./start');
 })();
diff --git a/frontend/src/plugin-loader.tsx b/frontend/src/plugin-loader.tsx
index 9e35ee46..7951e944 100644
--- a/frontend/src/plugin-loader.tsx
+++ b/frontend/src/plugin-loader.tsx
@@ -28,10 +28,10 @@ import { NotificationService } from './notification-service';
 import { InstallType, Plugin } from './plugin';
 import RouterHook from './router-hook';
 import { deinitSteamFixes, initSteamFixes } from './steamfixes';
-import { checkForUpdates } from './store';
+import { checkForPluginUpdates } from './store';
 import TabsHook from './tabs-hook';
 import Toaster from './toaster';
-import { VerInfo, callUpdaterMethod } from './updater';
+import { getVersionInfo } from './updater';
 import { getSetting, setSetting } from './utils/settings';
 import TranslationHelper, { TranslationClass } from './utils/TranslationHelper';
 
@@ -106,9 +106,7 @@ class PluginLoader extends Logger {
       .then(() => this.log('Initialized'));
   }
 
-  private getPluginsFromBackend = window.DeckyBackend.callable<[], { name: string; version: string }[]>(
-    'loader/get_plugins',
-  );
+  private getPluginsFromBackend = DeckyBackend.callable<[], { name: string; version: string }[]>('loader/get_plugins');
 
   private async loadPlugins() {
     // wait for SP window to exist before loading plugins
@@ -129,13 +127,13 @@ class PluginLoader extends Logger {
   }
 
   public async getUserInfo() {
-    const userInfo = await window.DeckyBackend.call<[], UserInfo>('utilities/get_user_info');
+    const userInfo = await DeckyBackend.call<[], UserInfo>('utilities/get_user_info');
     setSetting('user_info.user_name', userInfo.username);
     setSetting('user_info.user_home', userInfo.path);
   }
 
   public async updateVersion() {
-    const versionInfo = (await callUpdaterMethod('get_version')).result as VerInfo;
+    const versionInfo = await getVersionInfo();
     this.deckyState.setVersionInfo(versionInfo);
 
     return versionInfo;
@@ -164,7 +162,7 @@ class PluginLoader extends Logger {
   }
 
   public async checkPluginUpdates() {
-    const updates = await checkForUpdates(this.plugins);
+    const updates = await checkForPluginUpdates(this.plugins);
     this.deckyState.setUpdates(updates);
     return updates;
   }
@@ -199,8 +197,8 @@ class PluginLoader extends Logger {
         version={version}
         hash={hash}
         installType={install_type}
-        onOK={() => window.DeckyBackend.call<[string]>('utilities/confirm_plugin_install', request_id)}
-        onCancel={() => window.DeckyBackend.call<[string]>('utilities/cancel_plugin_install', request_id)}
+        onOK={() => DeckyBackend.call<[string]>('utilities/confirm_plugin_install', request_id)}
+        onCancel={() => DeckyBackend.call<[string]>('utilities/cancel_plugin_install', request_id)}
       />,
     );
   }
@@ -212,8 +210,8 @@ class PluginLoader extends Logger {
     showModal(
       <MultiplePluginsInstallModal
         requests={requests}
-        onOK={() => window.DeckyBackend.call<[string]>('utilities/confirm_plugin_install', request_id)}
-        onCancel={() => window.DeckyBackend.call<[string]>('utilities/cancel_plugin_install', request_id)}
+        onOK={() => DeckyBackend.call<[string]>('utilities/confirm_plugin_install', request_id)}
+        onCancel={() => DeckyBackend.call<[string]>('utilities/cancel_plugin_install', request_id)}
       />,
     );
   }
@@ -298,7 +296,7 @@ class PluginLoader extends Logger {
     let res = await fetch(`http://127.0.0.1:1337/plugins/${name}/frontend_bundle`, {
       credentials: 'include',
       headers: {
-        Authentication: window.deckyAuthToken,
+        Authentication: deckyAuthToken,
       },
     });
 
@@ -364,7 +362,7 @@ class PluginLoader extends Logger {
     this.warn(
       `Calling ${methodName} via callServerMethod, which is deprecated and will be removed in a future release. Please switch to the backend API.`,
     );
-    return await window.DeckyBackend.call<[methodName: string, kwargs: any], any>(
+    return await DeckyBackend.call<[methodName: string, kwargs: any], any>(
       'utilities/_call_legacy_utility',
       methodName,
       args,
@@ -429,7 +427,7 @@ class PluginLoader extends Logger {
     const pluginAPI = {
       backend: {
         call<Args extends any[] = any[], Return = void>(method: string, ...args: Args): Promise<Return> {
-          return window.DeckyBackend.call<[pluginName: string, method: string, ...args: Args], Return>(
+          return DeckyBackend.call<[pluginName: string, method: string, ...args: Args], Return>(
             'loader/call_plugin_method',
             pluginName,
             method,
@@ -448,7 +446,7 @@ class PluginLoader extends Logger {
       openFilePickerV2: this.openFilePickerV2,
       // Legacy
       async callPluginMethod(methodName: string, args = {}) {
-        return window.DeckyBackend.call<[pluginName: string, methodName: string, kwargs: any], any>(
+        return DeckyBackend.call<[pluginName: string, methodName: string, kwargs: any], any>(
           'loader/call_legacy_plugin_method',
           pluginName,
           methodName,
@@ -472,7 +470,7 @@ class PluginLoader extends Logger {
         }
         // this is terrible but a. we're going to redo this entire method anyway and b. it was already terrible
         try {
-          const ret = await window.DeckyBackend.call<
+          const ret = await DeckyBackend.call<
             [method: string, url: string, extra_opts?: any],
             { status: number; headers: { [key: string]: string }; body: string }
           >('utilities/http_request', method, url, req);
@@ -481,14 +479,12 @@ class PluginLoader extends Logger {
           return { success: false, result: e?.toString() };
         }
       },
-      executeInTab: window.DeckyBackend.callable<
+      executeInTab: DeckyBackend.callable<
         [tab: String, runAsync: Boolean, code: string],
         { success: boolean; result: any }
       >('utilities/execute_in_tab'),
-      injectCssIntoTab: window.DeckyBackend.callable<[tab: string, style: string], string>(
-        'utilities/inject_css_into_tab',
-      ),
-      removeCssFromTab: window.DeckyBackend.callable<[tab: string, cssId: string]>('utilities/remove_css_from_tab'),
+      injectCssIntoTab: DeckyBackend.callable<[tab: string, style: string], string>('utilities/inject_css_into_tab'),
+      removeCssFromTab: DeckyBackend.callable<[tab: string, cssId: string]>('utilities/remove_css_from_tab'),
     };
 
     return pluginAPI;
diff --git a/frontend/src/plugin.ts b/frontend/src/plugin.ts
index 21d45654..d750a63e 100644
--- a/frontend/src/plugin.ts
+++ b/frontend/src/plugin.ts
@@ -22,7 +22,7 @@ type installPluginArgs = [
   installType?: InstallType,
 ];
 
-export let installPlugin = window.DeckyBackend.callable<installPluginArgs>('utilities/install_plugin');
+export let installPlugin = DeckyBackend.callable<installPluginArgs>('utilities/install_plugin');
 
 type installPluginsArgs = [
   requests: {
@@ -34,6 +34,6 @@ type installPluginsArgs = [
   }[],
 ];
 
-export let installPlugins = window.DeckyBackend.callable<installPluginsArgs>('utilities/install_plugins');
+export let installPlugins = DeckyBackend.callable<installPluginsArgs>('utilities/install_plugins');
 
-export let uninstallPlugin = window.DeckyBackend.callable<[name: string]>('utilities/uninstall_plugin');
+export let uninstallPlugin = DeckyBackend.callable<[name: string]>('utilities/uninstall_plugin');
diff --git a/frontend/src/start.tsx b/frontend/src/start.tsx
index cc2c02c2..e8c489f2 100644
--- a/frontend/src/start.tsx
+++ b/frontend/src/start.tsx
@@ -6,15 +6,13 @@ import PluginLoader from './plugin-loader';
 import { DeckyUpdater } from './updater';
 
 declare global {
-  interface Window {
-    DeckyPluginLoader: PluginLoader;
-    DeckyUpdater?: DeckyUpdater;
-    importDeckyPlugin: Function;
-    deckyHasLoaded: boolean;
-    deckyHasConnectedRDT?: boolean;
-    deckyAuthToken: string;
-    DFL?: any;
-  }
+  export var DeckyPluginLoader: PluginLoader;
+  export var DeckyUpdater: DeckyUpdater | undefined; // TODO get rid of this
+  export var importDeckyPlugin: Function;
+  export var deckyHasLoaded: boolean;
+  export var deckyHasConnectedRDT: boolean | undefined;
+  export var deckyAuthToken: string;
+  export var DFL: any | undefined;
 }
 
 (async () => {
@@ -37,7 +35,7 @@ declare global {
       backend: {
         loadPath: 'http://127.0.0.1:1337/locales/{{lng}}.json',
         customHeaders: {
-          Authentication: window.deckyAuthToken,
+          Authentication: deckyAuthToken,
         },
         requestOptions: {
           credentials: 'include',
@@ -45,12 +43,12 @@ declare global {
       },
     });
 
-  window.DeckyPluginLoader?.dismountAll();
-  window.DeckyPluginLoader?.deinit();
+  window?.DeckyPluginLoader?.dismountAll();
+  window?.DeckyPluginLoader?.deinit();
   window.DeckyPluginLoader = new PluginLoader();
-  window.DeckyPluginLoader.init();
+  DeckyPluginLoader.init();
   window.importDeckyPlugin = function (name: string, version: string) {
-    window.DeckyPluginLoader?.importPlugin(name, version);
+    DeckyPluginLoader?.importPlugin(name, version);
   };
 })();
 
diff --git a/frontend/src/store.tsx b/frontend/src/store.tsx
index 5ae98ec4..3fcfdb2f 100644
--- a/frontend/src/store.tsx
+++ b/frontend/src/store.tsx
@@ -37,7 +37,7 @@ export async function getStore(): Promise<Store> {
 }
 
 export async function getPluginList(): Promise<StorePlugin[]> {
-  let version = await window.DeckyPluginLoader.updateVersion();
+  let version = await DeckyPluginLoader.updateVersion();
   let store = await getSetting<Store | null>('store', null);
   let customURL = await getSetting<string>('store-url', 'https://plugins.deckbrew.xyz/plugins');
   let storeURL;
@@ -112,7 +112,7 @@ export async function requestMultiplePluginInstalls(requests: PluginInstallReque
   );
 }
 
-export async function checkForUpdates(plugins: Plugin[]): Promise<PluginUpdateMapping> {
+export async function checkForPluginUpdates(plugins: Plugin[]): Promise<PluginUpdateMapping> {
   const serverData = await getPluginList();
   const updateMap = new Map<string, StorePluginVersion>();
   for (let plugin of plugins) {
diff --git a/frontend/src/updater.ts b/frontend/src/updater.ts
index 339b2b72..d6c23832 100644
--- a/frontend/src/updater.ts
+++ b/frontend/src/updater.ts
@@ -28,20 +28,7 @@ export interface VerInfo {
   updatable: boolean;
 }
 
-export async function callUpdaterMethod(methodName: string, args = {}) {
-  const response = await fetch(`http://127.0.0.1:1337/updater/${methodName}`, {
-    method: 'POST',
-    credentials: 'include',
-    headers: {
-      'Content-Type': 'application/json',
-      Authentication: window.deckyAuthToken,
-    },
-    body: JSON.stringify(args),
-  });
-
-  return response.json();
-}
-
-export async function finishUpdate() {
-  callUpdaterMethod('do_restart');
-}
+export const doUpdate = DeckyBackend.callable('updater/do_update');
+export const doRestart = DeckyBackend.callable('updater/do_restart');
+export const getVersionInfo = DeckyBackend.callable<[], VerInfo>('updater/get_version_info');
+export const checkForUpdates = DeckyBackend.callable<[], VerInfo>('updater/check_for_updates');
diff --git a/frontend/src/utils/settings.ts b/frontend/src/utils/settings.ts
index cba4887f..035b2c22 100644
--- a/frontend/src/utils/settings.ts
+++ b/frontend/src/utils/settings.ts
@@ -1,8 +1,8 @@
 export async function getSetting<T>(key: string, def: T): Promise<T> {
-  const res = await window.DeckyBackend.call<[string, T], T>('utilities/settings/get', key, def);
+  const res = await DeckyBackend.call<[string, T], T>('utilities/settings/get', key, def);
   return res;
 }
 
 export async function setSetting<T>(key: string, value: T): Promise<void> {
-  await window.DeckyBackend.call<[string, T], void>('utilities/settings/set', key, value);
+  await DeckyBackend.call<[string, T], void>('utilities/settings/set', key, value);
 }
diff --git a/frontend/src/wsrouter.ts b/frontend/src/wsrouter.ts
index d3bc45a0..32037eaf 100644
--- a/frontend/src/wsrouter.ts
+++ b/frontend/src/wsrouter.ts
@@ -3,14 +3,12 @@ import { sleep } from 'decky-frontend-lib';
 import Logger from './logger';
 
 declare global {
-  interface Window {
-    DeckyBackend: WSRouter;
-  }
+  export var DeckyBackend: WSRouter;
 }
 
 enum MessageType {
   ERROR = -1,
-  // Call-reply, Frontend -> Backend
+  // Call-reply, Frontend -> Backend -> Frontend
   CALL = 0,
   REPLY = 1,
   // Pub/Sub, Backend -> Frontend
@@ -22,8 +20,6 @@ interface CallMessage {
   args: any[];
   route: string;
   id: number;
-  // TODO implement this
-  // skipResponse?: boolean;
 }
 
 interface ReplyMessage {
@@ -61,7 +57,7 @@ export class WSRouter extends Logger {
   connect() {
     return (this.connectPromise = new Promise<void>((resolve) => {
       // Auth is a query param as JS WebSocket doesn't support headers
-      this.ws = new WebSocket(`ws://127.0.0.1:1337/ws?auth=${window.deckyAuthToken}`);
+      this.ws = new WebSocket(`ws://127.0.0.1:1337/ws?auth=${deckyAuthToken}`);
 
       this.ws.addEventListener('open', () => {
         this.debug('WS Connected');
@@ -129,7 +125,7 @@ export class WSRouter extends Logger {
           if (this.runningCalls.has(data.id)) {
             this.runningCalls.get(data.id)!.reject(data.error);
             this.runningCalls.delete(data.id);
-            this.debug(`Errored PY call ${data.id} with error`, data.error);
+            this.debug(`Rejected PY call ${data.id} with error`, data.error);
           }
           break;
 
@@ -143,7 +139,7 @@ export class WSRouter extends Logger {
   }
 
   // this.call<[number, number], string>('methodName', 1, 2);
-  call<Args extends any[] = any[], Return = void>(route: string, ...args: Args): Promise<Return> {
+  call<Args extends any[] = [], Return = void>(route: string, ...args: Args): Promise<Return> {
     const resolver = this.createPromiseResolver<Return>();
 
     const id = ++this.reqId;
@@ -157,7 +153,7 @@ export class WSRouter extends Logger {
     return resolver.promise;
   }
 
-  callable<Args extends any[] = any[], Return = void>(route: string): (...args: Args) => Promise<Return> {
+  callable<Args extends any[] = [], Return = void>(route: string): (...args: Args) => Promise<Return> {
     return (...args) => this.call<Args, Return>(route, ...args);
   }
 
-- 
cgit v1.2.3