require 'azure_backend'
require 'uri'

# Class to test the data disks that are attached to the specified VM
#
# @author Russell Seymour
#
# @attr_reader [Array<Hash>] params Array of hashes containing information about all the data disks attached to the machine
class AzureVmDataDisks < Inspec.resource(1)
  name 'azure_virtual_machine_datadisks'

  desc "
    This resource gather information about the data disks attached to a virtual machine
  "

  example "
    describe azure_vm_datadisks(host: 'example-01', resource_group: 'MyResourceGroup').where { (disk == 1 and size >= 10) } do
      it { should be true }
    end
  "

  attr_reader :params

  # Constructor to retrieve all the data disks for the specified machines and populate the params property
  #
  # @author Russell Seymour
  #
  # @param [Hash] opts Hashtable of options
  #     opts[:host] The name of the host in the resource group.  NOTE, this is the name as seen in Azure and not the name of the machine in the Operating System
  #     opts[:resource_group] Name of the resource group in which the host will be found
  def initialize(opts)
    @opts = opts
    @helpers = Helpers.new

    # Get the VM that needs to be interrogated
    vm = @helpers.get_vm(@opts[:name], @opts[:resource_group])

    # Parse the data disks
    @params = parse_data_disks(vm.storage_profile.data_disks)
  end

  # Create a FilterTable which can be used by controls to interogate the data disks
  filter = FilterTable.create
  filter.add_accessor(:where)
        .add_accessor(:entries)
        .add_accessor(:count)
        .add_accessor(:has_disks?)
        .add(:disk, field: 'disk')
        .add(:caching, field: 'caching')
        .add(:create_option, field: 'create_option')
        .add(:size, field: 'size')
        .add(:lun, field: 'lun')
        .add(:name, field: 'name')
        .add(:uri, field: 'uri')
        .add(:storage_account, field: 'storage_account')
        .add(:account_type, field: 'account_type')
        .add(:location, filed: 'location')

  filter.connect(self, :params)

  # Determine how many data disks have been applied to the machine
  #
  # @return [Integer] Number of data disks attached to the machine
  #
  def count
    entries.length
  end

  # Determine if any data disks are attached to the machine
  #
  # @return [Boolean] States if the VM has any disks attached
  #
  def has_data_disks?
    !entries.empty?
  end

  private

  # Parse the data disks and extract the necssary information
  #
  # @param [Array] data_disks Array of data disk objects
  #
  # @return [Array<Hash>] Array of hashes providing the information about the data disks attached to the machine
  #
  def parse_data_disks(data_disks)
    data_disks.each_with_index.map do |disk, index|
      parse_data_disk_item(disk, index)
    end.compact
  end

  # Parses each data disk item and extracts the information to be tested
  #
  # @return [Hash] Data disk information
  #
  def parse_data_disk_item(disk, index)

    # determine if using managed disks or a VHD
    if !disk.vhd.nil?

      # Parse the uri of the disk so that the storage account can be retrieved
      uri = URI.parse(disk.vhd.uri)

      {
        'disk' => index,
        'name' => disk.name,
        'size' => disk.disk_size_gb,

        'caching' => disk.caching,
        'create_option' => disk.create_option,
        'lun' => disk.lun,
        'uri' => disk.vhd.uri,
        'storage_account' => uri.host.split('.').first,
      }

    elsif !disk.managed_disk.nil?

      # Parse the ID so that the resource group and name are found
      id_parts = disk.managed_disk.id.split(%r{/})
      resource_group_name = id_parts[4]
      disk_name = id_parts.last

      # Use the compute client to get information about the disk
      managed_disk = @helpers.compute_mgmt.client.disks.get(resource_group_name, disk_name)

      # Build up the hash table to return
      {
        'disk' => index,
        'size' => managed_disk.disk_size_gb,
        'name' => managed_disk.name,

        'account_type' => managed_disk.type,
        'location' => managed_disk.location,
      }

    end
  end
end