lib/endeca/refinement.rb in primedia-endeca-0.9.6 vs lib/endeca/refinement.rb in primedia-endeca-0.9.7

@@ -1,10 +1,20 @@
 module Endeca
   class Refinement
     extend ClassToProc
     extend Readers
 
+    reader 'DimensionName' => :name,
+      'ExpansionLink' => :expansion_link,
+      'ContractionLink' => :contraction_link
+
+    integer_reader 'DimensionID' => :id
+
+    reader('DimensionValues' => :dimension_values) do |values|
+      values.map(&Dimension) if values
+    end
+
     attr_reader :raw
     def initialize(raw={})
       @raw = raw
     end
 
@@ -18,12 +28,11 @@
 
     def attributes
       (@raw['Dimensions'] || []).first || {}
     end
 
-    reader \
-      'DimensionName' => :name,
-      'ExpansionLink' => :to_params
+    def to_params
+      expansion_link || contraction_link
+    end
 
-    integer_reader 'DimensionID' => :id
   end
 end