NestedSet.php 32.6 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
<?php
/*
 *    $Id$
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * This software consists of voluntary contributions made by many individuals
 * and is licensed under the LGPL. For more information, see
 * <http://www.phpdoctrine.com>.
 */
/**
 * Doctrine_Node_NestedSet
 *
24 25 26 27 28 29 30 31
 * @package    Doctrine
 * @license    http://www.opensource.org/licenses/lgpl-license.php LGPL
 * @category   Object Relational Mapping
 * @link       www.phpdoctrine.com
 * @since      1.0
 * @version    $Revision$
 * @author     Joe Simms <joe.simms@websites4.com>
 * @author     Roman Borschel <roman@code-factory.org>     
32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
 */
class Doctrine_Node_NestedSet extends Doctrine_Node implements Doctrine_Node_Interface
{
    /**
     * test if node has previous sibling
     *
     * @return bool            
     */
    public function hasPrevSibling()
    {
        return $this->isValidNode($this->getPrevSibling());        
    }

    /**
     * test if node has next sibling
     *
     * @return bool            
     */ 
    public function hasNextSibling()
    {
        return $this->isValidNode($this->getNextSibling());        
    }

    /**
     * test if node has children
     *
     * @return bool            
     */
    public function hasChildren()
    {
        return (($this->getRightValue() - $this->getLeftValue() ) >1 );        
    }

    /**
     * test if node has parent
     *
     * @return bool            
     */
    public function hasParent()
    {
        return !$this->isRoot();
    }

    /**
     * gets record of prev sibling or empty record
     *
     * @return object     Doctrine_Record            
     */
    public function getPrevSibling()
    {
82
        $baseAlias = $this->_tree->getBaseAlias();
83
        $q = $this->_tree->getBaseQuery();
84
        $q = $q->addWhere("$baseAlias.rgt = ?", $this->getLeftValue() - 1);
85 86
        $q = $this->_tree->returnQueryWithRootId($q, $this->getRootValue());
        $result = $q->execute();
87

88 89 90 91 92 93 94 95 96
        if (count($result) <= 0) {
            return false;
        }
        
        if ($result instanceof Doctrine_Collection) {
            $sibling = $result->getFirst();
        } else if (is_array($result)) {
            $sibling = array_shift($result);
        }
97
        
98
        return $sibling;
99 100 101 102 103 104 105 106 107
    }

    /**
     * gets record of next sibling or empty record
     *
     * @return object     Doctrine_Record            
     */
    public function getNextSibling()
    {
108
        $baseAlias = $this->_tree->getBaseAlias();
109
        $q = $this->_tree->getBaseQuery();
110
        $q = $q->addWhere("$baseAlias.lft = ?", $this->getRightValue() + 1);
111 112
        $q = $this->_tree->returnQueryWithRootId($q, $this->getRootValue());
        $result = $q->execute();
113

114 115 116
        if (count($result) <= 0) {
            return false;
        }
117
        
118 119 120 121 122 123 124
        if ($result instanceof Doctrine_Collection) {
            $sibling = $result->getFirst();
        } else if (is_array($result)) {
            $sibling = array_shift($result);
        }
        
        return $sibling;
125 126 127 128 129 130 131 132 133 134 135
    }

    /**
     * gets siblings for node
     *
     * @return array     array of sibling Doctrine_Record objects            
     */
    public function getSiblings($includeNode = false)
    {
        $parent = $this->getParent();
        $siblings = array();
136 137 138
        if ($parent->exists()) {
            foreach ($parent->getNode()->getChildren() as $child) {
                if ($this->isEqualTo($child) && !$includeNode) {
139
                    continue;
140
                }
141
                $siblings[] = $child;
142
            }        
143 144 145 146 147 148 149 150 151 152 153
        }
        return $siblings;
    }

    /**
     * gets record of first child or empty record
     *
     * @return object     Doctrine_Record            
     */
    public function getFirstChild()
    {
154
        $baseAlias = $this->_tree->getBaseAlias();
155
        $q = $this->_tree->getBaseQuery();
156
        $q->addWhere("$baseAlias.lft = ?", $this->getLeftValue() + 1);
157 158 159 160 161 162
        $this->_tree->returnQueryWithRootId($q, $this->getRootValue());
        $result = $q->execute();

        if (count($result) <= 0) {
            return false;
        }
163
        
164 165 166 167 168
        if ($result instanceof Doctrine_Collection) {
            $child = $result->getFirst();
        } else if (is_array($result)) {
            $child = array_shift($result);
        }
169
        
170
        return $child;       
171 172 173 174 175 176 177 178 179
    }

    /**
     * gets record of last child or empty record
     *
     * @return object     Doctrine_Record            
     */
    public function getLastChild()
    {
180
        $baseAlias = $this->_tree->getBaseAlias();
181
        $q = $this->_tree->getBaseQuery();
182
        $q->addWhere("$baseAlias.rgt = ?", $this->getRightValue() - 1);
183 184
        $this->_tree->returnQueryWithRootId($q, $this->getRootValue());
        $result = $q->execute();
185

186 187 188 189 190 191 192 193 194
        if (count($result) <= 0) {
            return false;
        }
        
        if ($result instanceof Doctrine_Collection) {
            $child = $result->getFirst();
        } else if (is_array($result)) {
            $child = array_shift($result);
        }
195
        
196
        return $child;      
197 198 199 200 201
    }

    /**
     * gets children for node (direct descendants only)
     *
202
     * @return mixed The children of the node or FALSE if the node has no children.               
203 204 205
     */
    public function getChildren()
    { 
206
        return $this->getDescendants(1);
207 208 209 210 211
    }

    /**
     * gets descendants for node (direct descendants only)
     *
212 213 214
     * @return mixed  The descendants of the node or FALSE if the node has no descendants.
     * @todo Currently all descendants are fetched, no matter the depth. Maybe there is a better
     *       solution with less overhead.      
215
     */
216
    public function getDescendants($depth = null, $includeNode = false)
217
    {
218
        $baseAlias = $this->_tree->getBaseAlias();
219 220 221 222
        $q = $this->_tree->getBaseQuery();
        $params = array($this->record->get('lft'), $this->record->get('rgt'));
        
        if ($includeNode) {
223
            $q->addWhere("$baseAlias.lft >= ? AND $baseAlias.rgt <= ?", $params)->addOrderBy("$baseAlias.lft asc");
224
        } else {
225
            $q->addWhere("$baseAlias.lft > ? AND $baseAlias.rgt < ?", $params)->addOrderBy("$baseAlias.lft asc");
226 227 228
        }
        
        if ($depth !== null) {
229
            $q->addWhere("$baseAlias.level <= ?", $this->record['level'] + $depth);
230 231 232 233 234 235 236 237 238 239
        }
        
        $q = $this->_tree->returnQueryWithRootId($q, $this->getRootValue());
        $result = $q->execute();

        if (count($result) <= 0) {
            return false;
        }

        return $result;
240 241 242 243 244 245 246 247 248
    }

    /**
     * gets record of parent or empty record
     *
     * @return object     Doctrine_Record            
     */
    public function getParent()
    {
249
        $baseAlias = $this->_tree->getBaseAlias();
250
        $q = $this->_tree->getBaseQuery();
251 252
        $q->addWhere("$baseAlias.lft < ? AND $baseAlias.rgt > ?", array($this->getLeftValue(), $this->getRightValue()))
                ->addOrderBy("$baseAlias.rgt asc");
253 254
        $q = $this->_tree->returnQueryWithRootId($q, $this->getRootValue());
        $result = $q->execute();
255
        
256 257 258 259 260 261 262 263 264
        if (count($result) <= 0) {
            return false;
        }
               
        if ($result instanceof Doctrine_Collection) {
            $parent = $result->getFirst();
        } else if (is_array($result)) {
            $parent = array_shift($result);
        }
265 266 267 268 269 270 271
        
        return $parent;
    }

    /**
     * gets ancestors for node
     *
272 273 274
     * @param integer $deth  The depth 'upstairs'.
     * @return mixed  The ancestors of the node or FALSE if the node has no ancestors (this 
     *                basically means it's a root node).                
275
     */
276
    public function getAncestors($depth = null)
277
    {
278
        $baseAlias = $this->_tree->getBaseAlias();
279
        $q = $this->_tree->getBaseQuery();
280 281
        $q->addWhere("$baseAlias.lft < ? AND $baseAlias.rgt > ?", array($this->getLeftValue(), $this->getRightValue()))
                ->addOrderBy("$baseAlias.lft asc");
282
        if ($depth !== null) {
283
            $q->addWhere("$baseAlias.level >= ?", $this->record['level'] - $depth);
284 285
        }
        $q = $this->_tree->returnQueryWithRootId($q, $this->getRootValue());
286
        $ancestors = $q->execute();
287 288 289
        if (count($ancestors) <= 0) {
            return false;
        }
290 291 292 293 294 295 296 297 298 299 300 301 302 303
        return $ancestors;
    }

    /**
     * gets path to node from root, uses record::toString() method to get node names
     *
     * @param string     $seperator     path seperator
     * @param bool     $includeNode     whether or not to include node at end of path
     * @return string     string representation of path                
     */     
    public function getPath($seperator = ' > ', $includeRecord = false)
    {
        $path = array();
        $ancestors = $this->getAncestors();
304
        foreach ($ancestors as $ancestor) {
305 306
            $path[] = $ancestor->__toString();
        }
307
        if ($includeRecord) {
308
            $path[] = $this->getRecord()->__toString();
309 310
        }
            
311 312 313 314 315 316 317 318 319 320
        return implode($seperator, $path);
    }

    /**
     * gets number of children (direct descendants)
     *
     * @return int            
     */     
    public function getNumberChildren()
    {
321
        return count($this->getChildren());
322 323 324 325 326 327 328 329 330 331 332 333 334 335 336
    }

    /**
     * gets number of descendants (children and their children)
     *
     * @return int            
     */
    public function getNumberDescendants()
    {
        return ($this->getRightValue() - $this->getLeftValue() - 1) / 2;
    }

    /**
     * inserts node as parent of dest record
     *
337 338
     * @return bool
     * @todo Wrap in transaction          
339 340 341 342
     */
    public function insertAsParentOf(Doctrine_Record $dest)
    {
        // cannot insert a node that has already has a place within the tree
343
        if ($this->isValidNode()) {
344
            return false;
345
        }
346
        // cannot insert as parent of root
347
        if ($dest->getNode()->isRoot()) {
348
            return false;
349
        }
350 351 352
        $newRoot = $dest->getNode()->getRootValue();
        $this->shiftRLValues($dest->getNode()->getLeftValue(), 1, $newRoot);
        $this->shiftRLValues($dest->getNode()->getRightValue() + 2, 1, $newRoot);
353 354
        
        $newLeft = $dest->getNode()->getLeftValue();
zYne's avatar
zYne committed
355
        $newRight = $dest->getNode()->getRightValue() + 2;
356

357
        $this->record['level'] = $dest['level'] - 1;
358 359 360 361 362 363 364 365
        $this->insertNode($newLeft, $newRight, $newRoot);
        
        return true;
    }

    /**
     * inserts node as previous sibling of dest record
     *
366 367
     * @return bool
     * @todo Wrap in transaction       
368 369 370 371
     */
    public function insertAsPrevSiblingOf(Doctrine_Record $dest)
    {
        // cannot insert a node that has already has a place within the tree
372
        if ($this->isValidNode())
373 374 375 376 377 378 379
            return false;

        $newLeft = $dest->getNode()->getLeftValue();
        $newRight = $dest->getNode()->getLeftValue() + 1;
        $newRoot = $dest->getNode()->getRootValue();
        
        $this->shiftRLValues($newLeft, 2, $newRoot);
380
        $this->record['level'] = $dest['level'];
381 382 383 384 385 386 387 388 389 390 391
        $this->insertNode($newLeft, $newRight, $newRoot);
        // update destination left/right values to prevent a refresh
        // $dest->getNode()->setLeftValue($dest->getNode()->getLeftValue() + 2);
        // $dest->getNode()->setRightValue($dest->getNode()->getRightValue() + 2);
                        
        return true;
    }

    /**
     * inserts node as next sibling of dest record
     *
392 393
     * @return bool
     * @todo Wrap in transaction           
394 395 396 397
     */    
    public function insertAsNextSiblingOf(Doctrine_Record $dest)
    {
        // cannot insert a node that has already has a place within the tree
398
        if ($this->isValidNode())
399 400 401 402 403 404 405
            return false;

        $newLeft = $dest->getNode()->getRightValue() + 1;
        $newRight = $dest->getNode()->getRightValue() + 2;
        $newRoot = $dest->getNode()->getRootValue();

        $this->shiftRLValues($newLeft, 2, $newRoot);
406
        $this->record['level'] = $dest['level'];
407 408 409 410 411
        $this->insertNode($newLeft, $newRight, $newRoot);

        // update destination left/right values to prevent a refresh
        // no need, node not affected

412
        return true;
413 414 415 416 417
    }

    /**
     * inserts node as first child of dest record
     *
418 419
     * @return bool
     * @todo Wrap in transaction         
420 421 422 423
     */
    public function insertAsFirstChildOf(Doctrine_Record $dest)
    {
        // cannot insert a node that has already has a place within the tree
424
        if ($this->isValidNode())
425 426 427 428 429 430 431
            return false;

        $newLeft = $dest->getNode()->getLeftValue() + 1;
        $newRight = $dest->getNode()->getLeftValue() + 2;
        $newRoot = $dest->getNode()->getRootValue();

        $this->shiftRLValues($newLeft, 2, $newRoot);
432
        $this->record['level'] = $dest['level'] + 1;
433 434 435 436 437 438 439 440 441 442 443
        $this->insertNode($newLeft, $newRight, $newRoot);
        
        // update destination left/right values to prevent a refresh
        // $dest->getNode()->setRightValue($dest->getNode()->getRightValue() + 2);

        return true;
    }

    /**
     * inserts node as last child of dest record
     *
444 445
     * @return bool
     * @todo Wrap in transaction            
446 447 448 449
     */
    public function insertAsLastChildOf(Doctrine_Record $dest)
    {
        // cannot insert a node that has already has a place within the tree
450
        if ($this->isValidNode())
451 452 453 454 455 456 457
            return false;

        $newLeft = $dest->getNode()->getRightValue();
        $newRight = $dest->getNode()->getRightValue() + 1;
        $newRoot = $dest->getNode()->getRootValue();

        $this->shiftRLValues($newLeft, 2, $newRoot);
458
        $this->record['level'] = $dest['level'] + 1;
459 460 461 462 463 464 465
        $this->insertNode($newLeft, $newRight, $newRoot);

        // update destination left/right values to prevent a refresh
        // $dest->getNode()->setRightValue($dest->getNode()->getRightValue() + 2);
        
        return true;
    }
466 467 468
    
    /**
     * Accomplishes moving of nodes between different trees.
469
     * Used by the move* methods if the root values of the two nodes are different.
470 471 472 473
     *
     * @param Doctrine_Record $dest
     * @param unknown_type $newLeftValue
     * @param unknown_type $moveType
474
     * @todo Better exception handling/wrapping
475 476 477 478 479 480 481 482 483 484 485 486 487
     */
    private function _moveBetweenTrees(Doctrine_Record $dest, $newLeftValue, $moveType)
    {
        $conn = $this->record->getTable()->getConnection();
            
            try {
                $conn->beginTransaction();
                
                // Move between trees: Detach from old tree & insert into new tree
                $newRoot = $dest->getNode()->getRootValue();
                $oldRoot = $this->getRootValue();
                $oldLft = $this->getLeftValue();
                $oldRgt = $this->getRightValue();
488
                $oldLevel = $this->record['level'];
489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526
                
                // Prepare target tree for insertion, make room
                $this->shiftRlValues($newLeftValue, $oldRgt - $oldLft - 1, $newRoot);
                
                // Set new root id for this node
                $this->setRootValue($newRoot);
                $this->record->save();
                
                // Close gap in old tree
                $first = $oldRgt + 1;
                $delta = $oldLft - $oldRgt - 1;
                $this->shiftRLValues($first, $delta, $oldRoot);
                
                // Insert this node as a new node
                $this->setRightValue(0);
                $this->setLeftValue(0);
                
                switch ($moveType) {
                    case 'moveAsPrevSiblingOf':
                        $this->insertAsPrevSiblingOf($dest);
                    break;
                    case 'moveAsFirstChildOf':
                        $this->insertAsFirstChildOf($dest);
                    break;
                    case 'moveAsNextSiblingOf':
                        $this->insertAsNextSiblingOf($dest);
                    break;
                    case 'moveAsLastChildOf':
                        $this->insertAsLastChildOf($dest);
                    break;
                    default:
                        throw new Exception("Unknown move operation: $moveType.");
                }
                
                $diff = $oldRgt - $oldLft;
                $this->setRightValue($this->getLeftValue() + ($oldRgt - $oldLft));
                $this->record->save();
                
527 528 529
                $newLevel = $this->record['level'];
                $levelDiff = $newLevel - $oldLevel;
                
530 531
                // Relocate descendants of the node
                $diff = $this->getLeftValue() - $oldLft;
532
                $componentName = $this->_tree->getBaseComponent();
533 534
                $rootColName = $this->record->getTable()->getTree()->getAttribute('rootColumnName');

535 536
                // Update lft/rgt/root/level for all descendants
                $q = new Doctrine_Query($conn);
537
                $q = $q->update($componentName)
romanb's avatar
romanb committed
538 539 540 541
                        ->set($componentName . '.lft', 'lft + ?', $diff)
                        ->set($componentName . '.rgt', 'rgt + ?', $diff)
                        ->set($componentName . '.level', 'level + ?', $levelDiff)
                        ->set($componentName . '.' . $rootColName, '?', $newRoot)
542 543
                        ->where($componentName . '.lft > ? AND ' . $componentName . '.rgt < ?',
                        array($oldLft, $oldRgt));
544
                $q = $this->_tree->returnQueryWithRootId($q, $oldRoot);
545 546 547
                $q->execute();
                
                $conn->commit();
548
            } catch (Exception $e) {
549 550 551 552 553
                $conn->rollback();
                throw $e;
            }
    }
    
554 555
    /**
     * moves node as prev sibling of dest record
556
     * 
557 558 559
     */     
    public function moveAsPrevSiblingOf(Doctrine_Record $dest)
    {
560 561 562 563 564
        if ($dest->getNode()->getRootValue() != $this->getRootValue()) {
            // Move between trees
            $this->_moveBetweenTrees($dest, $dest->getNode()->getLeftValue(), __FUNCTION__);
        } else {
            // Move within the tree
565 566 567
            $oldLevel = $this->record['level'];
            $this->record['level'] = $dest['level'];
            $this->updateNode($dest->getNode()->getLeftValue(), $this->record['level'] - $oldLevel);
568
        }
569 570 571 572 573 574 575 576
    }

    /**
     * moves node as next sibling of dest record
     *        
     */
    public function moveAsNextSiblingOf(Doctrine_Record $dest)
    {
577 578 579 580 581
        if ($dest->getNode()->getRootValue() != $this->getRootValue()) {
            // Move between trees
            $this->_moveBetweenTrees($dest, $dest->getNode()->getRightValue() + 1, __FUNCTION__);
        } else {
            // Move within tree
582 583 584
            $oldLevel = $this->record['level'];
            $this->record['level'] = $dest['level'];
            $this->updateNode($dest->getNode()->getRightValue() + 1, $this->record['level'] - $oldLevel);
585
        }
586 587 588 589 590 591 592 593
    }

    /**
     * moves node as first child of dest record
     *            
     */
    public function moveAsFirstChildOf(Doctrine_Record $dest)
    {
594 595 596 597 598
        if ($dest->getNode()->getRootValue() != $this->getRootValue()) {
            // Move between trees
            $this->_moveBetweenTrees($dest, $dest->getNode()->getLeftValue() + 1, __FUNCTION__);
        } else {
            // Move within tree
599 600 601
            $oldLevel = $this->record['level'];
            $this->record['level'] = $dest['level'] + 1;
            $this->updateNode($dest->getNode()->getLeftValue() + 1, $this->record['level'] - $oldLevel);
602
        }
603 604 605 606 607 608 609 610
    }

    /**
     * moves node as last child of dest record
     *        
     */
    public function moveAsLastChildOf(Doctrine_Record $dest)
    {
611 612 613 614 615
        if ($dest->getNode()->getRootValue() != $this->getRootValue()) {
            // Move between trees
            $this->_moveBetweenTrees($dest, $dest->getNode()->getRightValue(), __FUNCTION__);
        } else {
            // Move within tree
616 617 618
            $oldLevel = $this->record['level'];
            $this->record['level'] = $dest['level'] + 1;
            $this->updateNode($dest->getNode()->getRightValue(), $this->record['level'] - $oldLevel);
619
        }
620
    }
621 622
    
    /**
623
     * Makes this node a root node. Only used in multiple-root trees.
624 625 626
     *
     * @todo Exception handling/wrapping
     */
627
    public function makeRoot($newRootId)
628 629 630 631 632 633 634 635 636
    {
        // TODO: throw exception instead?
        if ($this->getLeftValue() == 1 || !$this->record->getTable()->getTree()->getAttribute('hasManyRoots')) {
            return false;
        }
        
        $oldRgt = $this->getRightValue();
        $oldLft = $this->getLeftValue();
        $oldRoot = $this->getRootValue();
637
        $oldLevel = $this->record['level'];
638 639 640 641 642
        
        try {
            $conn = $this->record->getTable()->getConnection();
            $conn->beginTransaction();
            
643
            // Detach from old tree (close gap in old tree)
644 645 646 647
            $first = $oldRgt + 1;
            $delta = $oldLft - $oldRgt - 1;
            $this->shiftRLValues($first, $delta, $this->getRootValue());
            
648
            // Set new lft/rgt/root/level values for root node
649 650
            $this->setLeftValue(1);
            $this->setRightValue($oldRgt - $oldLft + 1);
651
            $this->setRootValue($newRootId);
652
            $this->record['level'] = 0;
653
            
654
            // Update descendants lft/rgt/root/level values
655
            $diff = 1 - $oldLft;
656
            $newRoot = $newRootId;
657
            $componentName = $this->_tree->getBaseComponent();
658
            $rootColName = $this->record->getTable()->getTree()->getAttribute('rootColumnName');
659
            $q = new Doctrine_Query($conn);
660
            $q = $q->update($componentName)
romanb's avatar
romanb committed
661 662 663 664
                    ->set($componentName . '.lft', 'lft + ?', $diff)
                    ->set($componentName . '.rgt', 'rgt + ?', $diff)
                    ->set($componentName . '.level', 'level - ?', $oldLevel)
                    ->set($componentName . '.' . $rootColName, '?', $newRoot)
665 666
                    ->where($componentName . '.lft > ? AND ' . $componentName . '.rgt < ?',
                    array($oldLft, $oldRgt));
667
            $q = $this->_tree->returnQueryWithRootId($q, $oldRoot);
668 669 670
            $q->execute();
            
            $conn->commit();
671 672
            
        } catch (Exception $e) {
673 674 675 676
            $conn->rollback();
            throw $e;
        }
    }
677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693

    /**
     * adds node as last child of record
     *        
     */
    public function addChild(Doctrine_Record $record)
    {
        $record->getNode()->insertAsLastChildOf($this->getRecord());
    }

    /**
     * determines if node is leaf
     *
     * @return bool            
     */
    public function isLeaf()
    {
zYne's avatar
zYne committed
694
        return (($this->getRightValue() - $this->getLeftValue()) == 1);
695 696 697 698 699 700 701 702 703
    }

    /**
     * determines if node is root
     *
     * @return bool            
     */
    public function isRoot()
    {
zYne's avatar
zYne committed
704
        return ($this->getLeftValue() == 1);
705 706 707 708 709 710 711 712 713
    }

    /**
     * determines if node is equal to subject node
     *
     * @return bool            
     */    
    public function isEqualTo(Doctrine_Record $subj)
    {
zYne's avatar
zYne committed
714 715 716 717
        return (($this->getLeftValue() == $subj->getNode()->getLeftValue()) &&
                ($this->getRightValue() == $subj->getNode()->getRightValue()) && 
                ($this->getRootValue() == $subj->getNode()->getRootValue())
                );
718 719 720 721 722
    }

    /**
     * determines if node is child of subject node
     *
zYne's avatar
zYne committed
723
     * @return bool
724 725 726
     */
    public function isDescendantOf(Doctrine_Record $subj)
    {
727 728 729
        return (($this->getLeftValue() > $subj->getNode()->getLeftValue()) &&
                ($this->getRightValue() < $subj->getNode()->getRightValue()) &&
                ($this->getRootValue() == $subj->getNode()->getRootValue()));
730 731 732 733 734 735 736 737 738
    }

    /**
     * determines if node is child of or sibling to subject node
     *
     * @return bool            
     */
    public function isDescendantOfOrEqualTo(Doctrine_Record $subj)
    {
739 740 741
        return (($this->getLeftValue() >= $subj->getNode()->getLeftValue()) &&
                ($this->getRightValue() <= $subj->getNode()->getRightValue()) &&
                ($this->getRootValue() == $subj->getNode()->getRootValue()));
742 743 744 745 746 747 748
    }

    /**
     * determines if node is valid
     *
     * @return bool            
     */
749
    public function isValidNode($record = null)
750
    {
751 752
        if ($record === null) {
          return ($this->getRightValue() > $this->getLeftValue());
753
        } else if ( $record instanceof Doctrine_Record ) {
754
          return ($record->getNode()->getRightValue() > $record->getNode()->getLeftValue());
755 756
        } else {
          return false;
757
        }
758 759 760 761
    }

    /**
     * deletes node and it's descendants
762
     * @todo Delete more efficiently. Wrap in transaction if needed.      
763 764 765 766
     */
    public function delete()
    {
        // TODO: add the setting whether or not to delete descendants or relocate children
767
        $oldRoot = $this->getRootValue();
768
        $q = $this->_tree->getBaseQuery();
769
        
770
        $baseAlias = $this->_tree->getBaseAlias();
771
        $componentName = $this->_tree->getBaseComponent();
772

773
        $q = $q->addWhere("$baseAlias.lft >= ? AND $baseAlias.rgt <= ?", array($this->getLeftValue(), $this->getRightValue()));
774

775
        $q = $this->record->getTable()->getTree()->returnQueryWithRootId($q, $oldRoot);
776 777 778 779 780 781 782
        
        $coll = $q->execute();

        $coll->delete();

        $first = $this->getRightValue() + 1;
        $delta = $this->getLeftValue() - $this->getRightValue() - 1;
783
        $this->shiftRLValues($first, $delta, $oldRoot);
784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805
        
        return true; 
    }

    /**
     * sets node's left and right values and save's it
     *
     * @param int     $destLeft     node left value
     * @param int        $destRight    node right value
     */    
    private function insertNode($destLeft = 0, $destRight = 0, $destRoot = 1)
    {
        $this->setLeftValue($destLeft);
        $this->setRightValue($destRight);
        $this->setRootValue($destRoot);
        $this->record->save();    
    }

    /**
     * move node's and its children to location $destLeft and updates rest of tree
     *
     * @param int     $destLeft    destination left value
806
     * @todo Wrap in transaction
807
     */
808
    private function updateNode($destLeft, $levelDiff)
809
    { 
810
        $componentName = $this->_tree->getBaseComponent();
811 812
        $left = $this->getLeftValue();
        $right = $this->getRightValue();
813
        $rootId = $this->getRootValue();
814 815 816

        $treeSize = $right - $left + 1;

817
        // Make room in the new branch
818
        $this->shiftRLValues($destLeft, $treeSize, $rootId);
819

820
        if ($left >= $destLeft) { // src was shifted too?
821 822 823 824
            $left += $treeSize;
            $right += $treeSize;
        }

825 826 827
        // update level for descendants
        $q = new Doctrine_Query();
        $q = $q->update($componentName)
828
                ->set($componentName . '.level', 'level + ?')
829
                ->where($componentName . '.lft > ? AND ' . $componentName . '.rgt < ?',
830
                        array($levelDiff, $left, $right));
831 832 833
        $q = $this->_tree->returnQueryWithRootId($q, $rootId);
        $q->execute();
        
834
        // now there's enough room next to target to move the subtree
835
        $this->shiftRLRange($left, $right, $destLeft - $left, $rootId);
836

837
        // correct values after source (close gap in old tree)
838
        $this->shiftRLValues($right + 1, -$treeSize, $rootId);
839 840 841 842 843 844 845 846 847 848 849

        $this->record->save();
        $this->record->refresh();
    }

    /**
     * adds '$delta' to all Left and Right values that are >= '$first'. '$delta' can also be negative.
     *
     * @param int $first         First node to be shifted
     * @param int $delta         Value to be shifted by, can be negative
     */    
zYne's avatar
zYne committed
850
    private function shiftRlValues($first, $delta, $rootId = 1)
851
    {
zYne's avatar
zYne committed
852 853
        $qLeft  = new Doctrine_Query();
        $qRight = new Doctrine_Query();
854 855

        // shift left columns
856
        $componentName = $this->_tree->getBaseComponent();
romanb's avatar
romanb committed
857
        $qLeft = $qLeft->update($componentName)
858 859
                                ->set($componentName . '.lft', 'lft + ?')
                                ->where($componentName . '.lft >= ?', array($delta, $first));
romanb's avatar
romanb committed
860
        
zYne's avatar
zYne committed
861
        $qLeft = $this->record->getTable()->getTree()->returnQueryWithRootId($qLeft, $rootId);
862 863 864 865
        
        $resultLeft = $qLeft->execute();
        
        // shift right columns
romanb's avatar
romanb committed
866
        $resultRight = $qRight->update($componentName)
867 868
                                ->set($componentName . '.rgt', 'rgt + ?')
                                ->where($componentName . '.rgt >= ?', array($delta, $first));
869

zYne's avatar
zYne committed
870
        $qRight = $this->record->getTable()->getTree()->returnQueryWithRootId($qRight, $rootId);
871 872 873 874 875 876 877 878 879 880 881 882

        $resultRight = $qRight->execute();
    }

    /**
     * adds '$delta' to all Left and Right values that are >= '$first' and <= '$last'. 
     * '$delta' can also be negative.
     *
     * @param int $first     First node to be shifted (L value)
     * @param int $last     Last node to be shifted (L value)
     * @param int $delta         Value to be shifted by, can be negative
     */ 
zYne's avatar
zYne committed
883
    private function shiftRlRange($first, $last, $delta, $rootId = 1)
884
    {
zYne's avatar
zYne committed
885 886
        $qLeft  = new Doctrine_Query();
        $qRight = new Doctrine_Query();
887 888

        // shift left column values
889
        $componentName = $this->_tree->getBaseComponent();
romanb's avatar
romanb committed
890
        $qLeft = $qLeft->update($componentName)
891 892
                                ->set($componentName . '.lft', 'lft + ?')
                                ->where($componentName . '.lft >= ? AND ' . $componentName . '.lft <= ?', array($delta, $first, $last));
893
        
zYne's avatar
zYne committed
894
        $qLeft = $this->record->getTable()->getTree()->returnQueryWithRootId($qLeft, $rootId);
895 896 897 898

        $resultLeft = $qLeft->execute();
        
        // shift right column values
romanb's avatar
romanb committed
899
        $qRight = $qRight->update($componentName)
900 901
                                ->set($componentName . '.rgt', 'rgt + ?')
                                ->where($componentName . '.rgt >= ? AND ' . $componentName . '.rgt <= ?', array($delta, $first, $last));
902

zYne's avatar
zYne committed
903
        $qRight = $this->record->getTable()->getTree()->returnQueryWithRootId($qRight, $rootId);
904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954

        $resultRight = $qRight->execute();
    }
    
    /**
     * gets record's left value
     *
     * @return int            
     */     
    public function getLeftValue()
    {
        return $this->record->get('lft');
    }

    /**
     * sets record's left value
     *
     * @param int            
     */     
    public function setLeftValue($lft)
    {
        $this->record->set('lft', $lft);        
    }

    /**
     * gets record's right value
     *
     * @return int            
     */     
    public function getRightValue()
    {
        return $this->record->get('rgt');        
    }

    /**
     * sets record's right value
     *
     * @param int            
     */    
    public function setRightValue($rgt)
    {
        $this->record->set('rgt', $rgt);         
    }

    /**
     * gets level (depth) of node in the tree
     *
     * @return int            
     */    
    public function getLevel()
    {
955
        if ( ! isset($this->record['level'])) {
956
            $baseAlias = $this->_tree->getBaseAlias();
957
            $componentName = $this->_tree->getBaseComponent();
958
            $q = $this->_tree->getBaseQuery();
959
            $q = $q->addWhere("$baseAlias.lft < ? AND $baseAlias.rgt > ?", array($this->getLeftValue(), $this->getRightValue()));
960

961
            $q = $this->_tree->returnQueryWithRootId($q, $this->getRootValue());
962 963 964
            
            $coll = $q->execute();

965
            $this->record['level'] = count($coll) ? count($coll) : 0;
966
        }
967
        return $this->record['level'];
968 969 970 971 972 973 974 975
    }

    /**
     * get records root id value
     *            
     */     
    public function getRootValue()
    {
976 977 978
        if ($this->_tree->getAttribute('hasManyRoots')) {
            return $this->record->get($this->_tree->getAttribute('rootColumnName'));
        }
979 980 981 982 983 984 985 986 987 988
        return 1;
    }

    /**
     * sets records root id value
     *
     * @param int            
     */
    public function setRootValue($value)
    {
989 990 991
        if ($this->_tree->getAttribute('hasManyRoots')) {
            $this->record->set($this->_tree->getAttribute('rootColumnName'), $value);   
        }    
992 993
    }
}