@agimon-ai/po-author 0.0.0-alpha.71 → 0.0.0-alpha.73

package/dist/extensions/native.mjs

@@ -958,6 +958,7 @@ const VIDEO_MIMES = {
 const IMAGE_SAVE_MAX_BYTES = 33554432;
 const PDF_MAX_BYTES = 33554432;
 const VIDEO_MAX_BYTES = 134217728;
+const READ_CHUNK_BYTES = 1048576;
 const extensionOf = (path) => path.toLowerCase().split("/").at(-1)?.split(".").slice(1).at(-1) ?? "";
 /** Reuse the native decoder's supported image paths for canvas-specific host layout. */
 function authorNativeIsImage(path) {
@@ -1078,49 +1079,74 @@ async function loadAuthorFile(sessionId, path, signal) {
 	if (format !== void 0) return await loadStructuredFile(sessionId, path, format, signal);
 	const artifact = path.startsWith(AUTHOR_ARTIFACT_PREFIX);
 	const query = { path: artifact ? path.slice(9) : path };
-	const result = artifact ? await artifactFiles.session(sessionId).content({
-		query,
-		signal
-	}) : await sessionFiles.session(sessionId).file({
-		query,
-		signal
-	});
-	if (!result.ok) throw new Error(result.error || "This file could not be read.");
-	const response = result.response;
-	if (!response) throw new Error("This file could not be read.");
+	const load = async (headers) => {
+		if (signal.aborted) throw new Error("Opening was cancelled.");
+		const result = artifact ? await artifactFiles.session(sessionId).content({
+			query,
+			headers,
+			signal
+		}) : await sessionFiles.session(sessionId).file({
+			query,
+			headers,
+			signal
+		});
+		const response = result.response;
+		if (result.status === 412) throw new Error("This file changed. Reopen it.");
+		const empty = result.status === 416 && response?.headers.get("content-range") === "bytes */0";
+		if (!result.ok && !empty) throw new Error(result.error || "This file could not be read.");
+		if (!response) throw new Error("This file could not be read.");
+		return response;
+	};
+	const response = await load({ Range: `bytes=0-1048575` });
 	const extension = extensionOf(query.path);
 	const mediaMime = extension === "pdf" ? "application/pdf" : VIDEO_MIMES[extension];
 	const limit = extension === "pdf" ? PDF_MAX_BYTES : mediaMime ? VIDEO_MAX_BYTES : MAX_FILE_BYTES;
 	if (Number(response.headers.get("content-length") ?? 0) > limit) throw new Error("This file is too large to open here.");
 	const mimeType = (response.headers.get("content-type")?.split(";")[0]?.trim() ?? "").toLowerCase();
 	const sourceSha256 = response.headers.get("X-File-SHA256") ?? void 0;
-	if (mediaMime !== void 0) {
-		const bytes = new Uint8Array(await response.arrayBuffer());
+	let bytes;
+	if (response.status === 206) {
+		const range = /^bytes 0-(\d+)\/(\d+)$/.exec(response.headers.get("content-range") ?? "");
+		const size = range ? Number(range[2]) : NaN;
+		const version = response.headers.get("etag");
+		if (!Number.isSafeInteger(size) || size < 1 || !version || version.startsWith("W/")) throw new Error("This file returned an invalid range or version.");
+		if (size > limit) throw new Error("This file is too large to open here.");
+		if (Number(range[1]) !== Math.min(READ_CHUNK_BYTES, size) - 1) throw new Error("This file returned an invalid range.");
+		bytes = new Uint8Array(size);
+		let part = response;
+		for (let offset = 0; offset < size; offset += READ_CHUNK_BYTES) {
+			const end = Math.min(offset + READ_CHUNK_BYTES, size) - 1;
+			if (offset > 0) part = await load({
+				Range: `bytes=${offset}-${end}`,
+				"If-Match": version
+			});
+			if (part.status !== 206 || part.headers.get("etag") !== version || part.headers.get("content-range") !== `bytes ${offset}-${end}/${size}`) throw new Error("This file changed or returned an invalid range.");
+			const chunk = new Uint8Array(await part.arrayBuffer());
+			if (signal.aborted) throw new Error("Opening was cancelled.");
+			if (chunk.byteLength !== end - offset + 1) throw new Error("This file response is incomplete.");
+			bytes.set(chunk, offset);
+		}
+	} else {
+		bytes = new Uint8Array(await response.arrayBuffer());
 		if (signal.aborted) throw new Error("Opening was cancelled.");
+		if (response.status === 416 && bytes.byteLength !== 0) throw new Error("This file response is incomplete.");
 		if (bytes.byteLength > limit) throw new Error("This file is too large to open here.");
-		return {
-			kind: extension === "pdf" ? "pdf" : "video",
-			bytes,
-			mimeType: mediaMime,
-			...sourceSha256 === void 0 ? {} : { sourceSha256 }
-		};
-	}
-	if (IMAGE_TYPES.has(mimeType)) {
-		const bytes = new Uint8Array(await response.arrayBuffer());
-		if (signal.aborted) throw new Error("Opening was cancelled.");
-		if (bytes.byteLength > MAX_FILE_BYTES) throw new Error("This file is too large to open here.");
-		return {
-			kind: "image",
-			data: encodeBase64(bytes),
-			mimeType,
-			...sourceSha256 === void 0 ? {} : { sourceSha256 }
-		};
 	}
+	if (mediaMime !== void 0) return {
+		kind: extension === "pdf" ? "pdf" : "video",
+		bytes,
+		mimeType: mediaMime,
+		...sourceSha256 === void 0 ? {} : { sourceSha256 }
+	};
+	if (IMAGE_TYPES.has(mimeType)) return {
+		kind: "image",
+		data: encodeBase64(bytes),
+		mimeType,
+		...sourceSha256 === void 0 ? {} : { sourceSha256 }
+	};
 	if (mimeType.startsWith("text/") || mimeType === "application/json" || TEXT_EXTENSIONS.has(extension)) {
-		const text = await response.text();
-		if (signal.aborted) throw new Error("Opening was cancelled.");
+		const text = new TextDecoder().decode(bytes);
 		if (new Blob([text]).size > MAX_FILE_BYTES) throw new Error("This file is too large to open here.");
-		const sourceSha256 = response.headers.get("X-File-SHA256") ?? void 0;
 		return {
 			kind: "text",
 			text,