prime-sdk 3.0.9-beta.0 → 3.1.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "prime-sdk",
-  "version": "3.0.9-beta.0",
+  "version": "3.1.0",
   "description": "swiss army knife",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/external/aws/s3/s3.service.ts

@@ -24,6 +24,9 @@ import {
   ListFiles,
   UploadFile,
 } from '../../../lib/storage/types/storage.type';
+import { Readable } from 'stream';
+
+const PRESIGNED_PUT_EXPIRES_IN_SECONDS = 3600;
 
 @Injectable()
 export class S3 implements IStorage {
@@ -34,6 +37,7 @@ export class S3 implements IStorage {
       Bucket: file.bucket,
       Key: `${file.path}/${file.nameFile}`,
       Body: file.file,
+      ...(file.contentType && { ContentType: file.contentType }),
       ...(file.metadata && { Metadata: file.metadata }),
       ...(file.tagging && {
         Tagging: file.tagging,
@@ -45,6 +49,72 @@ export class S3 implements IStorage {
     return response;
   }
 
+  async uploadFileViaPresignedUrl(file: UploadFile): Promise<void> {
+    const key = `${file.path}/${file.nameFile}`;
+    const contentType = file.contentType || 'application/octet-stream';
+    const body = await this.resolveFileBody(file.file);
+
+    const command = new PutObjectCommand({
+      Bucket: file.bucket,
+      Key: key,
+      ContentType: contentType,
+      ...(file.metadata && { Metadata: file.metadata }),
+      ...(file.tagging && { Tagging: file.tagging }),
+    });
+
+    const presignedUrl = await getSignedUrl(this.client, command, {
+      expiresIn: PRESIGNED_PUT_EXPIRES_IN_SECONDS,
+    });
+
+    const headers: Record<string, string> = {
+      'Content-Type': contentType,
+    };
+
+    if (file.metadata) {
+      for (const [k, v] of Object.entries(file.metadata)) {
+        headers[`x-amz-meta-${k}`] = v;
+      }
+    }
+
+    if (file.tagging) {
+      headers['x-amz-tagging'] = file.tagging;
+    }
+
+    const response = await fetch(presignedUrl, {
+      method: 'PUT',
+      headers,
+      body: new Uint8Array(body),
+    });
+
+    if (!response.ok) {
+      throw new Error(
+        `s3_presigned_upload_failed: ${response.status} ${response.statusText}`,
+      );
+    }
+  }
+
+  private async resolveFileBody(
+    file: string | Readable | Buffer,
+  ): Promise<Buffer> {
+    if (Buffer.isBuffer(file)) {
+      return file;
+    }
+
+    if (typeof file === 'string') {
+      return Buffer.from(file);
+    }
+
+    const chunks: Buffer[] = [];
+
+    return new Promise((resolve, reject) => {
+      file.on('data', (chunk) =>
+        chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk)),
+      );
+      file.on('end', () => resolve(Buffer.concat(chunks)));
+      file.on('error', reject);
+    });
+  }
+
   async generateSignUrlFile(file: GetFile): Promise<string> {
     const params = {
       Bucket: file.bucket,

package/src/lib/storage/interfaces/storage.interface.ts

@@ -17,6 +17,7 @@ import {
 
 export interface IStorage {
   uploadFile(file: UploadFile): Promise<PutObjectCommandOutput>;
+  uploadFileViaPresignedUrl(file: UploadFile): Promise<void>;
   generateSignUrlFile(file: GetFile): Promise<string>;
   getFile(file: GetFile): Promise<GetObjectCommandOutput>;
   listFiles(data: ListFiles): Promise<ListObjectsV2CommandOutput>;

package/src/lib/storage/storage.ts

@@ -26,6 +26,10 @@ export class Storage {
     return this.storage.uploadFile(file);
   }
 
+  async uploadFileViaPresignedUrl(file: UploadFile): Promise<void> {
+    return this.storage.uploadFileViaPresignedUrl(file);
+  }
+
   async generateSignUrlFile(file: GetFile): Promise<string> {
     return this.storage.generateSignUrlFile(file);
   }

package/src/lib/storage/types/storage.type.ts

@@ -5,6 +5,7 @@ export type UploadFile = {
   nameFile: string;
   file: string | Readable | Buffer;
   bucket: string;
+  contentType?: string;
   metadata?: Record<string, string>;
   tagging?: string;
 };

package/.yarnrc.yml

@@ -1 +0,0 @@
-nodeLinker: node-modules