prism/src/shared/windowCapabilities.ts

import type { WindowCapabilities, WindowDisplayServer } from '../types/windowCapabilities'

interface WindowCapabilityResolutionOptions {
  platform: string
  argv?: readonly string[]
  env?: Record<string, string | undefined>
}

export const DEFAULT_WINDOW_CAPABILITIES: WindowCapabilities = {
  displayServer: 'other',
  useNativeDragRegions: false,
  supportsProgrammaticReposition: true,
  supportsGeometryPersistence: true,
}

function readSwitchValue(argv: readonly string[], switchName: string): string | null {
  const exactPrefix = `--${switchName}=`

  for (let index = 0; index < argv.length; index += 1) {
    const value = argv[index]
    if (value.startsWith(exactPrefix)) {
      return value.slice(exactPrefix.length)
    }

    if (value === `--${switchName}`) {
      const nextValue = argv[index + 1]
      if (typeof nextValue === 'string' && !nextValue.startsWith('--')) {
        return nextValue
      }
      return ''
    }
  }

  return null
}

function resolveLinuxDisplayServer(
  argv: readonly string[],
  env: Record<string, string | undefined>,
): WindowDisplayServer {
  const ozonePlatform = readSwitchValue(argv, 'ozone-platform')?.toLowerCase() ?? null

  if (ozonePlatform === 'x11') {
    return 'x11'
  }

  if (ozonePlatform === 'wayland') {
    return 'wayland'
  }

  const sessionType = env.XDG_SESSION_TYPE?.toLowerCase() ?? ''
  const hasWaylandDisplay = Boolean(env.WAYLAND_DISPLAY)
  const hasX11Display = Boolean(env.DISPLAY)

  if (sessionType === 'wayland') {
    return 'wayland'
  }

  if (sessionType === 'x11') {
    return 'x11'
  }

  if (hasWaylandDisplay && !hasX11Display) {
    return 'wayland'
  }

  if (hasX11Display) {
    return 'x11'
  }

  if (hasWaylandDisplay) {
    return 'wayland'
  }

  return 'unknown'
}

export function resolveWindowCapabilities(options: WindowCapabilityResolutionOptions): WindowCapabilities {
  if (options.platform !== 'linux') {
    return { ...DEFAULT_WINDOW_CAPABILITIES }
  }

  const displayServer = resolveLinuxDisplayServer(options.argv ?? [], options.env ?? {})
  const isNativeWayland = displayServer === 'wayland'

  return {
    displayServer,
    useNativeDragRegions: isNativeWayland,
    supportsProgrammaticReposition: !isNativeWayland,
    supportsGeometryPersistence: !isNativeWayland,
  }
}