Update permissions

Author: alex-dna

src/models/DecisionTreeAnswer.php

@@ -51,12 +51,30 @@ public function getCMSFields()
 		return $fields;
 	}
 
+	/**
+	* Permissions
+	*/
+	public function canCreate($member = null) 
+	{
+		return singleton('ElementDecisionTree')->canCreate($member);
+	}
+
+	public function canView($member = null)
+	{
+		return singleton('ElementDecisionTree')->canCreate($member);
+	}
+
+	public function canEdit($member = null) 
+	{
+		return singleton('ElementDecisionTree')->canCreate($member);
+	}
+
 	/**
 	* Can only delete an answer that doesn't have a dependant question
 	*/
 	public function canDelete($member = null)
 	{
-		$canDelete = parent::canDelete($member);
+		$canDelete = singleton('ElementDecisionTree')->canDelete($member);
 		return ($canDelete && !$this->ResultingStep()->exists());
 	}
 

src/models/DecisionTreeStep.php

@@ -79,12 +79,30 @@ public function onBeforeWrite()
 		parent::onBeforeWrite();
 	}
 
+	/**
+	* Permissions
+	*/
+	public function canCreate($member = null) 
+	{
+		return singleton('ElementDecisionTree')->canCreate($member);
+	}
+
+	public function canView($member = null)
+	{
+		return singleton('ElementDecisionTree')->canCreate($member);
+	}
+
+	public function canEdit($member = null) 
+	{
+		return singleton('ElementDecisionTree')->canCreate($member);
+	}
+
 	/**
 	* Prevent deleting Step with answers that have dependant questions
 	*/
 	public function candelete($member = null)
 	{
-		$canDelete = parent::canDelete($member);
+		$canDelete = singleton('ElementDecisionTree')->canDelete($member);
 
 		foreach($this->Answers() as $answer) {
 			if (!$answer->canDelete()) {