import { test } from '../extends'

export function step(target: Function, context: ClassMethodDecoratorContext) {
  return function replacementMethod(this: Function, ...args: unknown[]) {
    const className = this.constructor.name
    const name = `${className.replace(/PageObject$/, '')}.${context.name as string}`
    return test.step(name, async () => {
      // eslint-disable-next-line ts/no-unsafe-return
      return await target.call(this, ...args)
    })
  }
}

export function boxedStep(target: Function, context: ClassMethodDecoratorContext) {
  return function replacementMethod(this: Function, ...args: unknown[]) {
    const className = this.constructor.name
    const name = `${className.replace(/PageObject$/, '')}.${context.name as string}`
    return test.step(name, async () => {
      // eslint-disable-next-line ts/no-unsafe-return
      return await target.call(this, ...args)
    }, { box: true })
  }
}