import { BlobServiceClient, Metadata } from '@azure/storage-blob';
import { v4 as uuidv4 } from 'uuid';
import { DefaultAzureCredential } from '@azure/identity';
import fs from 'fs';
import magic from 'stream-mmmagic';
import { resolve } from 'path';

export const upload = async ({
  containerName,
  fileName,
  filePath,
  useConnectionString,
  connectionString,
  accountName,
  metaData,
}: {
  containerName: string;
  fileName?: string;
  filePath: string;
  useConnectionString: boolean;
  connectionString?: string;
  accountName: string;
  metaData?: Metadata;
}) => {
  // get where the file is located
  filePath = resolve(filePath);
  console.log(filePath);
  console.log('upload started...');

  try {
    if (useConnectionString) {
      if (!connectionString) throw new Error('connectionString is required');
      const blobServiceClient = BlobServiceClient.fromConnectionString(
        connectionString
      );
      const containerClient = blobServiceClient.getContainerClient(
        containerName
      );
      if (containerClient.accountName !== accountName)
        throw new Error('Storage account name does not match');
      const containerExists = await containerClient.exists();
      if (!containerExists) await containerClient.create({ access: 'blob' });
      fileName ? fileName : (fileName = uuidv4());
      const blockBlobClient = containerClient.getBlockBlobClient(fileName);
      const file = fs.readFileSync(filePath);
      const input = fs.createReadStream(filePath);
      const [mime] = await magic.promise(input);
      console.log('uploading...');
      const uploadBlobResponse = await blockBlobClient.upload(
        file,
        file.length,
        {
          blobHTTPHeaders: {
            blobContentType: JSON.parse(JSON.stringify(mime)).type,
            blobContentEncoding: JSON.parse(JSON.stringify(mime)).encoding,
          },
          metadata: metaData,
        }
      );
      console.log('file uploaded!');
      const url = `${containerClient.url}/${fileName}`;
      return {
        data: { mime, url },
        success: true,
        message: 'File uploaded successfully',
        response: uploadBlobResponse,
      };
    } else {
      const sharedKeyCredential = new DefaultAzureCredential();
      const blobServiceClient = new BlobServiceClient(
        `https://${accountName}.blob.core.windows.net`,
        sharedKeyCredential
      );
      const containerClient = blobServiceClient.getContainerClient(
        containerName
      );
      const containerExists = await containerClient.exists();
      if (!containerExists) await containerClient.create({ access: 'blob' });
      fileName ? fileName : (fileName = uuidv4());
      const blockBlobClient = containerClient.getBlockBlobClient(fileName);
      const file = fs.readFileSync(filePath);
      const input = fs.createReadStream(filePath);
      const [mime] = await magic.promise(input);
      console.log('uploading...');
      const uploadBlobResponse = await blockBlobClient.upload(
        file,
        file.length,
        {
          blobHTTPHeaders: {
            blobContentType: JSON.parse(JSON.stringify(mime)).type,
            blobContentEncoding: JSON.parse(JSON.stringify(mime)).encoding,
          },
          metadata: metaData,
        }
      );
      const url = `${containerClient.url}/${fileName}`;
      console.log('file uploaded!');
      return {
        data: { mime, url },
        success: true,
        message: 'File uploaded successfully',
        response: uploadBlobResponse,
      };
    }
  } catch (err) {
    const error = err as Error;
    console.log('upload failed!');
    console.log(error.message);
    return { success: false, message: error.message };
  }
};