import {
  START_ENVIRONMENT_NAMES,
  tanStackStartRsbuild,
} from '@tanstack/start-plugin-core'
import { reactStartDefaultEntryPaths } from './shared'
import type { TanStackStartRsbuildPluginCoreOptions } from '@tanstack/start-plugin-core/rsbuild/types'
import type { TanStackStartRsbuildInputConfig } from '@tanstack/start-plugin-core'
import type { RsbuildPlugin } from '@rsbuild/core'

export function tanstackStart(
  options?: TanStackStartRsbuildInputConfig & { rsc?: { enabled?: boolean } },
): RsbuildPlugin {
  const rscEnabled = options?.rsc?.enabled ?? false

  let corePluginOpts: TanStackStartRsbuildPluginCoreOptions = {
    framework: 'react',
    defaultEntryPaths: reactStartDefaultEntryPaths,
    providerEnvironmentName: START_ENVIRONMENT_NAMES.server,
    ssrIsProvider: true,
  }

  if (rscEnabled) {
    const rscConfig = configureRscRsbuild()
    corePluginOpts = {
      ...corePluginOpts,
      providerEnvironmentName: rscConfig.providerEnvironmentName,
      ssrIsProvider: false,
      serializationAdapters: rscConfig.serializationAdapters,
      rsc: true,
    }
  }

  return tanStackStartRsbuild(corePluginOpts, options)
}

/**
 * Configure RSC options for the rsbuild adapter.
 *
 * Returns provider env, serialization adapters, and core RSC options.
 *
 * The SSR resolver strategy is intentionally NOT changed from 'default' —
 * in the rspack layered model, SSR and RSC live in the same
 * compilation so they share the same resolver file. No forwarding needed.
 */
function configureRscRsbuild(): {
  providerEnvironmentName: TanStackStartRsbuildPluginCoreOptions['providerEnvironmentName']
  serializationAdapters: TanStackStartRsbuildPluginCoreOptions['serializationAdapters']
} {
  return {
    providerEnvironmentName: START_ENVIRONMENT_NAMES.server,
    serializationAdapters: [
      {
        client: {
          module: '@tanstack/react-start/rsc/serialization/client',
          export: 'rscSerializationAdapter',
          isFactory: true,
        },
        server: {
          module: '@tanstack/react-start/rsc/serialization/server',
          export: 'rscSerializationAdapter',
          isFactory: true,
        },
      },
    ],
  }
}