51 lines
1.6 KiB
JavaScript
51 lines
1.6 KiB
JavaScript
|
import { fromBase64 } from "@smithy/util-base64";
|
||
|
|
export const streamCollector = (stream) => {
|
||
|
|
if (typeof Blob === "function" && stream instanceof Blob) {
|
||
|
|
return collectBlob(stream);
|
||
|
|
}
|
||
|
|
return collectStream(stream);
|
||
|
|
};
|
||
|
|
async function collectBlob(blob) {
|
||
|
|
const base64 = await readToBase64(blob);
|
||
|
|
const arrayBuffer = fromBase64(base64);
|
||
|
|
return new Uint8Array(arrayBuffer);
|
||
|
|
}
|
||
|
|
async function collectStream(stream) {
|
||
|
|
const chunks = [];
|
||
|
|
const reader = stream.getReader();
|
||
|
|
let isDone = false;
|
||
|
|
let length = 0;
|
||
|
|
while (!isDone) {
|
||
|
|
const { done, value } = await reader.read();
|
||
|
|
if (value) {
|
||
|
|
chunks.push(value);
|
||
|
|
length += value.length;
|
||
|
|
}
|
||
|
|
isDone = done;
|
||
|
|
}
|
||
|
|
const collected = new Uint8Array(length);
|
||
|
|
let offset = 0;
|
||
|
|
for (const chunk of chunks) {
|
||
|
|
collected.set(chunk, offset);
|
||
|
|
offset += chunk.length;
|
||
|
|
}
|
||
|
|
return collected;
|
||
|
|
}
|
||
|
|
function readToBase64(blob) {
|
||
|
|
return new Promise((resolve, reject) => {
|
||
|
|
const reader = new FileReader();
|
||
|
|
reader.onloadend = () => {
|
||
|
|
if (reader.readyState !== 2) {
|
||
|
|
return reject(new Error("Reader aborted too early"));
|
||
|
|
}
|
||
|
|
const result = (reader.result ?? "");
|
||
|
|
const commaIndex = result.indexOf(",");
|
||
|
|
const dataOffset = commaIndex > -1 ? commaIndex + 1 : result.length;
|
||
|
|
resolve(result.substring(dataOffset));
|
||
|
|
};
|
||
|
|
reader.onabort = () => reject(new Error("Read aborted"));
|
||
|
|
reader.onerror = () => reject(reader.error);
|
||
|
|
reader.readAsDataURL(blob);
|
||
|
|
});
|
||
|
|
}
|