Function classNames

packages/omi/src/class.ts:16–42

(...args: ArgumentArray)

type Argument = Value | Mapping | ArgumentArray | ReadonlyArgumentArray

export function classNames(...args: ArgumentArray): string {
  const classes: (string | number)[] = []

  for (let i = 0; i < args.length; i++) {
    const arg: Argument = args[i]
    if (!arg) continue

    const argType = typeof arg

    if (argType === 'string' || argType === 'number') {
      classes.push(arg as string | number)
    } else if (Array.isArray(arg) && arg.length) {
      const inner = classNames(...arg)
      if (inner) {
        classes.push(inner)
      }
    } else if (argType === 'object') {
      for (const key in arg as Mapping) {
        if (hasOwn.call(arg, key) && (arg as Mapping)[key]) {
          classes.push(key)
        }
      }
    }
  }

  return classes.join(' ')
}

type PropsMapping = {
  class?: Argument