ant-design/site/entry/utils.js

/* eslint-disable react/prefer-stateless-function, react/no-multi-comp */
import React from 'react';
import { IndexRedirect } from 'react-router';
import MainContent from '../component/MainContent';
import Article from '../component/Article';
import ComponentDoc from '../component/ComponentDoc';
import demosList from '../../_data/demos-list';
import { redirects } from '../website.config';

if (module.hot) {
  module.hot.accept('../../_data/demos-list', () => {});
}

function fileNameToPath(fileName) {
  const snippets = fileName
    .replace(/(\/index)?((\.zh-CN)|(\.en-US))?\.md$/i, '').split('/');
  return snippets[snippets.length - 1];
}

function getMenuItems(data, locale) {
  const menuMeta = Object.keys(data)
          .map((key) => data[key])
          .map((file) => {
            if (file.localized) {
              return file[locale].meta;
            }
            return file.meta;
          });

  const menuItems = {};
  menuMeta.sort((a, b) => {
    return parseInt(a.order, 10) - parseInt(b.order, 10);
  }).forEach((meta) => {
    const category = meta.category || 'topLevel';
    if (!menuItems[category]) {
      menuItems[category] = {};
    }

    const type = meta.type || 'topLevel';
    if (!menuItems[category][type]) {
      menuItems[category][type] = [];
    }

    menuItems[category][type].push(meta);
  });

  return menuItems;
}

export function generateContainer(data) {
  return class containerWrapper extends React.Component {
    static contextTypes = {
      intl: React.PropTypes.object,
    }

    render() {
      const locale = this.context.intl.locale;
      const menuItems = getMenuItems(data, locale);
      return (
        <MainContent {...this.props} menuItems={menuItems} />
      );
    }
  };
}

export function generateIndex(data) {
  const menuItems = getMenuItems(data, 'zh-CN'); // 以中文版配置为准
  const firstChild = menuItems.topLevel.topLevel.filter((item) => {
    return !item.disabled;
  })[0];
  return (
    <IndexRedirect key="index"
      to={fileNameToPath(firstChild.fileName)} />
  );
}

const pathToFile = {};
Object.keys(redirects).forEach((key) => pathToFile[redirects[key]] = key);
pathToFile['docs/react/changelog'] = './CHANGELOG'; // TODO

function getDoc(data, props) {
  const trimedPathname = props.location.pathname.replace(/^\//, '');
  const processedPathname = pathToFile[trimedPathname] || trimedPathname;
  const doc = data[`${processedPathname}.md`] ||
          data[`${processedPathname}/index.md`];
  return doc;
}

export function getChildrenWrapper(data) {
  return class childrenWrapper extends React.Component {
    static contextTypes = {
      intl: React.PropTypes.object,
    }

    render() {
      const props = this.props;
      const trimedPathname = props.location.pathname.replace(/^\//, '');
      const processedPathname = pathToFile[trimedPathname] || trimedPathname;
      const rawDoc = data[`${processedPathname}.md`] ||
              data[`${processedPathname}/index.md`];

      const locale = this.context.intl.locale;
      const doc = rawDoc.localized ? rawDoc[locale] : rawDoc;

      const hasDemos = demosList[doc.meta.fileName.replace(`.${locale}`, '')];
      return !hasDemos ?
        <Article {...props} content={doc} /> :
        <ComponentDoc {...props} doc={doc} />;
    }
  };
}

export function getEnterHandler(data) {
  return function handleEnter(nextState, replace) {
    const doc = getDoc(data, nextState);
    if (!doc) {
      replace('/404');
    }
  };
}