Chris@0: module CollectiveIdea #:nodoc: Chris@0: module Acts #:nodoc: Chris@0: module NestedSet #:nodoc: Chris@0: def self.included(base) Chris@0: base.extend(SingletonMethods) Chris@0: end Chris@0: Chris@0: # This acts provides Nested Set functionality. Nested Set is a smart way to implement Chris@0: # an _ordered_ tree, with the added feature that you can select the children and all of their Chris@0: # descendants with a single query. The drawback is that insertion or move need some complex Chris@0: # sql queries. But everything is done here by this module! Chris@0: # Chris@0: # Nested sets are appropriate each time you want either an orderd tree (menus, Chris@0: # commercial categories) or an efficient way of querying big trees (threaded posts). Chris@0: # Chris@0: # == API Chris@0: # Chris@0: # Methods names are aligned with acts_as_tree as much as possible, to make replacment from one Chris@0: # by another easier, except for the creation: Chris@0: # Chris@0: # in acts_as_tree: Chris@0: # item.children.create(:name => "child1") Chris@0: # Chris@0: # in acts_as_nested_set: Chris@0: # # adds a new item at the "end" of the tree, i.e. with child.left = max(tree.right)+1 Chris@0: # child = MyClass.new(:name => "child1") Chris@0: # child.save Chris@0: # # now move the item to its right place Chris@0: # child.move_to_child_of my_item Chris@0: # Chris@0: # You can pass an id or an object to: Chris@0: # * #move_to_child_of Chris@0: # * #move_to_right_of Chris@0: # * #move_to_left_of Chris@0: # Chris@0: module SingletonMethods Chris@0: # Configuration options are: Chris@0: # Chris@0: # * +:parent_column+ - specifies the column name to use for keeping the position integer (default: parent_id) Chris@0: # * +:left_column+ - column name for left boundry data, default "lft" Chris@0: # * +:right_column+ - column name for right boundry data, default "rgt" Chris@0: # * +:scope+ - restricts what is to be considered a list. Given a symbol, it'll attach "_id" Chris@0: # (if it hasn't been already) and use that as the foreign key restriction. You Chris@0: # can also pass an array to scope by multiple attributes. Chris@0: # Example: acts_as_nested_set :scope => [:notable_id, :notable_type] Chris@0: # * +:dependent+ - behavior for cascading destroy. If set to :destroy, all the Chris@0: # child objects are destroyed alongside this object by calling their destroy Chris@0: # method. If set to :delete_all (default), all the child objects are deleted Chris@0: # without calling their destroy method. Chris@0: # Chris@0: # See CollectiveIdea::Acts::NestedSet::ClassMethods for a list of class methods and Chris@0: # CollectiveIdea::Acts::NestedSet::InstanceMethods for a list of instance methods added Chris@0: # to acts_as_nested_set models Chris@0: def acts_as_nested_set(options = {}) Chris@0: options = { Chris@0: :parent_column => 'parent_id', Chris@0: :left_column => 'lft', Chris@0: :right_column => 'rgt', Chris@0: :order => 'id', Chris@0: :dependent => :delete_all, # or :destroy Chris@0: }.merge(options) Chris@0: Chris@0: if options[:scope].is_a?(Symbol) && options[:scope].to_s !~ /_id$/ Chris@0: options[:scope] = "#{options[:scope]}_id".intern Chris@0: end Chris@0: Chris@0: write_inheritable_attribute :acts_as_nested_set_options, options Chris@0: class_inheritable_reader :acts_as_nested_set_options Chris@0: Chris@0: include Comparable Chris@0: include Columns Chris@0: include InstanceMethods Chris@0: extend Columns Chris@0: extend ClassMethods Chris@0: Chris@0: # no bulk assignment Chris@0: attr_protected left_column_name.intern, Chris@0: right_column_name.intern, Chris@0: parent_column_name.intern Chris@0: Chris@0: before_create :set_default_left_and_right Chris@0: before_destroy :prune_from_tree Chris@0: Chris@0: # no assignment to structure fields Chris@0: [left_column_name, right_column_name, parent_column_name].each do |column| Chris@0: module_eval <<-"end_eval", __FILE__, __LINE__ Chris@0: def #{column}=(x) Chris@0: raise ActiveRecord::ActiveRecordError, "Unauthorized assignment to #{column}: it's an internal field handled by acts_as_nested_set code, use move_to_* methods instead." Chris@0: end Chris@0: end_eval Chris@0: end Chris@0: Chris@0: named_scope :roots, :conditions => {parent_column_name => nil}, :order => quoted_left_column_name Chris@0: named_scope :leaves, :conditions => "#{quoted_right_column_name} - #{quoted_left_column_name} = 1", :order => quoted_left_column_name Chris@0: if self.respond_to?(:define_callbacks) Chris@0: define_callbacks("before_move", "after_move") Chris@0: end Chris@0: Chris@0: Chris@0: end Chris@0: Chris@0: end Chris@0: Chris@0: module ClassMethods Chris@0: Chris@0: # Returns the first root Chris@0: def root Chris@0: roots.find(:first) Chris@0: end Chris@0: Chris@0: def valid? Chris@0: left_and_rights_valid? && no_duplicates_for_columns? && all_roots_valid? Chris@0: end Chris@0: Chris@0: def left_and_rights_valid? Chris@0: count( Chris@0: :joins => "LEFT OUTER JOIN #{quoted_table_name} AS parent ON " + Chris@0: "#{quoted_table_name}.#{quoted_parent_column_name} = parent.#{primary_key}", Chris@0: :conditions => Chris@0: "#{quoted_table_name}.#{quoted_left_column_name} IS NULL OR " + Chris@0: "#{quoted_table_name}.#{quoted_right_column_name} IS NULL OR " + Chris@0: "#{quoted_table_name}.#{quoted_left_column_name} >= " + Chris@0: "#{quoted_table_name}.#{quoted_right_column_name} OR " + Chris@0: "(#{quoted_table_name}.#{quoted_parent_column_name} IS NOT NULL AND " + Chris@0: "(#{quoted_table_name}.#{quoted_left_column_name} <= parent.#{quoted_left_column_name} OR " + Chris@0: "#{quoted_table_name}.#{quoted_right_column_name} >= parent.#{quoted_right_column_name}))" Chris@0: ) == 0 Chris@0: end Chris@0: Chris@0: def no_duplicates_for_columns? Chris@0: scope_string = Array(acts_as_nested_set_options[:scope]).map do |c| Chris@0: connection.quote_column_name(c) Chris@0: end.push(nil).join(", ") Chris@0: [quoted_left_column_name, quoted_right_column_name].all? do |column| Chris@0: # No duplicates Chris@0: find(:first, Chris@0: :select => "#{scope_string}#{column}, COUNT(#{column})", Chris@0: :group => "#{scope_string}#{column} Chris@0: HAVING COUNT(#{column}) > 1").nil? Chris@0: end Chris@0: end Chris@0: Chris@0: # Wrapper for each_root_valid? that can deal with scope. Chris@0: def all_roots_valid? Chris@0: if acts_as_nested_set_options[:scope] Chris@0: roots(:group => scope_column_names).group_by{|record| scope_column_names.collect{|col| record.send(col.to_sym)}}.all? do |scope, grouped_roots| Chris@0: each_root_valid?(grouped_roots) Chris@0: end Chris@0: else Chris@0: each_root_valid?(roots) Chris@0: end Chris@0: end Chris@0: Chris@0: def each_root_valid?(roots_to_validate) Chris@0: left = right = 0 Chris@0: roots_to_validate.all? do |root| chris@37: (root.left > left && root.right > right).tap do Chris@0: left = root.left Chris@0: right = root.right Chris@0: end Chris@0: end Chris@0: end Chris@0: Chris@0: # Rebuilds the left & rights if unset or invalid. Also very useful for converting from acts_as_tree. Chris@0: def rebuild! Chris@0: # Don't rebuild a valid tree. Chris@0: return true if valid? Chris@0: Chris@0: scope = lambda{|node|} Chris@0: if acts_as_nested_set_options[:scope] Chris@0: scope = lambda{|node| Chris@0: scope_column_names.inject(""){|str, column_name| Chris@0: str << "AND #{connection.quote_column_name(column_name)} = #{connection.quote(node.send(column_name.to_sym))} " Chris@0: } Chris@0: } Chris@0: end Chris@0: indices = {} Chris@0: Chris@0: set_left_and_rights = lambda do |node| Chris@0: # set left Chris@0: node[left_column_name] = indices[scope.call(node)] += 1 Chris@0: # find Chris@0: find(:all, :conditions => ["#{quoted_parent_column_name} = ? #{scope.call(node)}", node], :order => "#{quoted_left_column_name}, #{quoted_right_column_name}, #{acts_as_nested_set_options[:order]}").each{|n| set_left_and_rights.call(n) } Chris@0: # set right Chris@0: node[right_column_name] = indices[scope.call(node)] += 1 Chris@0: node.save! Chris@0: end Chris@0: Chris@0: # Find root node(s) Chris@0: root_nodes = find(:all, :conditions => "#{quoted_parent_column_name} IS NULL", :order => "#{quoted_left_column_name}, #{quoted_right_column_name}, #{acts_as_nested_set_options[:order]}").each do |root_node| Chris@0: # setup index for this scope Chris@0: indices[scope.call(root_node)] ||= 0 Chris@0: set_left_and_rights.call(root_node) Chris@0: end Chris@0: end Chris@0: end Chris@0: Chris@0: # Mixed into both classes and instances to provide easy access to the column names Chris@0: module Columns Chris@0: def left_column_name Chris@0: acts_as_nested_set_options[:left_column] Chris@0: end Chris@0: Chris@0: def right_column_name Chris@0: acts_as_nested_set_options[:right_column] Chris@0: end Chris@0: Chris@0: def parent_column_name Chris@0: acts_as_nested_set_options[:parent_column] Chris@0: end Chris@0: Chris@0: def scope_column_names Chris@0: Array(acts_as_nested_set_options[:scope]) Chris@0: end Chris@0: Chris@0: def quoted_left_column_name Chris@0: connection.quote_column_name(left_column_name) Chris@0: end Chris@0: Chris@0: def quoted_right_column_name Chris@0: connection.quote_column_name(right_column_name) Chris@0: end Chris@0: Chris@0: def quoted_parent_column_name Chris@0: connection.quote_column_name(parent_column_name) Chris@0: end Chris@0: Chris@0: def quoted_scope_column_names Chris@0: scope_column_names.collect {|column_name| connection.quote_column_name(column_name) } Chris@0: end Chris@0: end Chris@0: Chris@0: # Any instance method that returns a collection makes use of Rails 2.1's named_scope (which is bundled for Rails 2.0), so it can be treated as a finder. Chris@0: # Chris@0: # category.self_and_descendants.count Chris@0: # category.ancestors.find(:all, :conditions => "name like '%foo%'") Chris@0: module InstanceMethods Chris@0: # Value of the parent column Chris@0: def parent_id Chris@0: self[parent_column_name] Chris@0: end Chris@0: Chris@0: # Value of the left column Chris@0: def left Chris@0: self[left_column_name] Chris@0: end Chris@0: Chris@0: # Value of the right column Chris@0: def right Chris@0: self[right_column_name] Chris@0: end Chris@0: Chris@0: # Returns true if this is a root node. Chris@0: def root? Chris@0: parent_id.nil? Chris@0: end Chris@0: Chris@0: def leaf? Chris@0: new_record? || (right - left == 1) Chris@0: end Chris@0: Chris@0: # Returns true is this is a child node Chris@0: def child? Chris@0: !parent_id.nil? Chris@0: end Chris@0: Chris@0: # order by left column Chris@0: def <=>(x) Chris@0: left <=> x.left Chris@0: end Chris@0: Chris@0: # Redefine to act like active record Chris@0: def ==(comparison_object) Chris@0: comparison_object.equal?(self) || Chris@0: (comparison_object.instance_of?(self.class) && Chris@0: comparison_object.id == id && Chris@0: !comparison_object.new_record?) Chris@0: end Chris@0: Chris@0: # Returns root Chris@0: def root Chris@0: self_and_ancestors.find(:first) Chris@0: end Chris@0: Chris@0: # Returns the immediate parent Chris@0: def parent Chris@0: nested_set_scope.find_by_id(parent_id) if parent_id Chris@0: end Chris@0: Chris@0: # Returns the array of all parents and self Chris@0: def self_and_ancestors Chris@0: nested_set_scope.scoped :conditions => [ Chris@0: "#{self.class.table_name}.#{quoted_left_column_name} <= ? AND #{self.class.table_name}.#{quoted_right_column_name} >= ?", left, right Chris@0: ] Chris@0: end Chris@0: Chris@0: # Returns an array of all parents Chris@0: def ancestors Chris@0: without_self self_and_ancestors Chris@0: end Chris@0: Chris@0: # Returns the array of all children of the parent, including self Chris@0: def self_and_siblings Chris@0: nested_set_scope.scoped :conditions => {parent_column_name => parent_id} Chris@0: end Chris@0: Chris@0: # Returns the array of all children of the parent, except self Chris@0: def siblings Chris@0: without_self self_and_siblings Chris@0: end Chris@0: Chris@0: # Returns a set of all of its nested children which do not have children Chris@0: def leaves Chris@0: descendants.scoped :conditions => "#{self.class.table_name}.#{quoted_right_column_name} - #{self.class.table_name}.#{quoted_left_column_name} = 1" Chris@0: end Chris@0: Chris@0: # Returns the level of this object in the tree Chris@0: # root level is 0 Chris@0: def level Chris@0: parent_id.nil? ? 0 : ancestors.count Chris@0: end Chris@0: Chris@0: # Returns a set of itself and all of its nested children Chris@0: def self_and_descendants Chris@0: nested_set_scope.scoped :conditions => [ Chris@0: "#{self.class.table_name}.#{quoted_left_column_name} >= ? AND #{self.class.table_name}.#{quoted_right_column_name} <= ?", left, right Chris@0: ] Chris@0: end Chris@0: Chris@0: # Returns a set of all of its children and nested children Chris@0: def descendants Chris@0: without_self self_and_descendants Chris@0: end Chris@0: Chris@0: # Returns a set of only this entry's immediate children Chris@0: def children Chris@0: nested_set_scope.scoped :conditions => {parent_column_name => self} Chris@0: end Chris@0: Chris@0: def is_descendant_of?(other) Chris@0: other.left < self.left && self.left < other.right && same_scope?(other) Chris@0: end Chris@0: Chris@0: def is_or_is_descendant_of?(other) Chris@0: other.left <= self.left && self.left < other.right && same_scope?(other) Chris@0: end Chris@0: Chris@0: def is_ancestor_of?(other) Chris@0: self.left < other.left && other.left < self.right && same_scope?(other) Chris@0: end Chris@0: Chris@0: def is_or_is_ancestor_of?(other) Chris@0: self.left <= other.left && other.left < self.right && same_scope?(other) Chris@0: end Chris@0: Chris@0: # Check if other model is in the same scope Chris@0: def same_scope?(other) Chris@0: Array(acts_as_nested_set_options[:scope]).all? do |attr| Chris@0: self.send(attr) == other.send(attr) Chris@0: end Chris@0: end Chris@0: Chris@0: # Find the first sibling to the left Chris@0: def left_sibling Chris@0: siblings.find(:first, :conditions => ["#{self.class.table_name}.#{quoted_left_column_name} < ?", left], Chris@0: :order => "#{self.class.table_name}.#{quoted_left_column_name} DESC") Chris@0: end Chris@0: Chris@0: # Find the first sibling to the right Chris@0: def right_sibling Chris@0: siblings.find(:first, :conditions => ["#{self.class.table_name}.#{quoted_left_column_name} > ?", left]) Chris@0: end Chris@0: Chris@0: # Shorthand method for finding the left sibling and moving to the left of it. Chris@0: def move_left Chris@0: move_to_left_of left_sibling Chris@0: end Chris@0: Chris@0: # Shorthand method for finding the right sibling and moving to the right of it. Chris@0: def move_right Chris@0: move_to_right_of right_sibling Chris@0: end Chris@0: Chris@0: # Move the node to the left of another node (you can pass id only) Chris@0: def move_to_left_of(node) Chris@0: move_to node, :left Chris@0: end Chris@0: Chris@0: # Move the node to the left of another node (you can pass id only) Chris@0: def move_to_right_of(node) Chris@0: move_to node, :right Chris@0: end Chris@0: Chris@0: # Move the node to the child of another node (you can pass id only) Chris@0: def move_to_child_of(node) Chris@0: move_to node, :child Chris@0: end Chris@0: Chris@0: # Move the node to root nodes Chris@0: def move_to_root Chris@0: move_to nil, :root Chris@0: end Chris@0: Chris@0: def move_possible?(target) Chris@0: self != target && # Can't target self Chris@0: same_scope?(target) && # can't be in different scopes Chris@0: # !(left..right).include?(target.left..target.right) # this needs tested more Chris@0: # detect impossible move Chris@0: !((left <= target.left && right >= target.left) or (left <= target.right && right >= target.right)) Chris@0: end Chris@0: Chris@0: def to_text Chris@0: self_and_descendants.map do |node| Chris@0: "#{'*'*(node.level+1)} #{node.id} #{node.to_s} (#{node.parent_id}, #{node.left}, #{node.right})" Chris@0: end.join("\n") Chris@0: end Chris@0: Chris@0: protected Chris@0: Chris@0: def without_self(scope) Chris@0: scope.scoped :conditions => ["#{self.class.table_name}.#{self.class.primary_key} != ?", self] Chris@0: end Chris@0: Chris@0: # All nested set queries should use this nested_set_scope, which performs finds on Chris@0: # the base ActiveRecord class, using the :scope declared in the acts_as_nested_set Chris@0: # declaration. Chris@0: def nested_set_scope Chris@441: options = {:order => "#{self.class.table_name}.#{quoted_left_column_name}"} Chris@0: scopes = Array(acts_as_nested_set_options[:scope]) Chris@0: options[:conditions] = scopes.inject({}) do |conditions,attr| Chris@0: conditions.merge attr => self[attr] Chris@0: end unless scopes.empty? Chris@0: self.class.base_class.scoped options Chris@0: end Chris@0: Chris@0: # on creation, set automatically lft and rgt to the end of the tree Chris@0: def set_default_left_and_right Chris@0: maxright = nested_set_scope.maximum(right_column_name) || 0 Chris@0: # adds the new node to the right of all existing nodes Chris@0: self[left_column_name] = maxright + 1 Chris@0: self[right_column_name] = maxright + 2 Chris@0: end Chris@0: Chris@0: # Prunes a branch off of the tree, shifting all of the elements on the right Chris@0: # back to the left so the counts still work. Chris@0: def prune_from_tree Chris@441: return if right.nil? || left.nil? || leaf? || !self.class.exists?(id) Chris@0: Chris@0: delete_method = acts_as_nested_set_options[:dependent] == :destroy ? Chris@0: :destroy_all : :delete_all Chris@0: Chris@441: # TODO: should destroy children (not descendants) when deleted_method is :destroy_all Chris@0: self.class.base_class.transaction do Chris@210: reload_nested_set Chris@0: nested_set_scope.send(delete_method, Chris@0: ["#{quoted_left_column_name} > ? AND #{quoted_right_column_name} < ?", Chris@0: left, right] Chris@0: ) Chris@210: reload_nested_set Chris@210: diff = right - left + 1 Chris@0: nested_set_scope.update_all( Chris@0: ["#{quoted_left_column_name} = (#{quoted_left_column_name} - ?)", diff], Chris@0: ["#{quoted_left_column_name} >= ?", right] Chris@0: ) Chris@0: nested_set_scope.update_all( Chris@0: ["#{quoted_right_column_name} = (#{quoted_right_column_name} - ?)", diff], Chris@0: ["#{quoted_right_column_name} >= ?", right] Chris@0: ) Chris@0: end Chris@441: Chris@441: # Reload is needed because children may have updated their parent (self) during deletion. Chris@441: reload Chris@0: end Chris@0: Chris@0: # reload left, right, and parent Chris@0: def reload_nested_set Chris@0: reload(:select => "#{quoted_left_column_name}, " + Chris@0: "#{quoted_right_column_name}, #{quoted_parent_column_name}") Chris@0: end Chris@0: Chris@0: def move_to(target, position) Chris@0: raise ActiveRecord::ActiveRecordError, "You cannot move a new node" if self.new_record? Chris@0: return if callback(:before_move) == false Chris@0: transaction do Chris@0: if target.is_a? self.class.base_class Chris@0: target.reload_nested_set Chris@0: elsif position != :root Chris@0: # load object if node is not an object Chris@0: target = nested_set_scope.find(target) Chris@0: end Chris@0: self.reload_nested_set Chris@0: Chris@0: unless position == :root || move_possible?(target) Chris@0: raise ActiveRecord::ActiveRecordError, "Impossible move, target node cannot be inside moved tree." Chris@0: end Chris@0: Chris@0: bound = case position Chris@0: when :child; target[right_column_name] Chris@0: when :left; target[left_column_name] Chris@0: when :right; target[right_column_name] + 1 Chris@0: when :root; 1 Chris@0: else raise ActiveRecord::ActiveRecordError, "Position should be :child, :left, :right or :root ('#{position}' received)." Chris@0: end Chris@0: Chris@0: if bound > self[right_column_name] Chris@0: bound = bound - 1 Chris@0: other_bound = self[right_column_name] + 1 Chris@0: else Chris@0: other_bound = self[left_column_name] - 1 Chris@0: end Chris@0: Chris@0: # there would be no change Chris@0: return if bound == self[right_column_name] || bound == self[left_column_name] Chris@0: Chris@0: # we have defined the boundaries of two non-overlapping intervals, Chris@0: # so sorting puts both the intervals and their boundaries in order Chris@0: a, b, c, d = [self[left_column_name], self[right_column_name], bound, other_bound].sort Chris@0: Chris@0: new_parent = case position Chris@0: when :child; target.id Chris@0: when :root; nil Chris@0: else target[parent_column_name] Chris@0: end Chris@0: Chris@0: self.class.base_class.update_all([ Chris@0: "#{quoted_left_column_name} = CASE " + Chris@0: "WHEN #{quoted_left_column_name} BETWEEN :a AND :b " + Chris@0: "THEN #{quoted_left_column_name} + :d - :b " + Chris@0: "WHEN #{quoted_left_column_name} BETWEEN :c AND :d " + Chris@0: "THEN #{quoted_left_column_name} + :a - :c " + Chris@0: "ELSE #{quoted_left_column_name} END, " + Chris@0: "#{quoted_right_column_name} = CASE " + Chris@0: "WHEN #{quoted_right_column_name} BETWEEN :a AND :b " + Chris@0: "THEN #{quoted_right_column_name} + :d - :b " + Chris@0: "WHEN #{quoted_right_column_name} BETWEEN :c AND :d " + Chris@0: "THEN #{quoted_right_column_name} + :a - :c " + Chris@0: "ELSE #{quoted_right_column_name} END, " + Chris@0: "#{quoted_parent_column_name} = CASE " + Chris@0: "WHEN #{self.class.base_class.primary_key} = :id THEN :new_parent " + Chris@0: "ELSE #{quoted_parent_column_name} END", Chris@0: {:a => a, :b => b, :c => c, :d => d, :id => self.id, :new_parent => new_parent} Chris@0: ], nested_set_scope.proxy_options[:conditions]) Chris@0: end Chris@0: target.reload_nested_set if target Chris@0: self.reload_nested_set Chris@0: callback(:after_move) Chris@0: end Chris@0: Chris@0: end Chris@0: Chris@0: end Chris@0: end Chris@0: end