mirrors/signal-desktop
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages 2 Wiki Activity Actions

Use TUS utilities for backup upload

Browse source
This commit is contained in:
Fedor Indutny 2024-05-14 10:04:50 -07:00 • committed by GitHub
parent 4eb5458ace
commit 4fed756661
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
14 changed files with 356 additions and 228 deletions
Download patch file Download diff file Expand all files Collapse all files

13
ts/util/uploads/uploads.ts
View file

@ -1,12 +1,13 @@
// Copyright 2024 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
import fetch from 'node-fetch';
import { createReadStream, createWriteStream } from 'node:fs';
import { Writable } from 'node:stream';
import type { TusFileReader } from './tusProtocol';
import { pipeline } from 'node:stream/promises';
import type { TusFileReader, FetchFunctionType } from './tusProtocol';
import { tusResumeUpload, tusUpload } from './tusProtocol';
import { HTTPError } from '../../textsecure/Errors';
const defaultFileReader: TusFileReader = (filePath, offset) => {
export const defaultFileReader: TusFileReader = (filePath, offset) => {
return createReadStream(filePath, { start: offset });
};
@ -87,13 +88,15 @@ export async function _doDownload({
headers = {},
filePath,
signal,
fetchFn = fetch,
}: {
endpoint: string;
filePath: string;
headers?: Record<string, string>;
signal?: AbortSignal;
fetchFn?: FetchFunctionType;
}): Promise<void> {
const response = await fetch(endpoint, {
const response = await fetchFn(endpoint, {
method: 'GET',
signal,
redirect: 'error',
@ -106,7 +109,7 @@ export async function _doDownload({
throw new Error('Response has no body');
}
const writable = createWriteStream(filePath);
await response.body.pipeTo(Writable.toWeb(writable));
await pipeline(response.body, writable);
}
/**