EGI-FCTF/rOCCI-core

module Occi
  module InfrastructureExt
    # See `Occi::Infrastructure::Model` for details.
    #
    # @author Boris Parak <parak@cesnet.cz>
    class Model < Occi::Infrastructure::Model
      # Loads OGF's OCCI Infrastructure Ext Standard from `Occi::InfrastructureExt::Warehouse`.
      #
      # @example
      #    model = Occi::InfrastructureExt::Model.new
      #    model.load_infrastructure_ext!
      def load_infrastructure_ext!
        logger.debug 'Loading InfrastructureExt from InfrastructureExt::Warehouse'
        Occi::InfrastructureExt::Warehouse.bootstrap! self
        self << Occi::InfrastructureExt::Mixins::AvailabilityZone.new
        self << Occi::InfrastructureExt::Mixins::Region.new
        self << Occi::InfrastructureExt::Mixins::Floatingippool.new
        nil
      end

      # Returns an instance of `Occi::InfrastructureExt::InstanceBuilder` associated with this model.
      #
      # @return [Occi::InfrastructureExt::InstanceBuilder] instance of IB
      def instance_builder
        Occi::InfrastructureExt::InstanceBuilder.new(model: self)
      end

      # Returns all mixins dependent on the base `availability_zone` mixin defined by OGF.
      #
      # @return [Set] set of mixins dependent on `availability_zone`
      def find_availability_zones
        find_dependent Occi::InfrastructureExt::Mixins::AvailabilityZone.new
      end

      # Returns all mixins dependent on the base `region` mixin defined by OGF.
      #
      # @return [Set] set of mixins dependent on `region`
      def find_regions
        find_dependent Occi::InfrastructureExt::Mixins::Region.new
      end

      # Returns all mixins dependent on the base `floatingippool` mixin defined by OGF.
      #
      # @return [Set] set of mixins dependent on `floatingippool`
      def find_floatingippools
        find_dependent Occi::InfrastructureExt::Mixins::Floatingippool.new
      end
    end
  end
end