module Fog
  module Network
    class AzureRM
      class Real
        def create_virtual_network(name, location, resource_group_name, dns_list, subnet_address_list, network_address_list)
          virtual_network = define_vnet_object(location, name, network_address_list, dns_list, subnet_address_list)
          begin
            promise = @network_client.virtual_networks.create_or_update(resource_group_name, name, virtual_network)
            promise.value!
          rescue  MsRestAzure::AzureOperationError => e
            msg = "Exception creating Virtual Network #{name} in Resource Group: #{resource_group_name}. #{e.body['error']['message']}"
            raise msg
          end
        end

        private

        def define_vnet_object(location, network_name, network_address_list, dns_list, subnet_address_list)
          virtual_network = Azure::ARM::Network::Models::VirtualNetwork.new
          virtual_network.location = location
          virtual_network_properties = Azure::ARM::Network::Models::VirtualNetworkPropertiesFormat.new

          if network_address_list.nil?
            address_space = Azure::ARM::Network::Models::AddressSpace.new
            address_space.address_prefixes = ['10.2.0.0/16']
            virtual_network_properties.address_space = address_space
          else
            network_address_list = network_address_list.split(',')
            na_list = []
            (0...network_address_list.length).each do |i|
              na_list.push(network_address_list[i].strip)
            end
            address_space = Azure::ARM::Network::Models::AddressSpace.new
            address_space.address_prefixes = na_list
            virtual_network_properties.address_space = address_space
          end

          unless dns_list.nil?
            dns_list = dns_list.split(',')
            ns_list = []
            (0...dns_list.length).each do |i|
              ns_list.push(dns_list[i].strip)
            end
            dhcp_options = Azure::ARM::Network::Models::DhcpOptions.new
            dhcp_options.dns_servers = ns_list unless ns_list.nil?
            virtual_network_properties.dhcp_options = dhcp_options
          end

          unless subnet_address_list.nil?
            subnet_address_list = subnet_address_list.split(',')
            sub_nets = define_subnet_objects(network_name, subnet_address_list)
            virtual_network_properties.subnets = sub_nets
          end

          virtual_network.properties = virtual_network_properties
          virtual_network
        end

        def define_subnet_objects(network_name, subnet_address_list)
          sub_nets = []
          (0...subnet_address_list.length).each do |i|
            subnet_properties = Azure::ARM::Network::Models::SubnetPropertiesFormat.new
            subnet_properties.address_prefix = subnet_address_list[i].strip

            subnet = Azure::ARM::Network::Models::Subnet.new
            subnet.name = 'subnet_' + i.to_s + '_' + network_name
            subnet.properties = subnet_properties
            sub_nets.push(subnet)
          end
          sub_nets
        end
      end

      class Mock
        def create_virtual_network(_name, _location, _dns_list, _subnet_address_list, _network_address_list, _resource_group_name)
        end
      end
    end
  end
end