azurerm.rb in fog-azure-rm-0.0.6

- old
+ new
@@ -2,43 +2,45 @@
 require 'rest_client'
 require 'erb'
 require 'fog/azurerm/config'
 require 'fog/azurerm/utilities/general'
 require 'fog/azurerm/version'
-require 'fog/azurerm/core'
-require 'fog/azurerm/dns'
-require 'fog/azurerm/fog_azure_rm_exception'
-require 'fog/azurerm/resources'
-require 'fog/azurerm/storage'
-require 'fog/azurerm/network'
-require 'fog/azurerm/compute'
-require 'fog/azurerm/application_gateway'
-require 'fog/azurerm/traffic_manager'
+require 'fog/core'
 
 module Fog
+  module Credentials
+    autoload :AzureRM, File.expand_path('azurerm/credentials', __dir__)
+  end
+  module Compute
+    autoload :AzureRM, File.expand_path('azurerm/compute', __dir__)
+  end
+  module DNS
+    autoload :AzureRM, File.expand_path('azurerm/dns', __dir__)
+  end
+  module Network
+    autoload :AzureRM, File.expand_path('azurerm/network', __dir__)
+  end
+  module Resources
+    autoload :AzureRM, File.expand_path('azurerm/resources', __dir__)
+  end
+  module TrafficManager
+    autoload :AzureRM, File.expand_path('azurerm/traffic_manager', __dir__)
+  end
+  module Storage
+    autoload :AzureRM, File.expand_path('azurerm/storage', __dir__)
+  end
+  module ApplicationGateway
+    autoload :AzureRM, File.expand_path('azurerm/application_gateway', __dir__)
+  end
+
   # Main AzureRM fog Provider Module
   module AzureRM
-    def self.services
-      begin
-        array_of_services = []
-        ENV['BUNDLE_GEM'] = File.expand_path('../../lib', File.dirname(__FILE__))
-        gem_path = ENV['BUNDLE_GEM']
-        files = Dir.entries(File.join(gem_path, '/fog/azurerm')).select { |f| !File.directory? f }
-        files.each do |file|
-          next if file == 'config.rb'
-          next if file == 'core.rb'
-          next if file == 'credentials.rb'
-          next if file == 'docs'
-          next if file == 'models'
-          next if file == 'requests'
-          next if file == 'version.rb'
-          array_of_services.push(file.split('.').first.upcase)
-        end
-        array_of_services
-      rescue => e
-        Fog::Logger.warning(e.message)
-        raise e.message
-        # typically occurs if folder_to_count does not exist
-      end
-    end
+    extend Fog::Provider
+    service(:resources, 'Resources')
+    service(:dns, 'DNS')
+    service(:storage, 'Storage')
+    service(:network, 'Network')
+    service(:compute, 'Compute')
+    service(:application_gateway, 'ApplicationGateway')
+    service(:traffic_manager, 'TrafficManager')
   end
 end