handleTables.js

const {
  CELL, TABLE, ROW, LINK, TEXT,
} = require('../../enums');
const globalData = require('../data');

const { tableHead } = globalData;
const { iterateAndConcatValue, handleLinkReference } = require('../utils');

/**
 *
 * @param {Array} headNodes
 * @return {handleTables}
 */
const createHead = (headNodes) => {
  headNodes.forEach(({ type, children }) => {
    if (type !== CELL) return null;
    tableHead.push(iterateAndConcatValue(children));
    return tableHead;
  });
  return tableHead;
};

const handleCellLinkReference = (cell, idx) => {
  const relatedHead = tableHead[idx];
  return {
    [relatedHead]: handleLinkReference(cell, globalData.linkDefinitions),
  };
};

const handleCellTextHTML = (cell, idx) => {
  const relatedHead = tableHead[idx];
  return {
    [relatedHead]: cell.value,
  };
};

const handleCell = (cells, idx) => {
  // FIXME: handle <br/> separated multiple champions
  let result;
  if (cells.length) {
    cells.forEach((cell) => {
      switch (cell.type) {
        case LINK:
          result = handleCellLinkReference(cell, idx);
          return result;
        case TEXT:
          result = handleCellTextHTML(cell, idx);
          return result;
        default:
          return {};
      }
    });
  }
  return result;
};

const handleRows = (row) => {
  let obj = {};
  row.forEach(({ type, children }, idx) => {
    if (type !== CELL) return null;
    const rowLine = handleCell(children, idx);
    obj = { ...obj, ...rowLine };
  });
  return obj;
};

/**
 * @param {Object} table
 * @returns {Array}
 */
const handleTables = ({ align: { length }, type, children }) => {
  const arr = [];
  if (type !== TABLE) return null;
  children.forEach(({ children: tableRow, type: rowType }, idx) => {
    if (rowType !== ROW) return [];
    if (idx === 0) {
      createHead(tableRow);
      return [];
    }
    const row = handleRows(tableRow, length);
    arr.push(row);
    return arr;
  });
  return arr;
};

module.exports = handleTables;