Sha256: 0fa33d01bf9dd33aa4fa8b04e0bc42beda2024b58006f18ff790bc182c81a476

Contents?: true

Size: 1.3 KB

Versions: 8

Compression:

Stored size: 1.3 KB

Contents

module Mongoid::Acts::NestedSet

  module Rebuild

    # Rebuilds the left & rights if unset or invalid.  Also very useful for converting from acts_as_tree.
    # Warning: Very expensive!
    def rebuild!(options = {})
      # Don't rebuild a valid tree.
      return true if valid?

      scope = lambda{ |node| {} }
      if acts_as_nested_set_options[:scope]
        scope = lambda { |node| node.nested_set_scope.options.merge(node.nested_set_scope.selector) }
      end
      indices = {}

      set_left_and_rights = lambda do |node|
        # set left
        left = (indices[scope.call(node)] += 1)
        # find
        node.nested_set_scope.where(parent_field_name => node.id).asc(left_field_name).asc(right_field_name).each { |n| set_left_and_rights.call(n) }
        # set right
        right = (indices[scope.call(node)] += 1)

        node.class.collection.find(:_id => node.id).update(
          {"$set" => {left_field_name => left, right_field_name => right}},
          {:safe => true}
        )
      end

      # Find root node(s)
      root_nodes = self.where(parent_field_name => nil).asc(left_field_name).asc(right_field_name).asc(:_id).each do |root_node|
        # setup index for this scope
        indices[scope.call(root_node)] ||= 0
        set_left_and_rights.call(root_node)
      end
    end

  end
end

Version data entries

8 entries across 7 versions & 3 rubygems

Version Path
glebtv-mongoid_nested_set-0.4.2 lib/mongoid_nested_set/rebuild.rb
glebtv-mongoid_nested_set-0.4.1 lib/mongoid_nested_set/rebuild.rb
glebtv-mongoid_nested_set-0.4.0 lib/mongoid_nested_set/rebuild.rb
glebtv-mongoid_nested_set-0.3.0 lib/mongoid_nested_set/rebuild.rb
sunrise-cms-0.5.0.rc1 vendor/bundle/ruby/1.9.1/gems/mongoid_nested_set-0.2.1/lib/mongoid_nested_set/rebuild.rb
sunrise-cms-0.5.0.rc1 vendor/bundle/ruby/1.9.1/bundler/gems/mongoid_nested_set-d482b2642889/lib/mongoid_nested_set/rebuild.rb
mongoid_nested_set-0.2.1 lib/mongoid_nested_set/rebuild.rb
mongoid_nested_set-0.2.0 lib/mongoid_nested_set/rebuild.rb