import { customDecoratorFactory } from '@mwcp/share'
import type { MethodTypeUnknown } from '@waiting/shared-types'

import { KEY_Trace } from '../config.js'
import type { TraceOptions } from '../trace.service/index.trace.service.js'

import { DecoratorHandlerTrace } from './trace.decorator-handler.js'


/**
 * Declarative Trace Decorator
 */
export function Trace<M extends MethodTypeUnknown | undefined = undefined>(options?: TraceOptions<M>): MethodDecorator {

  const opts = typeof options === 'string'
    ? { spanName: options }
    : options

  return customDecoratorFactory({
    decoratorArgs: opts,
    decoratorKey: KEY_Trace,
    enableClassDecorator: false,
    classIgnoreIfMethodDecoratorKeys: [],
    methodIgnoreIfMethodDecoratorKeys: [],
    decoratorHandlerClass: DecoratorHandlerTrace,
  })
}