All files / roosterjs-content-model-dom/lib/modelToDom/handlers handleTable.ts

100% Statements 82/82
98.77% Branches 80/81
100% Functions 2/2
100% Lines 78/78

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 1651x 1x 1x 1x 1x 1x                     1x             112x   2x     110x   110x 9x   9x   101x   101x 78x     101x 101x   101x 101x 101x 101x     110x   110x 110x   110x 208x   208x   2x     206x 206x 206x   206x 187x 161x     187x     206x   463x 463x   463x 12x                 12x 12x 12x     12x     463x 454x   454x   454x   454x 454x 454x 454x   454x 4x   454x 4x     454x 4x     454x 4x     454x 408x 330x     408x 324x       454x 454x 407x 367x     407x 407x 407x           407x     454x     454x         110x   110x    
import { applyFormat } from '../utils/applyFormat';
import { hasMetadata } from '../../modelApi/metadata/updateMetadata';
import { isBlockEmpty } from '../../modelApi/common/isEmpty';
import { moveChildNodes } from '../../domUtils/moveChildNodes';
import { reuseCachedElement } from '../../domUtils/reuseCachedElement';
import { stackFormat } from '../utils/stackFormat';
import type {
    ContentModelBlockHandler,
    ContentModelTable,
    ModelToDomContext,
    TableSelection,
} from 'roosterjs-content-model-types';
 
/**
 * @internal
 */
export const handleTable: ContentModelBlockHandler<ContentModelTable> = (
    doc: Document,
    parent: Node,
    table: ContentModelTable,
    context: ModelToDomContext,
    refNode: Node | null
) => {
    if (isBlockEmpty(table)) {
        // Empty table, do not create TABLE element and just return
        return refNode;
    }
 
    let tableNode = context.allowCacheElement ? table.cachedElement : undefined;
 
    if (tableNode) {
        refNode = reuseCachedElement(parent, tableNode, refNode, context.rewriteFromModel);
 
        moveChildNodes(tableNode);
    } else {
        tableNode = doc.createElement('table');
 
        if (context.allowCacheElement) {
            table.cachedElement = tableNode;
        }
 
        parent.insertBefore(tableNode, refNode);
        context.rewriteFromModel.addedBlockElements.push(tableNode);
 
        applyFormat(tableNode, context.formatAppliers.block, table.format, context);
        applyFormat(tableNode, context.formatAppliers.table, table.format, context);
        applyFormat(tableNode, context.formatAppliers.tableBorder, table.format, context);
        applyFormat(tableNode, context.formatAppliers.dataset, table.dataset, context);
    }
 
    context.onNodeCreated?.(table, tableNode);
 
    const tbody = doc.createElement('tbody');
    tableNode.appendChild(tbody);
 
    for (let row = 0; row < table.rows.length; row++) {
        const tableRow = table.rows[row];
 
        if (tableRow.cells.length == 0) {
            // Skip empty row
            continue;
        }
 
        const tr = (context.allowCacheElement && tableRow.cachedElement) || doc.createElement('tr');
        tbody.appendChild(tr);
        moveChildNodes(tr);
 
        if (!tableRow.cachedElement) {
            if (context.allowCacheElement) {
                tableRow.cachedElement = tr;
            }
 
            applyFormat(tr, context.formatAppliers.tableRow, tableRow.format, context);
        }
 
        context.onNodeCreated?.(tableRow, tr);
 
        for (let col = 0; col < tableRow.cells.length; col++) {
            const cell = tableRow.cells[col];
 
            if (cell.isSelected) {
                const tableSelection: TableSelection = context.tableSelection || {
                    type: 'table',
                    table: tableNode,
                    firstColumn: col,
                    lastColumn: col,
                    firstRow: row,
                    lastRow: row,
                };
 
                Eif (tableSelection.table == tableNode) {
                    tableSelection.lastColumn = Math.max(tableSelection.lastColumn, col);
                    tableSelection.lastRow = Math.max(tableSelection.lastRow, row);
                }
 
                context.tableSelection = tableSelection;
            }
 
            if (!cell.spanAbove && !cell.spanLeft) {
                const tag = cell.isHeader ? 'th' : 'td';
                const td =
                    (context.allowCacheElement && cell.cachedElement) || doc.createElement(tag);
 
                tr.appendChild(td);
 
                let rowSpan = 1;
                let colSpan = 1;
                let width = table.widths[col];
                let height = tableRow.height;
 
                for (; table.rows[row + rowSpan]?.cells[col]?.spanAbove; rowSpan++) {
                    height += table.rows[row + rowSpan].height;
                }
                for (; tableRow.cells[col + colSpan]?.spanLeft; colSpan++) {
                    width += table.widths[col + colSpan];
                }
 
                if (rowSpan > 1) {
                    td.rowSpan = rowSpan;
                }
 
                if (colSpan > 1) {
                    td.colSpan = colSpan;
                }
 
                if (!cell.cachedElement || (cell.format.useBorderBox && hasMetadata(table))) {
                    if (width > 0 && !td.style.width) {
                        td.style.width = width + 'px';
                    }
 
                    if (height > 0 && !td.style.height) {
                        td.style.height = height + 'px';
                    }
                }
 
                stackFormat(context, tag, () => {
                    if (!cell.cachedElement) {
                        if (context.allowCacheElement) {
                            cell.cachedElement = td;
                        }
 
                        applyFormat(td, context.formatAppliers.block, cell.format, context);
                        applyFormat(td, context.formatAppliers.tableCell, cell.format, context);
                        applyFormat(
                            td,
                            context.formatAppliers.tableCellBorder,
                            cell.format,
                            context
                        );
                        applyFormat(td, context.formatAppliers.dataset, cell.dataset, context);
                    }
 
                    context.modelHandlers.blockGroupChildren(doc, td, cell, context);
                });
 
                context.onNodeCreated?.(cell, td);
            }
        }
    }
 
    context.domIndexer?.onTable(tableNode, table);
 
    return refNode;
};