Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| e115edd79e |
20
src/sss.ts
20
src/sss.ts
@@ -1,4 +1,4 @@
|
|||||||
import { S3Client } from "@aws-sdk/client-s3";
|
import { S3Client, CreateBucketCommand, HeadBucketCommand, S3ServiceException } from "@aws-sdk/client-s3";
|
||||||
import { Upload } from "@aws-sdk/lib-storage";
|
import { Upload } from "@aws-sdk/lib-storage";
|
||||||
import { createWriteStream } from "node:fs";
|
import { createWriteStream } from "node:fs";
|
||||||
import { Readable } from "node:stream";
|
import { Readable } from "node:stream";
|
||||||
@@ -24,7 +24,8 @@ export async function processAndStore(
|
|||||||
log("DEBUG", "Missing required variables for S3 Support - EXIT CODE 24");
|
log("DEBUG", "Missing required variables for S3 Support - EXIT CODE 24");
|
||||||
process.exit(24);
|
process.exit(24);
|
||||||
}
|
}
|
||||||
log('INFO', `Streaming to S3: ${cfg.S3_BUCKET}/${filename}`);
|
|
||||||
|
log("DEBUG", "Opening connection to S3 Server");
|
||||||
|
|
||||||
const client = new S3Client({
|
const client = new S3Client({
|
||||||
endpoint: cfg.S3_ENDPOINT,
|
endpoint: cfg.S3_ENDPOINT,
|
||||||
@@ -36,6 +37,21 @@ export async function processAndStore(
|
|||||||
forcePathStyle: true,
|
forcePathStyle: true,
|
||||||
});
|
});
|
||||||
|
|
||||||
|
// Check bucket exists, and create if needed
|
||||||
|
try {
|
||||||
|
await client.send(new HeadBucketCommand({ Bucket: cfg.S3_BUCKET }));
|
||||||
|
} catch (err) {
|
||||||
|
if (err instanceof S3ServiceException && err.$metadata.httpStatusCode === 404) {
|
||||||
|
log('INFO', `Bucket ${cfg.S3_BUCKET} does not exist, creating...`);
|
||||||
|
await client.send(new CreateBucketCommand({ Bucket: cfg.S3_BUCKET }));
|
||||||
|
} else {
|
||||||
|
log(`ERROR`, `Failed to verify bucket: ${cfg.S3_BUCKET}`);
|
||||||
|
process.exit(21);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
log('INFO', `Streaming to S3: ${cfg.S3_BUCKET}/${filename}`);
|
||||||
|
|
||||||
const upload = new Upload({
|
const upload = new Upload({
|
||||||
client,
|
client,
|
||||||
params: {
|
params: {
|
||||||
|
|||||||
Reference in New Issue
Block a user