import { createRequire } from 'node:module';
import * as path from 'node:path';

import type { Plugin } from 'vitest/config';
import { configDefaults, defineConfig } from 'vitest/config';

const require = createRequire(import.meta.url);
const runtimePkg = require.resolve('./src/internal.ts');
const internalPreactRoot = path.dirname(require.resolve('preact/package.json'));

function transformReactLynxPlugin(): Plugin {
  return {
    name: 'transformReactLynxPlugin',
    enforce: 'pre',
    transform(sourceText, sourcePath) {
      const { transformReactLynxSync } = require(
        '@lynx-js/react-transform',
      ) as typeof import('@lynx-js/react-transform');
      const relativePath = path.basename(sourcePath);

      if (!relativePath.endsWith('.jsx') && !relativePath.endsWith('.tsx')) {
        return {
          code: sourceText,
          map: null,
        };
      }

      const result = transformReactLynxSync(sourceText, {
        mode: 'test',
        pluginName: '',
        filename: relativePath,
        sourcemap: true,
        snapshot: {
          preserveJsx: false,
          runtimePkg,
          jsxImportSource: '@lynx-js/react',
          filename: 'test',
          target: 'MIXED',
          // Files named `*.legacy-slot.test.jsx` are compiled with the
          // legacy children + wrapper codegen (`compat.legacySlot`).
          legacySlot: relativePath.endsWith('.legacy-slot.test.jsx'),
        },
        dynamicImport: false,
        // snapshot: true,
        directiveDCE: false,
        defineDCE: false,
        shake: false,
        compat: false,
        worklet: false,
        refresh: false,
        cssScope: false,
      });

      return {
        code: result.code,
        map: result.map,
      };
    },
  };
}

export default defineConfig({
  plugins: [
    transformReactLynxPlugin(),
  ],
  resolve: {
    dedupe: ['preact'],
    alias: [
      { find: /^preact$/, replacement: path.join(internalPreactRoot, 'dist/preact.mjs') },
      { find: /^preact\/compat$/, replacement: path.join(internalPreactRoot, 'compat/dist/compat.mjs') },
      { find: /^preact\/hooks$/, replacement: path.join(internalPreactRoot, 'hooks/dist/hooks.mjs') },
      {
        find: /^preact\/jsx-dev-runtime$/,
        replacement: path.join(internalPreactRoot, 'jsx-runtime/dist/jsxRuntime.mjs'),
      },
      { find: /^preact\/jsx-runtime$/, replacement: path.join(internalPreactRoot, 'jsx-runtime/dist/jsxRuntime.mjs') },
      { find: '@lynx-js/react/compat', replacement: path.resolve(__dirname, './compat/index.js') },
      {
        find: '@lynx-js/react/worklet-runtime/init',
        replacement: path.resolve(__dirname, './src/worklet-runtime/index.ts'),
      },
      {
        find: '@lynx-js/react/worklet-runtime/bindings',
        replacement: path.resolve(__dirname, './src/worklet-runtime/bindings/index.ts'),
      },
      { find: '@lynx-js/react/runtime-components', replacement: path.resolve(__dirname, '../components/src/index.ts') },
      {
        find: /^@lynx-js\/react\/element-template\/internal$/,
        replacement: path.resolve(__dirname, './src/element-template/internal.ts'),
      },
      {
        find: /^@lynx-js\/react\/element-template$/,
        replacement: path.resolve(__dirname, './src/element-template/index.ts'),
      },
      { find: '@lynx-js/react/internal', replacement: path.resolve(__dirname, './src/internal.ts') },
      {
        find: /^@lynx-js\/react-signals\/lepus$/,
        replacement: path.resolve(__dirname, '../../react-signals/src/mainThread.ts'),
      },
      { find: '@lynx-js/react/jsx-dev-runtime', replacement: path.resolve(__dirname, './jsx-dev-runtime/index.js') },
      { find: '@lynx-js/react/jsx-runtime', replacement: path.resolve(__dirname, './jsx-runtime/index.js') },
      { find: /^@lynx-js\/react\/hooks$/, replacement: path.resolve(__dirname, './src/core/hooks/react.ts') },
      {
        find: /^@lynx-js\/react\/lepus\/hooks$/,
        replacement: path.resolve(__dirname, './src/core/hooks/mainThread.ts'),
      },
      { find: '@lynx-js/react/lepus', replacement: path.resolve(__dirname, './lepus/index.js') },
      {
        find: '@lynx-js/react/legacy-react-runtime',
        replacement: path.resolve(__dirname, './src/core/compat/legacy-react-runtime.ts'),
      },
      { find: '@lynx-js/react', replacement: path.resolve(__dirname, './src/index.ts') },
    ],
  },
  test: {
    name: 'react/runtime',
    exclude: [
      ...configDefaults.exclude,
      '**/__test__/element-template/**',
    ],
    server: {
      deps: {
        inline: [
          /@lynx-js\/internal-preact/,
        ],
      },
    },
    coverage: {
      exclude: [
        'debug',
        'jsx-runtime',
        'jsx-dev-runtime',
        'lazy/element-template*.js',
        'lepus/jsx-dev-runtime',
        'lepus/index.d.ts',
        'vitest.config.ts',
        '__test__/element-template/**',
        '__test__/snapshot/utils/**',
        '__test__/test-utils/**',
        'lib/**',
        'worklet-runtime/**',
        'src/element-template/**',
        'src/core/hooks/mainThread.ts',
        'src/core/hooks/mainThreadImpl.ts',
        'src/shared/component-stack.ts',
        'src/shared/profile.ts',
        'src/index.ts',
        'src/lynx-api.ts',
        'src/lynx.ts',
        'src/root.ts',
        'src/worklet-runtime/api/lepusQuerySelector.ts',
        'src/worklet-runtime/api/lynxApi.ts',
        'src/worklet-runtime/bindings/**',
        'src/worklet-runtime/global.ts',
        'src/worklet-runtime/index.ts',
        'src/worklet-runtime/listeners.ts',
        'src/worklet-runtime/types/**',
        'src/snapshot/debug/debug.ts',
        'src/snapshot/debug/profileHooks.ts',
        'src/snapshot/debug/utils.ts',
        'src/snapshot/lynx/calledByNative.ts',
        'src/snapshot/lynx/env.ts',
        'src/snapshot/lynx/tt.ts',
        'src/snapshot/compat/componentIs.ts',
        'src/snapshot/snapshot/types.ts',
        'src/snapshot/worklet/hmr.ts',

        '__test__/snapshot/page.test.jsx',
        '**/*.d.ts',
        '**/*.test-d.*',
      ],
      thresholds: {
        lines: 100,
        functions: 100,
        branches: 100,
        statements: 100,
      },
    },
    setupFiles: [
      path.join(__dirname, './__test__/snapshot/utils/globals.js'),
      path.join(__dirname, './__test__/snapshot/utils/setup.js'),
      path.join(__dirname, './__test__/snapshot/utils/runtimeProxy.ts'),
    ],
  },
});