dbgate/packages/datalib/src/PerspectiveDisplay.ts

import { getTableChildPerspectiveNodes, PerspectiveTableNode, PerspectiveTreeNode } from './PerspectiveTreeNode';
import _max from 'lodash/max';
import _range from 'lodash/max';
import _fill from 'lodash/fill';
import _findIndex from 'lodash/findIndex';

export class PerspectiveDisplayColumn {
  title: string;
  dataField: string;
  parentNodes: PerspectiveTreeNode[] = [];
  colSpanAtLevel = {};
  columnIndex = 0;
  dataNode: PerspectiveTreeNode = null;

  constructor(public display: PerspectiveDisplay) {}

  get rowSpan() {
    return this.display.columnLevelCount - this.parentNodes.length;
  }

  showParent(level: number) {
    return !!this.colSpanAtLevel[level];
  }

  getColSpan(level: number) {
    return this.colSpanAtLevel[level];
  }

  isVisible(level: number) {
    return level == this.columnLevel;
  }

  get columnLevel() {
    return this.parentNodes.length;
  }

  getParentName(level) {
    return this.parentNodes[level]?.title;
  }

  // hasParentNode(node: PerspectiveTreeNode) {
  //   return this.parentNodes.includes(node);
  // }
}

interface PerspectiveSubRowCollection {
  rows: CollectedPerspectiveDisplayRow[];
}

interface CollectedPerspectiveDisplayRow {
  // startIndex = 0;
  columnIndexes: number[];
  rowData: any[];
  // rowSpans: number[] = null;
  subRowCollections: PerspectiveSubRowCollection[];
}

export class PerspectiveDisplayRow {
  constructor(public display: PerspectiveDisplay) {
    this.rowData = _fill(Array(display.columns.length), undefined);
    this.rowSpans = _fill(Array(display.columns.length), 1);
  }

  getRow(rowIndex): PerspectiveDisplayRow {
    if (rowIndex == 0) return this;
    while (this.subrows.length < rowIndex) {
      this.subrows.push(new PerspectiveDisplayRow(this.display));
    }
    return this.subrows[rowIndex - 1];
  }

  subrows?: PerspectiveDisplayRow[] = [];

  rowData: any[] = [];
  rowSpans: number[] = null;
}

export class PerspectiveDisplay {
  columns: PerspectiveDisplayColumn[] = [];
  rows: PerspectiveDisplayRow[] = [];
  readonly columnLevelCount: number;

  constructor(public root: PerspectiveTreeNode, rows: any[]) {
    this.fillColumns(root.childNodes, []);
    this.columnLevelCount = _max(this.columns.map(x => x.parentNodes.length)) + 1;
    const collectedRows = this.collectRows(rows, root.childNodes);
    // console.log('COLLECTED', collectedRows);
    // this.mergeRows(collectedRows);
    this.mergeRows(collectedRows);
    // console.log('MERGED', this.rows);
  }

  fillColumns(children: PerspectiveTreeNode[], parentNodes: PerspectiveTreeNode[]) {
    for (const child of children) {
      if (child.isChecked) {
        this.processColumn(child, parentNodes);
      }
    }
  }

  processColumn(node: PerspectiveTreeNode, parentNodes: PerspectiveTreeNode[]) {
    if (node.isExpandable) {
      const countBefore = this.columns.length;
      this.fillColumns(node.childNodes, [...parentNodes, node]);

      if (this.columns.length > countBefore) {
        this.columns[countBefore].colSpanAtLevel[parentNodes.length] = this.columns.length - countBefore;
      }
    } else {
      const column = new PerspectiveDisplayColumn(this);
      column.title = node.columnTitle;
      column.dataField = node.dataField;
      column.parentNodes = parentNodes;
      column.display = this;
      column.columnIndex = this.columns.length;
      column.dataNode = node;
      this.columns.push(column);
    }
  }

  findColumnIndexFromNode(node: PerspectiveTreeNode) {
    return _findIndex(this.columns, x => x.dataNode.uniqueName == node.uniqueName);
  }

  collectRows(sourceRows: any[], nodes: PerspectiveTreeNode[]): CollectedPerspectiveDisplayRow[] {
    const columnNodes = nodes.filter(x => x.isChecked && !x.isExpandable);
    const treeNodes = nodes.filter(x => x.isChecked && x.isExpandable);

    const columnIndexes = columnNodes.map(node => this.findColumnIndexFromNode(node));

    // const nodeStartIndexes = new WeakMap();
    // for (const node of treeNodes) {
    //   const column = this.columns.find(x => x.hasParentNode(node));
    //   if (column) nodeStartIndexes.set(node, column.columnIndex);
    // }

    const res: CollectedPerspectiveDisplayRow[] = [];
    for (const sourceRow of sourceRows) {
      // console.log('PROCESS SOURCE', sourceRow);
      // row.startIndex = startIndex;
      const rowData = columnNodes.map(node => sourceRow[node.codeName]);
      const subRowCollections = [];

      for (const node of treeNodes) {
        // if (sourceRow.AlbumId == 1) {
        //   if (node.fieldName == 'ArtistIdRef') {
        //     console.log('XXX', sourceRow['ArtistIdRef']);
        //     console.log(require('lodash').keys(sourceRow))
        //     console.dir(sourceRow);
        //   }
        //   console.log(node.fieldName, sourceRow[node.fieldName], sourceRow);
        // }
        // console.log('sourceRow[node.fieldName]', sourceRow[node.fieldName]);
        if (sourceRow[node.fieldName]) {
          const subrows = {
            rows: this.collectRows(sourceRow[node.fieldName], node.childNodes),
          };
          subRowCollections.push(subrows);
        }
      }

      res.push({
        rowData,
        columnIndexes,
        subRowCollections,
      });
    }

    return res;
  }

  flushFlatRows(row: PerspectiveDisplayRow) {
    this.rows.push(row);
    for (const child of row.subrows) {
      this.flushFlatRows(child);
    }
  }

  fillRowSpans() {
    const lastFilledColumns = _fill(Array(this.columns.length), 0);
    let rowIndex = 0;
    for (const row of this.rows) {
      for (let i = 0; i < this.columns.length; i++) {
        if (row.rowData[i] !== undefined) {
          if (rowIndex - lastFilledColumns[i] > 1) {
            this.rows[lastFilledColumns[i]].rowSpans[i] = rowIndex - lastFilledColumns[i];
          }
          lastFilledColumns[i] = rowIndex;
        }
      }
      rowIndex++;
    }
  }

  mergeRows(collectedRows: CollectedPerspectiveDisplayRow[]) {
    const rows = [];
    for (const collectedRow of collectedRows) {
      const resultRow = new PerspectiveDisplayRow(this);
      this.mergeRow(collectedRow, resultRow);
      rows.push(resultRow);
    }
    for (const row of rows) {
      this.flushFlatRows(row);
    }
    for (const row of this.rows) {
      delete row.subrows;
    }
    this.fillRowSpans();
  }

  mergeRow(collectedRow: CollectedPerspectiveDisplayRow, resultRow: PerspectiveDisplayRow) {
    for (let i = 0; i < collectedRow.columnIndexes.length; i++) {
      resultRow.rowData[collectedRow.columnIndexes[i]] = collectedRow.rowData[i];
    }

    for (const subrows of collectedRow.subRowCollections) {
      let rowIndex = 0;
      for (const subrow of subrows.rows) {
        const targetRow = resultRow.getRow(rowIndex);
        this.mergeRow(subrow, targetRow);
        rowIndex++;
      }
    }
  }

  // rowToFlatRows(sourceRow: CollectedPerspectiveDisplayRow) {
  //   const res = [];

  //   const row = new PerspectiveDisplayRow(this);
  //   row.rowData = _fill(Array(this.columns.length), undefined);
  //   row.rowSpans = _fill(Array(this.columns.length), 1);
  //   res.push(row)

  //   for (let i = 0; i < sourceRow.columnIndexes.length; i++) {
  //     row.rowData[sourceRow.columnIndexes[i]] = sourceRow.rowData[i];
  //   }

  //   for(const subrows of sourceRow.subRowCollections) {
  //     let rowIndex=0;
  //     for(const subrow of subrows.rows) {
  //       if ()
  //       rowIndex++;

  //     }
  //   }

  //   return res;

  //   // while (true) {
  //   //   for (let colIndex = 0; colIndex < this.columns.length; colIndex++) {
  //   //     if (colIndex < sourceRow.startIndex) {
  //   //       continue;
  //   //     }
  //   //     if (colIndex < sourceRow.startIndex + sourceRow.rowData.length) {
  //   //       if (rowIndex == 0) {
  //   //         row.rowData[colIndex] = sourceRow.rowData[sourceRow.startIndex + colIndex];
  //   //         row.rowSpans[colIndex] = 1;
  //   //       } else {
  //   //         row.rowSpans[colIndex] += 1;
  //   //       }
  //   //     }
  //   //     const subrows = sourceRow.subRowCollections.find(x=>x.);
  //   //   }
  //   // }
  // }
}