feat(storage): auto-create S3 bucket on module init

This commit is contained in:
Lizandro Guarnizo
2026-06-04 21:33:54 -05:00
parent fff5e198b7
commit 8f2de37c0d
+23 -3
View File
@@ -1,9 +1,9 @@
import { Injectable, Logger } from "@nestjs/common";
import { Injectable, Logger, OnModuleInit } from "@nestjs/common";
import { ConfigService } from "@nestjs/config";
import * as fs from "fs";
import * as path from "path";
import { randomUUID } from "crypto";
import { S3Client, DeleteObjectCommand, PutObjectCommand } from "@aws-sdk/client-s3";
import { S3Client, DeleteObjectCommand, PutObjectCommand, CreateBucketCommand, HeadBucketCommand } from "@aws-sdk/client-s3";
import { getSignedUrl } from "@aws-sdk/s3-request-presigner";
import { GetObjectCommand } from "@aws-sdk/client-s3";
@@ -13,7 +13,7 @@ import { GetObjectCommand } from "@aws-sdk/client-s3";
* • Otherwise → local disk under ./uploads/ (development)
*/
@Injectable()
export class StorageService {
export class StorageService implements OnModuleInit {
private readonly logger = new Logger(StorageService.name);
private readonly uploadDir: string;
private readonly baseUrl: string;
@@ -50,6 +50,26 @@ export class StorageService {
}
}
async onModuleInit(): Promise<void> {
if (this.s3 && this.bucket) {
await this.ensureBucket();
}
}
private async ensureBucket(): Promise<void> {
try {
await this.s3!.send(new HeadBucketCommand({ Bucket: this.bucket! }));
this.logger.log(`[STORAGE] Bucket "${this.bucket}" exists`);
} catch (e: any) {
if (e.name === "NotFound" || e.$metadata?.httpStatusCode === 404) {
await this.s3!.send(new CreateBucketCommand({ Bucket: this.bucket! }));
this.logger.log(`[STORAGE] Created bucket "${this.bucket}"`);
} else {
this.logger.error(`[STORAGE] Bucket check failed: ${e.message}`);
}
}
}
get isS3Mode(): boolean { return !!this.s3; }
/**