zent/src/cascader/public-options-fns.ts

一套前端设计语言和基于React的实现

/**
 * Public helpers for manipulating cascader options
 */

import { clone as _clone, insertPath as _insertPath } from './forest';
import { IPublicCascaderItem } from './types';

interface IMergeFrame {
  x?: IPublicCascaderItem[];
  y?: IPublicCascaderItem[];
  xy: IPublicCascaderItem[];
}

/**
 * Shallow clone `options`, does not deep clone properties in each node.
 * @param options options to clone
 */
export function clone(options: IPublicCascaderItem[]): IPublicCascaderItem[] {
  return _clone(options, createNode);
}

/**
 * Insert `path` into `options`. __Mutates__ `options`.
 * @param options options for inserting
 * @param path path to insert
 */
export function insertPath(
  options: IPublicCascaderItem[],
  path: IPublicCascaderItem[]
) {
  return _insertPath(options, path, createNode);
}

/**
 * Return the target `node` following `path` in `options`.
 * @param options options to search for node
 * @param path path to target node
 */
export function getNode(
  options: IPublicCascaderItem[],
  path: IPublicCascaderItem[]
): IPublicCascaderItem | null {
  let node: IPublicCascaderItem = null;
  let children = options;

  for (let i = 0; i < path.length; i++) {
    const { value } = path[i];
    node = children.find(x => x.value === value);
    if (!node) {
      return null;
    }
    children = node.children;
  }

  return node;
}

/**
 * Merge `options` and `anotherOptions` into a new options array.
 * __Mutates__ `options`.
 * @param options options to merge into
 * @param anotherOptions options to merge from
 */
export function merge(
  options: IPublicCascaderItem[],
  anotherOptions: IPublicCascaderItem[]
): IPublicCascaderItem[] {
  const merged: IPublicCascaderItem[] = [];
  const stack: IMergeFrame[] = [
    {
      x: options,
      y: anotherOptions,
      xy: merged,
    },
  ];

  while (stack.length > 0) {
    const frame = stack.pop();
    const { x = [], y = [], xy } = frame;

    for (const i of x) {
      xy.push(i);
    }

    for (const i of y) {
      const { value } = i;
      const j = x.find(node => node.value === value);
      if (!j) {
        xy.push(i);
      } else {
        const x = j.children;
        j.children = [];
        stack.push({
          x,
          y: i.children,
          xy: j.children,
        });
      }
    }
  }

  return merged;
}

function createNode(node: IPublicCascaderItem) {
  return { ...node, children: [] };
}