annotate .svn/pristine/fd/fd672f720e53aab0a0184fd9c83dcdbc26d28825.svn-base @ 1524:82fac3dcf466 redmine-2.5-integration

Fix failure to interpret Javascript when autocompleting members for project
author Chris Cannam <chris.cannam@soundsoftware.ac.uk>
date Thu, 11 Sep 2014 10:24:38 +0100
parents 261b3d9a4903
children
rev   line source
Chris@1464 1 module CollectiveIdea #:nodoc:
Chris@1464 2 module Acts #:nodoc:
Chris@1464 3 module NestedSet #:nodoc:
Chris@1464 4
Chris@1464 5 # This acts provides Nested Set functionality. Nested Set is a smart way to implement
Chris@1464 6 # an _ordered_ tree, with the added feature that you can select the children and all of their
Chris@1464 7 # descendants with a single query. The drawback is that insertion or move need some complex
Chris@1464 8 # sql queries. But everything is done here by this module!
Chris@1464 9 #
Chris@1464 10 # Nested sets are appropriate each time you want either an orderd tree (menus,
Chris@1464 11 # commercial categories) or an efficient way of querying big trees (threaded posts).
Chris@1464 12 #
Chris@1464 13 # == API
Chris@1464 14 #
Chris@1464 15 # Methods names are aligned with acts_as_tree as much as possible to make replacment from one
Chris@1464 16 # by another easier.
Chris@1464 17 #
Chris@1464 18 # item.children.create(:name => "child1")
Chris@1464 19 #
Chris@1464 20
Chris@1464 21 # Configuration options are:
Chris@1464 22 #
Chris@1464 23 # * +:parent_column+ - specifies the column name to use for keeping the position integer (default: parent_id)
Chris@1464 24 # * +:left_column+ - column name for left boundry data, default "lft"
Chris@1464 25 # * +:right_column+ - column name for right boundry data, default "rgt"
Chris@1464 26 # * +:scope+ - restricts what is to be considered a list. Given a symbol, it'll attach "_id"
Chris@1464 27 # (if it hasn't been already) and use that as the foreign key restriction. You
Chris@1464 28 # can also pass an array to scope by multiple attributes.
Chris@1464 29 # Example: <tt>acts_as_nested_set :scope => [:notable_id, :notable_type]</tt>
Chris@1464 30 # * +:dependent+ - behavior for cascading destroy. If set to :destroy, all the
Chris@1464 31 # child objects are destroyed alongside this object by calling their destroy
Chris@1464 32 # method. If set to :delete_all (default), all the child objects are deleted
Chris@1464 33 # without calling their destroy method.
Chris@1464 34 # * +:counter_cache+ adds a counter cache for the number of children.
Chris@1464 35 # defaults to false.
Chris@1464 36 # Example: <tt>acts_as_nested_set :counter_cache => :children_count</tt>
Chris@1464 37 #
Chris@1464 38 # See CollectiveIdea::Acts::NestedSet::Model::ClassMethods for a list of class methods and
Chris@1464 39 # CollectiveIdea::Acts::NestedSet::Model for a list of instance methods added
Chris@1464 40 # to acts_as_nested_set models
Chris@1464 41 def acts_as_nested_set(options = {})
Chris@1464 42 options = {
Chris@1464 43 :parent_column => 'parent_id',
Chris@1464 44 :left_column => 'lft',
Chris@1464 45 :right_column => 'rgt',
Chris@1464 46 :dependent => :delete_all, # or :destroy
Chris@1464 47 :counter_cache => false,
Chris@1464 48 :order => 'id'
Chris@1464 49 }.merge(options)
Chris@1464 50
Chris@1464 51 if options[:scope].is_a?(Symbol) && options[:scope].to_s !~ /_id$/
Chris@1464 52 options[:scope] = "#{options[:scope]}_id".intern
Chris@1464 53 end
Chris@1464 54
Chris@1464 55 class_attribute :acts_as_nested_set_options
Chris@1464 56 self.acts_as_nested_set_options = options
Chris@1464 57
Chris@1464 58 include CollectiveIdea::Acts::NestedSet::Model
Chris@1464 59 include Columns
Chris@1464 60 extend Columns
Chris@1464 61
Chris@1464 62 belongs_to :parent, :class_name => self.base_class.to_s,
Chris@1464 63 :foreign_key => parent_column_name,
Chris@1464 64 :counter_cache => options[:counter_cache],
Chris@1464 65 :inverse_of => :children
Chris@1464 66 has_many :children, :class_name => self.base_class.to_s,
Chris@1464 67 :foreign_key => parent_column_name, :order => left_column_name,
Chris@1464 68 :inverse_of => :parent,
Chris@1464 69 :before_add => options[:before_add],
Chris@1464 70 :after_add => options[:after_add],
Chris@1464 71 :before_remove => options[:before_remove],
Chris@1464 72 :after_remove => options[:after_remove]
Chris@1464 73
Chris@1464 74 attr_accessor :skip_before_destroy
Chris@1464 75
Chris@1464 76 before_create :set_default_left_and_right
Chris@1464 77 before_save :store_new_parent
Chris@1464 78 after_save :move_to_new_parent
Chris@1464 79 before_destroy :destroy_descendants
Chris@1464 80
Chris@1464 81 # no assignment to structure fields
Chris@1464 82 [left_column_name, right_column_name].each do |column|
Chris@1464 83 module_eval <<-"end_eval", __FILE__, __LINE__
Chris@1464 84 def #{column}=(x)
Chris@1464 85 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@1464 86 end
Chris@1464 87 end_eval
Chris@1464 88 end
Chris@1464 89
Chris@1464 90 define_model_callbacks :move
Chris@1464 91 end
Chris@1464 92
Chris@1464 93 module Model
Chris@1464 94 extend ActiveSupport::Concern
Chris@1464 95
Chris@1464 96 module ClassMethods
Chris@1464 97 # Returns the first root
Chris@1464 98 def root
Chris@1464 99 roots.first
Chris@1464 100 end
Chris@1464 101
Chris@1464 102 def roots
Chris@1464 103 where(parent_column_name => nil).order(quoted_left_column_name)
Chris@1464 104 end
Chris@1464 105
Chris@1464 106 def leaves
Chris@1464 107 where("#{quoted_right_column_name} - #{quoted_left_column_name} = 1").order(quoted_left_column_name)
Chris@1464 108 end
Chris@1464 109
Chris@1464 110 def valid?
Chris@1464 111 left_and_rights_valid? && no_duplicates_for_columns? && all_roots_valid?
Chris@1464 112 end
Chris@1464 113
Chris@1464 114 def left_and_rights_valid?
Chris@1464 115 joins("LEFT OUTER JOIN #{quoted_table_name} AS parent ON " +
Chris@1464 116 "#{quoted_table_name}.#{quoted_parent_column_name} = parent.#{primary_key}").
Chris@1464 117 where(
Chris@1464 118 "#{quoted_table_name}.#{quoted_left_column_name} IS NULL OR " +
Chris@1464 119 "#{quoted_table_name}.#{quoted_right_column_name} IS NULL OR " +
Chris@1464 120 "#{quoted_table_name}.#{quoted_left_column_name} >= " +
Chris@1464 121 "#{quoted_table_name}.#{quoted_right_column_name} OR " +
Chris@1464 122 "(#{quoted_table_name}.#{quoted_parent_column_name} IS NOT NULL AND " +
Chris@1464 123 "(#{quoted_table_name}.#{quoted_left_column_name} <= parent.#{quoted_left_column_name} OR " +
Chris@1464 124 "#{quoted_table_name}.#{quoted_right_column_name} >= parent.#{quoted_right_column_name}))"
Chris@1464 125 ).count == 0
Chris@1464 126 end
Chris@1464 127
Chris@1464 128 def no_duplicates_for_columns?
Chris@1464 129 scope_string = Array(acts_as_nested_set_options[:scope]).map do |c|
Chris@1464 130 connection.quote_column_name(c)
Chris@1464 131 end.push(nil).join(", ")
Chris@1464 132 [quoted_left_column_name, quoted_right_column_name].all? do |column|
Chris@1464 133 # No duplicates
Chris@1464 134 select("#{scope_string}#{column}, COUNT(#{column})").
Chris@1464 135 group("#{scope_string}#{column}").
Chris@1464 136 having("COUNT(#{column}) > 1").
Chris@1464 137 first.nil?
Chris@1464 138 end
Chris@1464 139 end
Chris@1464 140
Chris@1464 141 # Wrapper for each_root_valid? that can deal with scope.
Chris@1464 142 def all_roots_valid?
Chris@1464 143 if acts_as_nested_set_options[:scope]
Chris@1464 144 roots.group(scope_column_names).group_by{|record| scope_column_names.collect{|col| record.send(col.to_sym)}}.all? do |scope, grouped_roots|
Chris@1464 145 each_root_valid?(grouped_roots)
Chris@1464 146 end
Chris@1464 147 else
Chris@1464 148 each_root_valid?(roots)
Chris@1464 149 end
Chris@1464 150 end
Chris@1464 151
Chris@1464 152 def each_root_valid?(roots_to_validate)
Chris@1464 153 left = right = 0
Chris@1464 154 roots_to_validate.all? do |root|
Chris@1464 155 (root.left > left && root.right > right).tap do
Chris@1464 156 left = root.left
Chris@1464 157 right = root.right
Chris@1464 158 end
Chris@1464 159 end
Chris@1464 160 end
Chris@1464 161
Chris@1464 162 # Rebuilds the left & rights if unset or invalid.
Chris@1464 163 # Also very useful for converting from acts_as_tree.
Chris@1464 164 def rebuild!(validate_nodes = true)
Chris@1464 165 # Don't rebuild a valid tree.
Chris@1464 166 return true if valid?
Chris@1464 167
Chris@1464 168 scope = lambda{|node|}
Chris@1464 169 if acts_as_nested_set_options[:scope]
Chris@1464 170 scope = lambda{|node|
Chris@1464 171 scope_column_names.inject(""){|str, column_name|
Chris@1464 172 str << "AND #{connection.quote_column_name(column_name)} = #{connection.quote(node.send(column_name.to_sym))} "
Chris@1464 173 }
Chris@1464 174 }
Chris@1464 175 end
Chris@1464 176 indices = {}
Chris@1464 177
Chris@1464 178 set_left_and_rights = lambda do |node|
Chris@1464 179 # set left
Chris@1464 180 node[left_column_name] = indices[scope.call(node)] += 1
Chris@1464 181 # find
Chris@1464 182 where(["#{quoted_parent_column_name} = ? #{scope.call(node)}", node]).order(acts_as_nested_set_options[:order]).each{|n| set_left_and_rights.call(n) }
Chris@1464 183 # set right
Chris@1464 184 node[right_column_name] = indices[scope.call(node)] += 1
Chris@1464 185 node.save!(:validate => validate_nodes)
Chris@1464 186 end
Chris@1464 187
Chris@1464 188 # Find root node(s)
Chris@1464 189 root_nodes = where("#{quoted_parent_column_name} IS NULL").order(acts_as_nested_set_options[:order]).each do |root_node|
Chris@1464 190 # setup index for this scope
Chris@1464 191 indices[scope.call(root_node)] ||= 0
Chris@1464 192 set_left_and_rights.call(root_node)
Chris@1464 193 end
Chris@1464 194 end
Chris@1464 195
Chris@1464 196 # Iterates over tree elements and determines the current level in the tree.
Chris@1464 197 # Only accepts default ordering, odering by an other column than lft
Chris@1464 198 # does not work. This method is much more efficent than calling level
Chris@1464 199 # because it doesn't require any additional database queries.
Chris@1464 200 #
Chris@1464 201 # Example:
Chris@1464 202 # Category.each_with_level(Category.root.self_and_descendants) do |o, level|
Chris@1464 203 #
Chris@1464 204 def each_with_level(objects)
Chris@1464 205 path = [nil]
Chris@1464 206 objects.each do |o|
Chris@1464 207 if o.parent_id != path.last
Chris@1464 208 # we are on a new level, did we decent or ascent?
Chris@1464 209 if path.include?(o.parent_id)
Chris@1464 210 # remove wrong wrong tailing paths elements
Chris@1464 211 path.pop while path.last != o.parent_id
Chris@1464 212 else
Chris@1464 213 path << o.parent_id
Chris@1464 214 end
Chris@1464 215 end
Chris@1464 216 yield(o, path.length - 1)
Chris@1464 217 end
Chris@1464 218 end
Chris@1464 219 end
Chris@1464 220
Chris@1464 221 # 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@1464 222 #
Chris@1464 223 # category.self_and_descendants.count
Chris@1464 224 # category.ancestors.find(:all, :conditions => "name like '%foo%'")
Chris@1464 225
Chris@1464 226 # Value of the parent column
Chris@1464 227 def parent_id
Chris@1464 228 self[parent_column_name]
Chris@1464 229 end
Chris@1464 230
Chris@1464 231 # Value of the left column
Chris@1464 232 def left
Chris@1464 233 self[left_column_name]
Chris@1464 234 end
Chris@1464 235
Chris@1464 236 # Value of the right column
Chris@1464 237 def right
Chris@1464 238 self[right_column_name]
Chris@1464 239 end
Chris@1464 240
Chris@1464 241 # Returns true if this is a root node.
Chris@1464 242 def root?
Chris@1464 243 parent_id.nil?
Chris@1464 244 end
Chris@1464 245
Chris@1464 246 def leaf?
Chris@1464 247 new_record? || (right - left == 1)
Chris@1464 248 end
Chris@1464 249
Chris@1464 250 # Returns true is this is a child node
Chris@1464 251 def child?
Chris@1464 252 !parent_id.nil?
Chris@1464 253 end
Chris@1464 254
Chris@1464 255 # Returns root
Chris@1464 256 def root
Chris@1464 257 self_and_ancestors.where(parent_column_name => nil).first
Chris@1464 258 end
Chris@1464 259
Chris@1464 260 # Returns the array of all parents and self
Chris@1464 261 def self_and_ancestors
Chris@1464 262 nested_set_scope.where([
Chris@1464 263 "#{self.class.quoted_table_name}.#{quoted_left_column_name} <= ? AND #{self.class.quoted_table_name}.#{quoted_right_column_name} >= ?", left, right
Chris@1464 264 ])
Chris@1464 265 end
Chris@1464 266
Chris@1464 267 # Returns an array of all parents
Chris@1464 268 def ancestors
Chris@1464 269 without_self self_and_ancestors
Chris@1464 270 end
Chris@1464 271
Chris@1464 272 # Returns the array of all children of the parent, including self
Chris@1464 273 def self_and_siblings
Chris@1464 274 nested_set_scope.where(parent_column_name => parent_id)
Chris@1464 275 end
Chris@1464 276
Chris@1464 277 # Returns the array of all children of the parent, except self
Chris@1464 278 def siblings
Chris@1464 279 without_self self_and_siblings
Chris@1464 280 end
Chris@1464 281
Chris@1464 282 # Returns a set of all of its nested children which do not have children
Chris@1464 283 def leaves
Chris@1464 284 descendants.where("#{self.class.quoted_table_name}.#{quoted_right_column_name} - #{self.class.quoted_table_name}.#{quoted_left_column_name} = 1")
Chris@1464 285 end
Chris@1464 286
Chris@1464 287 # Returns the level of this object in the tree
Chris@1464 288 # root level is 0
Chris@1464 289 def level
Chris@1464 290 parent_id.nil? ? 0 : ancestors.count
Chris@1464 291 end
Chris@1464 292
Chris@1464 293 # Returns a set of itself and all of its nested children
Chris@1464 294 def self_and_descendants
Chris@1464 295 nested_set_scope.where([
Chris@1464 296 "#{self.class.quoted_table_name}.#{quoted_left_column_name} >= ? AND #{self.class.quoted_table_name}.#{quoted_right_column_name} <= ?", left, right
Chris@1464 297 ])
Chris@1464 298 end
Chris@1464 299
Chris@1464 300 # Returns a set of all of its children and nested children
Chris@1464 301 def descendants
Chris@1464 302 without_self self_and_descendants
Chris@1464 303 end
Chris@1464 304
Chris@1464 305 def is_descendant_of?(other)
Chris@1464 306 other.left < self.left && self.left < other.right && same_scope?(other)
Chris@1464 307 end
Chris@1464 308
Chris@1464 309 def is_or_is_descendant_of?(other)
Chris@1464 310 other.left <= self.left && self.left < other.right && same_scope?(other)
Chris@1464 311 end
Chris@1464 312
Chris@1464 313 def is_ancestor_of?(other)
Chris@1464 314 self.left < other.left && other.left < self.right && same_scope?(other)
Chris@1464 315 end
Chris@1464 316
Chris@1464 317 def is_or_is_ancestor_of?(other)
Chris@1464 318 self.left <= other.left && other.left < self.right && same_scope?(other)
Chris@1464 319 end
Chris@1464 320
Chris@1464 321 # Check if other model is in the same scope
Chris@1464 322 def same_scope?(other)
Chris@1464 323 Array(acts_as_nested_set_options[:scope]).all? do |attr|
Chris@1464 324 self.send(attr) == other.send(attr)
Chris@1464 325 end
Chris@1464 326 end
Chris@1464 327
Chris@1464 328 # Find the first sibling to the left
Chris@1464 329 def left_sibling
Chris@1464 330 siblings.where(["#{self.class.quoted_table_name}.#{quoted_left_column_name} < ?", left]).
Chris@1464 331 order("#{self.class.quoted_table_name}.#{quoted_left_column_name} DESC").last
Chris@1464 332 end
Chris@1464 333
Chris@1464 334 # Find the first sibling to the right
Chris@1464 335 def right_sibling
Chris@1464 336 siblings.where(["#{self.class.quoted_table_name}.#{quoted_left_column_name} > ?", left]).first
Chris@1464 337 end
Chris@1464 338
Chris@1464 339 # Shorthand method for finding the left sibling and moving to the left of it.
Chris@1464 340 def move_left
Chris@1464 341 move_to_left_of left_sibling
Chris@1464 342 end
Chris@1464 343
Chris@1464 344 # Shorthand method for finding the right sibling and moving to the right of it.
Chris@1464 345 def move_right
Chris@1464 346 move_to_right_of right_sibling
Chris@1464 347 end
Chris@1464 348
Chris@1464 349 # Move the node to the left of another node (you can pass id only)
Chris@1464 350 def move_to_left_of(node)
Chris@1464 351 move_to node, :left
Chris@1464 352 end
Chris@1464 353
Chris@1464 354 # Move the node to the left of another node (you can pass id only)
Chris@1464 355 def move_to_right_of(node)
Chris@1464 356 move_to node, :right
Chris@1464 357 end
Chris@1464 358
Chris@1464 359 # Move the node to the child of another node (you can pass id only)
Chris@1464 360 def move_to_child_of(node)
Chris@1464 361 move_to node, :child
Chris@1464 362 end
Chris@1464 363
Chris@1464 364 # Move the node to root nodes
Chris@1464 365 def move_to_root
Chris@1464 366 move_to nil, :root
Chris@1464 367 end
Chris@1464 368
Chris@1464 369 def move_possible?(target)
Chris@1464 370 self != target && # Can't target self
Chris@1464 371 same_scope?(target) && # can't be in different scopes
Chris@1464 372 # !(left..right).include?(target.left..target.right) # this needs tested more
Chris@1464 373 # detect impossible move
Chris@1464 374 !((left <= target.left && right >= target.left) or (left <= target.right && right >= target.right))
Chris@1464 375 end
Chris@1464 376
Chris@1464 377 def to_text
Chris@1464 378 self_and_descendants.map do |node|
Chris@1464 379 "#{'*'*(node.level+1)} #{node.id} #{node.to_s} (#{node.parent_id}, #{node.left}, #{node.right})"
Chris@1464 380 end.join("\n")
Chris@1464 381 end
Chris@1464 382
Chris@1464 383 protected
Chris@1464 384
Chris@1464 385 def without_self(scope)
Chris@1464 386 scope.where(["#{self.class.quoted_table_name}.#{self.class.primary_key} != ?", self])
Chris@1464 387 end
Chris@1464 388
Chris@1464 389 # All nested set queries should use this nested_set_scope, which performs finds on
Chris@1464 390 # the base ActiveRecord class, using the :scope declared in the acts_as_nested_set
Chris@1464 391 # declaration.
Chris@1464 392 def nested_set_scope(options = {})
Chris@1464 393 options = {:order => "#{self.class.quoted_table_name}.#{quoted_left_column_name}"}.merge(options)
Chris@1464 394 scopes = Array(acts_as_nested_set_options[:scope])
Chris@1464 395 options[:conditions] = scopes.inject({}) do |conditions,attr|
Chris@1464 396 conditions.merge attr => self[attr]
Chris@1464 397 end unless scopes.empty?
Chris@1464 398 self.class.base_class.scoped options
Chris@1464 399 end
Chris@1464 400
Chris@1464 401 def store_new_parent
Chris@1464 402 @move_to_new_parent_id = send("#{parent_column_name}_changed?") ? parent_id : false
Chris@1464 403 true # force callback to return true
Chris@1464 404 end
Chris@1464 405
Chris@1464 406 def move_to_new_parent
Chris@1464 407 if @move_to_new_parent_id.nil?
Chris@1464 408 move_to_root
Chris@1464 409 elsif @move_to_new_parent_id
Chris@1464 410 move_to_child_of(@move_to_new_parent_id)
Chris@1464 411 end
Chris@1464 412 end
Chris@1464 413
Chris@1464 414 # on creation, set automatically lft and rgt to the end of the tree
Chris@1464 415 def set_default_left_and_right
Chris@1464 416 highest_right_row = nested_set_scope(:order => "#{quoted_right_column_name} desc").limit(1).lock(true).first
Chris@1464 417 maxright = highest_right_row ? (highest_right_row[right_column_name] || 0) : 0
Chris@1464 418 # adds the new node to the right of all existing nodes
Chris@1464 419 self[left_column_name] = maxright + 1
Chris@1464 420 self[right_column_name] = maxright + 2
Chris@1464 421 end
Chris@1464 422
Chris@1464 423 def in_tenacious_transaction(&block)
Chris@1464 424 retry_count = 0
Chris@1464 425 begin
Chris@1464 426 transaction(&block)
Chris@1464 427 rescue ActiveRecord::StatementInvalid => error
Chris@1464 428 raise unless connection.open_transactions.zero?
Chris@1464 429 raise unless error.message =~ /Deadlock found when trying to get lock|Lock wait timeout exceeded/
Chris@1464 430 raise unless retry_count < 10
Chris@1464 431 retry_count += 1
Chris@1464 432 logger.info "Deadlock detected on retry #{retry_count}, restarting transaction"
Chris@1464 433 sleep(rand(retry_count)*0.1) # Aloha protocol
Chris@1464 434 retry
Chris@1464 435 end
Chris@1464 436 end
Chris@1464 437
Chris@1464 438 # Prunes a branch off of the tree, shifting all of the elements on the right
Chris@1464 439 # back to the left so the counts still work.
Chris@1464 440 def destroy_descendants
Chris@1464 441 return if right.nil? || left.nil? || skip_before_destroy
Chris@1464 442
Chris@1464 443 in_tenacious_transaction do
Chris@1464 444 reload_nested_set
Chris@1464 445 # select the rows in the model that extend past the deletion point and apply a lock
Chris@1464 446 self.class.base_class.
Chris@1464 447 select("id").
Chris@1464 448 where("#{quoted_left_column_name} >= ?", left).
Chris@1464 449 lock(true).
Chris@1464 450 all
Chris@1464 451
Chris@1464 452 if acts_as_nested_set_options[:dependent] == :destroy
Chris@1464 453 descendants.each do |model|
Chris@1464 454 model.skip_before_destroy = true
Chris@1464 455 model.destroy
Chris@1464 456 end
Chris@1464 457 else
Chris@1464 458 nested_set_scope.delete_all(
Chris@1464 459 ["#{quoted_left_column_name} > ? AND #{quoted_right_column_name} < ?",
Chris@1464 460 left, right]
Chris@1464 461 )
Chris@1464 462 end
Chris@1464 463
Chris@1464 464 # update lefts and rights for remaining nodes
Chris@1464 465 diff = right - left + 1
Chris@1464 466 nested_set_scope.update_all(
Chris@1464 467 ["#{quoted_left_column_name} = (#{quoted_left_column_name} - ?)", diff],
Chris@1464 468 ["#{quoted_left_column_name} > ?", right]
Chris@1464 469 )
Chris@1464 470 nested_set_scope.update_all(
Chris@1464 471 ["#{quoted_right_column_name} = (#{quoted_right_column_name} - ?)", diff],
Chris@1464 472 ["#{quoted_right_column_name} > ?", right]
Chris@1464 473 )
Chris@1464 474
Chris@1464 475 reload
Chris@1464 476 # Don't allow multiple calls to destroy to corrupt the set
Chris@1464 477 self.skip_before_destroy = true
Chris@1464 478 end
Chris@1464 479 end
Chris@1464 480
Chris@1464 481 # reload left, right, and parent
Chris@1464 482 def reload_nested_set
Chris@1464 483 reload(
Chris@1464 484 :select => "#{quoted_left_column_name}, #{quoted_right_column_name}, #{quoted_parent_column_name}",
Chris@1464 485 :lock => true
Chris@1464 486 )
Chris@1464 487 end
Chris@1464 488
Chris@1464 489 def move_to(target, position)
Chris@1464 490 raise ActiveRecord::ActiveRecordError, "You cannot move a new node" if self.new_record?
Chris@1464 491 run_callbacks :move do
Chris@1464 492 in_tenacious_transaction do
Chris@1464 493 if target.is_a? self.class.base_class
Chris@1464 494 target.reload_nested_set
Chris@1464 495 elsif position != :root
Chris@1464 496 # load object if node is not an object
Chris@1464 497 target = nested_set_scope.find(target)
Chris@1464 498 end
Chris@1464 499 self.reload_nested_set
Chris@1464 500
Chris@1464 501 unless position == :root || move_possible?(target)
Chris@1464 502 raise ActiveRecord::ActiveRecordError, "Impossible move, target node cannot be inside moved tree."
Chris@1464 503 end
Chris@1464 504
Chris@1464 505 bound = case position
Chris@1464 506 when :child; target[right_column_name]
Chris@1464 507 when :left; target[left_column_name]
Chris@1464 508 when :right; target[right_column_name] + 1
Chris@1464 509 when :root; 1
Chris@1464 510 else raise ActiveRecord::ActiveRecordError, "Position should be :child, :left, :right or :root ('#{position}' received)."
Chris@1464 511 end
Chris@1464 512
Chris@1464 513 if bound > self[right_column_name]
Chris@1464 514 bound = bound - 1
Chris@1464 515 other_bound = self[right_column_name] + 1
Chris@1464 516 else
Chris@1464 517 other_bound = self[left_column_name] - 1
Chris@1464 518 end
Chris@1464 519
Chris@1464 520 # there would be no change
Chris@1464 521 return if bound == self[right_column_name] || bound == self[left_column_name]
Chris@1464 522
Chris@1464 523 # we have defined the boundaries of two non-overlapping intervals,
Chris@1464 524 # so sorting puts both the intervals and their boundaries in order
Chris@1464 525 a, b, c, d = [self[left_column_name], self[right_column_name], bound, other_bound].sort
Chris@1464 526
Chris@1464 527 # select the rows in the model between a and d, and apply a lock
Chris@1464 528 self.class.base_class.select('id').lock(true).where(
Chris@1464 529 ["#{quoted_left_column_name} >= :a and #{quoted_right_column_name} <= :d", {:a => a, :d => d}]
Chris@1464 530 )
Chris@1464 531
Chris@1464 532 new_parent = case position
Chris@1464 533 when :child; target.id
Chris@1464 534 when :root; nil
Chris@1464 535 else target[parent_column_name]
Chris@1464 536 end
Chris@1464 537
Chris@1464 538 self.nested_set_scope.update_all([
Chris@1464 539 "#{quoted_left_column_name} = CASE " +
Chris@1464 540 "WHEN #{quoted_left_column_name} BETWEEN :a AND :b " +
Chris@1464 541 "THEN #{quoted_left_column_name} + :d - :b " +
Chris@1464 542 "WHEN #{quoted_left_column_name} BETWEEN :c AND :d " +
Chris@1464 543 "THEN #{quoted_left_column_name} + :a - :c " +
Chris@1464 544 "ELSE #{quoted_left_column_name} END, " +
Chris@1464 545 "#{quoted_right_column_name} = CASE " +
Chris@1464 546 "WHEN #{quoted_right_column_name} BETWEEN :a AND :b " +
Chris@1464 547 "THEN #{quoted_right_column_name} + :d - :b " +
Chris@1464 548 "WHEN #{quoted_right_column_name} BETWEEN :c AND :d " +
Chris@1464 549 "THEN #{quoted_right_column_name} + :a - :c " +
Chris@1464 550 "ELSE #{quoted_right_column_name} END, " +
Chris@1464 551 "#{quoted_parent_column_name} = CASE " +
Chris@1464 552 "WHEN #{self.class.base_class.primary_key} = :id THEN :new_parent " +
Chris@1464 553 "ELSE #{quoted_parent_column_name} END",
Chris@1464 554 {:a => a, :b => b, :c => c, :d => d, :id => self.id, :new_parent => new_parent}
Chris@1464 555 ])
Chris@1464 556 end
Chris@1464 557 target.reload_nested_set if target
Chris@1464 558 self.reload_nested_set
Chris@1464 559 end
Chris@1464 560 end
Chris@1464 561
Chris@1464 562 end
Chris@1464 563
Chris@1464 564 # Mixed into both classes and instances to provide easy access to the column names
Chris@1464 565 module Columns
Chris@1464 566 def left_column_name
Chris@1464 567 acts_as_nested_set_options[:left_column]
Chris@1464 568 end
Chris@1464 569
Chris@1464 570 def right_column_name
Chris@1464 571 acts_as_nested_set_options[:right_column]
Chris@1464 572 end
Chris@1464 573
Chris@1464 574 def parent_column_name
Chris@1464 575 acts_as_nested_set_options[:parent_column]
Chris@1464 576 end
Chris@1464 577
Chris@1464 578 def scope_column_names
Chris@1464 579 Array(acts_as_nested_set_options[:scope])
Chris@1464 580 end
Chris@1464 581
Chris@1464 582 def quoted_left_column_name
Chris@1464 583 connection.quote_column_name(left_column_name)
Chris@1464 584 end
Chris@1464 585
Chris@1464 586 def quoted_right_column_name
Chris@1464 587 connection.quote_column_name(right_column_name)
Chris@1464 588 end
Chris@1464 589
Chris@1464 590 def quoted_parent_column_name
Chris@1464 591 connection.quote_column_name(parent_column_name)
Chris@1464 592 end
Chris@1464 593
Chris@1464 594 def quoted_scope_column_names
Chris@1464 595 scope_column_names.collect {|column_name| connection.quote_column_name(column_name) }
Chris@1464 596 end
Chris@1464 597 end
Chris@1464 598
Chris@1464 599 end
Chris@1464 600 end
Chris@1464 601 end