noahspan-flying/api/src/file/file.service.ts

import { BlobServiceClient, BlockBlobClient } from '@azure/storage-blob';
import { Injectable } from '@nestjs/common';
import { ConfigService } from '@nestjs/config';

@Injectable() export class FileService {
  constructor(private readonly configService: ConfigService) {}

  private containerName: string;

  private streamToBuffer(readableStream: NodeJS.ReadableStream) {
    return new Promise<Buffer>((resolve, reject) => {
      const chunks = [];

      readableStream.on('data', (data) => {
        chunks.push(data instanceof Buffer ? data : Buffer.from(data));
      });
      readableStream.on('end', () => {
        resolve(Buffer.concat(chunks));
      });
      readableStream.on('error', reject)
    })
  }

  async getBlobServiceInstance() {
    const connectionString = this.configService.get<string>('azureStorageConnectionString');
    const blobServiceClient: BlobServiceClient = await BlobServiceClient.fromConnectionString(connectionString)

    return blobServiceClient;
  }

  async getBlobClient(fileName: string): Promise<BlockBlobClient> {
    const blobService = await this.getBlobServiceInstance();
    const containerName = this.containerName;
    const containerClient = blobService.getContainerClient(containerName);
    const blockBlobClient = containerClient.getBlockBlobClient(fileName);

    return blockBlobClient;
  }

  async uploadFile(file: Express.Multer.File, containerName: string, rowKey: string): Promise<string> {
    this.containerName = containerName;

    const blockBlobClient = await this.getBlobClient(`${rowKey}/${file.originalname}`);
    const fileUrl = blockBlobClient.url;
    
    await blockBlobClient.uploadData(file.buffer);

    return fileUrl;
  }

  async downloadFile(containerName: string, rowKey: string, fileName: string): Promise<string> {
    this.containerName = containerName;

    const blockBlobClient = await this.getBlobClient(`${rowKey}/${fileName}`);
    const downloadBlockBlobResponse = await blockBlobClient.download();
    const downloaded: string = (await this.streamToBuffer(downloadBlockBlobResponse.readableStreamBody)).toString()
    
    return downloaded
  }

  async deleteFile(containerName: string, rowKey:string, fileName: string): Promise<void> {
      this.containerName = containerName;

      const blockBlobClient = await this.getBlobClient(`${rowKey}/${fileName}`);

      await blockBlobClient.deleteIfExists();
  }
}