miniSCP/ui/src/app/rpc.service.ts

import { Injectable } from '@angular/core';
import { Electroview } from 'electrobun/view';
import type { AppRPC } from '@shared/rpc';
import type { TransferSnapshot } from '@shared/types';

// Matches MAX_REQUEST_TIME_MS on the bun side — SFTP ops can be slow.
const MAX_REQUEST_TIME_MS = 10 * 60_000;

export type SessionClosedEvent = { sessionId: string; reason: string };

@Injectable({ providedIn: 'root' })
export class RpcService {
  /** False when the page is opened in a plain browser instead of the app. */
  readonly available: boolean;

  // Push messages are delivered through plain callbacks, NOT signals: a signal
  // consumed from an effect that also writes signals re-triggers itself
  // (this froze the app), and signal coalescing would drop progress snapshots.
  private transferListeners = new Set<(s: TransferSnapshot) => void>();
  private closedListeners = new Set<(e: SessionClosedEvent) => void>();
  private menuListeners = new Set<(action: string) => void>();

  private rpc: ReturnType<typeof Electroview.defineRPC<AppRPC>> | undefined;
  private ready: Promise<void> = Promise.resolve();

  constructor() {
    this.available = typeof window !== 'undefined' && '__electrobun' in window;
    if (!this.available) return;

    this.rpc = Electroview.defineRPC<AppRPC>({
      maxRequestTime: MAX_REQUEST_TIME_MS,
      handlers: {
        messages: {
          transferUpdate: (snapshot) => {
            for (const listener of this.transferListeners) listener(snapshot);
          },
          sessionClosed: (event) => {
            for (const listener of this.closedListeners) listener(event);
          },
          menuAction: ({ action }) => {
            for (const listener of this.menuListeners) listener(action);
          },
        },
      },
    });
    const electroview = new Electroview({ rpc: this.rpc });

    // Requests fired before the RPC WebSocket finishes connecting are lost, so
    // everything below gates on it being open (services fire calls at startup).
    const socket = electroview.bunSocket;
    if (socket && socket.readyState !== WebSocket.OPEN) {
      this.ready = new Promise((resolve) => {
        socket.addEventListener('open', () => resolve(), { once: true });
        socket.addEventListener('error', () => resolve(), { once: true });
      });
    }
  }

  onTransferUpdate(listener: (snapshot: TransferSnapshot) => void): void {
    this.transferListeners.add(listener);
  }

  onSessionClosed(listener: (event: SessionClosedEvent) => void): void {
    this.closedListeners.add(listener);
  }

  onMenuAction(listener: (action: string) => void): void {
    this.menuListeners.add(listener);
  }

  /** Typed request proxy: `await rpc.request.listDir({...})`. Waits for the bridge. */
  get request() {
    const rpc = this.rpc;
    if (!rpc) throw new Error('RPC is only available inside the desktop app');
    const ready = this.ready;
    return new Proxy({} as (typeof rpc)['request'], {
      get:
        (_target, method: string) =>
        async (params?: unknown) => {
          await ready;
          const methods = rpc.request as unknown as Record<
            string,
            (p?: unknown) => Promise<unknown>
          >;
          return methods[method](params);
        },
    });
  }

  /** Fire-and-forget messages to the bun process (e.g. workspacesSync). */
  get send() {
    const rpc = this.rpc;
    if (!rpc) throw new Error('RPC is only available inside the desktop app');
    const ready = this.ready;
    return new Proxy({} as (typeof rpc)['send'], {
      get:
        (_target, message: string) =>
        (payload?: unknown) => {
          void ready.then(() => {
            const messages = rpc.send as unknown as Record<string, (p?: unknown) => void>;
            messages[message](payload);
          });
        },
    });
  }
}