lib/garb/resource.rb in garb-0.7.4 vs lib/garb/resource.rb in garb-0.7.5

@@ -38,10 +38,14 @@
 
     def clear_filters
       @filter_parameters = FilterParameters.new
     end
 
+    def set_segment_id(id)
+      @segment = "gaid::#{id.to_i}"
+    end
+
     def results(profile, opts = {}, &block)
       @profile = profile.is_a?(Profile) ? profile : Profile.first(profile, opts.fetch(:session, Session))
 
       if @profile
         @start_date = opts.fetch(:start_date, Time.now - MONTH)
@@ -65,16 +69,21 @@
       {'ids' => @profile.table_id,
         'start-date' => format_time(@start_date),
         'end-date' => format_time(@end_date)}
     end
 
+    def segment_params
+      @segment ? {'segment' => @segment} : {}
+    end
+
     def params
       [
         metrics.to_params,
         dimensions.to_params,
         sort.to_params,
         filters.to_params,
-        page_params
+        page_params,
+        segment_params
       ].inject(default_params) do |p, i|
         p.merge(i)
       end
     end