Huang Xin
61 lines
1.4 KiB
TypeScript
61 lines
1.4 KiB
TypeScript
import { AwsClient } from 'aws4fetch';
|
|
|
|
const getR2Client = () => {
|
|
return new AwsClient({
|
|
service: 's3',
|
|
region: process.env['R2_REGION'] || 'auto',
|
|
accessKeyId: process.env['R2_ACCESS_KEY_ID']!,
|
|
secretAccessKey: process.env['R2_SECRET_ACCESS_KEY']!,
|
|
});
|
|
};
|
|
|
|
const getR2Url = () => {
|
|
const R2_ACCOUNT_ID = process.env['R2_ACCOUNT_ID']!;
|
|
return `https://${R2_ACCOUNT_ID}.r2.cloudflarestorage.com`;
|
|
};
|
|
|
|
export const getDownloadSignedUrl = async (
|
|
bucketName: string,
|
|
fileKey: string,
|
|
expiresIn: number,
|
|
) => {
|
|
return (
|
|
await getR2Client().sign(
|
|
new Request(`${getR2Url()}/${bucketName}/${fileKey}?X-Amz-Expires=${expiresIn}`),
|
|
{
|
|
aws: { signQuery: true },
|
|
},
|
|
)
|
|
).url.toString();
|
|
};
|
|
|
|
export const getUploadSignedUrl = async (
|
|
bucketName: string,
|
|
fileKey: string,
|
|
contentLength: number,
|
|
expiresIn: number,
|
|
) => {
|
|
return (
|
|
await getR2Client().sign(
|
|
new Request(
|
|
`${getR2Url()}/${bucketName}/${fileKey}?X-Amz-Expires=${expiresIn}&X-Amz-SignedHeaders=content-length`,
|
|
{
|
|
method: 'PUT',
|
|
headers: {
|
|
'Content-Length': contentLength.toString(),
|
|
},
|
|
},
|
|
),
|
|
{
|
|
aws: { signQuery: true },
|
|
},
|
|
)
|
|
).url.toString();
|
|
};
|
|
|
|
export const deleteObject = async (bucketName: string, fileKey: string) => {
|
|
return await getR2Client().fetch(`${getR2Url()}/${bucketName}/${fileKey}`, {
|
|
method: 'DELETE',
|
|
});
|
|
};
|