import { fileURLToPath } from 'node:url';
import path from 'node:path';
import preact from '@preact/preset-vite';
import { defineConfig } from 'vite';

const rootDir = fileURLToPath(new URL('.', import.meta.url));

function searchIndexPolicyPlugin(isDemo: boolean) {
  return {
    name: 'nodewarden-search-index-policy',
    transformIndexHtml(html: string) {
      if (isDemo) return html;
      return html.replace(
        '<meta name="viewport" content="width=device-width, initial-scale=1.0" />',
        '<meta name="viewport" content="width=device-width, initial-scale=1.0" />\n    <meta name="robots" content="noindex, nofollow, noarchive, nosnippet" />'
      );
    },
    generateBundle() {
      this.emitFile({
        type: 'asset',
        fileName: 'robots.txt',
        source: isDemo
          ? 'User-agent: *\nAllow: /\n'
          : 'User-agent: *\nDisallow: /\n',
      });
    },
  };
}

export default defineConfig(({ mode }) => {
  const isDemo = mode === 'demo';

  return {
    root: rootDir,
    plugins: [preact(), searchIndexPolicyPlugin(isDemo)],
    define: {
      __NODEWARDEN_DEMO__: JSON.stringify(isDemo),
    },
    resolve: {
      alias: {
        '@/lib/demo': path.resolve(rootDir, isDemo ? 'src/lib/demo.ts' : 'src/lib/demo.empty.ts'),
        '@/lib/demo-brand-icons': path.resolve(
          rootDir,
          isDemo ? 'src/lib/demo-brand-icons.ts' : 'src/lib/demo.empty.ts'
        ),
        '@': path.resolve(rootDir, 'src'),
        '@shared': path.resolve(rootDir, '../shared'),
      },
    },
    build: {
      outDir: path.resolve(rootDir, '../dist'),
      emptyOutDir: true,
      sourcemap: false,
      target: 'esnext',
      rollupOptions: {
        output: {
          manualChunks(id) {
            if (id.includes('/node_modules/')) {
              return 'vendor';
            }

            const normalized = id.replace(/\\/g, '/');

            const localeMatch = normalized.match(/\/src\/lib\/i18n\/locales\/(.+)\.ts$/);
            if (localeMatch) {
              if (localeMatch[1] === 'en') return undefined;
              return `i18n-${localeMatch[1]}`;
            }

            if (normalized.includes('/src/lib/i18n.ts')) {
              return 'i18n-core';
            }

            if (
              normalized.includes('/src/components/AuthViews.tsx') ||
              normalized.includes('/src/components/PublicSendPage.tsx') ||
              normalized.includes('/src/components/RecoverTwoFactorPage.tsx') ||
              normalized.includes('/src/components/JwtWarningPage.tsx') ||
              normalized.includes('/src/lib/app-auth.ts')
            ) {
              return 'auth-suite';
            }

            if (
              !isDemo &&
              (
                normalized.includes('/src/components/ImportPage.tsx') ||
                normalized.includes('/src/lib/import-') ||
                normalized.includes('/src/lib/export-formats.ts') ||
                normalized.includes('/src/components/SendsPage.tsx') ||
                normalized.includes('/src/components/TotpCodesPage.tsx') ||
                normalized.includes('/src/components/BackupCenterPage.tsx') ||
                normalized.includes('/src/components/backup-center/') ||
                normalized.includes('/src/components/SettingsPage.tsx') ||
                normalized.includes('/src/components/SecurityDevicesPage.tsx') ||
                normalized.includes('/src/components/AdminPage.tsx')
              )
            ) {
              return 'workspace-suite';
            }

            return undefined;
          },
        },
      },
    },
    server: {
      port: 5173,
      fs: {
        allow: [path.resolve(rootDir, '..')],
      },
      proxy: {
        '/api': 'http://127.0.0.1:8787',
        '/identity': 'http://127.0.0.1:8787',
        '/setup': 'http://127.0.0.1:8787',
        '/icons': 'http://127.0.0.1:8787',
        '/config': 'http://127.0.0.1:8787',
        '/notifications': 'http://127.0.0.1:8787',
        '/.well-known': 'http://127.0.0.1:8787',
      },
    },
  };
});