axiom-types 0.0.1 → 0.0.2

data/lib/axiom/types/array.rb

@@ -4,10 +4,21 @@ module Axiom
   module Types
 
     # Represents an array type
-    class Array < Object
+    class Array < Collection
       primitive       ::Array
       coercion_method :to_array
 
+      # Test if the type is a base type
+      #
+      # @return [Boolean]
+      #
+      # @api private
+      def self.base?
+        equal?(Array)
+      end
+
+      private_class_method :base?
+
     end # class Array
   end # module Types
 end # module Axiom

data/lib/axiom/types/boolean.rb

@@ -9,6 +9,23 @@ module Axiom
 
       includes true, false
 
+      # Infer the type if the primitive class matches
+      #
+      # @param [Object] object
+      #
+      # @return [Class<Axiom::Types::Boolean>]
+      #   returned if the primitive class matches
+      # @return [nil]
+      #   returned if the primitive class does not match
+      #
+      # @api private
+      def self.infer_from_primitive_class(object)
+        self if TrueClass.singleton_class  === object ||
+                FalseClass.singleton_class === object
+      end
+
+      private_class_method :infer_from_primitive_class
+
     end # class Boolean
   end # module Types
 end # module Axiom

data/lib/axiom/types/collection.rb

@@ -0,0 +1,133 @@
+# encoding: utf-8
+
+module Axiom
+  module Types
+
+    # Represents a collection type
+    class Collection < Object
+      primitive      ::Enumerable
+      accept_options :member_type
+
+      member_type Object
+
+      # Infer the type of the object
+      #
+      # @example with a type
+      #   Axiom::Types::Array.infer(Axiom::Types::Array)
+      #   # => Axiom::Types::Array
+      #
+      # @example with a primitive class
+      #   Axiom::Types::Collection.infer(::Array)
+      #   # => Axiom::Types::Array
+      #
+      # @example with a primitive instance
+      #   Axiom::Types::Array.infer(Array[])
+      #   # => Axiom::Types::Array
+      #
+      # @example with a primitive instance and a member type
+      #   Axiom::Types::Collection.infer(Array[Axiom::Types::String])
+      #   # => Axiom::Types::Array subclass w/String member type
+      #
+      # @example with a primitive instance and a member primitive
+      #   Axiom::Types::Collection.infer(Array[String])
+      #   # => Axiom::Types::Array subclass w/String member type
+      #
+      # @param [Object] object
+      #
+      # @return [Class<Axiom::Types::Collection>]
+      #   returned if the type matches
+      # @return [nil]
+      #   returned if the type does not match
+      #
+      # @api public
+      def self.infer(object)
+        case object
+        when primitive
+          infer_from_primitive_instance(object)
+        else
+          super
+        end
+      end
+
+      # Finalize by setting up constraints for the member
+      #
+      # @return [Class<Axiom::Types::Collection>]
+      #
+      # @api private
+      def self.finalize
+        return self if frozen?
+        member_type.finalize
+        matches_member_type
+        super
+      end
+
+      # Infer the type from a primitive instance
+      #
+      # @param [Object] object
+      #
+      # @return [Class<Axiom::Types::Collection>]
+      #   returned if the primitive instance matches
+      # @return [nil]
+      #   returned if the primitive instance does not match
+      #
+      # @api private
+      def self.infer_from_primitive_instance(object)
+        member      = object.first
+        member_type = Types.infer(member) || Object
+        infer_from(member_type) || new_from(member_type)
+      end
+
+      # Infer the type from the member_type
+      #
+      # @param [Class<Axiom::Types::Object>] member_type
+      #
+      # @return [Class<Axiom::Types::Collection>]
+      #   returned if the member_type matches
+      # @return [nil]
+      #   returned if the member_type does not match
+      #
+      # @api private
+      def self.infer_from(member_type)
+        self if self.member_type.equal?(member_type)
+      end
+
+      # Instantiate a new type from a base type
+      #
+      # @param [Class<Axiom::Types::Object>] member_type
+      #
+      # @return [Class<Axiom::Types::Collection>]
+      #   returned if a base type
+      # @return [nil]
+      #   returned if not a base type
+      #
+      # @api private
+      def self.new_from(member_type)
+        new { member_type(member_type) } if base?
+      end
+
+      # Test if the type is a base type
+      #
+      # @return [Boolean]
+      #
+      # @api private
+      def self.base?
+        # noop
+      end
+
+      # Add a constraints for the member
+      #
+      # @return [undefined]
+      #
+      # @api private
+      def self.matches_member_type
+        constraint do |object|
+          object.all? { |member| member_type.include?(member) }
+        end
+      end
+
+      private_class_method :infer_from_primitive_instance, :infer_from,
+        :new_from, :base?, :matches_member_type
+
+    end # class Collection
+  end # module Types
+end # module Axiom

data/lib/axiom/types/hash.rb

@@ -12,9 +12,28 @@ module Axiom
       key_type   Object
       value_type Object
 
+      # Infer the type of the object
+      #
+      # @param [Object] object
+      #
+      # @return [Class<Axiom::Types::Hash>]
+      #   returned if the type matches
+      # @return [nil]
+      #   returned if the type does not match
+      #
+      # @api public
+      def self.infer(object)
+        case object
+        when primitive
+          infer_from_primitive_instance(object)
+        else
+          super
+        end
+      end
+
       # Finalize by setting up constraints for the key and value
       #
-      # @return [Axiom::Types::Object]
+      # @return [Class<Axiom::Types::Hash>]
       #
       # @api private
       def self.finalize
@@ -25,6 +44,64 @@ module Axiom
         super
       end
 
+      # Infer the type from a primitive instance
+      #
+      # @param [Object] object
+      #
+      # @return [Class<Axiom::Types::Hash>]
+      #   returned if the primitive instance matches
+      # @return [nil]
+      #   returned if the primitive instance does not match
+      #
+      # @api private
+      def self.infer_from_primitive_instance(object)
+        key, value = object.first
+        key_type   = Types.infer(key)   || Object
+        value_type = Types.infer(value) || Object
+        infer_from(key_type, value_type) || new_from(key_type, value_type)
+      end
+
+      # Infer the type from the key_type and value_type
+      #
+      # @param [Class<Axiom::Types::Object>] key_type
+      # @param [Class<Axiom::Types::Object>] value_type
+      #
+      # @return [Class<Axiom::Types::Hash>]
+      #   returned if the key_type and value_type match
+      # @return [nil]
+      #   returned if the key_type and value_type do not match
+      #
+      # @api private
+      def self.infer_from(key_type, value_type)
+        if self.key_type.equal?(key_type) && self.value_type.equal?(value_type)
+          self
+        end
+      end
+
+      # Instantiate a new type from a base type
+      #
+      # @param [Class<Axiom::Types::Object>] key_type
+      # @param [Class<Axiom::Types::Object>] value_type
+      #
+      # @return [Class<Axiom::Types::Hash>]
+      #   returned if a base type
+      # @return [nil]
+      #   returned if not a base type
+      #
+      # @api private
+      def self.new_from(key_type, value_type)
+        new { key_type(key_type).value_type(value_type) } if base?
+      end
+
+      # Test if the type is a base type
+      #
+      # @return [Boolean]
+      #
+      # @api private
+      def self.base?
+        equal?(Hash)
+      end
+
       # Add a constraints for the key and value
       #
       # @return [undefined]
@@ -38,7 +115,8 @@ module Axiom
         end
       end
 
-      private_class_method :matches_key_and_value_types
+      private_class_method :infer_from_primitive_instance, :infer_from,
+        :new_from, :base?, :matches_key_and_value_types
 
     end # class Hash
   end # module Types

data/lib/axiom/types/object.rb

@@ -9,9 +9,26 @@ module Axiom
       primitive       ::Object.superclass || ::Object
       coercion_method :to_object
 
+      # Infer the type of the object
+      #
+      # @example
+      #   Axiom::Types::Object.infer(::Object)  # => Axiom::Types::Object
+      #
+      # @param [Object] object
+      #
+      # @return [Class<Axiom::Types::Object>]
+      #   returned if the type matches
+      # @return [nil]
+      #   returned if the type does not match
+      #
+      # @api public
+      def self.infer(object)
+        super || infer_from_primitive_class(object)
+      end
+
       # Finalize by setting up a primitive constraint
       #
-      # @return [Axiom::Types::Object]
+      # @return [Class<Axiom::Types::Object>]
       #
       # @api private
       def self.finalize
@@ -20,6 +37,21 @@ module Axiom
         super
       end
 
+      # Infer the type if the primitive class matches
+      #
+      # @param [Object] object
+      #
+      # @return [Class<Axiom::Types::Object>]
+      #   returned if the primitive class matches
+      # @return [nil]
+      #   returned if the primitive class does not match
+      #
+      # @api private
+      def self.infer_from_primitive_class(object)
+        self if object.respond_to?(:ancestors) &&
+                object.ancestors.include?(primitive)
+      end
+
       # Add a constraint for the primitive
       #
       # @return [undefined]
@@ -29,7 +61,7 @@ module Axiom
         constraint(&primitive.method(:===))
       end
 
-      private_class_method :inherits_from_primitive
+      private_class_method :infer_from_primitive_class, :inherits_from_primitive
 
     end # class Object
   end # module Types

data/lib/axiom/types/set.rb

@@ -4,10 +4,21 @@ module Axiom
   module Types
 
     # Represents a set type
-    class Set < Object
+    class Set < Collection
       primitive       ::Set
       coercion_method :to_set
 
+      # Test if the type is a base type
+      #
+      # @return [Boolean]
+      #
+      # @api private
+      def self.base?
+        equal?(Set)
+      end
+
+      private_class_method :base?
+
     end # class Set
   end # module Types
 end # module Axiom

data/lib/axiom/types/support/options.rb

@@ -46,6 +46,18 @@ module Axiom
         self
       end
 
+      # Set the option if it is not already set
+      #
+      # @param [Symbol] option
+      # @param [Object] value
+      #
+      # @return [self]
+      #
+      # @api private
+      def set_option(option, value)
+        public_send(option, value) unless public_send(option)
+      end
+
     private
 
       # Adds descendant to descendants array and inherits default options
@@ -58,7 +70,7 @@ module Axiom
       def inherited(descendant)
         super
         options.each do |option, value|
-          descendant.add_accepted_option(option).public_send(option, value)
+          descendant.add_accepted_option(option).set_option(option, value)
         end
       end
 

data/lib/axiom/types/type.rb

@@ -10,6 +10,21 @@ module Axiom
       accept_options :constraint
       constraint Tautology
 
+      # Infer the type of the object
+      #
+      # @example
+      #  type = Axiom::Types::Type.infer(Axiom::Types::Integer)
+      #  # => Axiom::Types::Integer
+      #
+      # @param [Object] object
+      #
+      # @return [Class<Axiom::Types::Type>]
+      #
+      # @api public
+      def self.infer(object)
+        self if equal?(object)
+      end
+
       # Instantiate a new Axiom::Types::Type subclass
       #
       # @example
@@ -37,7 +52,7 @@ module Axiom
 
       # Finalize by deep freezing
       #
-      # @return [Axiom::Types::Type]
+      # @return [Class<Axiom::Types::Type>]
       #
       # @api private
       def self.finalize
@@ -89,7 +104,7 @@ module Axiom
       # @yieldreturn [Boolean]
       #   true if the object matches the type constraint
       #
-      # @return [self]
+      # @return [Class<Axiom::Types::Type>]
       #
       # @api public
       def self.constraint(constraint = Undefined, &block)

data/lib/axiom/types/version.rb

@@ -4,7 +4,7 @@ module Axiom
   module Types
 
     # Gem version
-    VERSION = '0.0.1'.freeze
+    VERSION = '0.0.2'.freeze
 
   end # module Types
 end # module Axiom