From d4535da73341dc28e7782bf8681846a33e764dde Mon Sep 17 00:00:00 2001
From: Camila Belo <camilaibs@gmail.com>
Date: Fri, 9 Jul 2021 17:30:16 +0200
Subject: [PATCH] Initial Azure migrate method

Signed-off-by: Camila Belo <camilaibs@gmail.com>
---
 .../src/stages/publish/azureBlobStorage.ts    | 62 +++++++++++++++++++
 1 file changed, 62 insertions(+)

diff --git a/packages/techdocs-common/src/stages/publish/azureBlobStorage.ts b/packages/techdocs-common/src/stages/publish/azureBlobStorage.ts
index 3ed75f532e..ddc70d5b46 100644
--- a/packages/techdocs-common/src/stages/publish/azureBlobStorage.ts
+++ b/packages/techdocs-common/src/stages/publish/azureBlobStorage.ts
@@ -289,4 +289,66 @@ export class AzureBlobStoragePublish implements PublisherBase {
       .getBlockBlobClient(`${entityRootDir}/index.html`)
       .exists();
   }
+
+  protected toLowerCase(originalName: string): string {
+    const [namespace, kind, name, ...parts] = originalName.split('/');
+    const lowerNamespace = namespace.toLowerCase();
+    const lowerKind = kind.toLowerCase();
+    const lowerName = name.toLowerCase();
+    return [lowerNamespace, lowerKind, lowerName, ...parts].join('/');
+  }
+
+  protected async renameBlob(
+    originalName: string,
+    newName: string,
+    removeOriginal = false,
+  ): Promise<void> {
+    const container = this.storageClient.getContainerClient(this.containerName);
+    const blob = container.getBlobClient(newName);
+    const { url } = container.getBlobClient(originalName);
+    const response = await blob.beginCopyFromURL(url);
+    await response.pollUntilDone();
+    if (removeOriginal) {
+      await container.deleteBlob(originalName);
+    }
+  }
+
+  protected async renameBlobToLowerCase(
+    originalName: string,
+    removeOriginal: boolean,
+  ) {
+    const newName = this.toLowerCase(originalName);
+    if (originalName === newName) return;
+    try {
+      this.logger.debug(`Migrating ${originalName}`);
+      await this.renameBlob(originalName, newName, removeOriginal);
+    } catch (e) {
+      this.logger.warn(`Unable to migrate ${originalName}: ${e.message}`);
+    }
+  }
+
+  /**
+   * todo: Put this documentation somewhere the user will actually see and read
+   * it.
+   */
+  async migrateDocsCase({
+    removeOriginal = false,
+    concurrency = 25,
+  }): Promise<void> {
+    const promises = [];
+    const limiter = limiterFactory(concurrency);
+    const container = this.storageClient.getContainerClient(this.containerName);
+
+    for await (const blob of container.listBlobsFlat()) {
+      promises.push(
+        limiter(
+          this.renameBlobToLowerCase.bind(this),
+          blob.name,
+          removeOriginal,
+        ),
+      );
+    }
+
+    await Promise.all(promises);
+  }
 }