import '@contentlayer2/utils/effect/Tracing/Enable'

import * as core from '@contentlayer2/core'
import { errorToString } from '@contentlayer2/utils'
import { E, OT, pipe, S, T } from '@contentlayer2/utils/effect'
import type * as esbuild from 'esbuild'
import type { WebpackOptionsNormalized } from 'webpack'

import { checkConstraints } from './check-constraints.js'

export type NextPluginOptions = {
  configPath?: string | undefined
  esbuildOptions?: Pick<esbuild.BuildOptions, 'external'> | undefined
}

/** Seems like the next.config.js export function might be executed multiple times, so we need to make sure we only run it once */
let contentlayerInitialized = false

const runContentlayerDev = async ({ configPath, esbuildOptions }: NextPluginOptions) => {
  if (contentlayerInitialized) return
  contentlayerInitialized = true

  await pipe(
    core.getConfigWatch({ configPath, esbuildOptions }),
    S.tapSkipFirstRight(() => T.log(`Contentlayer config change detected. Updating type definitions and data...`)),
    S.tapRight((config) => (config.source.options.disableImportAliasWarning ? T.unit : T.fork(core.validateTsconfig))),
    S.chainSwitchMapEitherRight((config) => core.generateDotpkgStream({ config, verbose: false, isDev: true })),
    S.tap(E.fold((error) => T.log(errorToString(error)), core.logGenerateInfo)),
    S.runDrain,
    runMain,
  )
}

const runContentlayerBuild = async ({ configPath, esbuildOptions }: NextPluginOptions) => {
  if (contentlayerInitialized) return
  contentlayerInitialized = true

  await pipe(
    core.getConfig({ configPath, esbuildOptions }),
    T.chain((config) => core.generateDotpkg({ config, verbose: false })),
    T.tap(core.logGenerateInfo),
    OT.withSpan('next-contentlayer:runContentlayerBuild'),
    runMain,
  )
}

const runMain = core.runMain({ tracingServiceName: 'next-contentlayer', verbose: process.env.CL_DEBUG !== undefined })

export const runBeforeWebpackCompile = async ({
  mode,
  pluginOptions,
  devServerStartedRef,
}: {
  mode: WebpackOptionsNormalized['mode']
  pluginOptions: NextPluginOptions
  devServerStartedRef: { current: boolean }
}) => {
  const isNextDev = mode === 'development'
  const isBuild = mode === 'production'

  const { configPath, esbuildOptions } = pluginOptions

  if (isBuild) {
    checkConstraints()
    await runContentlayerBuild({ configPath, esbuildOptions })
  } else if (isNextDev && !devServerStartedRef.current) {
    devServerStartedRef.current = true
    // TODO also block here until first Contentlayer run is complete
    runContentlayerDev({ configPath, esbuildOptions })
  }
}