feat: support providing endpoint for output S3 (#235)

Author: birme

src/TranscribeService/TranscribeService.ts

@@ -133,11 +133,13 @@ export class TranscribeService {
     format,
     bucket,
     key,
-    region
+    region,
+    endpoint
   }: TTranscribeRemoteFile & {
     bucket: string;
     key: string;
     region?: string;
+    endpoint?: string;
   }): Promise<void> {
     try {
       this.workerState = State.ACTIVE;
@@ -155,6 +157,7 @@ export class TranscribeService {
         key,
         format,
         region,
+        endpoint,
         content: JSON.stringify(resp)
       });
       // TODO: Find a way to not be dependent on the need to download the file locally

src/api.ts

@@ -157,6 +157,9 @@ const transcribeS3: FastifyPluginCallback<Options> = (fastify, _opts, next) => {
             },
             region: {
               type: 'string'
+            },
+            endpoint: {
+              type: 'string'
             }
           },
           required: ['url', 'bucket', 'key']

src/aws/upload.ts

@@ -38,7 +38,8 @@ export async function uploadToS3({
   bucket,
   key,
   format,
-  region
+  region,
+  endpoint
 }: TUploadToS3): Promise<void> {
   let fileFormat = 'txt';
   if (['json', 'verbose_json'].includes(format)) {
@@ -48,7 +49,12 @@ export async function uploadToS3({
   } else if (format === 'vtt') {
     fileFormat = 'vtt';
   }
-  const client = new S3Client({ region: region ?? process.env.AWS_REGION });
+  const customEndpoint = endpoint ?? process.env.AWS_S3_ENDPOINT;
+  const client = new S3Client({
+    region: region ?? process.env.AWS_REGION,
+    forcePathStyle: customEndpoint ? true : false,
+    endpoint: customEndpoint
+  });
   const upload = new Upload({
     client,
     params: { Bucket: bucket, Key: `${key}.${fileFormat}`, Body: content }