compassql/src/result.ts

CompassQL visualization query language

import {GroupBy} from './query/groupby';

/**
 * An ordered tree structure for storing query results.
 */
export interface ResultTree<T> {
  name: string;
  path: string;
  items: (ResultTree<T> | T)[];
  groupBy?: GroupBy;
  orderGroupBy?: string | string[];
}

export function isResultTree<T>(item: ResultTree<T> | T): item is ResultTree<T> {
  return (<ResultTree<T>>item).items !== undefined;
}

export function getTopResultTreeItem<T>(specQuery: ResultTree<T>): T {
  let topItem = specQuery.items[0];
  while (topItem && isResultTree(topItem)) {
    topItem = topItem.items[0];
  }
  return <T>topItem;
}

export function mapLeaves<T, U>(group: ResultTree<T>, f: (item: T) => U): ResultTree<U> {
  return {
    ...group,
    items: group.items.map(item => (isResultTree(item) ? mapLeaves(item, f) : f(item)))
  };
}