Enum classes should have types (#67)

lib/literal/constructors.rb

@@ -3,18 +3,21 @@
 module Literal::Constructors
 	SpecialisedEnums = {
 		String => Class.new(Literal::Enum) do
-			def type = String
+			@type = String
+
 			alias_method :to_s, :value
 			alias_method :to_str, :value
 		end,
 
 		Integer => Class.new(Literal::Enum) do
-			def type = Integer
+			@type = Integer
+
 			alias_method :to_i, :value
 		end,
 
 		Array => Class.new(Literal::Enum) do
-			def type = Array
+			@type = Array
+
 			alias_method :to_a, :value
 			alias_method :to_ary, :value
 		end
@@ -27,7 +30,7 @@ def Array(type)
 
 	def Enum(type)
 		SpecialisedEnums[type] || Class.new(Literal::Enum) do
-			define_method(:type) { type }
+			@type = type
 		end
 	end
 

lib/literal/enum.rb

@@ -22,9 +22,12 @@ def respond_to_missing?(name, include_private = false)
 		end
 
 		def inherited(subclass)
+			type = @type
+
 			subclass.instance_exec do
 				@values = {}
 				@members = []
+				@type = type
 			end
 		end
 
@@ -96,6 +99,10 @@ def name
 		"#{self.class.name}::#{@name}"
 	end
 
+	def type
+		self.class.type
+	end
+
 	alias_method :inspect, :name
 
 	def siblings

lib/literal/enum_type.rb

@@ -12,25 +12,4 @@ def inspect
 	def ===(value)
 		Literal::Enum === value && value.type == @type
 	end
-
-	def define(&)
-		type = @type
-
-		Class.new(Literal::Enum) do
-			@type = type
-
-			if Integer == type
-				alias_method :to_i, :value
-			elsif String == type
-				alias_method :to_s, :value
-			elsif Array == type
-				alias_method :to_a, :value
-			elsif Hash == type
-				alias_method :to_h, :value
-			end
-
-			class_exec(&)
-			deep_freeze
-		end
-	end
 end

test/literal/enum.test.rb

@@ -34,6 +34,10 @@ def toggle = Switch::On
 	assert Color.frozen?
 end
 
+test "the enum class has a type" do
+	expect(Color.type) == Integer
+end
+
 test "the enum class is enumerable" do
 	expect(Color).to_be_an Enumerable
 	expect(Color.map(&:value)) == [0, 1, 3, 4]