Add missing SAS credential support in Blob and Queue extension

devlooped · Jan 20, 2023 · a022406 · a022406
1 parent 7b443fd
commit a022406
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/CloudStorageAccount/BlobAccountExtensions.cs b/src/CloudStorageAccount/BlobAccountExtensions.cs
@@ -1,5 +1,6 @@
 using System;
 using System.ComponentModel;
+using Azure;
 using Azure.Storage;
 using Azure.Storage.Blobs;
 
@@ -33,6 +34,8 @@ public static BlobServiceClient CreateBlobServiceClient(this CloudStorageAccount
                 new StorageSharedKeyCredential(account.Credentials.AccountName, account.Credentials.AccountKey));
         else if (account.Credentials.IsAnonymous)
             return new BlobServiceClient(account.BlobEndpoint);
+        else if (account.Credentials.IsSAS)
+            return new BlobServiceClient(account.BlobEndpoint, new AzureSasCredential(account.Credentials.Signature!));
 
         throw new InvalidOperationException("Account credentials are not supported for Blob client.");
     }

diff --git a/src/CloudStorageAccount/QueueAccountExtensions.cs b/src/CloudStorageAccount/QueueAccountExtensions.cs
@@ -1,5 +1,6 @@
 using System;
 using System.ComponentModel;
+using Azure;
 using Azure.Storage;
 using Azure.Storage.Queues;
 
@@ -33,6 +34,8 @@ public static QueueServiceClient CreateQueueServiceClient(this CloudStorageAccou
                 new StorageSharedKeyCredential(account.Credentials.AccountName, account.Credentials.AccountKey));
         else if (account.Credentials.IsAnonymous)
             return new QueueServiceClient(account.QueueEndpoint);
+        else if (account.Credentials.IsSAS)
+            return new QueueServiceClient(account.QueueEndpoint, new AzureSasCredential(account.Credentials.Signature!));
 
         throw new InvalidOperationException("Account credentials are not supported for Queue client.");
     }