Chris@0: <?php
Chris@0: 
Chris@0: /*
Chris@0:  * This file is part of the Prophecy.
Chris@0:  * (c) Konstantin Kudryashov <ever.zet@gmail.com>
Chris@0:  *     Marcello Duarte <marcello.duarte@gmail.com>
Chris@0:  *
Chris@0:  * For the full copyright and license information, please view the LICENSE
Chris@0:  * file that was distributed with this source code.
Chris@0:  */
Chris@0: 
Chris@0: namespace Prophecy\Argument\Token;
Chris@0: 
Chris@0: /**
Chris@0:  * Array elements count token.
Chris@0:  *
Chris@0:  * @author Boris Mikhaylov <kaguxmail@gmail.com>
Chris@0:  */
Chris@0: 
Chris@0: class ArrayCountToken implements TokenInterface
Chris@0: {
Chris@0:     private $count;
Chris@0: 
Chris@0:     /**
Chris@0:      * @param integer $value
Chris@0:      */
Chris@0:     public function __construct($value)
Chris@0:     {
Chris@0:         $this->count = $value;
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * Scores 6 when argument has preset number of elements.
Chris@0:      *
Chris@0:      * @param $argument
Chris@0:      *
Chris@0:      * @return bool|int
Chris@0:      */
Chris@0:     public function scoreArgument($argument)
Chris@0:     {
Chris@0:         return $this->isCountable($argument) && $this->hasProperCount($argument) ? 6 : false;
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * Returns false.
Chris@0:      *
Chris@0:      * @return boolean
Chris@0:      */
Chris@0:     public function isLast()
Chris@0:     {
Chris@0:         return false;
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * Returns string representation for token.
Chris@0:      *
Chris@0:      * @return string
Chris@0:      */
Chris@0:     public function __toString()
Chris@0:     {
Chris@0:         return sprintf('count(%s)', $this->count);
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * Returns true if object is either array or instance of \Countable
Chris@0:      *
Chris@0:      * @param $argument
Chris@0:      * @return bool
Chris@0:      */
Chris@0:     private function isCountable($argument)
Chris@0:     {
Chris@0:         return (is_array($argument) || $argument instanceof \Countable);
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * Returns true if $argument has expected number of elements
Chris@0:      *
Chris@0:      * @param array|\Countable $argument
Chris@0:      *
Chris@0:      * @return bool
Chris@0:      */
Chris@0:     private function hasProperCount($argument)
Chris@0:     {
Chris@0:         return $this->count === count($argument);
Chris@0:     }
Chris@0: }