mirror of
https://github.com/skidoodle/albert.lol.git
synced 2025-02-15 06:09:15 +01:00
46 lines
1.1 KiB
TypeScript
46 lines
1.1 KiB
TypeScript
import { NextApiRequest, NextApiResponse } from 'next';
|
|
import aws from 'aws-sdk';
|
|
|
|
const { BUCKET, ACCESS_KEY, SECRET_KEY, ENDPOINT, REGION } = process.env;
|
|
|
|
export default async function Storage(req: NextApiRequest, res: NextApiResponse) {
|
|
aws.config.s3 = {
|
|
accessKeyId: ACCESS_KEY,
|
|
secretAccessKey: SECRET_KEY,
|
|
region: REGION,
|
|
endpoint: ENDPOINT,
|
|
signatureVersion: 'v4',
|
|
};
|
|
|
|
let isTruncated: boolean | undefined = true;
|
|
let startAfter;
|
|
|
|
let objects = 0;
|
|
let size = 0;
|
|
|
|
const s3 = new aws.S3();
|
|
|
|
while (isTruncated) {
|
|
let params: any = { Bucket: BUCKET };
|
|
|
|
if (startAfter) {
|
|
params.StartAfter = startAfter;
|
|
}
|
|
const data = await s3.listObjectsV2(params).promise();
|
|
|
|
data.Contents?.forEach((object: any) => {
|
|
objects++;
|
|
size += object.Size! / 1024 / 1024 / 1024;
|
|
});
|
|
|
|
isTruncated = data.IsTruncated;
|
|
if (isTruncated) {
|
|
startAfter = data.Contents!.slice(-1)[0].Key;
|
|
}
|
|
}
|
|
res.setHeader(
|
|
'Cache-Control',
|
|
'public, s-maxage=10, stale-while-revalidate=59'
|
|
);
|
|
res.json({ object: objects, size: Number(size.toFixed(2)) });
|
|
}
|