ant-design-vue/components/_util/classNames.ts

import { isArray, isString, isObject } from './util';

export type ClassArray = ClassValue[];

export interface ClassDictionary {
  [id: string]: any;
}

export type ClassValue =
  | string
  | number
  | ClassDictionary
  | ClassArray
  | undefined
  | null
  | boolean;

function classNames(...args: ClassValue[]): string {
  const classes: string[] = [];
  for (let i = 0; i < args.length; i++) {
    const value = args[i];
    if (!value) {
      continue;
    }
    if (isString(value)) {
      classes.push(value);
    } else if (isArray(value)) {
      for (let i = 0; i < value.length; i++) {
        const inner = classNames(value[i]);
        if (inner) {
          classes.push(inner);
        }
      }
    } else if (isObject(value)) {
      for (const name in value) {
        if (value[name]) {
          classes.push(name);
        }
      }
    }
  }
  return classes.join(' ');
}

export default classNames;