No commit message

ORM_Tree.php

@@ -5,6 +5,7 @@ class ORM_Tree_Core{
 	protected $class;
 
 	protected $parent_column='parent_id';
+
 	public function __construct(ORM $model)
 	{
 		$this->model=$model;
@@ -14,28 +15,34 @@ public function set_parent_column($value)
 	{
 		$this->parent_column=$value;
 	}
-
+	public function get_parent_column()
+	{
+		return $this->parent_column;
+	}
 	public function get_children()
 	{
-		return $this->model->find_all_by_parent_id($this->model->id);
+		$method='find_all_by_'.$this->parent_column;
+		return $this->model->$method($this->model->id);
 	}
 	public function has_parent()
 	{
-		return !empty($this->model->parent_id);
+		return !empty($this->model->{$this->parent_column});
 	}
 	public function get_parent(){
 		if($this->has_parent())
-			return $this->model->find_all_by_id($this->model->parent_id);
+			return $this->model->find_all_by_id($this->model->{$this->parent_column});
 
 		return false;
 	}
 	public function get_siblings(){
+		$method='find_all_by_'.$this->parent_column;		
 		$this->model->where(array('id!='=>$this->model->id));
-		return $this->model->find_all_by_parent_id($this->model->parent_id);
+		return $this->model->$method($this->model->{$this->parent_column});
 	}
 	public function get_self_and_siblings()
 	{
-		return $this->model->find_all_by_parent_id($this->model->parent_id);
+		$method='find_all_by_'.$this->parent_column;
+		return $this->model->$method($this->model->{$this->parent_column});
 	}
 	public function add_child(ORM $child)
 	{
@@ -47,7 +54,7 @@ public function add_child(ORM $child)
 			$this->model->save();
 		}
 
-		$child->parent_id=$this->model->id;
+		$child->{$this->parent_column}=$this->model->id;
 		return $child->save();	
 
 	}