@chenchaolong/plugin-trade-compliance-workbench 1.0.127 → 1.0.129

package/dist/lib/remote-components/trade-compliance-workbench/editor-source.mjs

@@ -7,12 +7,17 @@ import '@univerjs/preset-sheets-core/lib/index.css'
 import '@univerjs/preset-sheets-drawing/lib/index.css'
 import ExcelJS from 'exceljs'
 
+const LIGHTWEIGHT_EDITOR_MAX_ESTIMATED_CELLS = 25000
+const LIGHTWEIGHT_EDITOR_MAX_IMAGE_COUNT = 8
+const LIGHTWEIGHT_EDITOR_MAX_IMAGE_BYTES = 5 * 1024 * 1024
+
 window.TradeComplianceExcelEditor = {
   async open({ container, base64, onSave }) {
     const original = new ExcelJS.Workbook()
     await original.xlsx.load(base64ToBytes(base64))
-    const snapshot = toUniverSnapshot(original)
-    const workbookImages = extractWorkbookImages(original)
+    const lightweightMode = shouldUseLightweightEditor(original)
+    const snapshot = toUniverSnapshot(original, { includeStyles: !lightweightMode })
+    const workbookImages = lightweightMode ? [] : extractWorkbookImages(original)
     const { univer, univerAPI } = createUniver({
       locale: 'zhCN',
       locales: { zhCN },
@@ -36,6 +41,14 @@ window.TradeComplianceExcelEditor = {
   }
 }
 
+function shouldUseLightweightEditor(workbook) {
+  const estimatedCells = workbook.worksheets.reduce((total, worksheet) => total + Math.max(worksheet.rowCount, 1) * Math.max(worksheet.columnCount, 1), 0)
+  const media = workbook.model.media || []
+  const imageCount = workbook.worksheets.reduce((total, worksheet) => total + worksheet.getImages().length, 0)
+  const imageBytes = media.reduce((total, item) => total + Number(item?.buffer?.byteLength || item?.buffer?.length || 0), 0)
+  return estimatedCells > LIGHTWEIGHT_EDITOR_MAX_ESTIMATED_CELLS || imageCount > LIGHTWEIGHT_EDITOR_MAX_IMAGE_COUNT || imageBytes > LIGHTWEIGHT_EDITOR_MAX_IMAGE_BYTES
+}
+
 function extractWorkbookImages(workbook) {
   const media = workbook.model.media || []
   const byIndex = new Map(media.map((item, index) => [item.index ?? index, item]))
@@ -93,7 +106,8 @@ async function insertWorkbookImages(univerAPI, images) {
   }
 }
 
-function toUniverSnapshot(workbook) {
+function toUniverSnapshot(workbook, options = {}) {
+  const includeStyles = options.includeStyles !== false
   const sheets = {}; const sheetOrder = []
   workbook.worksheets.forEach((worksheet, index) => {
     const id = `sheet-${index + 1}`; sheetOrder.push(id); const cellData = {}
@@ -101,7 +115,7 @@ function toUniverSnapshot(workbook) {
       const rowIndex = rowNumber - 1; const columnIndex = columnNumber - 1
       cellData[rowIndex] ||= {}
       const raw = excelCellDisplayValue(cell.value)
-      cellData[rowIndex][columnIndex] = { v: raw instanceof Date ? raw.toISOString().slice(0, 10) : raw == null ? '' : String(raw), t: typeof raw === 'number' ? 2 : 1, s: cellStyle(cell) }
+      cellData[rowIndex][columnIndex] = { v: raw instanceof Date ? raw.toISOString().slice(0, 10) : raw == null ? '' : String(raw), t: typeof raw === 'number' ? 2 : 1, s: includeStyles ? cellStyle(cell) : undefined }
     }))
     sheets[id] = { id, name: worksheet.name, rowCount: Math.max(worksheet.rowCount + 20, 100), columnCount: Math.max(worksheet.columnCount + 5, 26), cellData, mergeData: mergeData(worksheet), rowData: rowData(worksheet), columnData: columnData(worksheet) }
   })