58 lines
3.3 KiB
JavaScript
58 lines
3.3 KiB
JavaScript
import assert from "node:assert/strict";
|
|
import test from "node:test";
|
|
import { strToU8, unzipSync, zipSync } from "fflate";
|
|
import { compactPartnerBatchWorkbookForUpload } from "../koc-portal/app/batch-workbook-upload.ts";
|
|
|
|
const imageBytes = (marker, size) => {
|
|
const bytes = new Uint8Array(size);
|
|
bytes.set([0x89, 0x50, 0x4e, 0x47, marker]);
|
|
for (let index = 5; index < bytes.length; index += 1) bytes[index] = marker;
|
|
return bytes;
|
|
};
|
|
|
|
test("slims oversized WPS workbooks without removing backfill screenshots", () => {
|
|
const worksheet = `<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
|
|
<worksheet xmlns="http://schemas.openxmlformats.org/spreadsheetml/2006/main">
|
|
<sheetData>
|
|
<row r="1">
|
|
<c r="A1" t="inlineStr"><is><t>序号（不能改）</t></is></c>
|
|
<c r="D1" t="inlineStr"><is><t>图片1</t></is></c>
|
|
<c r="F1" t="inlineStr"><is><t>笔记截图</t></is></c>
|
|
<c r="G1" t="inlineStr"><is><t>数据分析截图（单篇笔记数据分析截图）</t></is></c>
|
|
</row>
|
|
<row r="2">
|
|
<c r="D2" t="str"><f>_xlfn.DISPIMG("SOURCE",1)</f><v>=DISPIMG("SOURCE",1)</v></c>
|
|
<c r="F2" t="str"><f>_xlfn.DISPIMG("PUBLISH",1)</f><v>=DISPIMG("PUBLISH",1)</v></c>
|
|
</row>
|
|
</sheetData>
|
|
</worksheet>`;
|
|
const cellImages = `<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
|
|
<etc:cellImages xmlns:etc="http://www.wps.cn/officeDocument/2017/etCustomData" xmlns:xdr="http://schemas.openxmlformats.org/drawingml/2006/spreadsheetDrawing" xmlns:a="http://schemas.openxmlformats.org/drawingml/2006/main" xmlns:r="http://schemas.openxmlformats.org/officeDocument/2006/relationships">
|
|
<etc:cellImage><xdr:pic><xdr:nvPicPr><xdr:cNvPr id="1" name="SOURCE"/></xdr:nvPicPr><xdr:blipFill><a:blip r:embed="rId1"/></xdr:blipFill></xdr:pic></etc:cellImage>
|
|
<etc:cellImage><xdr:pic><xdr:nvPicPr><xdr:cNvPr id="2" name="PUBLISH"/></xdr:nvPicPr><xdr:blipFill><a:blip r:embed="rId2"/></xdr:blipFill></xdr:pic></etc:cellImage>
|
|
</etc:cellImages>`;
|
|
const relationships = `<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
|
|
<Relationships xmlns="http://schemas.openxmlformats.org/package/2006/relationships">
|
|
<Relationship Id="rId1" Type="http://schemas.openxmlformats.org/officeDocument/2006/relationships/image" Target="media/source.png"/>
|
|
<Relationship Id="rId2" Type="http://schemas.openxmlformats.org/officeDocument/2006/relationships/image" Target="media/publish.png"/>
|
|
</Relationships>`;
|
|
const workbook = zipSync(
|
|
{
|
|
"xl/worksheets/sheet1.xml": strToU8(worksheet),
|
|
"xl/cellimages.xml": strToU8(cellImages),
|
|
"xl/_rels/cellimages.xml.rels": strToU8(relationships),
|
|
"xl/media/source.png": [imageBytes(1, 2_000_000), { level: 0 }],
|
|
"xl/media/publish.png": [imageBytes(2, 2_000), { level: 0 }],
|
|
},
|
|
{ level: 0 },
|
|
);
|
|
|
|
const compacted = compactPartnerBatchWorkbookForUpload(workbook);
|
|
const entries = unzipSync(compacted.bytes);
|
|
assert.equal(entries["xl/media/source.png"], undefined);
|
|
assert.deepEqual(entries["xl/media/publish.png"], imageBytes(2, 2_000));
|
|
assert.equal(compacted.removedMediaCount, 1);
|
|
assert.equal(compacted.preservedScreenshotCount, 1);
|
|
assert.ok(compacted.bytes.byteLength < workbook.byteLength / 10);
|
|
});
|