X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/9d57643812a8ee94022541c1119a69374565a6dd..af2b13c2c1c220326241784b1d237d85ff388426:/app/models/relation_member.rb

diff --git a/app/models/relation_member.rb b/app/models/relation_member.rb
index 79102853e..389e2ce52 100644
--- a/app/models/relation_member.rb
+++ b/app/models/relation_member.rb
@@ -1,30 +1,26 @@
 class RelationMember < ActiveRecord::Base
-  set_table_name 'current_relation_members'
+  set_table_name 'current_relation_members'  
+  set_primary_keys :relation_id, :sequence_id
 
-  # problem with RelationMember is that it may link to any one 
-  # object (a node, a way, another relation), and belongs_to is
-  # not flexible enough for that. So we do this, which is ugly,
-  # but fortunately rails won't actually run the SQL behind that
-  # unless someone really accesses .node, .way, or
-  # .relation - which is what we do below based on member_type.
-  # (and no: the :condition on belongs_to doesn't work here as
-  # it is a condition on the *referenced* object not the 
-  # *referencing* object!)
-  
-  belongs_to :node, :foreign_key => "member_id"
-  belongs_to :way, :foreign_key => "member_id"
-  belongs_to :relation, :foreign_key => "member_id"
+  belongs_to :relation
+  belongs_to :member, :polymorphic => true
 
-  # so we define this "member" function that returns whatever it
-  # is.
- 
-  def member()
-    return (member_type == "node") ? node : (member_type == "way") ? way : relation
+  after_find :set_class_from_type
+  after_initialize :set_class_from_type
+  before_save :set_type_from_class
+
+  def member_type=(type)
+    self[:member_type] = type
+    self[:member_class] = type.capitalize
   end
 
-  # NOTE - relations are SUBJECTS of memberships. The fact that nodes, 
-  # ways, and relations can be the OBJECT of a membership,
-  # i.e. a node/way/relation can be referenced throgh a
-  # RelationMember object, is NOT modelled in rails, i.e. these links
-  # have to be resolved manually, on demand. 
+private
+
+  def set_class_from_type
+    self[:member_class] = self.member_type.classify unless self.member_type.nil?
+  end
+
+  def set_type_from_class
+    self.member_type = self[:member_class].classify
+  end
 end