vite.config.ts

import path from 'node:path'
import { createConfig } from '@arianrhodsandlot/vite-plus-config'
import devServer, { defaultOptions } from '@hono/vite-dev-server'
import { reactRouter } from '@react-router/dev/vite'
import tailwindcss from '@tailwindcss/vite'
import { defaults, noop } from 'es-toolkit/compat'
import { $, execaNode } from 'execa'
import fs from 'fs-extra'
import { DateTime } from 'luxon'
import devtoolsJson from 'vite-plugin-devtools-json'
import { defineConfig, type Plugin, type UserConfig } from 'vite-plus'
import { getTargetRuntime, logServerInfo, prepareWranglerConfig } from './scripts/utils.ts'
import { getDirectories } from './src/constants/env.ts'

defaults(process.env, {
  RETROASSEMBLY_BUILD_TIME_VITE_BUILD_TIME: DateTime.now().setZone('utc').toISO(),
  RETROASSEMBLY_BUILD_TIME_VITE_VERSION: await getVersion(),
  RETROASSEMBLY_RUN_TIME_PORT: '8000',
})

async function getVersion() {
  return getEnvVersion() ?? (await getGitDescription()) ?? ''
}

function getEnvVersion() {
  const envNames = ['WORKERS_CI_COMMIT_SHA']
  for (const envName of envNames) {
    if (process.env[envName]) {
      return process.env[envName].slice(0, 7)
    }
  }
}

async function getGitDescription() {
  try {
    const {
      stdout: [revision],
    } = await $({ lines: true })`git describe --tags`
    return revision
  } catch {}
}

function serverInfo() {
  const plugin: Plugin = {
    configureServer(server) {
      const { httpServer } = server
      server.printUrls = noop
      server.bindCLIShortcuts = noop
      server.config.logger.info = noop
      httpServer?.on('listening', () => {
        const address = httpServer?.address()
        if (address && typeof address === 'object') {
          logServerInfo('localhost', address.port, true)
        }
      })
    },
    name: 'log-server-info',
  }
  return plugin
}

export default defineConfig(async (env) => {
  const envPort = process.env.RETROASSEMBLY_RUN_TIME_PORT || process.env.PORT
  const port = envPort ? Math.trunc(Number(envPort)) || 8000 : 8000
  const plugins = [tailwindcss({ optimize: false }), reactRouter(), [devtoolsJson()], serverInfo()]
  const config: UserConfig = {
    build: { chunkSizeWarningLimit: 1024 },
    clearScreen: false,
    envPrefix: 'RETROASSEMBLY_BUILD_TIME_VITE_',
    pack: {
      clean: false,
      entry: ['scripts/serve.ts', 'src/server/node.ts'],
      outDir: 'dist',
      suppressWarnings: [/module level directive may not be preserved/iu],
    },
    plugins: plugins as UserConfig['plugins'],
    server: {
      allowedHosts: true,
      hmr: { overlay: true },
      host: true,
      open: true,
      port,
    },
    staged: {
      'pnpm-lock.yaml': 'node --run=check-lockfile',
    },
  }

  if (getTargetRuntime() === 'workerd') {
    if (env.command === 'serve') {
      await $`wrangler d1 migrations apply retroassembly_library`
    }
    await prepareWranglerConfig()
    const { cloudflare } = await import('@cloudflare/vite-plugin')
    plugins.push(cloudflare({ viteEnvironment: { name: 'ssr' } }))
    config.resolve = {
      alias: {
        '@entry.server.tsx': path.resolve(
          'node_modules',
          'react-router-hono-fullstack-template',
          'app',
          'entry.server.tsx',
        ),
      },
    }
  } else {
    if (env.command === 'serve') {
      const { storageDirectory } = getDirectories()
      await fs.ensureDir(storageDirectory)
      await execaNode`./src/utils/server/migration/initalization.ts`
      const serverAdapterPlugin = devServer({
        entry: path.resolve('src', 'server', 'node-dev.ts'),
        exclude: [
          ...defaultOptions.exclude,
          /^\/src\/.+/u,
          '/.well-known/appspecific/com.chrome.devtools.json',
          /\?(?:inline|url|no-inline|raw|import(?:&(?:inline|url|no-inline|raw))*)$/u,
        ],
        injectClientScript: false,
      })
      delete serverAdapterPlugin.handleHotUpdate
      plugins.push([serverAdapterPlugin])
    }
    config.resolve = {
      alias: {
        '@entry.server.tsx': path.resolve(
          'node_modules',
          '@react-router',
          'dev',
          'dist',
          'config',
          'defaults',
          'entry.server.node.tsx',
        ),
      },
    }
  }

  return createConfig(config)
})