vite.config.js

import { fileURLToPath, URL } from 'node:url'
import fs from 'node:fs'
import path from 'node:path'

import vue from '@vitejs/plugin-vue'
import { defineConfig } from 'vite'
import { buildBackgroundStyle, injectApiBase, injectTitle, parseCspOrigins, stripCspMeta } from './src/utils/csp.js'

const devProxyTarget = process.env.VITE_DEV_PROXY_TARGET || 'https://localhost:8787'
const rootDir = fileURLToPath(new URL('.', import.meta.url))
const frontendDir = fileURLToPath(new URL('./src/frontend', import.meta.url))

const createWorkerProxy = () => ({
  target: devProxyTarget,
  changeOrigin: true,
  secure: false,
  ws: true
})

function loadEnvFile() {
  const envPath = path.resolve(rootDir, '.env')
  const env = {}
  if (!fs.existsSync(envPath)) return env

  for (const line of fs.readFileSync(envPath, 'utf8').split('\n')) {
    const trimmed = line.trim()
    if (!trimmed || trimmed.startsWith('#')) continue
    const eqIndex = trimmed.indexOf('=')
    if (eqIndex === -1) continue
    const key = trimmed.slice(0, eqIndex).trim()
    let value = trimmed.slice(eqIndex + 1).trim()
    if ((value.startsWith('"') && value.endsWith('"')) || (value.startsWith("'") && value.endsWith("'"))) {
      value = value.slice(1, -1)
    }
    env[key] = value
  }
  return env
}

function envPlugin() {
  const env = loadEnvFile()
  const apiDomains = [
    ...parseCspOrigins(env.API_BASE || ''),
    ...parseCspOrigins(env.CSP_API || ''),
  ]

  return {
    name: 'env-inject',
    transformIndexHtml(html) {
      let transformed = stripCspMeta(html)
      transformed = injectTitle(transformed, env.TITLE || '')
      transformed = injectApiBase(transformed, apiDomains)
      if (env.BACKGROUND_IMAGE) {
        transformed = transformed.replace('</head>', `${buildBackgroundStyle(env.BACKGROUND_IMAGE)}\n</head>`)
      }
      return transformed
    },
  }
}

export default defineConfig({
  root: frontendDir,
  publicDir: fileURLToPath(new URL('./public', import.meta.url)),
  plugins: [vue(), envPlugin()],
  base: process.env.VITE_BASE || '/',
  resolve: {
    alias: {
      '@': frontendDir
    }
  },
  build: {
    outDir: fileURLToPath(new URL('./dist', import.meta.url)),
    assetsDir: 'static',
    emptyOutDir: true,
    rollupOptions: {
      output: {
        entryFileNames: 'static/[name]-[hash].js',
        chunkFileNames: 'static/[name]-[hash].js',
        assetFileNames: 'static/[name]-[hash].[ext]'
      }
    }
  },
  server: {
    port: 5173,
    proxy: {
      '/api': createWorkerProxy(),
      '/admin/api': createWorkerProxy(),
      '/update': createWorkerProxy(),
      '/updateDatabase': createWorkerProxy(),
      '/clearHistory': createWorkerProxy(),
      '/__do': createWorkerProxy()
    }
  }
})