@antv/s2
Version:
effective spreadsheet render core lib
152 lines • 6.87 kB
JavaScript
import { escape, map, max } from 'lodash';
import { LINE_SEPARATOR, ROOT_NODE_ID, TAB_SEPARATOR } from '../../../common';
import { CopyMIMEType, } from '../../../common/interface/export';
// 把 string[][] 矩阵转换成 CopyablePlain
export const matrixPlainTextTransformer = (dataMatrix, separator = TAB_SEPARATOR) => {
return {
type: CopyMIMEType.PLAIN,
content: map(dataMatrix, (line) => line.join(separator)).join(LINE_SEPARATOR),
};
};
// 把 string[][] 矩阵转换成 CopyableItem
export const matrixHtmlTransformer = (dataMatrix) => {
function createTableData(data, tagName) {
return data
.map((cell) => `<${tagName}>${escape(cell)}</${tagName}>`)
.join('');
}
function createBody(data, tagName) {
return data
.map((row) => `<${tagName}>${createTableData(row, 'td')}</${tagName}>`)
.join('');
}
const body = createBody(dataMatrix, 'tr');
return {
type: CopyMIMEType.HTML,
content: `<meta charset="utf-8"><table><tbody>${body}</tbody></table>`,
};
};
export const Transformers = {
[CopyMIMEType.PLAIN]: matrixPlainTextTransformer,
[CopyMIMEType.HTML]: matrixHtmlTransformer,
};
// 生成矩阵：https://gw.alipayobjects.com/zos/antfincdn/bxBVt0nXx/a182c1d4-81bf-469f-b868-8b2e29acfc5f.png
export const assembleMatrix = ({ rowMatrix, colMatrix, dataMatrix, cornerMatrix, }) => {
var _a, _b, _c, _d, _e, _f;
const rowWidth = (_b = (_a = rowMatrix === null || rowMatrix === void 0 ? void 0 : rowMatrix[0]) === null || _a === void 0 ? void 0 : _a.length) !== null && _b !== void 0 ? _b : 0;
const colHeight = (_c = colMatrix === null || colMatrix === void 0 ? void 0 : colMatrix.length) !== null && _c !== void 0 ? _c : 0;
const dataWidth = (_e = (_d = dataMatrix[0]) === null || _d === void 0 ? void 0 : _d.length) !== null && _e !== void 0 ? _e : 0;
const dataHeight = (_f = dataMatrix.length) !== null && _f !== void 0 ? _f : 0;
const matrixWidth = rowWidth + dataWidth;
const matrixHeight = colHeight + dataHeight;
let matrix = Array.from(Array(matrixHeight), () => new Array(matrixWidth));
matrix = map(matrix, (heightArr, y) => map(heightArr, (_, x) => {
var _a, _b;
if (x >= 0 && x < rowWidth && y >= 0 && y < colHeight) {
return (_b = (_a = cornerMatrix === null || cornerMatrix === void 0 ? void 0 : cornerMatrix[y]) === null || _a === void 0 ? void 0 : _a[x]) !== null && _b !== void 0 ? _b : '';
}
if (x >= rowWidth && x <= matrixWidth && y >= 0 && y < colHeight) {
return colMatrix[y][x - rowWidth];
}
if (x >= 0 && x < rowWidth && y >= colHeight && y < matrixHeight) {
return rowMatrix === null || rowMatrix === void 0 ? void 0 : rowMatrix[y - colHeight][x];
}
if (x >= rowWidth &&
x <= matrixWidth &&
y >= colHeight &&
y < matrixHeight) {
return dataMatrix[y - colHeight][x - rowWidth];
}
return undefined;
}));
return matrix;
};
export function getMaxRowLen(matrix) {
var _a;
return (_a = max(map(matrix, (row) => row.length))) !== null && _a !== void 0 ? _a : 0;
}
/**
* 补全 matrix 中的元素个数, 使得每一行的元素个数一致，以最大的行元素个数为准
* @param {SimpleData[][]} matrix
* @return {SimpleData[][]}
*/
export function completeMatrix(matrix) {
const maxRowLen = getMaxRowLen(matrix);
return map(matrix, (row) => {
const diff = maxRowLen - row.length;
return diff ? [...row, ...new Array(diff).fill('')] : row;
});
}
export const getFormatOptions = (formatOptions) => {
var _a, _b;
if (typeof formatOptions === 'object') {
return {
formatHeader: (_a = formatOptions.formatHeader) !== null && _a !== void 0 ? _a : false,
formatData: (_b = formatOptions.formatData) !== null && _b !== void 0 ? _b : false,
};
}
return {
formatHeader: formatOptions !== null && formatOptions !== void 0 ? formatOptions : false,
formatData: formatOptions !== null && formatOptions !== void 0 ? formatOptions : false,
};
};
function getTransformer(customTransformer) {
if (!customTransformer) {
return Transformers;
}
const customTransformersTemp = customTransformer(Transformers);
return {
[CopyMIMEType.PLAIN]: customTransformersTemp[CopyMIMEType.PLAIN] ||
Transformers[CopyMIMEType.PLAIN],
[CopyMIMEType.HTML]: customTransformersTemp[CopyMIMEType.HTML] ||
Transformers[CopyMIMEType.HTML],
};
}
// 因为 copy 和 export 在配置上有一定差异，此方法用于抹平差异
export function unifyConfig({ spreadsheet: { options: { interaction }, }, config: { formatOptions = false, separator = TAB_SEPARATOR, selectedCells = [], customTransformer, async = false, }, isExport, }) {
var _a, _b;
const { copy } = interaction;
const { formatData, formatHeader } = isExport
? getFormatOptions(formatOptions)
: {
formatData: (_a = copy === null || copy === void 0 ? void 0 : copy.withFormat) !== null && _a !== void 0 ? _a : false,
formatHeader: (_b = copy === null || copy === void 0 ? void 0 : copy.withFormat) !== null && _b !== void 0 ? _b : false,
};
const transformers = getTransformer(isExport ? customTransformer : copy === null || copy === void 0 ? void 0 : copy.customTransformer);
return {
separator,
selectedCells,
transformers,
formatData,
formatHeader,
async,
};
}
/**
* 获取到当前节点所有数据
* @param leafNode
*/
export const getNodeFormatData = (leafNode) => {
const line = [];
const getNodeFormatterLabel = (node) => {
var _a, _b, _c;
// node.id === KEY_ROOT_NODE 时，为 S2 内的虚拟根节点，导出的内容不需要考虑此节点
if (node.id === ROOT_NODE_ID) {
return;
}
const formatter = (_c = (_b = (_a = node.spreadsheet) === null || _a === void 0 ? void 0 : _a.dataSet) === null || _b === void 0 ? void 0 : _b.getFieldFormatter) === null || _c === void 0 ? void 0 : _c.call(_b, node.field);
const value = formatter === null || formatter === void 0 ? void 0 : formatter(node.value);
line.unshift(value);
if (node === null || node === void 0 ? void 0 : node.parent) {
return getNodeFormatterLabel(node.parent);
}
};
getNodeFormatterLabel(leafNode);
return line;
};
export const getHeaderNodeFromMeta = (meta, spreadsheet) => {
const { rowIndex, colIndex } = meta;
const { facet } = spreadsheet;
return [facet.getRowNodeByIndex(rowIndex), facet.getColNodeByIndex(colIndex)];
};
//# sourceMappingURL=common.js.map