orca/src/relay/workspace-session-handler.ts

import { existsSync, mkdirSync, readFileSync, renameSync, writeFileSync } from 'node:fs'
import { homedir } from 'node:os'
import { dirname, join } from 'node:path'
import type { RelayDispatcher } from './dispatcher'
import { publishWorkspaceSnapshotChange } from './workspace-snapshot-publication'

type RemoteWorkspaceSnapshot = {
  namespace: string
  revision: number
  updatedAt: number
  schemaVersion: number
  session: Record<string, unknown>
}

type ConnectedClient = {
  clientId: string
  name: string
  lastSeenAt: number
}

type PatchResult =
  | { ok: true; snapshot: RemoteWorkspaceSnapshot }
  | {
      ok: false
      reason: 'stale-revision' | 'unavailable'
      snapshot?: RemoteWorkspaceSnapshot
      message?: string
    }

const SNAPSHOT_SCHEMA_VERSION = 1
const PRESENCE_TTL_MS = 45_000

function emptySession(): Record<string, unknown> {
  return {
    activeRepoId: null,
    activeWorktreeId: null,
    activeTabId: null,
    tabsByWorktree: {},
    terminalLayoutsByTabId: {}
  }
}

function sanitizeNamespace(namespace: unknown): string {
  const raw = typeof namespace === 'string' && namespace.trim() ? namespace.trim() : 'default'
  return raw.replace(/[^a-zA-Z0-9._-]/g, '_').slice(0, 160) || 'default'
}

function sanitizeClientName(value: string): string {
  return value.replace(/\s+/g, ' ').trim().slice(0, 80)
}

function sanitizeClientId(value: string): string {
  return value.trim().slice(0, 200)
}

export class WorkspaceSessionHandler {
  private readonly clientsByNamespace = new Map<string, Map<string, ConnectedClient>>()

  constructor(
    private dispatcher: RelayDispatcher,
    private baseDir = join(homedir(), '.orca', 'sessions')
  ) {
    this.dispatcher.onRequest('workspace.get', (params) => this.get(params))
    this.dispatcher.onRequest('workspace.patch', (params) => this.patch(params))
    this.dispatcher.onRequest('workspace.presence', (params) => this.presence(params))
  }

  private snapshotPath(namespace: string): string {
    return join(this.baseDir, `${namespace}.json`)
  }

  private read(namespace: string): RemoteWorkspaceSnapshot {
    const path = this.snapshotPath(namespace)
    if (!existsSync(path)) {
      return {
        namespace,
        revision: 0,
        updatedAt: 0,
        schemaVersion: SNAPSHOT_SCHEMA_VERSION,
        session: emptySession()
      }
    }

    try {
      const parsed = JSON.parse(readFileSync(path, 'utf-8')) as Partial<RemoteWorkspaceSnapshot>
      return {
        namespace,
        revision:
          typeof parsed.revision === 'number' && Number.isFinite(parsed.revision)
            ? parsed.revision
            : 0,
        updatedAt:
          typeof parsed.updatedAt === 'number' && Number.isFinite(parsed.updatedAt)
            ? parsed.updatedAt
            : 0,
        schemaVersion:
          typeof parsed.schemaVersion === 'number' && Number.isFinite(parsed.schemaVersion)
            ? parsed.schemaVersion
            : SNAPSHOT_SCHEMA_VERSION,
        session:
          parsed.session && typeof parsed.session === 'object' && !Array.isArray(parsed.session)
            ? (parsed.session as Record<string, unknown>)
            : emptySession()
      }
    } catch {
      return {
        namespace,
        revision: 0,
        updatedAt: 0,
        schemaVersion: SNAPSHOT_SCHEMA_VERSION,
        session: emptySession()
      }
    }
  }

  private write(snapshot: RemoteWorkspaceSnapshot): void {
    const path = this.snapshotPath(snapshot.namespace)
    mkdirSync(dirname(path), { recursive: true, mode: 0o700 })
    const tmpPath = `${path}.tmp`
    writeFileSync(tmpPath, JSON.stringify(snapshot, null, 2), { mode: 0o600 })
    renameSync(tmpPath, path)
  }

  private async get(params: Record<string, unknown>): Promise<RemoteWorkspaceSnapshot> {
    return this.read(sanitizeNamespace(params.namespace))
  }

  private async patch(params: Record<string, unknown>): Promise<PatchResult> {
    const namespace = sanitizeNamespace(params.namespace)
    const current = this.read(namespace)
    const baseRevision = Number(params.baseRevision)
    if (Number.isFinite(baseRevision) && baseRevision !== current.revision) {
      return { ok: false, reason: 'stale-revision', snapshot: current }
    }

    const patch = params.patch as { kind?: unknown; session?: unknown } | undefined
    if (
      !patch ||
      patch.kind !== 'replace-session' ||
      !patch.session ||
      typeof patch.session !== 'object' ||
      Array.isArray(patch.session)
    ) {
      return { ok: false, reason: 'unavailable', message: 'Invalid workspace patch' }
    }

    const snapshot: RemoteWorkspaceSnapshot = {
      namespace,
      revision: current.revision + 1,
      updatedAt: Date.now(),
      schemaVersion: SNAPSHOT_SCHEMA_VERSION,
      session: patch.session as Record<string, unknown>
    }
    this.write(snapshot)
    publishWorkspaceSnapshotChange(
      this.dispatcher,
      {
        namespace,
        snapshot,
        sourceClientId: typeof params.clientId === 'string' ? params.clientId : undefined
      },
      namespace
    )
    return { ok: true, snapshot }
  }

  private async presence(params: Record<string, unknown>): Promise<{ clients: ConnectedClient[] }> {
    const namespace = sanitizeNamespace(params.namespace)
    const clientId = typeof params.clientId === 'string' ? sanitizeClientId(params.clientId) : ''
    const name = typeof params.clientName === 'string' ? sanitizeClientName(params.clientName) : ''
    const clients = this.clientsByNamespace.get(namespace) ?? new Map<string, ConnectedClient>()
    this.clientsByNamespace.set(namespace, clients)

    const now = Date.now()
    for (const [id, client] of clients) {
      if (now - client.lastSeenAt > PRESENCE_TTL_MS) {
        clients.delete(id)
      }
    }
    if (clientId) {
      clients.set(clientId, {
        clientId,
        name: name || 'Unknown device',
        lastSeenAt: now
      })
    }

    return {
      clients: Array.from(clients.values()).sort((a, b) => b.lastSeenAt - a.lastSeenAt)
    }
  }
}