module VagrantPlugins
  module ProviderVirtualBox
    module Model
      # Represents a storage controller for VirtualBox. Storage controllers
      # have a type, a name, and can have hard disks or optical drives attached.
      class StorageController

        SATA_CONTROLLER_TYPES = ["IntelAhci"].map(&:freeze).freeze
        IDE_CONTROLLER_TYPES = ["PIIX4", "PIIX3", "ICH6"].map(&:freeze).freeze

        # The name of the storage controller.
        #
        # @return [String]
        attr_reader :name

        # The specific type of controller.
        #
        # @return [String]
        attr_reader :type

        # The storage bus associated with the storage controller, which can be
        # inferred from its specific type.
        #
        # @return [String]
        attr_reader :storage_bus

        # The maximum number of avilable ports for the storage controller. For
        # SATA controllers, this indicates the number of disks that can be
        # attached. For IDE controllers, this indicates that n*2 disks can be
        # attached (primary/secondary).
        #
        # @return [Integer]
        attr_reader :maxportcount

        # The list of disks/ISOs attached to each storage controller.
        #
        # @return [Array<Hash>]
        attr_reader :attachments

        def initialize(name, type, maxportcount, attachments)
          @name         = name
          @type         = type

          if SATA_CONTROLLER_TYPES.include?(@type)
            @storage_bus = 'SATA'
          elsif IDE_CONTROLLER_TYPES.include?(@type)
            @storage_bus = 'IDE'
          else
            @storage_bus = 'Unknown'
          end

          @maxportcount = maxportcount.to_i

          attachments ||= []
          @attachments  = attachments
        end
      end
    end
  end
end