mirror of
https://github.com/excalidraw/excalidraw.git
synced 2025-05-03 10:00:07 -04:00
first implementation of itxt on image.ts
This commit is contained in:
parent
04d0b00b95
commit
22869ee58c
2 changed files with 63 additions and 19 deletions
|
@ -1,4 +1,5 @@
|
|||
import tEXt from "png-chunk-text";
|
||||
import { encode as encodeITXt, decode as decodeITXt } from "png-chunk-itxt";
|
||||
import encodePng from "png-chunks-encode";
|
||||
import decodePng from "png-chunks-extract";
|
||||
|
||||
|
@ -11,48 +12,89 @@ import { encode, decode } from "./encode";
|
|||
// PNG
|
||||
// -----------------------------------------------------------------------------
|
||||
|
||||
export const getTEXtChunk = async (
|
||||
export const getMetadataChunk = async (
|
||||
blob: Blob,
|
||||
): Promise<{ keyword: string; text: string } | null> => {
|
||||
const chunks = decodePng(new Uint8Array(await blobToArrayBuffer(blob)));
|
||||
const metadataChunk = chunks.find((chunk) => chunk.name === "tEXt");
|
||||
if (metadataChunk) {
|
||||
return tEXt.decode(metadataChunk.data);
|
||||
|
||||
const iTXtChunk = chunks.find((chunk) => chunk.name === "iTXt");
|
||||
if (iTXtChunk) {
|
||||
try {
|
||||
const decoded = decodeITXt(iTXtChunk.data);
|
||||
return { keyword: decoded.keyword, text: decoded.text };
|
||||
} catch (error) {
|
||||
console.error("Failed to decode iTXt chunk:", error);
|
||||
}
|
||||
}
|
||||
|
||||
const tEXtChunk = chunks.find((chunk) => chunk.name === "tEXt");
|
||||
if (tEXtChunk) {
|
||||
return tEXt.decode(tEXtChunk.data);
|
||||
}
|
||||
|
||||
return null;
|
||||
};
|
||||
|
||||
export const encodePngMetadata = async ({
|
||||
blob,
|
||||
metadata,
|
||||
useITXt = true,
|
||||
}: {
|
||||
blob: Blob;
|
||||
metadata: string;
|
||||
useITXt?: boolean;
|
||||
}) => {
|
||||
const chunks = decodePng(new Uint8Array(await blobToArrayBuffer(blob)));
|
||||
|
||||
const metadataChunk = tEXt.encode(
|
||||
MIME_TYPES.excalidraw,
|
||||
JSON.stringify(
|
||||
encode({
|
||||
text: metadata,
|
||||
compress: true,
|
||||
}),
|
||||
),
|
||||
const filteredChunks = chunks.filter(
|
||||
(chunk) =>
|
||||
!(chunk.name === "tEXt" &&
|
||||
tEXt.decode(chunk.data).keyword === MIME_TYPES.excalidraw) &&
|
||||
!(chunk.name === "iTXt" &&
|
||||
decodeITXt(chunk.data).keyword === MIME_TYPES.excalidraw)
|
||||
);
|
||||
// insert metadata before last chunk (iEND)
|
||||
chunks.splice(-1, 0, metadataChunk);
|
||||
|
||||
return new Blob([encodePng(chunks)], { type: MIME_TYPES.png });
|
||||
const encodedData = JSON.stringify(
|
||||
encode({
|
||||
text: metadata,
|
||||
compress: true,
|
||||
}),
|
||||
);
|
||||
|
||||
let metadataChunk;
|
||||
try {
|
||||
if (useITXt) {
|
||||
metadataChunk = encodeITXt(
|
||||
MIME_TYPES.excalidraw,
|
||||
encodedData,
|
||||
{
|
||||
compressed: false, //Already compressed in encode
|
||||
language: "en",
|
||||
translated: ""
|
||||
}
|
||||
);
|
||||
} else {
|
||||
throw new Error("Fallback to tEXt");
|
||||
}
|
||||
} catch (error) {
|
||||
console.warn("iTXt encoding failed, falling back to tEXt:", error);
|
||||
metadataChunk = tEXt.encode(
|
||||
MIME_TYPES.excalidraw,
|
||||
encodedData,
|
||||
);
|
||||
}
|
||||
|
||||
filteredChunks.splice(-1, 0, metadataChunk);
|
||||
|
||||
return new Blob([encodePng(filteredChunks)], { type: MIME_TYPES.png });
|
||||
};
|
||||
|
||||
export const decodePngMetadata = async (blob: Blob) => {
|
||||
const metadata = await getTEXtChunk(blob);
|
||||
const metadata = await getMetadataChunk(blob);
|
||||
if (metadata?.keyword === MIME_TYPES.excalidraw) {
|
||||
try {
|
||||
const encodedData = JSON.parse(metadata.text);
|
||||
if (!("encoded" in encodedData)) {
|
||||
// legacy, un-encoded scene JSON
|
||||
if (
|
||||
"type" in encodedData &&
|
||||
encodedData.type === EXPORT_DATA_TYPES.excalidraw
|
||||
|
|
|
@ -101,9 +101,10 @@ export const exportToBlob = async (
|
|||
mimeType?: string;
|
||||
quality?: number;
|
||||
exportPadding?: number;
|
||||
useITXt?: boolean;
|
||||
},
|
||||
): Promise<Blob> => {
|
||||
let { mimeType = MIME_TYPES.png, quality } = opts;
|
||||
let { mimeType = MIME_TYPES.png, quality, useITXt = true } = opts;
|
||||
|
||||
if (mimeType === MIME_TYPES.png && typeof quality === "number") {
|
||||
console.warn(`"quality" will be ignored for "${MIME_TYPES.png}" mimeType`);
|
||||
|
@ -150,6 +151,7 @@ export const exportToBlob = async (
|
|||
opts.files || {},
|
||||
"local",
|
||||
),
|
||||
useITXt,
|
||||
});
|
||||
}
|
||||
resolve(blob);
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue