specpress/lib/md2docx/handlers/jsonTableHandler.js

Export PDF and/or DOCX files from a subset of Markdown, ASN.1 and JSON files

const { readFileSync } = require('fs')
const { resolve } = require('path')
const { Table, TableRow, TableCell, Paragraph, WidthType, BorderStyle } = require('docx')
const { buildSpanMap, normalizeJsonTable } = require('../../common/buildSpanMap')

const cellBorders = {
  top: { style: BorderStyle.SINGLE, size: 1 },
  bottom: { style: BorderStyle.SINGLE, size: 1 },
  left: { style: BorderStyle.SINGLE, size: 1 },
  right: { style: BorderStyle.SINGLE, size: 1 }
}


function alignStyle(col, isHeader) {
  if (isHeader) return 'TAH'
  const align = col && col.align
  if (align === 'center') return 'TAC'
  if (align === 'right') return 'TAR'
  return 'TAL'
}

/**
 * Converts parsed JSON table data to a docx Table.
 *
 * @param {object} data - JSON object with `columns` and `rows` arrays.
 * @returns {Promise<Table>} docx Table element.
 */
async function jsonToDocxTable(data) {
  data = normalizeJsonTable(data)
  if (!data || !data.rows || data.rows.length === 0) return null
  const { buildInlineRuns } = require('../md2docx')

  const columns = data.columns || []
  const rowCount = data.rows.length
  const colCount = Math.max(columns.length, ...data.rows.map(r => r.length))
  const spanMap = buildSpanMap(data.rows, rowCount, colCount)
  const tableRows = []

  const MarkdownIt = require('markdown-it')
  const cellMd = new MarkdownIt({ html: true })

  async function cellRuns(text, isNote) {
    const tokens = cellMd.parseInline(text, {})
    const children = tokens.length > 0 ? tokens[0].children : []
    return buildInlineRuns(children || [], isNote ? { replaceFirstColonSpace: true } : {})
  }

  const hasHeaders = columns.some(col => col.name)
  if (hasHeaders) {
    const cells = []
    for (let c = 0; c < colCount; c++) {
      const runs = await cellRuns(String(columns[c]?.name ?? ''))
      cells.push(new TableCell({
        children: [new Paragraph({ children: runs, style: 'TAH' })],
        borders: cellBorders
      }))
    }
    tableRows.push(new TableRow({ children: cells }))
  }

  const noteRe = /^(\*\*)?NOTE(\*\*)?( \d+)?:/

  for (let r = 0; r < rowCount; r++) {
    const cells = []
    for (let c = 0; c < colCount; c++) {
      const span = spanMap[r][c]
      if (span.skip) continue
      const text = span.value === null || span.value === undefined ? '' : String(span.value)
      const baseStyle = alignStyle(columns[c], false)
      const lines = text.split('\n')
      const paragraphs = []
      for (const line of lines) {
        const isNote = noteRe.test(line.trim())
        const runs = await cellRuns(line, isNote)
        paragraphs.push(new Paragraph({ children: runs, style: isNote ? 'TAN' : baseStyle }))
      }
      const cellOpts = { children: paragraphs, borders: cellBorders }
      if (span.colspan > 1) cellOpts.columnSpan = span.colspan
      if (span.rowspan > 1) cellOpts.rowSpan = span.rowspan
      cells.push(new TableCell(cellOpts))
    }
    tableRows.push(new TableRow({ children: cells }))
  }

  return new Table({ rows: tableRows, width: { size: 100, type: WidthType.PERCENTAGE } })
}

/**
 * Reads a JSON file and converts it to a docx Table.
 *
 * @param {string} filePath - Absolute path to the JSON file.
 * @returns {Promise<Table|null>} docx Table element, or null if data is empty.
 */
async function jsonFileToDocxTable(filePath) {
  const content = readFileSync(filePath, 'utf8')
  const data = JSON.parse(content)
  if (Array.isArray(data)) {
    const tables = []
    for (const item of data) {
      const t = await jsonToDocxTable(item)
      if (t) tables.push(t)
    }
    return tables
  }
  return jsonToDocxTable(data)
}

module.exports = { jsonToDocxTable, jsonFileToDocxTable }