ant-design/.dumi/theme/builtins/ComponentTokenTable/index.tsx

/* eslint import/no-unresolved: 0 */
import { ConfigProvider, Table } from 'antd';
import { getDesignToken } from 'antd-token-previewer';
import tokenMeta from 'antd/es/version/token-meta.json';
import tokenData from 'antd/es/version/token.json';
import React from 'react';
import useLocale from '../../../hooks/useLocale';
import useSiteToken from '../../../hooks/useSiteToken';
import { useColumns } from '../TokenTable';

const defaultToken = getDesignToken();

const locales = {
  cn: {
    token: 'Token 名称',
    description: '描述',
    type: '类型',
    value: '默认值',
  },
  en: {
    token: 'Token Name',
    description: 'Description',
    type: 'Type',
    value: 'Default Value',
  },
};

interface SubTokenTableProps {
  defaultOpen?: boolean;
  title: string;
  tokens: string[];
}

function SubTokenTable({ defaultOpen, tokens, title }: SubTokenTableProps) {
  const [, lang] = useLocale(locales);
  const { token } = useSiteToken();
  const columns = useColumns();

  if (!tokens.length) {
    return null;
  }

  const data = tokens
    .sort((token1, token2) => {
      const hasColor1 = token1.toLowerCase().includes('color');
      const hasColor2 = token2.toLowerCase().includes('color');

      if (hasColor1 && !hasColor2) {
        return -1;
      }

      if (!hasColor1 && hasColor2) {
        return 1;
      }

      return token1 < token2 ? -1 : 1;
    })
    .map((name) => {
      const meta = tokenMeta[name];

      if (!meta) {
        return null;
      }

      return {
        name,
        desc: lang === 'cn' ? meta.desc : meta.descEn,
        type: meta.type,
        value: (defaultToken as any)[name],
      };
    })
    .filter((info) => info);

  return (
    // Reuse `.markdown` style
    <details className="markdown" open={defaultOpen || process.env.NODE_ENV !== 'production'}>
      <summary>
        <h3 style={{ display: 'inline' }}>{title}</h3>
      </summary>
      <ConfigProvider
        theme={{
          token: {
            borderRadius: 0,
          },
        }}
      >
        <Table
          size="middle"
          columns={columns}
          bordered
          dataSource={data}
          style={{ marginBottom: token.margin }}
          pagination={false}
        />
      </ConfigProvider>
    </details>
  );
}

export interface ComponentTokenTableProps {
  component: string;
}

function ComponentTokenTable({ component }: ComponentTokenTableProps) {
  const [mergedGlobalTokens] = React.useMemo(() => {
    const globalTokenSet = new Set<string>();
    let componentTokens: Record<string, string> = {};

    component.split(',').forEach((comp) => {
      const { global: globalTokens = [], component: singleComponentTokens = [] } =
        tokenData[comp] || {};

      globalTokens.forEach((token: string) => {
        globalTokenSet.add(token);
      });

      componentTokens = {
        ...componentTokens,
        ...singleComponentTokens,
      };
    });

    return [Array.from(globalTokenSet), componentTokens];
  }, [component]);

  return (
    <>
      {/* Component Token 先不展示 */}
      {/* <SubTokenTable title="Component Token" tokens={mergedComponentTokens} defaultOpen /> */}
      <SubTokenTable title="Global Token" tokens={mergedGlobalTokens} />
    </>
  );
}

export default React.memo(ComponentTokenTable);