lib/fog/azurerm/resources.rb in fog-azure-rm-0.1.2 vs lib/fog/azurerm/resources.rb in fog-azure-rm-0.2.0

@@ -4,23 +4,27 @@
     class AzureRM < Fog::Service
       requires :tenant_id
       requires :client_id
       requires :client_secret
       requires :subscription_id
+      recognizes :environment
 
       request_path 'fog/azurerm/requests/resources'
       request :create_resource_group
       request :list_resource_groups
       request :delete_resource_group
       request :get_resource_group
+      request :check_resource_group_exists
       request :create_deployment
       request :delete_deployment
       request :list_deployments
       request :get_deployment
+      request :check_deployment_exists
       request :delete_resource_tag
       request :list_tagged_resources
       request :tag_resource
+      request :check_azure_resource_exists
 
       model_path 'fog/azurerm/models/resources'
       model :resource_group
       collection :resource_groups
       model :deployment
@@ -51,12 +55,16 @@
           rescue LoadError => e
             retry if require('rubygems')
             raise e.message
           end
 
-          credentials = Fog::Credentials::AzureRM.get_credentials(options[:tenant_id], options[:client_id], options[:client_secret])
-          @rmc = ::Azure::ARM::Resources::ResourceManagementClient.new(credentials, resource_manager_endpoint_url)
+          options[:environment] = 'AzureCloud' if options[:environment].nil?
+
+          credentials = Fog::Credentials::AzureRM.get_credentials(options[:tenant_id], options[:client_id], options[:client_secret], options[:environment])
+          telemetry = "fog-azure-rm/#{Fog::AzureRM::VERSION}"
+          @rmc = ::Azure::ARM::Resources::ResourceManagementClient.new(credentials, resource_manager_endpoint_url(options[:environment]))
           @rmc.subscription_id = options[:subscription_id]
+          @rmc.add_user_agent_information(telemetry)
         end
       end
     end
   end
 end