Support for other cloud regions

Author: JoeDupuis

CHANGELOG.md

@@ -3,6 +3,7 @@
 - Add get_container_properties
 - Add create_container
 - Add delete_container
+- Support for Azure China, US Gov and Germany
 
 ## [0.5.1] 2024-09-09
 

lib/azure_blob/client.rb

@@ -17,10 +17,11 @@ class Client
     def initialize(account_name:, access_key:, container:, **options)
       @account_name = account_name
       @container = container
+      @cloud_regions = options[:cloud_regions]&.to_sym || :global
 
       @signer = !access_key.nil? && !access_key.empty?  ?
         AzureBlob::SharedKeySigner.new(account_name:, access_key:) :
-        AzureBlob::EntraIdSigner.new(account_name:, **options.slice(:principal_id))
+        AzureBlob::EntraIdSigner.new(account_name:, host:, **options.slice(:principal_id))
     end
 
     # Create a blob of type block. Will automatically split the the blob in multiple block and send the blob in pieces (blocks) if the blob is too big.
@@ -331,10 +332,10 @@ def put_blob_single(key, content, options = {})
       Http.new(uri, headers, metadata: options[:metadata], signer:).put(content.read)
     end
 
-    attr_reader :account_name, :signer, :container, :http
-
     def host
-      "https://#{account_name}.blob.core.windows.net"
+      @host ||= "https://#{account_name}.blob.#{CLOUD_REGIONS_SUFFIX[cloud_regions]}"
     end
+
+    attr_reader :account_name, :signer, :container, :http, :cloud_regions
   end
 end

lib/azure_blob/const.rb

@@ -5,4 +5,10 @@ module AzureBlob
   MAX_UPLOAD_SIZE = 256 * 1024 * 1024 # 256 Megabytes
   DEFAULT_BLOCK_SIZE = 128 * 1024 * 1024 # 128 Megabytes
   BLOB_SERVICE = "b"
+  CLOUD_REGIONS_SUFFIX = {
+    global: "core.windows.net",
+    cn: "core.chinacloudapi.cn",
+    de: "core.cloudapi.de",
+    usgovt: "core.usgovcloudapi.net",
+  }
 end

lib/azure_blob/entra_id_signer.rb

@@ -13,9 +13,10 @@ class EntraIdSigner # :nodoc:
     attr_reader :token
     attr_reader :account_name
 
-    def initialize(account_name:, principal_id: nil)
+    def initialize(account_name:, host:, principal_id: nil)
       @token = AzureBlob::IdentityToken.new(principal_id:)
       @account_name = account_name
+      @host = host
     end
 
     def authorization_header(uri:, verb:, headers: {})

lib/azure_blob/user_delegation_key.rb

@@ -6,7 +6,7 @@ class UserDelegationKey # :nodoc:
     EXPIRATION_BUFFER = 3600 # 1 hours
     def initialize(account_name:, signer:)
       @uri = URI.parse(
-        "https://#{account_name}.blob.core.windows.net/?restype=service&comp=userdelegationkey"
+        "#{signer.host}/?restype=service&comp=userdelegationkey"
       )
 
       @signer = signer