From 322f47898e80af3fcdc3cb3db35e177d8216a2d2 Mon Sep 17 00:00:00 2001
From: Jose and Yehuda <wycats@gmail.com>
Date: Sat, 15 Oct 2011 18:27:56 +0200
Subject: Add association_ids

---
 activemodel/lib/active_model/serializer.rb | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

(limited to 'activemodel/lib/active_model')

diff --git a/activemodel/lib/active_model/serializer.rb b/activemodel/lib/active_model/serializer.rb
index 6e89eec09c..99a007de31 100644
--- a/activemodel/lib/active_model/serializer.rb
+++ b/activemodel/lib/active_model/serializer.rb
@@ -18,12 +18,25 @@ module ActiveModel
             serializer.new(item, scope).serializable_hash
           end
         end
+
+        def serialize_ids(collection, scope)
+          # use named scopes if they are present
+          return collection.ids if collection.respond_to?(:ids)
+
+          collection.map do |item|
+            item.read_attribute_for_serialization(:id)
+          end
+        end
       end
 
       class HasOne < Config
         def serialize(object, scope)
           serializer.new(object, scope).serializable_hash
         end
+
+        def serialize_ids(object, scope)
+          object.read_attribute_for_serialization(:id)
+        end
       end
     end
 
@@ -80,7 +93,11 @@ module ActiveModel
     end
 
     def serializable_hash
-      hash = attributes
+      attributes.merge(associations)
+    end
+
+    def associations
+      hash = {}
 
       _associations.each do |association|
         associated_object = send(association.name)
@@ -90,6 +107,17 @@ module ActiveModel
       hash
     end
 
+    def association_ids
+      hash = {}
+
+      _associations.each do |association|
+        associated_object = send(association.name)
+        hash[association.name] = association.serialize_ids(associated_object, scope)
+      end
+
+      hash
+    end
+
     def attributes
       hash = {}
 
-- 
cgit v1.2.3