Add the addClause() method.

1 files changed, 20 insertions, 1 deletions

diff --git a/Model/Behavior.php b/Model/Behavior.php
index 362f9f4..efa7c31 100644
--- a/Model/Behavior.php
+++ b/Model/Behavior.php
@@ -141,7 +141,7 @@ class Behavior extends Clause {
 
 
     /**
-     * Get a specific clause.
+     * Get or create a specific clause.
      *
      * @access  public
      * @param   string  $clause    Clause (without leading arobase).
@@ -212,6 +212,25 @@ class Behavior extends Clause {
     }
 
     /**
+     * Add a clause.
+     *
+     * @access  public
+     * @param   \Hoa\Praspel\Model\Clause  $clause    Clause.
+     * @return  \Hoa\Praspel\Model\Clause
+     */
+    public function addClause ( Clause $clause ) {
+
+        $name = $clause->getName();
+
+        if(false === in_array($name, static::getAllowedClauses()))
+            throw new \Hoa\Praspel\Exception\Model(
+                'Clause @%s is not allowed in @%s.',
+                0, array($name, $this->getId()));
+
+        return $this->_clauses[$name] = $clause;
+    }
+
+    /**
      * Check if a clause already exists, i.e. has been declared.
      *
      * @access  public