feat: move more electron dependencies into their own service

Author: jonathanlab

apps/code/src/main/db/service.ts

@@ -1,12 +1,13 @@
 import path from "node:path";
+import type { IStoragePaths } from "@posthog/platform/storage-paths";
 import Database from "better-sqlite3";
 import {
   type BetterSQLite3Database,
   drizzle,
 } from "drizzle-orm/better-sqlite3";
 import { migrate } from "drizzle-orm/better-sqlite3/migrator";
-import { app } from "electron";
-import { injectable, postConstruct, preDestroy } from "inversify";
+import { inject, injectable, postConstruct, preDestroy } from "inversify";
+import { MAIN_TOKENS } from "../di/tokens";
 import { logger } from "../utils/logger";
 
 import * as schema from "./schema";
@@ -20,6 +21,11 @@ export class DatabaseService {
   private _db: BetterSQLite3Database<typeof schema> | null = null;
   private _sqlite: InstanceType<typeof Database> | null = null;
 
+  constructor(
+    @inject(MAIN_TOKENS.StoragePaths)
+    private readonly storagePaths: IStoragePaths,
+  ) {}
+
   get db(): BetterSQLite3Database<typeof schema> {
     if (!this._db) {
       throw new Error("Database not initialized — call initialize() first");
@@ -29,7 +35,7 @@ export class DatabaseService {
 
   @postConstruct()
   initialize(): void {
-    const dbPath = path.join(app.getPath("userData"), "posthog-code.db");
+    const dbPath = path.join(this.storagePaths.appDataPath, "posthog-code.db");
     log.info("Opening database", {
       path: dbPath,
       migrationsFolder: MIGRATIONS_FOLDER,

apps/code/src/main/di/container.ts

@@ -9,6 +9,12 @@ import { SuspensionRepositoryImpl } from "../db/repositories/suspension-reposito
 import { WorkspaceRepository } from "../db/repositories/workspace-repository";
 import { WorktreeRepository } from "../db/repositories/worktree-repository";
 import { DatabaseService } from "../db/service";
+import { ElectronAppMeta } from "../platform-adapters/electron-app-meta";
+import { ElectronClipboard } from "../platform-adapters/electron-clipboard";
+import { ElectronDialog } from "../platform-adapters/electron-dialog";
+import { ElectronFileIcon } from "../platform-adapters/electron-file-icon";
+import { ElectronSecureStorage } from "../platform-adapters/electron-secure-storage";
+import { ElectronStoragePaths } from "../platform-adapters/electron-storage-paths";
 import { ElectronUrlLauncher } from "../platform-adapters/electron-url-launcher";
 import { AgentAuthAdapter } from "../services/agent/auth-adapter";
 import { AgentService } from "../services/agent/service";
@@ -54,6 +60,12 @@ export const container = new Container({
 });
 
 container.bind(MAIN_TOKENS.UrlLauncher).to(ElectronUrlLauncher);
+container.bind(MAIN_TOKENS.StoragePaths).to(ElectronStoragePaths);
+container.bind(MAIN_TOKENS.AppMeta).to(ElectronAppMeta);
+container.bind(MAIN_TOKENS.Dialog).to(ElectronDialog);
+container.bind(MAIN_TOKENS.Clipboard).to(ElectronClipboard);
+container.bind(MAIN_TOKENS.FileIcon).to(ElectronFileIcon);
+container.bind(MAIN_TOKENS.SecureStorage).to(ElectronSecureStorage);
 
 container.bind(MAIN_TOKENS.DatabaseService).to(DatabaseService);
 container

apps/code/src/main/di/tokens.ts

@@ -7,6 +7,12 @@
 export const MAIN_TOKENS = Object.freeze({
   // Platform ports (host-agnostic interfaces from @posthog/platform)
   UrlLauncher: Symbol.for("Platform.UrlLauncher"),
+  StoragePaths: Symbol.for("Platform.StoragePaths"),
+  AppMeta: Symbol.for("Platform.AppMeta"),
+  Dialog: Symbol.for("Platform.Dialog"),
+  Clipboard: Symbol.for("Platform.Clipboard"),
+  FileIcon: Symbol.for("Platform.FileIcon"),
+  SecureStorage: Symbol.for("Platform.SecureStorage"),
 
   // Stores
   SettingsStore: Symbol.for("Main.SettingsStore"),

apps/code/src/main/platform-adapters/electron-app-meta.ts

@@ -0,0 +1,14 @@
+import type { IAppMeta } from "@posthog/platform/app-meta";
+import { app } from "electron";
+import { injectable } from "inversify";
+
+@injectable()
+export class ElectronAppMeta implements IAppMeta {
+  public get version(): string {
+    return app.getVersion();
+  }
+
+  public get isProduction(): boolean {
+    return app.isPackaged;
+  }
+}

apps/code/src/main/platform-adapters/electron-clipboard.ts

@@ -0,0 +1,10 @@
+import type { IClipboard } from "@posthog/platform/clipboard";
+import { clipboard } from "electron";
+import { injectable } from "inversify";
+
+@injectable()
+export class ElectronClipboard implements IClipboard {
+  public async writeText(text: string): Promise<void> {
+    clipboard.writeText(text);
+  }
+}

apps/code/src/main/platform-adapters/electron-dialog.ts

@@ -0,0 +1,61 @@
+import type {
+  ConfirmOptions,
+  DialogSeverity,
+  IDialog,
+  PickFileOptions,
+} from "@posthog/platform/dialog";
+import {
+  BrowserWindow,
+  dialog,
+  type MessageBoxOptions,
+  type OpenDialogOptions,
+} from "electron";
+import { injectable } from "inversify";
+
+type OpenDialogProperty = NonNullable<OpenDialogOptions["properties"]>[number];
+
+function severityToType(severity?: DialogSeverity): MessageBoxOptions["type"] {
+  return severity ?? "none";
+}
+
+function buildProperties(options: PickFileOptions): OpenDialogProperty[] {
+  const properties: OpenDialogProperty[] = [
+    options.directories ? "openDirectory" : "openFile",
+    "treatPackageAsDirectory",
+  ];
+  if (options.multiple) properties.push("multiSelections");
+  if (options.createDirectories) properties.push("createDirectory");
+  return properties;
+}
+
+@injectable()
+export class ElectronDialog implements IDialog {
+  public async confirm(options: ConfirmOptions): Promise<number> {
+    const parent = BrowserWindow.getFocusedWindow();
+    const electronOptions: MessageBoxOptions = {
+      type: severityToType(options.severity),
+      title: options.title,
+      message: options.message,
+      detail: options.detail,
+      buttons: options.options,
+      defaultId: options.defaultIndex,
+      cancelId: options.cancelIndex,
+    };
+    const result = parent
+      ? await dialog.showMessageBox(parent, electronOptions)
+      : await dialog.showMessageBox(electronOptions);
+    return result.response;
+  }
+
+  public async pickFile(options: PickFileOptions): Promise<string[]> {
+    const parent = BrowserWindow.getFocusedWindow();
+    const electronOptions: OpenDialogOptions = {
+      title: options.title,
+      properties: buildProperties(options),
+    };
+    const result = parent
+      ? await dialog.showOpenDialog(parent, electronOptions)
+      : await dialog.showOpenDialog(electronOptions);
+    return result.canceled ? [] : result.filePaths;
+  }
+}

apps/code/src/main/platform-adapters/electron-file-icon.ts

@@ -0,0 +1,34 @@
+import type { IFileIcon } from "@posthog/platform/file-icon";
+import { app } from "electron";
+import { injectable } from "inversify";
+
+type FileIconModule = typeof import("file-icon");
+
+@injectable()
+export class ElectronFileIcon implements IFileIcon {
+  private fileIconModule: FileIconModule | undefined;
+
+  public async getAsDataUrl(filePath: string): Promise<string | null> {
+    try {
+      if (process.platform === "darwin") {
+        const mod = await this.loadFileIconModule();
+        const uint8Array = await mod.fileIconToBuffer(filePath, { size: 64 });
+        const base64 = Buffer.from(uint8Array).toString("base64");
+        return `data:image/png;base64,${base64}`;
+      }
+
+      const icon = await app.getFileIcon(filePath, { size: "normal" });
+      const base64 = icon.toPNG().toString("base64");
+      return `data:image/png;base64,${base64}`;
+    } catch {
+      return null;
+    }
+  }
+
+  private async loadFileIconModule(): Promise<FileIconModule> {
+    if (!this.fileIconModule) {
+      this.fileIconModule = await import("file-icon");
+    }
+    return this.fileIconModule;
+  }
+}

apps/code/src/main/platform-adapters/electron-secure-storage.ts

@@ -0,0 +1,19 @@
+import type { ISecureStorage } from "@posthog/platform/secure-storage";
+import { safeStorage } from "electron";
+import { injectable } from "inversify";
+
+@injectable()
+export class ElectronSecureStorage implements ISecureStorage {
+  public isAvailable(): boolean {
+    return safeStorage.isEncryptionAvailable();
+  }
+
+  public async encryptString(text: string): Promise<Uint8Array> {
+    const buffer = safeStorage.encryptString(text);
+    return new Uint8Array(buffer.buffer, buffer.byteOffset, buffer.byteLength);
+  }
+
+  public async decryptString(data: Uint8Array): Promise<string> {
+    return safeStorage.decryptString(Buffer.from(data));
+  }
+}

apps/code/src/main/platform-adapters/electron-storage-paths.ts

@@ -0,0 +1,14 @@
+import type { IStoragePaths } from "@posthog/platform/storage-paths";
+import { app } from "electron";
+import { injectable } from "inversify";
+
+@injectable()
+export class ElectronStoragePaths implements IStoragePaths {
+  public get appDataPath(): string {
+    return app.getPath("userData");
+  }
+
+  public get logsPath(): string {
+    return app.getPath("logs");
+  }
+}

apps/code/src/main/services/context-menu/service.ts

@@ -1,13 +1,8 @@
+import type { IDialog } from "@posthog/platform/dialog";
 import type { DetectedApplication } from "@shared/types";
-import {
-  dialog,
-  Menu,
-  type MenuItemConstructorOptions,
-  nativeImage,
-} from "electron";
+import { Menu, type MenuItemConstructorOptions, nativeImage } from "electron";
 import { inject, injectable } from "inversify";
 import { MAIN_TOKENS } from "../../di/tokens";
-import { getMainWindow } from "../../trpc/context";
 import type { ExternalAppsService } from "../external-apps/service";
 import type {
   ArchivedTaskAction,
@@ -46,6 +41,8 @@ export class ContextMenuService {
   constructor(
     @inject(MAIN_TOKENS.ExternalAppsService)
     private readonly externalAppsService: ExternalAppsService,
+    @inject(MAIN_TOKENS.Dialog)
+    private readonly dialog: IDialog,
   ) {}
 
   private async getExternalAppsData() {
@@ -348,17 +345,15 @@ export class ContextMenuService {
   }
 
   private async confirm(options: ConfirmOptions): Promise<boolean> {
-    const win = getMainWindow();
-    const result = await dialog.showMessageBox({
-      ...(win ? { parent: win } : {}),
-      type: "question",
+    const response = await this.dialog.confirm({
+      severity: "question",
       title: options.title,
       message: options.message,
       detail: options.detail,
-      buttons: ["Cancel", options.confirmLabel],
-      defaultId: 1,
-      cancelId: 0,
+      options: ["Cancel", options.confirmLabel],
+      defaultIndex: 1,
+      cancelIndex: 0,
     });
-    return result.response === 1;
+    return response === 1;
   }
 }