import type { NextConfigComplete } from '../server/config-shared'
import type { Span } from '../trace'

import * as Log from './output/log'
import createSpinner from './spinner'
import isError from '../lib/is-error'
import type { Telemetry } from '../telemetry/storage'
import { EVENT_BUILD_FEATURE_USAGE } from '../telemetry/events/build'
import { hrtimeBigIntDurationToString } from './duration-to-string'

// TODO: refactor this to account for more compiler lifecycle events
// such as beforeProductionBuild, but for now this is the only one that is needed
export async function runAfterProductionCompile({
  config,
  buildSpan,
  telemetry,
  metadata,
}: {
  config: NextConfigComplete
  buildSpan: Span
  telemetry: Telemetry
  metadata: {
    projectDir: string
    distDir: string
  }
}): Promise<void> {
  const run = config.compiler.runAfterProductionCompile
  if (!run) {
    return
  }
  telemetry.record([
    {
      eventName: EVENT_BUILD_FEATURE_USAGE,
      payload: {
        featureName: 'runAfterProductionCompile',
        invocationCount: 1,
      },
    },
  ])
  const afterBuildSpinner = createSpinner(
    'Running next.config.js provided runAfterProductionCompile'
  )

  try {
    const startTime = process.hrtime.bigint()
    await buildSpan
      .traceChild('after-production-compile')
      .traceAsyncFn(async () => {
        await run(metadata)
      })
    const duration = process.hrtime.bigint() - startTime
    const formattedDuration = hrtimeBigIntDurationToString(duration)
    Log.event(`Completed runAfterProductionCompile in ${formattedDuration}`)
  } catch (err) {
    // Handle specific known errors differently if needed
    if (isError(err)) {
      Log.error(`Failed to run runAfterProductionCompile: ${err.message}`)
    }

    throw err
  } finally {
    afterBuildSpinner?.stop()
  }
}