forked from akai/readest
Huang Xin
46 lines
1.6 KiB
TypeScript
46 lines
1.6 KiB
TypeScript
import { s3Storage } from './s3';
|
|
import { r2Storage } from './r2';
|
|
import { getStorageType } from './storage';
|
|
|
|
export const getDownloadSignedUrl = async (
|
|
fileKey: string,
|
|
expiresIn: number,
|
|
bucketName?: string,
|
|
) => {
|
|
const storageType = getStorageType();
|
|
if (storageType === 'r2') {
|
|
bucketName = bucketName || process.env['R2_BUCKET_NAME'] || '';
|
|
return await r2Storage.getDownloadSignedUrl(bucketName, fileKey, expiresIn);
|
|
} else {
|
|
bucketName = bucketName || process.env['S3_BUCKET_NAME'] || '';
|
|
return await s3Storage.getDownloadSignedUrl(bucketName, fileKey, expiresIn);
|
|
}
|
|
};
|
|
|
|
export const getUploadSignedUrl = async (
|
|
fileKey: string,
|
|
contentLength: number,
|
|
expiresIn: number,
|
|
bucketName?: string,
|
|
) => {
|
|
const storageType = getStorageType();
|
|
if (storageType === 'r2') {
|
|
bucketName = bucketName || process.env['R2_BUCKET_NAME'] || '';
|
|
return await r2Storage.getUploadSignedUrl(bucketName, fileKey, contentLength, expiresIn);
|
|
} else {
|
|
bucketName = bucketName || process.env['S3_BUCKET_NAME'] || '';
|
|
return await s3Storage.getUploadSignedUrl(bucketName, fileKey, contentLength, expiresIn);
|
|
}
|
|
};
|
|
|
|
export const deleteObject = async (fileKey: string, bucketName?: string) => {
|
|
const storageType = getStorageType();
|
|
if (storageType === 'r2') {
|
|
bucketName = bucketName || process.env['R2_BUCKET_NAME'] || '';
|
|
return await r2Storage.deleteObject(bucketName, fileKey);
|
|
} else {
|
|
bucketName = bucketName || process.env['S3_BUCKET_NAME'] || '';
|
|
return await s3Storage.deleteObject(bucketName, fileKey);
|
|
}
|
|
};
|