Add value and name listing

Author: thegorgon

README.md

@@ -105,6 +105,13 @@ COLORS[:RED] === COLORS.RED # => true
 COLORS['RED'] === COLORS.RED # => true
 ```
 
+You can also access all the enums using `#entries` or just get a list of names (as symbols) or values using `#values` or `#names`.
+
+```
+COLORS.entries # => [COLORS.RED, COLORS.GREEN, COLORS.BLUE]
+COLORS.values # => [:red, :green, :blue], equivalent to #pluck(:value)
+COLORS.names # => [:RED, :GREEN, :BLUE], equivalent to #pluck(:name)
+
 ## Contributing
 
 Pull requests welcome!

lib/enum_type/enum.rb

@@ -65,6 +65,14 @@ def [](value)
       @enums[value.to_sym] || @enums_by_value[value]
     end
 
+    def values
+      @enums_by_value.keys
+    end
+
+    def names
+      @enums.keys
+    end
+
     def respond_to_missing?(method_name, *arguments, &block)
       method_name =~ /[A-Z]+/ && arguments.empty? && block.nil? || super
     end

spec/enum_type_spec.rb

@@ -42,7 +42,7 @@
       end
 
       it 'allows iterating over values' do
-        expect(enum.map { |e| e }).to eq [enum.RED, enum.GREEN, enum.BLUE]
+        expect(enum.entries).to eq [enum.RED, enum.GREEN, enum.BLUE]
         expect(enum.map(&:value)).to eq %i[red green blue]
       end
 
@@ -57,6 +57,14 @@
       it 'allows looking up by value' do
         expect(enum[:red]).to equal enum.RED
       end
+
+      it 'allows getting the list of values' do
+        expect(enum.values).to eq %i[red green blue]
+      end
+
+      it 'allows getting the list of names as symbols' do
+        expect(enum.names).to eq %i[RED GREEN BLUE]
+      end
     end
 
     context 'with array attributes' do
@@ -94,26 +102,34 @@
       end
 
       it 'allows iterating over values' do
-        expect(enum.map { |e| e }).to eq [enum.RED, enum.GREEN, enum.BLUE]
+        expect(enum.entries).to eq [enum.RED, enum.GREEN, enum.BLUE]
         expect(enum.map(&:value)).to eq %i[red green blue]
         expect(enum.map(&:hex)).to eq %w[#f00 #0f0 #00f]
         expect(enum.map(&:rgb)).to eq [[255, 0, 0], [0, 255, 0], [0, 0, 255]]
       end
+
+      it 'allows getting the list of values' do
+        expect(enum.values).to eq %i[red green blue]
+      end
+
+      it 'allows getting the list of names as symbols' do
+        expect(enum.names).to eq %i[RED GREEN BLUE]
+      end
     end
 
     context 'with hash attributes' do
       let(:enum) do
-        EnumType.create(value: Symbol, hex: String, rgb: Array) do
-          RED(:red, '#f00', [255, 0, 0])
-          GREEN(:green, '#0f0', [0, 255, 0])
-          BLUE(:blue, '#00f', [0, 0, 255])
+        EnumType.create(value: String, hex: String, rgb: Array) do
+          RED('red', '#f00', [255, 0, 0])
+          GREEN('green', '#0f0', [0, 255, 0])
+          BLUE('blue', '#00f', [0, 0, 255])
         end
       end
 
       it 'defines all the enum types with the right value' do
-        expect(enum.RED.value).to eq :red
-        expect(enum.GREEN.value).to eq :green
-        expect(enum.BLUE.value).to eq :blue
+        expect(enum.RED.value).to eq 'red'
+        expect(enum.GREEN.value).to eq 'green'
+        expect(enum.BLUE.value).to eq 'blue'
       end
 
       it 'defines all the enum types with the right name' do
@@ -136,11 +152,19 @@
       end
 
       it 'allows iterating over values' do
-        expect(enum.map { |e| e }).to eq [enum.RED, enum.GREEN, enum.BLUE]
-        expect(enum.map(&:value)).to eq %i[red green blue]
+        expect(enum.entries).to eq [enum.RED, enum.GREEN, enum.BLUE]
+        expect(enum.map(&:value)).to eq %w[red green blue]
         expect(enum.map(&:hex)).to eq %w[#f00 #0f0 #00f]
         expect(enum.map(&:rgb)).to eq [[255, 0, 0], [0, 255, 0], [0, 0, 255]]
       end
+
+      it 'allows getting the list of values' do
+        expect(enum.values).to eq %w[red green blue]
+      end
+
+      it 'allows getting the list of names as symbols' do
+        expect(enum.names).to eq %i[RED GREEN BLUE]
+      end
     end
 
     context 'with hash attributes and an invalid definition' do