decorators.ts

import {debounce as db, throttle as th, ThrottleOptions} from './index'

export function throttle(wait = 0, opts: ThrottleOptions = {}) {
  return (proto: unknown, name: string, descriptor: PropertyDescriptor) => {
    if (!descriptor || typeof descriptor.value !== 'function') {
      throw new Error('debounce can only decorate functions')
    }
    const fn = descriptor.value
    descriptor.value = th(fn, wait, opts)
    Object.defineProperty(proto, name, descriptor)
  }
}

export function debounce(wait = 0, opts: ThrottleOptions = {}) {
  return (proto: unknown, name: string, descriptor: PropertyDescriptor) => {
    if (!descriptor || typeof descriptor.value !== 'function') {
      throw new Error('debounce can only decorate functions')
    }
    const fn = descriptor.value
    descriptor.value = db(fn, wait, opts)
    Object.defineProperty(proto, name, descriptor)
  }
}