toeverything/AFFiNE
1
1
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2024-12-27 03:14:11 +03:00
Code Issues Packages Projects Releases Wiki Activity

feat: detect if blob too large (#1738)

Browse Source
This commit is contained in:
Himself65 2023-03-29 02:39:23 -05:00 committed by GitHub
parent c1e8818db4
commit 9c6fb82c82
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 28 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
packages/env/src/constant.ts vendored
View File

@ -17,6 +17,7 @@ export const enum MessageCode {
leaveWorkspaceFailed,
downloadWorkspaceFailed,
refreshTokenError,
blobTooLarge,
}
export const Messages = {
@ -65,4 +66,11 @@ export const Messages = {
[MessageCode.refreshTokenError]: {
message: 'Refresh token failed',
},
} as const;
[MessageCode.blobTooLarge]: {
message: 'Blob too large',
},
} as const satisfies {
[key in MessageCode]: {
message: string;
};
};

15
packages/workspace/src/affine/__tests__/api.spec.ts
View File

@ -101,6 +101,21 @@ describe('api', () => {
document.removeEventListener('affine-error', listener);
});
test('blob too large', async () => {
let called = false;
try {
await workspaceApis.uploadBlob(
'test',
new ArrayBuffer(1024 * 1024 * 1024 + 1),
'image/png'
);
} catch (e) {
called = true;
expect(e).toBeInstanceOf(RequestError);
}
expect(called, 'throw error').toBe(true);
});
test('refresh token', async () => {
const storage = getLoginStorage();
assertExists(storage);

4
packages/workspace/src/affine/api/index.ts
View File

@ -302,6 +302,10 @@ export function createWorkspaceApis(prefixUrl = '/') {
): Promise<string> => {
const auth = getLoginStorage();
assertExists(auth);
const mb = arrayBuffer.byteLength / 1048576;
if (mb > 10) {
throw new RequestError(MessageCode.blobTooLarge);
}
return fetch(prefixUrl + 'api/blob', {
method: 'PUT',
body: arrayBuffer,