1: <?php
 2: namespace Pharborist\Functions;
 3: 
 4: use Pharborist\DocCommentTrait;
 5: use Pharborist\ParenTrait;
 6: use Pharborist\Token;
 7: use Pharborist\TokenNode;
 8: 
 9: trait FunctionTrait {
10:   use ParameterTrait;
11:   use DocCommentTrait;
12:   use ParenTrait;
13: 
14:   /**
15:    * @var TokenNode
16:    */
17:   protected $reference;
18: 
19:   /**
20:    * @return TokenNode
21:    */
22:   public function getReference() {
23:     return $this->reference;
24:   }
25: 
26:   /**
27:    * @param boolean $is_reference
28:    * @return $this
29:    */
30:   public function setReference($is_reference) {
31:     if ($is_reference) {
32:       if (!isset($this->reference)) {
33:         /** @var \Pharborist\Functions\FunctionDeclarationNode|\Pharborist\Objects\ClassMethodNode|\Pharborist\Objects\InterfaceMethodNode $this */
34:         $this->name->before(Token::reference());
35:       }
36:     }
37:     else {
38:       if (isset($this->reference)) {
39:         $this->reference->remove();
40:       }
41:     }
42:     return $this;
43:   }
44: }
45: