Something went wrong. Try again.
leaflet fork for copy edit / bug fixes forked from leaflet.pub/leaflet
Something went wrong. Try again.
TypeScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321import { Replicache } from "replicache";import { ReplicacheMutators } from "../replicache";import { supabaseBrowserClient } from "supabase/browserClient";import type { FilterAttributes } from "src/replicache/attributes";import type { FactInput } from "src/replicache/mutations";import { rgbaToThumbHash, thumbHashToDataURL } from "thumbhash";import { v7 } from "uuid";import { encodeBitmapToWebP } from "./imageEncoding";import { setImageUploadStatus, clearImageUploadStatus,} from "./imageUploadStatus";
// Maps a public image src to an in-memory object URL, for optimistic display// before the upload completes.export const localImages = new Map<string, string>();
// Caps concurrent addImage pipelines so a 30-image paste does not pin the main// thread on decode/encode/thumbhash work or saturate uplink bandwidth.const MAX_CONCURRENT_IMAGE_UPLOADS = 4;let active = 0;const waiters: Array<() => void> = [];function acquireSlot(): Promise<void> { if (active < MAX_CONCURRENT_IMAGE_UPLOADS) { active++; return Promise.resolve(); } return new Promise<void>((resolve) => { waiters.push(() => { active++; resolve(); }); });}function releaseSlot() { active--; const next = waiters.shift(); if (next) next();}
type ImageArgs = { entityID: string; attribute: keyof FilterAttributes<{ type: "image" }>; // Skip recording the image facts in undo history. Callers that create the // containing block themselves (e.g. paste) group their own undo entry. ignoreUndo?: boolean;};
export type PreparedImage = { // Optimistic block/image fact. Commit alongside the block's structural facts // so the image appears in one render, with no empty-block flash. imageFact: FactInput & { ignoreUndo?: true }; // Uploads to storage, writes the final image fact, and releases the // concurrency slot. Call after imageFact commits. finishUpload: () => Promise<void>;};
// Decodes the image and stages its optimistic preview, returning the fact to// commit plus a continuation that uploads. Separating the two lets the caller// batch the image fact with the block's structural facts. Holds the concurrency// slot until finishUpload resolves.export async function prepareImage( file: File, rep: Replicache<ReplicacheMutators>, args: ImageArgs,): Promise<PreparedImage> { await acquireSlot(); let released = false; const release = () => { if (released) return; released = true; releaseSlot(); }; try { let client = supabaseBrowserClient(); let isAnimated = await isAnimatedFormat(file); let fileID = v7() + (isAnimated ? "." + file.name.split(".").pop() : ""); let url = client.storage.from("minilink-user-assets").getPublicUrl(fileID) .data.publicUrl;
// Decode once for dimensions + thumbhash. localImages must be set before the // fact commits so the block renders the preview, not the fallback, on first // paint. const bitmap = await createImageBitmap(file); const width = bitmap.width; const height = bitmap.height; const thumbhash = computeThumbHashFromBitmap(bitmap); localImages.set(url, URL.createObjectURL(file)); setImageUploadStatus(url, { state: "uploading" });
const imageFact: FactInput & { ignoreUndo?: true } = { entity: args.entityID, attribute: "block/image", ignoreUndo: args.ignoreUndo || undefined, data: { fallback: thumbhash, type: "image", local: rep.clientID, src: url, height, width, }, };
const finishUpload = async () => { try { let uploadBlob: Blob = file; if (!isAnimated) { try { uploadBlob = await encodeBitmapToWebP(bitmap, { quality: 0.92, preferLossless: file.type === "image/png", }); } catch { // Encoding is an optimization; upload the original bytes if it // fails. } } bitmap.close(); await uploadImageAndFinalize({ rep, fileID, url, blob: uploadBlob, file, entityID: args.entityID, attribute: args.attribute, thumbhash, width, height, ignoreUndo: args.ignoreUndo, }); } finally { release(); } };
return { imageFact, finishUpload }; } catch (e) { release(); throw e; }}
type StorageError = { message?: string; statusCode?: string | number };
// A 409 means an earlier attempt (whose response we lost) already stored the// object, so the upload has effectively succeeded.const isAlreadyUploaded = (error: StorageError) => String(error.statusCode) === "409" || !!error.message?.toLowerCase().includes("already exists");
// Uploads an image to storage and swaps the optimistic `local` fact for the// final one. storage-js reports failures as a returned `{ error }` rather than// throwing, so every failure funnels through here: one automatic retry, then// the failure is logged and parked in useImageUploadStatus with a manual// retry. The `local` fact stays put on failure so the preview (object URL// here, thumbhash for other clients) keeps rendering instead of a broken// image.export async function uploadImageAndFinalize(opts: { rep: Replicache<ReplicacheMutators>; fileID: string; url: string; blob: Blob; file: File; entityID: string; attribute: keyof FilterAttributes<{ type: "image" }>; thumbhash: string; width: number; height: number; ignoreUndo?: boolean;}) { const client = supabaseBrowserClient(); const attempt = async (): Promise<StorageError | null> => { try { const { error } = await client.storage .from("minilink-user-assets") .upload(opts.fileID, opts.blob, { // storage-js expects seconds here, not a header value — anything // else is stored as a malformed Cache-Control and defeats the CDN. cacheControl: "31536000", }); if (error && !isAlreadyUploaded(error)) return error; return null; } catch (e) { return { message: e instanceof Error ? e.message : String(e) }; } }; const finalize = async (ignoreUndo?: boolean) => { await opts.rep.mutate.assertFact({ entity: opts.entityID, attribute: opts.attribute, ignoreUndo: ignoreUndo || undefined, data: { fallback: opts.thumbhash, type: "image", src: opts.url, height: opts.height, width: opts.width, }, }); clearImageUploadStatus(opts.url); }; const fail = (error: StorageError, attempts: string) => { logUploadFailure({ src: opts.url, fileName: opts.file.name, fileType: opts.file.type, fileSize: opts.file.size, uploadSize: opts.blob.size, attempts, error: { message: error.message, statusCode: error.statusCode }, }); setImageUploadStatus(opts.url, { state: "failed", retry }); }; const retry = async () => { setImageUploadStatus(opts.url, { state: "uploading" }); const error = await attempt(); if (error) return fail(error, "manual-retry"); // Manual retries land long after the user action that created the block, // so they should never be their own undo step. await finalize(true); };
setImageUploadStatus(opts.url, { state: "uploading" }); let error = await attempt(); if (error) error = await attempt(); if (error) return fail(error, "initial+auto-retry"); await finalize(opts.ignoreUndo);}
// Browser consoles are invisible to us, so mirror upload failures to the// server where they land in function logs.function logUploadFailure(details: { src: string; fileName: string; fileType: string; fileSize: number; uploadSize: number; attempts: string; error: StorageError;}) { console.error("Image upload failed", details); try { fetch("/api/log_upload_failure", { method: "POST", keepalive: true, headers: { "Content-Type": "application/json" }, body: JSON.stringify(details), }).catch(() => {}); } catch {}}
// For callers whose block already renders (image upload UI, canvas drop):// commit the preview fact, then upload.export async function addImage( file: File, rep: Replicache<ReplicacheMutators>, args: ImageArgs,) { let { imageFact, finishUpload } = await prepareImage(file, rep, args); await rep.mutate.assertFact(imageFact); await finishUpload();}
function computeThumbHashFromBitmap(bitmap: ImageBitmap): string { const canvas = document.createElement("canvas"); const context = canvas.getContext("2d") as CanvasRenderingContext2D; const maxDimension = 100; let width = bitmap.width; let height = bitmap.height;
if (width > height) { if (width > maxDimension) { height *= maxDimension / width; width = maxDimension; } } else { if (height > maxDimension) { width *= maxDimension / height; height = maxDimension; } }
canvas.width = width; canvas.height = height; context.drawImage(bitmap, 0, 0, width, height);
const imageData = context.getImageData(0, 0, width, height); return thumbHashToDataURL( rgbaToThumbHash(imageData.width, imageData.height, imageData.data), );}
async function isAnimatedFormat(file: File): Promise<boolean> { if (file.type === "image/gif" || file.type === "image/apng") return true; if (file.type === "image/webp") { // WebP files have an "ANIM" chunk header when animated. We only need the // first ~40 bytes to find it. try { const head = await file.slice(0, 64).arrayBuffer(); const bytes = new Uint8Array(head); // Look for the four-byte ASCII sequence "ANIM" after the RIFF/WEBP // preamble. for (let i = 12; i < bytes.length - 3; i++) { if ( bytes[i] === 0x41 && bytes[i + 1] === 0x4e && bytes[i + 2] === 0x49 && bytes[i + 3] === 0x4d ) { return true; } } } catch { // Defensive: if we can't read the header, assume static webp. } } return false;}