openreplay/frontend/app/player/web/network/loadFiles.ts

89 lines
2.3 KiB
TypeScript

import APIClient from 'App/api_client';
const ALLOWED_404 = 'No-file-and-this-is-ok';
const NO_BACKUP_FILE = 'No-efs-file';
export const NO_URLS = 'No-urls-provided';
export async function loadFiles(
urls: string[],
onData: (data: Uint8Array) => void,
canSkip: boolean = false
): Promise<void> {
if (!urls.length) {
throw NO_URLS;
}
try {
for (let url of urls) {
await loadFile(url, onData, urls.length > 1 ? url !== urls[0] : canSkip);
}
return Promise.resolve();
} catch (e) {
return Promise.reject(e);
}
}
export async function loadFile(
url: string,
onData: (data: Uint8Array) => void,
canSkip: boolean = false
): Promise<void> {
return window
.fetch(url)
.then((response) => processAPIStreamResponse(response, canSkip))
.then((data) => onData(data))
.catch((e) => {
if (e === ALLOWED_404) {
return;
} else {
throw e;
}
});
}
export async function requestEFSDom(sessionId: string) {
return await requestEFSMobFile(sessionId + '/dom.mob');
}
export async function requestEFSDevtools(sessionId: string) {
return await requestEFSMobFile(sessionId + '/devtools.mob');
}
export async function requestTarball(url: string) {
const res = await window.fetch(url);
if (res.ok) {
const buf = await res.arrayBuffer();
return new Uint8Array(buf);
} else {
throw new Error(res.status.toString());
}
}
async function requestEFSMobFile(filename: string) {
const api = new APIClient();
const siteId = document.location.href.match(
/https:\/\/[a-z.-]+\/([a-z0-9]+)\/[a-z-]+\/[0-9]+/
)?.[1];
if (siteId) {
api.forceSiteId(siteId);
api.setSiteIdCheck(() => siteId);
}
const res = await api.fetch('/unprocessed/' + filename);
if (res.status >= 400) {
throw NO_BACKUP_FILE;
}
return await processAPIStreamResponse(res, false);
}
const processAPIStreamResponse = (response: Response, skippable: boolean) => {
return new Promise<ArrayBuffer>((res, rej) => {
if (response.status === 404 && skippable) {
return rej(ALLOWED_404);
}
if (response.status >= 400) {
return rej(
`Bad file status code ${response.status}. Url: ${response.url}`
);
}
res(response.arrayBuffer());
}).then(async (buf) => new Uint8Array(buf));
};