import type { TransformContext, ValidatedContext } from '../../types'

export async function transformStage(
  ctx: ValidatedContext,
  transformFn: ((data: unknown) => unknown) | undefined,
): Promise<TransformContext> {
  if (!transformFn) {
    return { key: ctx.key, data: ctx.data, warnings: ctx.warnings }
  }
  const transformed = transformFn(ctx.data)
  return { key: ctx.key, data: transformed, warnings: ctx.warnings }
}