vite.base.config.ts

import { readFileSync } from 'node:fs'
import { builtinModules } from 'node:module'
import type { AddressInfo } from 'node:net'
import { join } from 'node:path'
import type { ConfigEnv, Plugin, UserConfig } from 'vite'

import pkg from './package.json'

export const builtins = [
  'electron',
  'bufferutil',
  'utf-8-validate',
  ...builtinModules.map((m) => [m, `node:${m}`]).flat(),
]

// ESM-only packages (those with "type": "module" in their package.json)
// can't be loaded via CJS require(). In dev mode we externalize all deps
// for speed, so these must be detected and kept bundled by Vite instead.
function isEsmOnlyPackage(name: string): boolean {
  try {
    const pkgJsonPath = join(
      process.cwd(),
      'node_modules',
      name,
      'package.json'
    )
    const depPkg = JSON.parse(readFileSync(pkgJsonPath, 'utf-8')) as {
      type?: string
    }
    return depPkg.type === 'module'
  } catch {
    return false
  }
}

// Dev: externalize all CJS-compatible deps for fast builds and HMR.
// Prod: only externalize builtins so Vite bundles everything — the
// forge Vite plugin strips node_modules from the ASAR, so externalized
// npm packages would be missing at runtime.
export function getExternal(command: 'serve' | 'build') {
  if (command === 'build') return builtins
  return [
    ...builtins,
    ...Object.keys(
      'dependencies' in pkg ? (pkg.dependencies as Record<string, unknown>) : {}
    ).filter((dep) => !isEsmOnlyPackage(dep)),
  ]
}

export function getBuildConfig(env: ConfigEnv<'build'>): UserConfig {
  const { root, mode, command } = env

  return {
    root,
    mode,
    build: {
      // Prevent multiple builds from interfering with each other.
      emptyOutDir: false,
      // 🚧 Multiple builds may conflict.
      outDir: '.vite/build',
      watch: command === 'serve' ? {} : null,
      minify: command === 'build',
      sourcemap: command === 'serve',
      // Match the Chromium in our Electron version. Legacy defaults (chrome87,
      // …) make esbuild fail when transpiling modern deps (destructuring/rest).
      target: 'esnext',
    },
    esbuild: {
      target: 'esnext',
    },
    clearScreen: false,
  }
}

export function getDefineKeys(names: string[]) {
  const define: { [name: string]: K6StudioRuntimeKeys } = {}

  return names.reduce((acc, name) => {
    const NAME = name.toUpperCase()
    const keys: K6StudioRuntimeKeys = {
      VITE_DEV_SERVER_URL: `${NAME}_VITE_DEV_SERVER_URL`,
      VITE_NAME: `${NAME}_VITE_NAME`,
      SENTRY_DSN: 'SENTRY_DSN',
    }

    return { ...acc, [name]: keys }
  }, define)
}

export function getBuildDefine(env: ConfigEnv<'build'>) {
  const { command, forgeConfig } = env
  const names = forgeConfig.renderer
    .filter(({ name }) => name != null)
    .map(({ name }) => name)
  const defineKeys = getDefineKeys(names)
  const define = Object.entries(defineKeys).reduce(
    (acc, [name, keys]) => {
      const { VITE_DEV_SERVER_URL, VITE_NAME, SENTRY_DSN } = keys
      const def = {
        [VITE_DEV_SERVER_URL]:
          command === 'serve'
            ? JSON.stringify(process.env[VITE_DEV_SERVER_URL])
            : undefined,
        [VITE_NAME]: JSON.stringify(name),
        [SENTRY_DSN]: JSON.stringify(process.env.SENTRY_DSN),
      }
      return { ...acc, ...def }
    },
    // eslint-disable-next-line @typescript-eslint/no-explicit-any
    {} as Record<string, any>
  )

  return define
}

export function pluginExposeRenderer(name: string): Plugin {
  const { VITE_DEV_SERVER_URL } = getDefineKeys([name])[name]!

  return {
    name: '@electron-forge/plugin-vite:expose-renderer',
    configureServer(server) {
      process.viteDevServers ??= {}
      // Expose server for preload scripts hot reload.
      process.viteDevServers[name] = server

      server.httpServer?.once('listening', () => {
        const addressInfo = server.httpServer!.address() as AddressInfo
        // Expose env constant for main process use.
        process.env[VITE_DEV_SERVER_URL] =
          `http://localhost:${addressInfo?.port}`
      })
    },
  }
}

export function pluginHotRestart(command: 'reload' | 'restart'): Plugin {
  return {
    name: '@electron-forge/plugin-vite:hot-restart',
    closeBundle() {
      if (command === 'reload') {
        for (const server of Object.values(process.viteDevServers)) {
          // Preload scripts hot reload.
          server.ws.send({ type: 'full-reload' })
        }
      } else {
        // Main process hot restart.
        // https://github.com/electron/forge/blob/v7.2.0/packages/api/core/src/api/start.ts#L216-L223
        process.stdin.emit('data', 'rs')
      }
    },
  }
}