TreeBehavior.php 35 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964
  1. <?php
  2. /**
  3. * Tree behavior class.
  4. *
  5. * Enables a model object to act as a node-based tree.
  6. *
  7. * PHP 5
  8. *
  9. * CakePHP : Rapid Development Framework (http://cakephp.org)
  10. * Copyright 2005-2011, Cake Software Foundation, Inc.
  11. *
  12. * Licensed under The MIT License
  13. * Redistributions of files must retain the above copyright notice.
  14. *
  15. * @copyright Copyright 2005-2011, Cake Software Foundation, Inc. (http://cakefoundation.org)
  16. * @link http://cakephp.org CakePHP Project
  17. * @package Cake.Model.Behavior
  18. * @since CakePHP v 1.2.0.4487
  19. * @license MIT License (http://www.opensource.org/licenses/mit-license.php)
  20. */
  21. /**
  22. * Tree Behavior.
  23. *
  24. * Enables a model object to act as a node-based tree. Using Modified Preorder Tree Traversal
  25. *
  26. * @see http://en.wikipedia.org/wiki/Tree_traversal
  27. * @package Cake.Model.Behavior
  28. * @link http://book.cakephp.org/view/1339/Tree
  29. */
  30. class TreeBehavior extends ModelBehavior {
  31. /**
  32. * Errors
  33. *
  34. * @var array
  35. */
  36. public $errors = array();
  37. /**
  38. * Defaults
  39. *
  40. * @var array
  41. */
  42. protected $_defaults = array(
  43. 'parent' => 'parent_id', 'left' => 'lft', 'right' => 'rght',
  44. 'scope' => '1 = 1', 'type' => 'nested', '__parentChange' => false, 'recursive' => -1
  45. );
  46. /**
  47. * Initiate Tree behavior
  48. *
  49. * @param Model $Model instance of model
  50. * @param array $config array of configuration settings.
  51. * @return void
  52. */
  53. public function setup($Model, $config = array()) {
  54. if (!is_array($config)) {
  55. $config = array('type' => $config);
  56. }
  57. $settings = array_merge($this->_defaults, $config);
  58. if (in_array($settings['scope'], $Model->getAssociated('belongsTo'))) {
  59. $data = $Model->getAssociated($settings['scope']);
  60. $parent = $Model->{$settings['scope']};
  61. $settings['scope'] = $Model->alias . '.' . $data['foreignKey'] . ' = ' . $parent->alias . '.' . $parent->primaryKey;
  62. $settings['recursive'] = 0;
  63. }
  64. $this->settings[$Model->alias] = $settings;
  65. }
  66. /**
  67. * After save method. Called after all saves
  68. *
  69. * Overriden to transparently manage setting the lft and rght fields if and only if the parent field is included in the
  70. * parameters to be saved.
  71. *
  72. * @param Model $Model Model instance.
  73. * @param boolean $created indicates whether the node just saved was created or updated
  74. * @return boolean true on success, false on failure
  75. */
  76. public function afterSave($Model, $created) {
  77. extract($this->settings[$Model->alias]);
  78. if ($created) {
  79. if ((isset($Model->data[$Model->alias][$parent])) && $Model->data[$Model->alias][$parent]) {
  80. return $this->_setParent($Model, $Model->data[$Model->alias][$parent], $created);
  81. }
  82. } elseif ($__parentChange) {
  83. $this->settings[$Model->alias]['__parentChange'] = false;
  84. return $this->_setParent($Model, $Model->data[$Model->alias][$parent]);
  85. }
  86. }
  87. /**
  88. * Before delete method. Called before all deletes
  89. *
  90. * Will delete the current node and all children using the deleteAll method and sync the table
  91. *
  92. * @param Model $Model Model instance
  93. * @param boolean $cascade
  94. * @return boolean true to continue, false to abort the delete
  95. */
  96. public function beforeDelete($Model, $cascade = true) {
  97. extract($this->settings[$Model->alias]);
  98. list($name, $data) = array($Model->alias, $Model->read());
  99. $data = $data[$name];
  100. if (!$data[$right] || !$data[$left]) {
  101. return true;
  102. }
  103. $diff = $data[$right] - $data[$left] + 1;
  104. if ($diff > 2) {
  105. if (is_string($scope)) {
  106. $scope = array($scope);
  107. }
  108. $scope[]["{$Model->alias}.{$left} BETWEEN ? AND ?"] = array($data[$left] + 1, $data[$right] - 1);
  109. $Model->deleteAll($scope);
  110. }
  111. $this->_sync($Model, $diff, '-', '> ' . $data[$right]);
  112. return true;
  113. }
  114. /**
  115. * Before save method. Called before all saves
  116. *
  117. * Overriden to transparently manage setting the lft and rght fields if and only if the parent field is included in the
  118. * parameters to be saved. For newly created nodes with NO parent the left and right field values are set directly by
  119. * this method bypassing the setParent logic.
  120. *
  121. * @since 1.2
  122. * @param Model $Model Model instance
  123. * @return boolean true to continue, false to abort the save
  124. */
  125. public function beforeSave($Model) {
  126. extract($this->settings[$Model->alias]);
  127. $this->_addToWhitelist($Model, array($left, $right));
  128. if (!$Model->id) {
  129. if (array_key_exists($parent, $Model->data[$Model->alias]) && $Model->data[$Model->alias][$parent]) {
  130. $parentNode = $Model->find('first', array(
  131. 'conditions' => array($scope, $Model->escapeField() => $Model->data[$Model->alias][$parent]),
  132. 'fields' => array($Model->primaryKey, $right), 'recursive' => $recursive
  133. ));
  134. if (!$parentNode) {
  135. return false;
  136. }
  137. list($parentNode) = array_values($parentNode);
  138. $Model->data[$Model->alias][$left] = 0; //$parentNode[$right];
  139. $Model->data[$Model->alias][$right] = 0; //$parentNode[$right] + 1;
  140. } else {
  141. $edge = $this->_getMax($Model, $scope, $right, $recursive);
  142. $Model->data[$Model->alias][$left] = $edge + 1;
  143. $Model->data[$Model->alias][$right] = $edge + 2;
  144. }
  145. } elseif (array_key_exists($parent, $Model->data[$Model->alias])) {
  146. if ($Model->data[$Model->alias][$parent] != $Model->field($parent)) {
  147. $this->settings[$Model->alias]['__parentChange'] = true;
  148. }
  149. if (!$Model->data[$Model->alias][$parent]) {
  150. $Model->data[$Model->alias][$parent] = null;
  151. $this->_addToWhitelist($Model, $parent);
  152. } else {
  153. $values = $Model->find('first', array(
  154. 'conditions' => array($scope,$Model->escapeField() => $Model->id),
  155. 'fields' => array($Model->primaryKey, $parent, $left, $right ), 'recursive' => $recursive)
  156. );
  157. if ($values === false) {
  158. return false;
  159. }
  160. list($node) = array_values($values);
  161. $parentNode = $Model->find('first', array(
  162. 'conditions' => array($scope, $Model->escapeField() => $Model->data[$Model->alias][$parent]),
  163. 'fields' => array($Model->primaryKey, $left, $right), 'recursive' => $recursive
  164. ));
  165. if (!$parentNode) {
  166. return false;
  167. }
  168. list($parentNode) = array_values($parentNode);
  169. if (($node[$left] < $parentNode[$left]) && ($parentNode[$right] < $node[$right])) {
  170. return false;
  171. } elseif ($node[$Model->primaryKey] == $parentNode[$Model->primaryKey]) {
  172. return false;
  173. }
  174. }
  175. }
  176. return true;
  177. }
  178. /**
  179. * Get the number of child nodes
  180. *
  181. * If the direct parameter is set to true, only the direct children are counted (based upon the parent_id field)
  182. * If false is passed for the id parameter, all top level nodes are counted, or all nodes are counted.
  183. *
  184. * @param Model $Model Model instance
  185. * @param mixed $id The ID of the record to read or false to read all top level nodes
  186. * @param boolean $direct whether to count direct, or all, children
  187. * @return integer number of child nodes
  188. * @link http://book.cakephp.org/view/1347/Counting-children
  189. */
  190. public function childCount($Model, $id = null, $direct = false) {
  191. if (is_array($id)) {
  192. extract (array_merge(array('id' => null), $id));
  193. }
  194. if ($id === null && $Model->id) {
  195. $id = $Model->id;
  196. } elseif (!$id) {
  197. $id = null;
  198. }
  199. extract($this->settings[$Model->alias]);
  200. if ($direct) {
  201. return $Model->find('count', array('conditions' => array($scope, $Model->escapeField($parent) => $id)));
  202. }
  203. if ($id === null) {
  204. return $Model->find('count', array('conditions' => $scope));
  205. } elseif ($Model->id === $id && isset($Model->data[$Model->alias][$left]) && isset($Model->data[$Model->alias][$right])) {
  206. $data = $Model->data[$Model->alias];
  207. } else {
  208. $data = $Model->find('first', array('conditions' => array($scope, $Model->escapeField() => $id), 'recursive' => $recursive));
  209. if (!$data) {
  210. return 0;
  211. }
  212. $data = $data[$Model->alias];
  213. }
  214. return ($data[$right] - $data[$left] - 1) / 2;
  215. }
  216. /**
  217. * Get the child nodes of the current model
  218. *
  219. * If the direct parameter is set to true, only the direct children are returned (based upon the parent_id field)
  220. * If false is passed for the id parameter, top level, or all (depending on direct parameter appropriate) are counted.
  221. *
  222. * @param Model $Model Model instance
  223. * @param mixed $id The ID of the record to read
  224. * @param boolean $direct whether to return only the direct, or all, children
  225. * @param mixed $fields Either a single string of a field name, or an array of field names
  226. * @param string $order SQL ORDER BY conditions (e.g. "price DESC" or "name ASC") defaults to the tree order
  227. * @param integer $limit SQL LIMIT clause, for calculating items per page.
  228. * @param integer $page Page number, for accessing paged data
  229. * @param integer $recursive The number of levels deep to fetch associated records
  230. * @return array Array of child nodes
  231. * @link http://book.cakephp.org/view/1346/Children
  232. */
  233. public function children($Model, $id = null, $direct = false, $fields = null, $order = null, $limit = null, $page = 1, $recursive = null) {
  234. if (is_array($id)) {
  235. extract (array_merge(array('id' => null), $id));
  236. }
  237. $overrideRecursive = $recursive;
  238. if ($id === null && $Model->id) {
  239. $id = $Model->id;
  240. } elseif (!$id) {
  241. $id = null;
  242. }
  243. $name = $Model->alias;
  244. extract($this->settings[$Model->alias]);
  245. if (!is_null($overrideRecursive)) {
  246. $recursive = $overrideRecursive;
  247. }
  248. if (!$order) {
  249. $order = $Model->alias . '.' . $left . ' asc';
  250. }
  251. if ($direct) {
  252. $conditions = array($scope, $Model->escapeField($parent) => $id);
  253. return $Model->find('all', compact('conditions', 'fields', 'order', 'limit', 'page', 'recursive'));
  254. }
  255. if (!$id) {
  256. $conditions = $scope;
  257. } else {
  258. $result = array_values((array)$Model->find('first', array(
  259. 'conditions' => array($scope, $Model->escapeField() => $id),
  260. 'fields' => array($left, $right),
  261. 'recursive' => $recursive
  262. )));
  263. if (empty($result) || !isset($result[0])) {
  264. return array();
  265. }
  266. $conditions = array($scope,
  267. $Model->escapeField($right) . ' <' => $result[0][$right],
  268. $Model->escapeField($left) . ' >' => $result[0][$left]
  269. );
  270. }
  271. return $Model->find('all', compact('conditions', 'fields', 'order', 'limit', 'page', 'recursive'));
  272. }
  273. /**
  274. * A convenience method for returning a hierarchical array used for HTML select boxes
  275. *
  276. * @param Model $Model Model instance
  277. * @param mixed $conditions SQL conditions as a string or as an array('field' =>'value',...)
  278. * @param string $keyPath A string path to the key, i.e. "{n}.Post.id"
  279. * @param string $valuePath A string path to the value, i.e. "{n}.Post.title"
  280. * @param string $spacer The character or characters which will be repeated
  281. * @param integer $recursive The number of levels deep to fetch associated records
  282. * @return array An associative array of records, where the id is the key, and the display field is the value
  283. * @link http://book.cakephp.org/view/1348/generatetreelist
  284. */
  285. public function generateTreeList($Model, $conditions = null, $keyPath = null, $valuePath = null, $spacer = '_', $recursive = null) {
  286. $overrideRecursive = $recursive;
  287. extract($this->settings[$Model->alias]);
  288. if (!is_null($overrideRecursive)) {
  289. $recursive = $overrideRecursive;
  290. }
  291. if ($keyPath == null && $valuePath == null && $Model->hasField($Model->displayField)) {
  292. $fields = array($Model->primaryKey, $Model->displayField, $left, $right);
  293. } else {
  294. $fields = null;
  295. }
  296. if ($keyPath == null) {
  297. $keyPath = '{n}.' . $Model->alias . '.' . $Model->primaryKey;
  298. }
  299. if ($valuePath == null) {
  300. $valuePath = array('{0}{1}', '{n}.tree_prefix', '{n}.' . $Model->alias . '.' . $Model->displayField);
  301. } elseif (is_string($valuePath)) {
  302. $valuePath = array('{0}{1}', '{n}.tree_prefix', $valuePath);
  303. } else {
  304. $valuePath[0] = '{' . (count($valuePath) - 1) . '}' . $valuePath[0];
  305. $valuePath[] = '{n}.tree_prefix';
  306. }
  307. $order = $Model->alias . '.' . $left . ' asc';
  308. $results = $Model->find('all', compact('conditions', 'fields', 'order', 'recursive'));
  309. $stack = array();
  310. foreach ($results as $i => $result) {
  311. while ($stack && ($stack[count($stack) - 1] < $result[$Model->alias][$right])) {
  312. array_pop($stack);
  313. }
  314. $results[$i]['tree_prefix'] = str_repeat($spacer,count($stack));
  315. $stack[] = $result[$Model->alias][$right];
  316. }
  317. if (empty($results)) {
  318. return array();
  319. }
  320. return Set::combine($results, $keyPath, $valuePath);
  321. }
  322. /**
  323. * Get the parent node
  324. *
  325. * reads the parent id and returns this node
  326. *
  327. * @param Model $Model Model instance
  328. * @param mixed $id The ID of the record to read
  329. * @param string|array $fields
  330. * @param integer $recursive The number of levels deep to fetch associated records
  331. * @return array|boolean Array of data for the parent node
  332. * @link http://book.cakephp.org/view/1349/getparentnode
  333. */
  334. public function getParentNode($Model, $id = null, $fields = null, $recursive = null) {
  335. if (is_array($id)) {
  336. extract (array_merge(array('id' => null), $id));
  337. }
  338. $overrideRecursive = $recursive;
  339. if (empty ($id)) {
  340. $id = $Model->id;
  341. }
  342. extract($this->settings[$Model->alias]);
  343. if (!is_null($overrideRecursive)) {
  344. $recursive = $overrideRecursive;
  345. }
  346. $parentId = $Model->find('first', array('conditions' => array($Model->primaryKey => $id), 'fields' => array($parent), 'recursive' => -1));
  347. if ($parentId) {
  348. $parentId = $parentId[$Model->alias][$parent];
  349. $parent = $Model->find('first', array('conditions' => array($Model->escapeField() => $parentId), 'fields' => $fields, 'recursive' => $recursive));
  350. return $parent;
  351. }
  352. return false;
  353. }
  354. /**
  355. * Get the path to the given node
  356. *
  357. * @param Model $Model Model instance
  358. * @param mixed $id The ID of the record to read
  359. * @param mixed $fields Either a single string of a field name, or an array of field names
  360. * @param integer $recursive The number of levels deep to fetch associated records
  361. * @return array Array of nodes from top most parent to current node
  362. * @link http://book.cakephp.org/view/1350/getpath
  363. */
  364. public function getPath($Model, $id = null, $fields = null, $recursive = null) {
  365. if (is_array($id)) {
  366. extract (array_merge(array('id' => null), $id));
  367. }
  368. $overrideRecursive = $recursive;
  369. if (empty ($id)) {
  370. $id = $Model->id;
  371. }
  372. extract($this->settings[$Model->alias]);
  373. if (!is_null($overrideRecursive)) {
  374. $recursive = $overrideRecursive;
  375. }
  376. $result = $Model->find('first', array('conditions' => array($Model->escapeField() => $id), 'fields' => array($left, $right), 'recursive' => $recursive));
  377. if ($result) {
  378. $result = array_values($result);
  379. } else {
  380. return null;
  381. }
  382. $item = $result[0];
  383. $results = $Model->find('all', array(
  384. 'conditions' => array($scope, $Model->escapeField($left) . ' <=' => $item[$left], $Model->escapeField($right) . ' >=' => $item[$right]),
  385. 'fields' => $fields, 'order' => array($Model->escapeField($left) => 'asc'), 'recursive' => $recursive
  386. ));
  387. return $results;
  388. }
  389. /**
  390. * Reorder the node without changing the parent.
  391. *
  392. * If the node is the last child, or is a top level node with no subsequent node this method will return false
  393. *
  394. * @param Model $Model Model instance
  395. * @param mixed $id The ID of the record to move
  396. * @param integer|boolean $number how many places to move the node or true to move to last position
  397. * @return boolean true on success, false on failure
  398. * @link http://book.cakephp.org/view/1352/moveDown
  399. */
  400. public function moveDown($Model, $id = null, $number = 1) {
  401. if (is_array($id)) {
  402. extract (array_merge(array('id' => null), $id));
  403. }
  404. if (!$number) {
  405. return false;
  406. }
  407. if (empty ($id)) {
  408. $id = $Model->id;
  409. }
  410. extract($this->settings[$Model->alias]);
  411. list($node) = array_values($Model->find('first', array(
  412. 'conditions' => array($scope, $Model->escapeField() => $id),
  413. 'fields' => array($Model->primaryKey, $left, $right, $parent), 'recursive' => $recursive
  414. )));
  415. if ($node[$parent]) {
  416. list($parentNode) = array_values($Model->find('first', array(
  417. 'conditions' => array($scope, $Model->escapeField() => $node[$parent]),
  418. 'fields' => array($Model->primaryKey, $left, $right), 'recursive' => $recursive
  419. )));
  420. if (($node[$right] + 1) == $parentNode[$right]) {
  421. return false;
  422. }
  423. }
  424. $nextNode = $Model->find('first', array(
  425. 'conditions' => array($scope, $Model->escapeField($left) => ($node[$right] + 1)),
  426. 'fields' => array($Model->primaryKey, $left, $right), 'recursive' => $recursive)
  427. );
  428. if ($nextNode) {
  429. list($nextNode) = array_values($nextNode);
  430. } else {
  431. return false;
  432. }
  433. $edge = $this->_getMax($Model, $scope, $right, $recursive);
  434. $this->_sync($Model, $edge - $node[$left] + 1, '+', 'BETWEEN ' . $node[$left] . ' AND ' . $node[$right]);
  435. $this->_sync($Model, $nextNode[$left] - $node[$left], '-', 'BETWEEN ' . $nextNode[$left] . ' AND ' . $nextNode[$right]);
  436. $this->_sync($Model, $edge - $node[$left] - ($nextNode[$right] - $nextNode[$left]), '-', '> ' . $edge);
  437. if (is_int($number)) {
  438. $number--;
  439. }
  440. if ($number) {
  441. $this->moveDown($Model, $id, $number);
  442. }
  443. return true;
  444. }
  445. /**
  446. * Reorder the node without changing the parent.
  447. *
  448. * If the node is the first child, or is a top level node with no previous node this method will return false
  449. *
  450. * @param Model $Model Model instance
  451. * @param mixed $id The ID of the record to move
  452. * @param integer|boolean $number how many places to move the node, or true to move to first position
  453. * @return boolean true on success, false on failure
  454. * @link http://book.cakephp.org/view/1353/moveUp
  455. */
  456. public function moveUp($Model, $id = null, $number = 1) {
  457. if (is_array($id)) {
  458. extract (array_merge(array('id' => null), $id));
  459. }
  460. if (!$number) {
  461. return false;
  462. }
  463. if (empty ($id)) {
  464. $id = $Model->id;
  465. }
  466. extract($this->settings[$Model->alias]);
  467. list($node) = array_values($Model->find('first', array(
  468. 'conditions' => array($scope, $Model->escapeField() => $id),
  469. 'fields' => array($Model->primaryKey, $left, $right, $parent ), 'recursive' => $recursive
  470. )));
  471. if ($node[$parent]) {
  472. list($parentNode) = array_values($Model->find('first', array(
  473. 'conditions' => array($scope, $Model->escapeField() => $node[$parent]),
  474. 'fields' => array($Model->primaryKey, $left, $right), 'recursive' => $recursive
  475. )));
  476. if (($node[$left] - 1) == $parentNode[$left]) {
  477. return false;
  478. }
  479. }
  480. $previousNode = $Model->find('first', array(
  481. 'conditions' => array($scope, $Model->escapeField($right) => ($node[$left] - 1)),
  482. 'fields' => array($Model->primaryKey, $left, $right),
  483. 'recursive' => $recursive
  484. ));
  485. if ($previousNode) {
  486. list($previousNode) = array_values($previousNode);
  487. } else {
  488. return false;
  489. }
  490. $edge = $this->_getMax($Model, $scope, $right, $recursive);
  491. $this->_sync($Model, $edge - $previousNode[$left] +1, '+', 'BETWEEN ' . $previousNode[$left] . ' AND ' . $previousNode[$right]);
  492. $this->_sync($Model, $node[$left] - $previousNode[$left], '-', 'BETWEEN ' .$node[$left] . ' AND ' . $node[$right]);
  493. $this->_sync($Model, $edge - $previousNode[$left] - ($node[$right] - $node[$left]), '-', '> ' . $edge);
  494. if (is_int($number)) {
  495. $number--;
  496. }
  497. if ($number) {
  498. $this->moveUp($Model, $id, $number);
  499. }
  500. return true;
  501. }
  502. /**
  503. * Recover a corrupted tree
  504. *
  505. * The mode parameter is used to specify the source of info that is valid/correct. The opposite source of data
  506. * will be populated based upon that source of info. E.g. if the MPTT fields are corrupt or empty, with the $mode
  507. * 'parent' the values of the parent_id field will be used to populate the left and right fields. The missingParentAction
  508. * parameter only applies to "parent" mode and determines what to do if the parent field contains an id that is not present.
  509. *
  510. * @todo Could be written to be faster, *maybe*. Ideally using a subquery and putting all the logic burden on the DB.
  511. * @param Model $Model Model instance
  512. * @param string $mode parent or tree
  513. * @param mixed $missingParentAction 'return' to do nothing and return, 'delete' to
  514. * delete, or the id of the parent to set as the parent_id
  515. * @return boolean true on success, false on failure
  516. * @link http://book.cakephp.org/view/1628/Recover
  517. */
  518. public function recover($Model, $mode = 'parent', $missingParentAction = null) {
  519. if (is_array($mode)) {
  520. extract (array_merge(array('mode' => 'parent'), $mode));
  521. }
  522. extract($this->settings[$Model->alias]);
  523. $Model->recursive = $recursive;
  524. if ($mode == 'parent') {
  525. $Model->bindModel(array('belongsTo' => array('VerifyParent' => array(
  526. 'className' => $Model->alias,
  527. 'foreignKey' => $parent,
  528. 'fields' => array($Model->primaryKey, $left, $right, $parent),
  529. ))));
  530. $missingParents = $Model->find('list', array(
  531. 'recursive' => 0,
  532. 'conditions' => array($scope, array(
  533. 'NOT' => array($Model->escapeField($parent) => null), $Model->VerifyParent->escapeField() => null
  534. ))
  535. ));
  536. $Model->unbindModel(array('belongsTo' => array('VerifyParent')));
  537. if ($missingParents) {
  538. if ($missingParentAction == 'return') {
  539. foreach ($missingParents as $id => $display) {
  540. $this->errors[] = 'cannot find the parent for ' . $Model->alias . ' with id ' . $id . '(' . $display . ')';
  541. }
  542. return false;
  543. } elseif ($missingParentAction == 'delete') {
  544. $Model->deleteAll(array($Model->primaryKey => array_flip($missingParents)));
  545. } else {
  546. $Model->updateAll(array($parent => $missingParentAction), array($Model->escapeField($Model->primaryKey) => array_flip($missingParents)));
  547. }
  548. }
  549. $count = 1;
  550. foreach ($Model->find('all', array('conditions' => $scope, 'fields' => array($Model->primaryKey), 'order' => $left)) as $array) {
  551. $Model->id = $array[$Model->alias][$Model->primaryKey];
  552. $lft = $count++;
  553. $rght = $count++;
  554. $Model->save(array($left => $lft, $right => $rght), array('callbacks' => false));
  555. }
  556. foreach ($Model->find('all', array('conditions' => $scope, 'fields' => array($Model->primaryKey, $parent), 'order' => $left)) as $array) {
  557. $Model->create();
  558. $Model->id = $array[$Model->alias][$Model->primaryKey];
  559. $this->_setParent($Model, $array[$Model->alias][$parent]);
  560. }
  561. } else {
  562. $db = ConnectionManager::getDataSource($Model->useDbConfig);
  563. foreach ($Model->find('all', array('conditions' => $scope, 'fields' => array($Model->primaryKey, $parent), 'order' => $left)) as $array) {
  564. $path = $this->getPath($Model, $array[$Model->alias][$Model->primaryKey]);
  565. if ($path == null || count($path) < 2) {
  566. $parentId = null;
  567. } else {
  568. $parentId = $path[count($path) - 2][$Model->alias][$Model->primaryKey];
  569. }
  570. $Model->updateAll(array($parent => $db->value($parentId, $parent)), array($Model->escapeField() => $array[$Model->alias][$Model->primaryKey]));
  571. }
  572. }
  573. return true;
  574. }
  575. /**
  576. * Reorder method.
  577. *
  578. * Reorders the nodes (and child nodes) of the tree according to the field and direction specified in the parameters.
  579. * This method does not change the parent of any node.
  580. *
  581. * Requires a valid tree, by default it verifies the tree before beginning.
  582. *
  583. * Options:
  584. *
  585. * - 'id' id of record to use as top node for reordering
  586. * - 'field' Which field to use in reordeing defaults to displayField
  587. * - 'order' Direction to order either DESC or ASC (defaults to ASC)
  588. * - 'verify' Whether or not to verify the tree before reorder. defaults to true.
  589. *
  590. * @param Model $Model Model instance
  591. * @param array $options array of options to use in reordering.
  592. * @return boolean true on success, false on failure
  593. * @link http://book.cakephp.org/view/1355/reorder
  594. * @link http://book.cakephp.org/view/1629/Reorder
  595. */
  596. public function reorder($Model, $options = array()) {
  597. $options = array_merge(array('id' => null, 'field' => $Model->displayField, 'order' => 'ASC', 'verify' => true), $options);
  598. extract($options);
  599. if ($verify && !$this->verify($Model)) {
  600. return false;
  601. }
  602. $verify = false;
  603. extract($this->settings[$Model->alias]);
  604. $fields = array($Model->primaryKey, $field, $left, $right);
  605. $sort = $field . ' ' . $order;
  606. $nodes = $this->children($Model, $id, true, $fields, $sort, null, null, $recursive);
  607. $cacheQueries = $Model->cacheQueries;
  608. $Model->cacheQueries = false;
  609. if ($nodes) {
  610. foreach ($nodes as $node) {
  611. $id = $node[$Model->alias][$Model->primaryKey];
  612. $this->moveDown($Model, $id, true);
  613. if ($node[$Model->alias][$left] != $node[$Model->alias][$right] - 1) {
  614. $this->reorder($Model, compact('id', 'field', 'order', 'verify'));
  615. }
  616. }
  617. }
  618. $Model->cacheQueries = $cacheQueries;
  619. return true;
  620. }
  621. /**
  622. * Remove the current node from the tree, and reparent all children up one level.
  623. *
  624. * If the parameter delete is false, the node will become a new top level node. Otherwise the node will be deleted
  625. * after the children are reparented.
  626. *
  627. * @param Model $Model Model instance
  628. * @param mixed $id The ID of the record to remove
  629. * @param boolean $delete whether to delete the node after reparenting children (if any)
  630. * @return boolean true on success, false on failure
  631. * @link http://book.cakephp.org/view/1354/removeFromTree
  632. */
  633. public function removeFromTree($Model, $id = null, $delete = false) {
  634. if (is_array($id)) {
  635. extract (array_merge(array('id' => null), $id));
  636. }
  637. extract($this->settings[$Model->alias]);
  638. list($node) = array_values($Model->find('first', array(
  639. 'conditions' => array($scope, $Model->escapeField() => $id),
  640. 'fields' => array($Model->primaryKey, $left, $right, $parent),
  641. 'recursive' => $recursive
  642. )));
  643. if ($node[$right] == $node[$left] + 1) {
  644. if ($delete) {
  645. return $Model->delete($id);
  646. } else {
  647. $Model->id = $id;
  648. return $Model->saveField($parent, null);
  649. }
  650. } elseif ($node[$parent]) {
  651. list($parentNode) = array_values($Model->find('first', array(
  652. 'conditions' => array($scope, $Model->escapeField() => $node[$parent]),
  653. 'fields' => array($Model->primaryKey, $left, $right),
  654. 'recursive' => $recursive
  655. )));
  656. } else {
  657. $parentNode[$right] = $node[$right] + 1;
  658. }
  659. $db = ConnectionManager::getDataSource($Model->useDbConfig);
  660. $Model->updateAll(
  661. array($parent => $db->value($node[$parent], $parent)),
  662. array($Model->escapeField($parent) => $node[$Model->primaryKey])
  663. );
  664. $this->_sync($Model, 1, '-', 'BETWEEN ' . ($node[$left] + 1) . ' AND ' . ($node[$right] - 1));
  665. $this->_sync($Model, 2, '-', '> ' . ($node[$right]));
  666. $Model->id = $id;
  667. if ($delete) {
  668. $Model->updateAll(
  669. array(
  670. $Model->escapeField($left) => 0,
  671. $Model->escapeField($right) => 0,
  672. $Model->escapeField($parent) => null
  673. ),
  674. array($Model->escapeField() => $id)
  675. );
  676. return $Model->delete($id);
  677. } else {
  678. $edge = $this->_getMax($Model, $scope, $right, $recursive);
  679. if ($node[$right] == $edge) {
  680. $edge = $edge - 2;
  681. }
  682. $Model->id = $id;
  683. return $Model->save(
  684. array($left => $edge + 1, $right => $edge + 2, $parent => null),
  685. array('callbacks' => false)
  686. );
  687. }
  688. }
  689. /**
  690. * Check if the current tree is valid.
  691. *
  692. * Returns true if the tree is valid otherwise an array of (type, incorrect left/right index, message)
  693. *
  694. * @param Model $Model Model instance
  695. * @return mixed true if the tree is valid or empty, otherwise an array of (error type [index, node],
  696. * [incorrect left/right index,node id], message)
  697. * @link http://book.cakephp.org/view/1630/Verify
  698. */
  699. public function verify($Model) {
  700. extract($this->settings[$Model->alias]);
  701. if (!$Model->find('count', array('conditions' => $scope))) {
  702. return true;
  703. }
  704. $min = $this->_getMin($Model, $scope, $left, $recursive);
  705. $edge = $this->_getMax($Model, $scope, $right, $recursive);
  706. $errors = array();
  707. for ($i = $min; $i <= $edge; $i++) {
  708. $count = $Model->find('count', array('conditions' => array(
  709. $scope, 'OR' => array($Model->escapeField($left) => $i, $Model->escapeField($right) => $i)
  710. )));
  711. if ($count != 1) {
  712. if ($count == 0) {
  713. $errors[] = array('index', $i, 'missing');
  714. } else {
  715. $errors[] = array('index', $i, 'duplicate');
  716. }
  717. }
  718. }
  719. $node = $Model->find('first', array('conditions' => array($scope, $Model->escapeField($right) . '< ' . $Model->escapeField($left)), 'recursive' => 0));
  720. if ($node) {
  721. $errors[] = array('node', $node[$Model->alias][$Model->primaryKey], 'left greater than right.');
  722. }
  723. $Model->bindModel(array('belongsTo' => array('VerifyParent' => array(
  724. 'className' => $Model->alias,
  725. 'foreignKey' => $parent,
  726. 'fields' => array($Model->primaryKey, $left, $right, $parent)
  727. ))));
  728. foreach ($Model->find('all', array('conditions' => $scope, 'recursive' => 0)) as $instance) {
  729. if (is_null($instance[$Model->alias][$left]) || is_null($instance[$Model->alias][$right])) {
  730. $errors[] = array('node', $instance[$Model->alias][$Model->primaryKey],
  731. 'has invalid left or right values');
  732. } elseif ($instance[$Model->alias][$left] == $instance[$Model->alias][$right]) {
  733. $errors[] = array('node', $instance[$Model->alias][$Model->primaryKey],
  734. 'left and right values identical');
  735. } elseif ($instance[$Model->alias][$parent]) {
  736. if (!$instance['VerifyParent'][$Model->primaryKey]) {
  737. $errors[] = array('node', $instance[$Model->alias][$Model->primaryKey],
  738. 'The parent node ' . $instance[$Model->alias][$parent] . ' doesn\'t exist');
  739. } elseif ($instance[$Model->alias][$left] < $instance['VerifyParent'][$left]) {
  740. $errors[] = array('node', $instance[$Model->alias][$Model->primaryKey],
  741. 'left less than parent (node ' . $instance['VerifyParent'][$Model->primaryKey] . ').');
  742. } elseif ($instance[$Model->alias][$right] > $instance['VerifyParent'][$right]) {
  743. $errors[] = array('node', $instance[$Model->alias][$Model->primaryKey],
  744. 'right greater than parent (node ' . $instance['VerifyParent'][$Model->primaryKey] . ').');
  745. }
  746. } elseif ($Model->find('count', array('conditions' => array($scope, $Model->escapeField($left) . ' <' => $instance[$Model->alias][$left], $Model->escapeField($right) . ' >' => $instance[$Model->alias][$right]), 'recursive' => 0))) {
  747. $errors[] = array('node', $instance[$Model->alias][$Model->primaryKey], 'The parent field is blank, but has a parent');
  748. }
  749. }
  750. if ($errors) {
  751. return $errors;
  752. }
  753. return true;
  754. }
  755. /**
  756. * Sets the parent of the given node
  757. *
  758. * The force parameter is used to override the "don't change the parent to the current parent" logic in the event
  759. * of recovering a corrupted table, or creating new nodes. Otherwise it should always be false. In reality this
  760. * method could be private, since calling save with parent_id set also calls setParent
  761. *
  762. * @param Model $Model Model instance
  763. * @param mixed $parentId
  764. * @param boolean $created
  765. * @return boolean true on success, false on failure
  766. */
  767. protected function _setParent($Model, $parentId = null, $created = false) {
  768. extract($this->settings[$Model->alias]);
  769. list($node) = array_values($Model->find('first', array(
  770. 'conditions' => array($scope, $Model->escapeField() => $Model->id),
  771. 'fields' => array($Model->primaryKey, $parent, $left, $right),
  772. 'recursive' => $recursive
  773. )));
  774. $edge = $this->_getMax($Model, $scope, $right, $recursive, $created);
  775. if (empty ($parentId)) {
  776. $this->_sync($Model, $edge - $node[$left] + 1, '+', 'BETWEEN ' . $node[$left] . ' AND ' . $node[$right], $created);
  777. $this->_sync($Model, $node[$right] - $node[$left] + 1, '-', '> ' . $node[$left], $created);
  778. } else {
  779. $values = $Model->find('first', array(
  780. 'conditions' => array($scope, $Model->escapeField() => $parentId),
  781. 'fields' => array($Model->primaryKey, $left, $right),
  782. 'recursive' => $recursive
  783. ));
  784. if ($values === false) {
  785. return false;
  786. }
  787. $parentNode = array_values($values);
  788. if (empty($parentNode) || empty($parentNode[0])) {
  789. return false;
  790. }
  791. $parentNode = $parentNode[0];
  792. if (($Model->id == $parentId)) {
  793. return false;
  794. } elseif (($node[$left] < $parentNode[$left]) && ($parentNode[$right] < $node[$right])) {
  795. return false;
  796. }
  797. if (empty ($node[$left]) && empty ($node[$right])) {
  798. $this->_sync($Model, 2, '+', '>= ' . $parentNode[$right], $created);
  799. $result = $Model->save(
  800. array($left => $parentNode[$right], $right => $parentNode[$right] + 1, $parent => $parentId),
  801. array('validate' => false, 'callbacks' => false)
  802. );
  803. $Model->data = $result;
  804. } else {
  805. $this->_sync($Model, $edge - $node[$left] +1, '+', 'BETWEEN ' . $node[$left] . ' AND ' . $node[$right], $created);
  806. $diff = $node[$right] - $node[$left] + 1;
  807. if ($node[$left] > $parentNode[$left]) {
  808. if ($node[$right] < $parentNode[$right]) {
  809. $this->_sync($Model, $diff, '-', 'BETWEEN ' . $node[$right] . ' AND ' . ($parentNode[$right] - 1), $created);
  810. $this->_sync($Model, $edge - $parentNode[$right] + $diff + 1, '-', '> ' . $edge, $created);
  811. } else {
  812. $this->_sync($Model, $diff, '+', 'BETWEEN ' . $parentNode[$right] . ' AND ' . $node[$right], $created);
  813. $this->_sync($Model, $edge - $parentNode[$right] + 1, '-', '> ' . $edge, $created);
  814. }
  815. } else {
  816. $this->_sync($Model, $diff, '-', 'BETWEEN ' . $node[$right] . ' AND ' . ($parentNode[$right] - 1), $created);
  817. $this->_sync($Model, $edge - $parentNode[$right] + $diff + 1, '-', '> ' . $edge, $created);
  818. }
  819. }
  820. }
  821. return true;
  822. }
  823. /**
  824. * get the maximum index value in the table.
  825. *
  826. * @param Model $Model
  827. * @param string $scope
  828. * @param string $right
  829. * @param integer $recursive
  830. * @param boolean $created
  831. * @return integer
  832. */
  833. protected function _getMax($Model, $scope, $right, $recursive = -1, $created = false) {
  834. $db = ConnectionManager::getDataSource($Model->useDbConfig);
  835. if ($created) {
  836. if (is_string($scope)) {
  837. $scope .= " AND {$Model->alias}.{$Model->primaryKey} <> ";
  838. $scope .= $db->value($Model->id, $Model->getColumnType($Model->primaryKey));
  839. } else {
  840. $scope['NOT'][$Model->alias . '.' . $Model->primaryKey] = $Model->id;
  841. }
  842. }
  843. $name = $Model->alias . '.' . $right;
  844. list($edge) = array_values($Model->find('first', array(
  845. 'conditions' => $scope,
  846. 'fields' => $db->calculate($Model, 'max', array($name, $right)),
  847. 'recursive' => $recursive
  848. )));
  849. return (empty($edge[$right])) ? 0 : $edge[$right];
  850. }
  851. /**
  852. * get the minimum index value in the table.
  853. *
  854. * @param Model $Model
  855. * @param string $scope
  856. * @param string $left
  857. * @param integer $recursive
  858. * @return integer
  859. */
  860. protected function _getMin($Model, $scope, $left, $recursive = -1) {
  861. $db = ConnectionManager::getDataSource($Model->useDbConfig);
  862. $name = $Model->alias . '.' . $left;
  863. list($edge) = array_values($Model->find('first', array(
  864. 'conditions' => $scope,
  865. 'fields' => $db->calculate($Model, 'min', array($name, $left)),
  866. 'recursive' => $recursive
  867. )));
  868. return (empty($edge[$left])) ? 0 : $edge[$left];
  869. }
  870. /**
  871. * Table sync method.
  872. *
  873. * Handles table sync operations, Taking account of the behavior scope.
  874. *
  875. * @param Model $Model
  876. * @param integer $shift
  877. * @param string $dir
  878. * @param array $conditions
  879. * @param boolean $created
  880. * @param string $field
  881. * @return void
  882. */
  883. protected function _sync($Model, $shift, $dir = '+', $conditions = array(), $created = false, $field = 'both') {
  884. $ModelRecursive = $Model->recursive;
  885. extract($this->settings[$Model->alias]);
  886. $Model->recursive = $recursive;
  887. if ($field == 'both') {
  888. $this->_sync($Model, $shift, $dir, $conditions, $created, $left);
  889. $field = $right;
  890. }
  891. if (is_string($conditions)) {
  892. $conditions = array("{$Model->alias}.{$field} {$conditions}");
  893. }
  894. if (($scope != '1 = 1' && $scope !== true) && $scope) {
  895. $conditions[] = $scope;
  896. }
  897. if ($created) {
  898. $conditions['NOT'][$Model->alias . '.' . $Model->primaryKey] = $Model->id;
  899. }
  900. $Model->updateAll(array($Model->alias . '.' . $field => $Model->escapeField($field) . ' ' . $dir . ' ' . $shift), $conditions);
  901. $Model->recursive = $ModelRecursive;
  902. }
  903. }