vite.config.ts

import type { RouteType } from './src/appRoutes.js';
import { getRoutes } from './src/appRoutes.js';
import { uvPath } from '@titaniumnetwork-dev/ultraviolet';
import { createBareServer } from '@tomphttp/bare-server-node';
import react from '@vitejs/plugin-react-swc';
import { createRequire } from 'node:module';
import { resolve } from 'node:path';
import type { ConfigEnv } from 'vite';
import { defineConfig, loadEnv } from 'vite';
import { ViteMinifyPlugin } from 'vite-plugin-minify';
import { viteStaticCopy } from 'vite-plugin-static-copy';
import svgr from 'vite-plugin-svgr';

// https://vitejs.dev/config/
export default function viteConfig({ mode }: ConfigEnv) {
	Object.assign(process.env, loadEnv(mode, process.cwd()));
	process.env.VITE_PUBLIC_PATH ??= '';

	const require = createRequire(import.meta.url);

	for (const required of [
		'VITE_ROUTER',
		'VITE_SUPPORT_EMAIL',
		'VITE_HAT_BADGE',
		'VITE_DEFAULT_PROXY',
		'VITE_TN_DISCORD_URL',
		'VITE_BARE_APIS',
		'VITE_RH_API',
		'VITE_DB_API',
		'VITE_THEATRE_CDN',
	])
		if (!(required in process.env))
			throw new TypeError(`Missing environment variable: ${required}`);

	const routes = getRoutes(
		process.env.VITE_ROUTER as RouteType,
		process.env.VITE_PUBLIC_PATH,
	);

	return defineConfig({
		server: {
			proxy: {
				'/cdn/': 'https://holyubofficial.net/',
				'/db/': 'https://holyubofficial.net/',
			},
		},
		build: {
			sourcemap: true,
			rollupOptions: {
				output: {
					chunkFileNames: '[hash].js',
				},
				plugins: [
					{
						name: 'copy index.html to all the hot routes',
						async generateBundle(options, bundle) {
							const index = bundle['index.html'];

							for (const file of routes)
								bundle[file.file] = {
									...index,
									fileName: file.file,
								};
						},
					},
				],
			},
		},
		plugins: [
			{
				name: 'serve index.html for all the hot routes',
				configureServer(server) {
					server.middlewares.use((req, _res, next) => {
						let url = req.url;
						const search = url.indexOf('?');
						if (search !== -1) url = url.slice(0, search);
						if (routes.some((route) => route.path === url))
							req.url = '/index.html';
						next();
					});
				},
			},
			{
				name: 'bare server',
				configureServer(server) {
					const bare = createBareServer('/bare/');
					server.middlewares.use((req, res, next) => {
						if (bare.shouldRoute(req)) bare.routeRequest(req, res);
						else next();
					});

					const upgraders = server.httpServer.listeners(
						'upgrade',
					) as Parameters<(typeof server)['httpServer']['on']>[1][];

					// remover other listeners
					for (const upgrader of upgraders)
						server.httpServer.off('upgrade', upgrader);

					server.httpServer.on('upgrade', (req, socket, head) => {
						if (bare.shouldRoute(req)) bare.routeUpgrade(req, socket, head);
						else for (const upgrader of upgraders) upgrader(req, socket, head);
					});
				},
			},
			react(),
			svgr(),
			viteStaticCopy({
				targets: [
					{
						src: uvPath,
						dest: '',
						rename: 'uv',
						overwrite: false,
					},
					{
						src: resolve(require.resolve('@ruffle-rs/ruffle'), '..'),
						dest: '',
						rename: 'ruffle',
					},
				],
			}),
			// minify the output HTML
			ViteMinifyPlugin(),
		],
	});
}