import type { NameTransform } from 'style-dictionary/types'

type Transformer = NameTransform['transform']

export const prefixTransform: (transformer: Transformer) => Transformer =
  transformer =>
  (...args) => {
    const [, config] = args
    const prefix = config.prefix ? `${config.prefix}-` : ''
    const value = transformer(...args)

    if (value instanceof Promise) {
      return value.then(val => `${prefix}${val}`)
    }
    return `${prefix}${value}`
  }