<?php

namespace Phabel\Target\Php70;

use Phabel\Plugin;
use PhpParser\Node;
use PhpParser\Node\Expr\ArrayDimFetch;
use PhpParser\Node\Expr\ClassConstFetch;
use PhpParser\Node\Expr\Isset_;
use PhpParser\Node\Expr\Yield_;

/**
 * Fix certain compound access statements.
 */
class CompoundAccess extends Plugin
{
    /**
     * Replace non-simple isset calls.
     *
     * @param Isset_ $node Isset node
     *
     * @return void
     */
    public function enterIsset(Isset_ $node): void
    {
        foreach ($node->vars as &$var) {
            if (!$var instanceof ArrayDimFetch) {
                continue;
            }
            if (!$var->var instanceof ClassConstFetch) {
                continue;
            }
            $var->var = self::callPoly('returnMe', $var->var);
        }
    }
    /**
     * Fix yield array access.
     *
     * @param ArrayDimFetch $node Node
     * 
     * @return void
     */
    public function enterArrayYield(ArrayDimFetch $node): void
    {
        if (!$node->var instanceof Node\Expr\Yield_) {
            return;
        }
        $node->var = self::callPoly('returnMe', $node->var);
    }
    /**
     * Fix yield array access
     *
     * @param Yield_ $node Yield
     * 
     * @return void
     */
    public function enterYield(Yield_ $node): void
    {
        $value = &$node->value;
        if ($value instanceof Node\Expr\Variable && $value->name !== "this") {
            return;
        }
        if ($value instanceof Node\Expr\FuncCall ||
                $value instanceof Node\Expr\MethodCall ||
                $value instanceof Node\Expr\StaticCall ||
                $value instanceof Node\Scalar
            ) {
            return;
        }
        $value = self::callPoly('returnMe', $value);
    }
    /**
     * Returns the data provided.
     *
     * @param mixed $data Data
     *
     * @return mixed
     *
     * @template T
     *
     * @psalm-param T $data data
     *
     * @psalm-return T
     */
    public static function returnMe($data)
    {
        return $data;
    }
}