@antv/s2
Version:
effective spreadsheet render core lib
189 lines • 7.27 kB
JavaScript
import { __awaiter } from "tslib";
import { concat, every, first, isEmpty, last } from 'lodash';
import { CellType, DATA_CELL_ID_SEPARATOR, InteractionStateName, } from '../../../common';
import { CopyMIMEType, } from '../../../common/interface/export';
import { getSelectedCols, getSelectedRows } from '../method';
import { copyToClipboard } from '../utils';
import { asyncProcessSelectedAllPivot, processSelectedPivotByDataCell, processSelectedPivotByHeader, } from './pivot-data-cell-copy';
import { getBrushHeaderCopyable } from './pivot-header-copy';
import { asyncProcessSelectedAllTable, processSelectedTableByDataCell, processSelectedTableByHeader, } from './table-copy';
/**
* 返回选中数据单元格生成的二维数组（ CellMeta[][]）
* @param { CellMeta[] } cells
*/
const getSelectedCellsMeta = (cells) => {
if (!(cells === null || cells === void 0 ? void 0 : cells.length)) {
return [];
}
const [minCell, maxCell] = [
{ row: Infinity, col: Infinity },
{ row: 0, col: 0 },
];
// get left-top cell and right-bottom cell position
cells.forEach((e) => {
const { rowIndex, colIndex } = e;
minCell.col = Math.min(colIndex, minCell.col);
minCell.row = Math.min(rowIndex, minCell.row);
maxCell.col = Math.max(colIndex, maxCell.col);
maxCell.row = Math.max(rowIndex, maxCell.row);
});
const [rowLen, colLen] = [
maxCell.row - minCell.row + 1,
maxCell.col - minCell.col + 1,
];
const twoDimDataArray = new Array(rowLen)
.fill('')
.map(() => new Array(colLen).fill(''));
cells.forEach((e) => {
const { rowIndex, colIndex } = e;
const [diffRow, diffCol] = [rowIndex - minCell.row, colIndex - minCell.col];
twoDimDataArray[diffRow][diffCol] = e;
});
return twoDimDataArray;
};
const processSelectedByHeader = (spreadsheet, selectedRows) => {
if (spreadsheet.isPivotMode()) {
return processSelectedPivotByHeader(spreadsheet, selectedRows);
}
return processSelectedTableByHeader(spreadsheet, selectedRows);
};
function getIsBrushHeader(interactedCells) {
return isEmpty(interactedCells)
? false
: every(interactedCells, (cell) => cell.cellType === CellType.ROW_CELL ||
cell.cellType === CellType.COL_CELL);
}
function processSelectedByData(selectedCellsMeta, selectedColMetas, selectedRowMetas, spreadsheet) {
if (spreadsheet.isPivotMode()) {
return processSelectedPivotByDataCell({
spreadsheet,
selectedCells: selectedCellsMeta,
headerSelectedCells: concat(selectedColMetas, selectedRowMetas),
});
}
return processSelectedTableByDataCell({
spreadsheet,
selectedCells: selectedCellsMeta,
headerSelectedCells: selectedColMetas,
});
}
function getDataCellCopyable(spreadsheet, cells) {
let data;
const selectedCols = getSelectedCols(cells);
const selectedRows = getSelectedRows(cells);
if (spreadsheet.interaction.getCurrentStateName() ===
InteractionStateName.ALL_SELECTED) {
data = processSelectedByHeader(spreadsheet, []);
}
else if (selectedCols.length) {
// 选中某列
data = processSelectedByHeader(spreadsheet, selectedCols);
}
else if (selectedRows.length) {
// 选中某行
data = processSelectedByHeader(spreadsheet, selectedRows);
}
else {
if (!cells.length) {
return [
{
type: CopyMIMEType.PLAIN,
content: '',
},
{
type: CopyMIMEType.HTML,
content: '',
},
];
}
// normal selected
const selectedCellsMeta = getSelectedCellsMeta(cells);
const selectedColMetas = selectedCellsMeta[0].map((cellMeta) => {
var _a;
return Object.assign(Object.assign({}, cellMeta), { id: (cellMeta === null || cellMeta === void 0 ? void 0 : cellMeta.colId) ||
last((_a = cellMeta === null || cellMeta === void 0 ? void 0 : cellMeta.id) === null || _a === void 0 ? void 0 : _a.split(DATA_CELL_ID_SEPARATOR)) ||
'', type: CellType.COL_CELL });
});
const selectedRowMetas = selectedCellsMeta.map((cellMeta) => {
var _a, _b, _c;
return Object.assign(Object.assign({}, cellMeta[0]), { id: ((_a = cellMeta[0]) === null || _a === void 0 ? void 0 : _a.rowId) ||
first((_c = (_b = cellMeta[0]) === null || _b === void 0 ? void 0 : _b.id) === null || _c === void 0 ? void 0 : _c.split(DATA_CELL_ID_SEPARATOR)) ||
'', type: CellType.ROW_CELL });
});
data = processSelectedByData(selectedCellsMeta, selectedColMetas, selectedRowMetas, spreadsheet);
}
return data;
}
// 刷选复制使用
export const getSelectedData = (spreadsheet) => {
var _a;
const interaction = spreadsheet.interaction;
const cells = interaction.getState().cells || [];
// 通过判断当前存在交互的单元格，来区分圈选行/列头 还是 点选行/列头
const interactedCells = (_a = interaction.getInteractedCells()) !== null && _a !== void 0 ? _a : [];
const isBrushHeader = getIsBrushHeader(interactedCells);
// 行列头圈选复制 和 单元格复制不同
const data = isBrushHeader
? getBrushHeaderCopyable(interactedCells)
: getDataCellCopyable(spreadsheet, cells);
if (data) {
copyToClipboard(data);
}
return data;
};
// 异步全量导出
export const asyncProcessAllSelected = (params) => {
const { sheetInstance } = params;
const check = sheetInstance.enableAsyncExport();
if (check instanceof Error) {
// eslint-disable-next-line no-console
console.warn(check);
throw check;
}
if (sheetInstance.isPivotMode()) {
return asyncProcessSelectedAllPivot(params);
}
return asyncProcessSelectedAllTable(params);
};
/**
* 异步获取文本数据 (text/plain)
* @example
const data = await asyncGetAllPlainData({
sheetInstance: s2,
split: '\t',
formatOptions: true,
});
*/
export const asyncGetAllPlainData = (params) => __awaiter(void 0, void 0, void 0, function* () {
const result = yield asyncProcessAllSelected(params);
return result[0].content;
});
/**
* 异步获取富文本数据 (text/html)
* @example
const data = await asyncGetAllHtmlData({
sheetInstance: s2,
split: '\t',
formatOptions: true,
});
*/
export const asyncGetAllHtmlData = (params) => __awaiter(void 0, void 0, void 0, function* () {
const result = yield asyncProcessAllSelected(params);
return result[1].content;
});
/**
* 异步获取数据
* - 文本 (text/plain)
* - 富文本 (text/html)
* @example
const data = await asyncGetAllData({
sheetInstance: s2,
split: '\t',
formatOptions: true,
});
*/
export const asyncGetAllData = (params) => __awaiter(void 0, void 0, void 0, function* () {
const result = yield asyncProcessAllSelected(params);
return result;
});
//# sourceMappingURL=core.js.map