pdfmake/js/helpers/node.js

Client/server side PDF printing in pure JavaScript

"use strict";

exports.__esModule = true;
exports.getNodeId = getNodeId;
exports.getNodeMargin = getNodeMargin;
exports.stringifyNode = stringifyNode;
var _variableType = require("./variableType");
function fontStringify(key, val) {
  if (key === 'font') {
    return 'font';
  }
  return val;
}

/**
 * Convert node to readable string
 *
 * @param {object} node
 * @returns {string}
 */
function stringifyNode(node) {
  return JSON.stringify(node, fontStringify);
}

/**
 * @param {object} node
 * @returns {?string}
 */
function getNodeId(node) {
  if (node.id) {
    return node.id;
  }
  if (Array.isArray(node.text)) {
    for (let n of node.text) {
      let nodeId = getNodeId(n);
      if (nodeId) {
        return nodeId;
      }
    }
  }
  return null;
}

/**
 * @param {object} node
 * @param {object} styleStack object is instance of PDFDocument
 * @returns {?Array}
 */
function getNodeMargin(node, styleStack) {
  function processSingleMargins(node, currentMargin, defaultMargin = 0) {
    if (node.marginLeft !== undefined || node.marginTop !== undefined || node.marginRight !== undefined || node.marginBottom !== undefined) {
      return [node.marginLeft ?? currentMargin[0] ?? defaultMargin, node.marginTop ?? currentMargin[1] ?? defaultMargin, node.marginRight ?? currentMargin[2] ?? defaultMargin, node.marginBottom ?? currentMargin[3] ?? defaultMargin];
    }
    return currentMargin;
  }
  function flattenStyleArray(styleArray, styleStack, visited = new Set()) {
    styleArray = Array.isArray(styleArray) ? styleArray : [styleArray];

    // style is not valid array of strings
    if (!styleArray.every(item => (0, _variableType.isString)(item))) {
      return {};
    }
    let flattenedStyles = {};
    for (let index = 0; index < styleArray.length; index++) {
      let styleName = styleArray[index];
      let style = styleStack.styleDictionary[styleName];

      // style not found
      if (style === undefined) {
        continue;
      }
      if (visited.has(styleName)) {
        continue;
      }
      visited.add(styleName);
      if (style.extends !== undefined) {
        flattenedStyles = {
          ...flattenStyleArray(style.extends, styleStack, visited),
          ...flattenedStyles
        };
      }
      if (style.margin !== undefined) {
        flattenedStyles = {
          margin: convertMargin(style.margin)
        };
        continue;
      }
      flattenedStyles = {
        margin: processSingleMargins(style, flattenedStyles.margin ?? {}, undefined)
      };
    }
    return flattenedStyles;
  }
  function convertMargin(margin) {
    if ((0, _variableType.isNumber)(margin)) {
      margin = [margin, margin, margin, margin];
    } else if (Array.isArray(margin)) {
      if (margin.length === 2) {
        margin = [margin[0], margin[1], margin[0], margin[1]];
      }
    }
    return margin;
  }
  let margin = [undefined, undefined, undefined, undefined];
  if (node.style) {
    let styleArray = Array.isArray(node.style) ? node.style : [node.style];
    let flattenedStyleArray = flattenStyleArray(styleArray, styleStack);
    if (flattenedStyleArray) {
      margin = processSingleMargins(flattenedStyleArray, margin);
    }
    if (flattenedStyleArray.margin) {
      margin = convertMargin(flattenedStyleArray.margin);
    }
  }
  margin = processSingleMargins(node, margin);
  if (node.margin !== undefined) {
    margin = convertMargin(node.margin);
  }
  if (margin[0] === undefined && margin[1] === undefined && margin[2] === undefined && margin[3] === undefined) {
    return null;
  }
  return margin;
}