1 files changed, 23 insertions, 4 deletions
diff --git a/lib/arel/nodes/window.rb b/lib/arel/nodes/window.rb
index 60259e8c05..fee8eeff7a 100644
--- a/lib/arel/nodes/window.rb
+++ b/lib/arel/nodes/window.rb
@@ -1,10 +1,12 @@
 module Arel
   module Nodes
     class Window < Arel::Nodes::Node
-      attr_accessor :orders, :framing
+      attr_accessor :orders, :framing, :partitions
 
       def initialize
         @orders = []
+        @partitions = []
+        @framing = nil
       end
 
       def order *expr
@@ -15,16 +17,32 @@ module Arel
         self
       end
 
+      def partition *expr
+        # FIXME: We SHOULD NOT be converting these to SqlLiteral automatically
+        @partitions.concat expr.map { |x|
+          String === x || Symbol === x ? Nodes::SqlLiteral.new(x.to_s) : x
+        }
+        self
+      end
+
       def frame(expr)
         @framing = expr
       end
 
       def rows(expr = nil)
-        frame(Rows.new(expr))
+        if @framing
+          Rows.new(expr)
+        else
+          frame(Rows.new(expr))
+        end
       end
 
       def range(expr = nil)
-        frame(Range.new(expr))
+        if @framing
+          Range.new(expr)
+        else
+          frame(Range.new(expr))
+        end
       end
 
       def initialize_copy other
@@ -39,7 +57,8 @@ module Arel
       def eql? other
         self.class == other.class &&
           self.orders == other.orders &&
-          self.framing == other.framing
+          self.framing == other.framing &&
+          self.partitions == other.partitions
       end
       alias :== :eql?
     end