require 'fog/azurerm/core'

module Fog
  module Network
    # Fog Service Class for AzureRM
    class AzureRM < Fog::Service
      requires :tenant_id
      requires :client_id
      requires :client_secret
      requires :subscription_id

      request_path 'fog/azurerm/requests/network'
      request :create_virtual_network
      request :delete_virtual_network
      request :list_virtual_networks
      request :check_for_virtual_network
      request :create_public_ip
      request :delete_public_ip
      request :list_public_ips
      request :check_for_public_ip
      request :create_subnet
      request :list_subnets
      request :delete_subnet
      request :create_network_interface
      request :delete_network_interface
      request :list_network_interfaces
      request :create_load_balancer
      request :delete_load_balancer
      request :list_load_balancers
      request :create_network_security_group
      request :delete_network_security_group
      request :list_network_security_groups
      request :create_application_gateway
      request :delete_application_gateway
      request :list_application_gateways
      request :create_traffic_manager_profile
      request :get_traffic_manager_profile
      request :delete_traffic_manager_profile
      request :list_traffic_manager_profiles
      request :create_traffic_manager_endpoint
      request :delete_traffic_manager_endpoint

      model_path 'fog/azurerm/models/network'
      model :virtual_network
      collection :virtual_networks
      model :public_ip
      collection :public_ips
      model :subnet
      collection :subnets
      model :network_interface
      collection :network_interfaces
      model :load_balancer
      collection :load_balancers
      model :frontend_ip_configuration
      model :inbound_nat_pool
      model :inbound_nat_rule
      model :load_balancing_rule
      model :probe
      model :network_security_group
      collection :network_security_groups
      model :network_security_rule
      model :application_gateway
      collection :application_gateways
      model :application_gateway_backend_address_pool
      model :application_gateway_backend_http_setting
      model :application_gateway_frontend_ip_configuration
      model :application_gateway_frontend_port
      model :application_gateway_ip_configuration
      model :application_gateway_http_listener
      model :path_rule
      model :application_gateway_probe
      model :application_gateway_request_routing_rule
      model :application_gateway_ssl_certificate
      model :application_gateway_url_path_map
      model :traffic_manager_profile
      collection :traffic_manager_profiles
      model :traffic_manager_end_point
      collection :traffic_manager_end_points

      # Mock class for Network Service
      class Mock
        def initialize(_options = {})
          begin
            require 'azure_mgmt_network'
          rescue LoadError => e
            retry if require('rubygems')
            raise e.message
          end
        end
      end

      # Real class for Network Service
      class Real
        def initialize(options)
          begin
            require 'azure_mgmt_network'
          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])
          @network_client = ::Azure::ARM::Network::NetworkManagementClient.new(credentials)
          @network_client.subscription_id = options[:subscription_id]
          @tenant_id = options[:tenant_id]
          @client_id = options[:client_id]
          @client_secret = options[:client_secret]
          @subscription_id = options[:subscription_id]
        end
      end
    end
  end
end