add storage module

Author: daniwebdev

.gitignore

@@ -36,4 +36,6 @@ lerna-debug.log*
 
 
 # Adding for this project
-.env
+.env
+
+src/service-account.json

package.json

@@ -21,6 +21,7 @@
     "nest": "nest"
   },
   "dependencies": {
+    "@google-cloud/storage": "^6.12.0",
     "@nestjs-modules/mailer": "^1.9.1",
     "@nestjs/cache-manager": "^2.1.0",
     "@nestjs/common": "^10.0.0",

src/apps/app.module.ts

@@ -1,5 +1,5 @@
 import { Module } from '@nestjs/common';
-import { ConfigModule, ConfigService } from '@nestjs/config';
+import { ConfigModule } from '@nestjs/config';
 import { JwtModule } from '@nestjs/jwt';
 import { TypeOrmModule } from '@nestjs/typeorm';
 import { defaultTypeOrmConfig } from 'src/config/typeorm.config';

src/modules/storage/providers/gcs.provider.ts

@@ -0,0 +1,47 @@
+import { join } from "path";
+import { Bucket, StorageOptions, Storage as GCS } from "@google-cloud/storage";
+import { ProviderStorage } from "./provider.abstract";
+
+export class GoogleCloudStorage implements ProviderStorage<Bucket> {
+
+    private _instance: Bucket = null;
+
+    constructor(
+        public storageOption: StorageOptions,
+        public config: {
+            bucketName: string,
+            prefix: string
+        }
+    ) {
+    
+    }
+
+    instance(): Bucket {
+        if(this._instance == null) {
+            this._instance = new GCS(this.storageOption).bucket('tradeapp-f8e74.appspot.com');
+        }
+
+        return this._instance;
+    }
+
+    async put(destination: string, buffer: Buffer, mimetype?: string) {
+
+        const file = this.instance().file(destination);
+
+        await file.save(buffer, {
+            gzip: true,
+        });
+
+        return file;
+    }
+
+    async move(origin:string, destination: string) {
+        console.log(`${this.config.prefix}/${origin}`)
+        console.log(`${this.config.prefix}/${destination}`)
+
+        await this.instance().file(`${this.config.prefix}/${origin}`).move(`${this.config.prefix}/${destination}`);
+
+        return destination;
+    }
+
+}

src/modules/storage/providers/provider.abstract.ts

@@ -0,0 +1,5 @@
+export abstract class ProviderStorage<Instance> {
+    abstract put(destination: string, buffer: Buffer, mimetype?: string): Promise<any>;
+    abstract move(origin:string, destination: string): Promise<any>
+    abstract instance(): Instance | any
+}

src/modules/storage/storage.module.ts

@@ -0,0 +1,12 @@
+import { Module } from '@nestjs/common';
+import { ConfigModule } from '@nestjs/config';
+import { StorageService } from './storage.service';
+
+@Module({
+  imports: [
+    ConfigModule
+  ],
+  providers: [StorageService],
+  exports: [StorageService]
+})
+export class StorageModule {}

src/modules/storage/storage.service.ts

@@ -0,0 +1,68 @@
+import { Injectable } from '@nestjs/common';
+import { ConfigService } from '@nestjs/config';
+import { join } from 'path';
+import { GoogleCloudStorage } from './providers/gcs.provider';
+import { ProviderStorage } from './providers/provider.abstract';
+
+@Injectable()
+export class StorageService {
+
+    private environment = 'development';
+
+    private providerInstance: ProviderStorage<any>;
+
+    private domainURL: string;
+    private provider: string;
+    private prefixToReplace: any;
+
+    constructor(
+        private configService: ConfigService,
+    ) {
+        this.environment = this.configService.get('STORAGE_UPLOAD_PREFIX');
+
+        this.prefixToReplace = {
+            'gcs': 'gcs://',
+            's3': 's3://',
+            'cloudinary': 'cld://'
+        }
+    }
+
+    setProvider(provider: 'gcs') {
+        this.provider = provider;
+
+        switch(provider) {
+
+            case 'gcs': {
+                this.domainURL = this.configService.get('GCS_DOMAIN_URL') ?? 'http://___domain.com';
+                this.providerInstance = new GoogleCloudStorage({
+                    keyFile: join(__dirname, '..', '..', 'service-account.json'),
+                    projectId: this.configService.get('GCS_PROJECT_ID') ?? 'intervest-io',
+                }, { bucketName: this.configService.get('GCS_BUCKET_NAME') ??'intervest-io.appspot.com', prefix: this.environment })
+            }
+
+            default: {
+                throw Error('Storage Provider not available.')
+            }
+        }
+
+        return this;
+    }
+
+    async put(destination: string, buffer: Buffer, mimetype?: string) {
+
+        return await this.providerInstance.put(destination, buffer, mimetype)
+
+    }
+
+    async move(origin:string, destination: string) {
+
+        return  await this.providerInstance.move(origin, destination);
+    }
+
+    url(path: string) {
+        const currentPath = path.replace(this.prefixToReplace[this.provider], `${this.domainURL}/`);
+
+        return currentPath;
+    }
+    
+}