diff options
author | Bharat Mediratta <bharat@menalto.com> | 2008-10-31 22:12:14 +0000 |
---|---|---|
committer | Bharat Mediratta <bharat@menalto.com> | 2008-10-31 22:12:14 +0000 |
commit | eba717f95f586d2538007bd18da6e9b32b076c30 (patch) | |
tree | 15fc596a270f9de0d163c66c96e3c65fca5ee100 /kohana/libraries/ORM_Tree.php | |
parent | fff10f8b70376ef25722bd867df26bc5aefced43 (diff) |
Merge over vendor code.
git-svn-id: http://gallery.svn.sourceforge.net/svnroot/gallery/trunk/eval/gx/gallery3/trunk@18408 57fcd75e-5312-0410-8df3-f5eb6fbb1595
Diffstat (limited to 'kohana/libraries/ORM_Tree.php')
-rw-r--r-- | kohana/libraries/ORM_Tree.php | 76 |
1 files changed, 76 insertions, 0 deletions
diff --git a/kohana/libraries/ORM_Tree.php b/kohana/libraries/ORM_Tree.php new file mode 100644 index 00000000..ff5f62f1 --- /dev/null +++ b/kohana/libraries/ORM_Tree.php @@ -0,0 +1,76 @@ +<?php defined('SYSPATH') or die('No direct script access.'); +/** + * Object Relational Mapping (ORM) "tree" extension. Allows ORM objects to act + * as trees, with parents and children. + * + * $Id$ + * + * @package Core + * @author Kohana Team + * @copyright (c) 2007-2008 Kohana Team + * @license http://kohanaphp.com/license.html + */ +class ORM_Tree_Core extends ORM { + + // Name of the child + protected $children; + + // Parent keyword name + protected $parent_key = 'parent_id'; + + /** + * Overload ORM::__get to support "parent" and "children" properties. + * + * @param string column name + * @return mixed + */ + public function __get($column) + { + if ($column === 'parent') + { + if (empty($this->related[$column])) + { + // Load child model + $model = ORM::factory(inflector::singular($this->children)); + + if (isset($this->object[$this->parent_key])) + { + // Find children of this parent + $model->where($this->parent_key, $this->object[$this->parent_key])->find(); + } + + $this->related[$column] = $model; + } + + return $this->related[$column]; + } + elseif ($column === 'children') + { + if (empty($this->related[$column])) + { + $model = ORM::factory(inflector::singular($this->children)); + + if ($this->children === $this->table_name) + { + // Load children within this table + $this->related[$column] = $model + ->where($this->parent_key, $this->object[$this->primary_key]) + ->find_all(); + } + else + { + // Find first selection of children + $this->related[$column] = $model + ->where($this->foreign_key(), $this->object[$this->primary_key]) + ->where($this->parent_key, NULL) + ->find_all(); + } + } + + return $this->related[$column]; + } + + return parent::__get($column); + } + +} // End ORM Tree
\ No newline at end of file |