MethodNode.php in Zircon Profile 8

Namespace

Prophecy\Doubler\Generator\Node

File

vendor/phpspec/prophecy/src/Prophecy/Doubler/Generator/Node/MethodNode.php

<?php

/*
 * This file is part of the Prophecy.
 * (c) Konstantin Kudryashov <[e-mail]>
 *     Marcello Duarte <[e-mail]>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */
namespace Prophecy\Doubler\Generator\Node;

use Prophecy\Exception\InvalidArgumentException;

/**
 * Method node.
 *
 * @author Konstantin Kudryashov <[e-mail]>
 */
class MethodNode {
  private $name;
  private $code;
  private $visibility = 'public';
  private $static = false;
  private $returnsReference = false;
  private $returnType;

  /**
   * @var ArgumentNode[]
   */
  private $arguments = array();

  /**
   * @param string $name
   * @param string $code
   */
  public function __construct($name, $code = null) {
    $this->name = $name;
    $this->code = $code;
  }
  public function getVisibility() {
    return $this->visibility;
  }

  /**
   * @param string $visibility
   */
  public function setVisibility($visibility) {
    $visibility = strtolower($visibility);
    if (!in_array($visibility, array(
      'public',
      'private',
      'protected',
    ))) {
      throw new InvalidArgumentException(sprintf('`%s` method visibility is not supported.', $visibility));
    }
    $this->visibility = $visibility;
  }
  public function isStatic() {
    return $this->static;
  }
  public function setStatic($static = true) {
    $this->static = (bool) $static;
  }
  public function returnsReference() {
    return $this->returnsReference;
  }
  public function setReturnsReference() {
    $this->returnsReference = true;
  }
  public function getName() {
    return $this->name;
  }
  public function addArgument(ArgumentNode $argument) {
    $this->arguments[] = $argument;
  }

  /**
   * @return ArgumentNode[]
   */
  public function getArguments() {
    return $this->arguments;
  }
  public function hasReturnType() {
    return null !== $this->returnType;
  }

  /**
   * @param string $type
   */
  public function setReturnType($type = null) {
    switch ($type) {
      case '':
        $this->returnType = null;
        break;
      case 'string':
      case 'float':
      case 'int':
      case 'bool':
      case 'array':
      case 'callable':
        $this->returnType = $type;
        break;
      case 'double':
      case 'real':
        $this->returnType = 'float';
        break;
      case 'boolean':
        $this->returnType = 'bool';
        break;
      case 'integer':
        $this->returnType = 'int';
        break;
      default:
        $this->returnType = '\\' . ltrim($type, '\\');
    }
  }
  public function getReturnType() {
    return $this->returnType;
  }

  /**
   * @param string $code
   */
  public function setCode($code) {
    $this->code = $code;
  }
  public function getCode() {
    if ($this->returnsReference) {
      return "throw new \\Prophecy\\Exception\\Doubler\\ReturnByReferenceException('Returning by reference not supported', get_class(\$this), '{$this->name}');";
    }
    return (string) $this->code;
  }
  public function useParentCode() {
    $this->code = sprintf('return parent::%s(%s);', $this
      ->getName(), implode(', ', array_map(function (ArgumentNode $arg) {
      return '$' . $arg
        ->getName();
    }, $this->arguments)));
  }

}

Classes

Name	Description
MethodNode	Method node.