feat: webcontentsview extension ipcpromise

src/env.d.ts

@@ -1,6 +1,9 @@
 /// <reference types="vite/client" />
 /// <reference types="@modyfi/vite-plugin-yaml/modules" />
 /// <reference types="electron-vite/node" />
+/// <reference path="../node_modules/electron/electron.d.ts" />
+
+import type { IpcPromiseResult } from "@shared/utils/promises";
 
 interface ImportMetaEnv {
   readonly MAIN_APP_SECRET: string;
@@ -10,4 +13,12 @@ interface ImportMeta {
   readonly env: ImportMetaEnv;
 }
 
-type StringLiteral<KnownValues extends string> = (string & {}) | KnownValues;
+declare global {
+  declare namespace Electron {
+    interface WebContentsView {
+      invoke<T = any>(channelName: string, data: any): IpcPromiseResult<T>;
+    }
+  }
+}
+
+type StringLiteral<KnownValues extends string> = (string & {}) | KnownValues;

src/main/utils/electron.ts

@@ -1,7 +1,8 @@
 import { is } from "@electron-toolkit/utils";
 import { Logger } from "@shared/utils/console";
 import logger from "@shared/utils/Logger";
-import { app } from "electron";
+import { ipcPromise } from "@shared/utils/promises";
+import { app, WebContentsView } from "electron";
 import { isProduction } from "./devUtils";
 
 let isInitialized = false;
@@ -21,6 +22,8 @@ export function initializeCustomElectronEnvironment() {
 
   if (import.meta.env.PROD && !process.env.DEBUG) Logger.enableProductionMode();
   process.env.NODE_ENV = import.meta.env.MODE;
-
+  WebContentsView.prototype.invoke = function<T>(channel: string, data: any) {
+    return ipcPromise<T>(this, channel, data);
+  }
   isInitialized = true;
 }

src/main/utils/mappedWindow.ts

@@ -1,14 +1,13 @@
-import { ipcPromise } from "@shared/utils/promises";
 import { BrowserWindow, WebContentsView } from "electron";
 import IPC_EVENT_NAMES from "./eventNames";
 
-type TrackControlTypes = StringLiteral<("play" | "pause" | "next" | "prev" | "toggle")>
+type TrackControlTypes = StringLiteral<"play" | "pause" | "next" | "prev" | "toggle">;
 type TrackControlFn = <T = any>(type: TrackControlTypes) => Promise<T>;
 export interface BrowserWindowViews<T, TView extends WebContentsView = WebContentsView> {
   main: BrowserWindow;
   views: { [key: string]: TView } & T;
   sendToAllViews(ev: string, ...args: any[]): void;
-  sendTrackControl: TrackControlFn
+  sendTrackControl: TrackControlFn;
 }
 
 export function getViewObject(bwv: { [key: string]: WebContentsView }) {
@@ -27,9 +26,9 @@ export function createWindowContext<T, TView extends WebContentsView = WebConten
     views: { [key: string]: TView } & T = _data.views || ({} as any);
     async sendTrackControl<T = any>(type: TrackControlTypes) {
       const view = this.views.youtubeView;
-      if (!view) return Promise.reject(new Error("View not found"))
-      return await ipcPromise<T>(view, IPC_EVENT_NAMES.TRACK_CONTROL, {type})
-    };
+      if (!view) return Promise.reject(new Error("View not found"));
+      return await view.invoke<T>(IPC_EVENT_NAMES.TRACK_CONTROL, { type });
+    }
     sendToAllViews(ev: string, ...args: any[]): void {
       return (Object.values(this.views) as TView[])
         .filter(

src/shared/utils/promises.ts

@@ -26,4 +26,5 @@ export async function ipcPromise<T = any, R = T>(view: WebContentsView, channel:
       reject(new Error("IPC Promise timeout."))
     }, 10000);
   })
-}
+}
+export type IpcPromiseResult<T> = ReturnType<typeof ipcPromise<T>>;