ComfyUI_frontend/scripts/vite-define-shim.ts

/**
 * Shim for Vite define variables to make them available during Playwright test execution
 * This file should be imported before any code that uses Vite define variables
 */

// Define global constants that Vite would normally replace at build time
declare global {
  const __COMFYUI_FRONTEND_VERSION__: string
  const __SENTRY_ENABLED__: boolean
  const __SENTRY_DSN__: string
  const __ALGOLIA_APP_ID__: string
  const __ALGOLIA_API_KEY__: string
  const __USE_PROD_CONFIG__: boolean
  const __DEV_SERVER_COMFYUI_URL__: string
  const __DISTRIBUTION__: 'desktop' | 'localhost' | 'cloud'
  const __IS_NIGHTLY__: boolean
}

type GlobalWithDefines = typeof globalThis & {
  __COMFYUI_FRONTEND_VERSION__: string
  __SENTRY_ENABLED__: boolean
  __SENTRY_DSN__: string
  __ALGOLIA_APP_ID__: string
  __ALGOLIA_API_KEY__: string
  __USE_PROD_CONFIG__: boolean
  __DEV_SERVER_COMFYUI_URL__: string
  __DISTRIBUTION__: 'desktop' | 'localhost' | 'cloud'
  __IS_NIGHTLY__: boolean
  window?: Record<string, unknown>
}

const globalWithDefines = globalThis as GlobalWithDefines

// Set default values for Playwright test environment
globalWithDefines.__COMFYUI_FRONTEND_VERSION__ =
  process.env.npm_package_version || '1.0.0'
globalWithDefines.__SENTRY_ENABLED__ = false
globalWithDefines.__SENTRY_DSN__ = ''
globalWithDefines.__ALGOLIA_APP_ID__ = ''
globalWithDefines.__ALGOLIA_API_KEY__ = ''
globalWithDefines.__USE_PROD_CONFIG__ = false
globalWithDefines.__DEV_SERVER_COMFYUI_URL__ = ''
globalWithDefines.__DISTRIBUTION__ = 'localhost'
globalWithDefines.__IS_NIGHTLY__ = false

// Provide a minimal window shim for Node environment
// This is needed for code that checks window existence during imports
if (typeof window === 'undefined') {
  globalWithDefines.window = {}
}

export {}