<?php

declare (strict_types=1);
namespace Rector\DowngradePhp80\Rector\Expression;

use PhpParser\Node;
use PhpParser\Node\Expr;
use PhpParser\Node\Expr\ArrayDimFetch;
use PhpParser\Node\Expr\ArrowFunction;
use PhpParser\Node\Expr\Assign;
use PhpParser\Node\Expr\BinaryOp\Coalesce;
use PhpParser\Node\Expr\BinaryOp\Identical;
use PhpParser\Node\Expr\BooleanNot;
use PhpParser\Node\Expr\CallLike;
use PhpParser\Node\Expr\Closure;
use PhpParser\Node\Expr\ConstFetch;
use PhpParser\Node\Expr\Isset_;
use PhpParser\Node\Expr\PropertyFetch;
use PhpParser\Node\Expr\Ternary;
use PhpParser\Node\Expr\Throw_;
use PhpParser\Node\Expr\Variable;
use PhpParser\Node\Name;
use PhpParser\Node\Stmt;
use PhpParser\Node\Stmt\Expression;
use PhpParser\Node\Stmt\If_;
use PhpParser\Node\Stmt\Return_;
use Rector\NodeAnalyzer\CoalesceAnalyzer;
use Rector\NodeFactory\NamedVariableFactory;
use Rector\NodeManipulator\BinaryOpManipulator;
use Rector\Php72\NodeFactory\AnonymousFunctionFactory;
use Rector\PhpParser\Node\BetterNodeFinder;
use Rector\Rector\AbstractRector;
use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
/**
 * @changelog https://wiki.php.net/rfc/throw_expression
 *
 * @see \Rector\Tests\DowngradePhp80\Rector\Expression\DowngradeThrowExprRector\DowngradeThrowExprRectorTest
 */
final class DowngradeThrowExprRector extends AbstractRector
{
    /**
     * @readonly
     */
    private CoalesceAnalyzer $coalesceAnalyzer;
    /**
     * @readonly
     */
    private BinaryOpManipulator $binaryOpManipulator;
    /**
     * @readonly
     */
    private BetterNodeFinder $betterNodeFinder;
    /**
     * @readonly
     */
    private AnonymousFunctionFactory $anonymousFunctionFactory;
    /**
     * @readonly
     */
    private NamedVariableFactory $namedVariableFactory;
    public function __construct(CoalesceAnalyzer $coalesceAnalyzer, BinaryOpManipulator $binaryOpManipulator, BetterNodeFinder $betterNodeFinder, AnonymousFunctionFactory $anonymousFunctionFactory, NamedVariableFactory $namedVariableFactory)
    {
        $this->coalesceAnalyzer = $coalesceAnalyzer;
        $this->binaryOpManipulator = $binaryOpManipulator;
        $this->betterNodeFinder = $betterNodeFinder;
        $this->anonymousFunctionFactory = $anonymousFunctionFactory;
        $this->namedVariableFactory = $namedVariableFactory;
    }
    public function getRuleDefinition(): RuleDefinition
    {
        return new RuleDefinition('Downgrade throw expression', [new CodeSample(<<<'CODE_SAMPLE'
echo $variable ?? throw new RuntimeException();
CODE_SAMPLE
, <<<'CODE_SAMPLE'
if (! isset($variable)) {
    throw new RuntimeException();
}

echo $variable;
CODE_SAMPLE
)]);
    }
    /**
     * @return array<class-string<Node>>
     */
    public function getNodeTypes(): array
    {
        return [ArrowFunction::class, Expression::class, Return_::class];
    }
    /**
     * @param ArrowFunction|Expression|Return_ $node
     * @return Node|Node[]|null
     */
    public function refactor(Node $node)
    {
        if ($node instanceof ArrowFunction) {
            return $this->refactorArrowFunctionReturn($node);
        }
        if ($node instanceof Return_) {
            return $this->refactorReturn($node);
        }
        if ($node->expr instanceof Throw_) {
            return null;
        }
        if ($node->expr instanceof Assign) {
            $resultNode = $this->refactorAssign($node->expr);
            if ($resultNode !== null) {
                return $resultNode;
            }
        }
        if ($node->expr instanceof CallLike && !$node->expr->isFirstClassCallable()) {
            $args = $node->expr->getArgs();
            foreach ($args as $arg) {
                if ($arg->value instanceof Ternary && $arg->value->else instanceof Throw_) {
                    $refactorTernary = $this->refactorTernary($arg->value, null, \true);
                    if (is_array($refactorTernary) && $refactorTernary[0] instanceof Expression && $refactorTernary[0]->expr instanceof Assign) {
                        $arg->value = $refactorTernary[0]->expr->var;
                        return array_merge($refactorTernary, [$node]);
                    }
                }
            }
            return null;
        }
        if ($node->expr instanceof Coalesce) {
            return $this->refactorCoalesce($node->expr, null);
        }
        if ($node->expr instanceof Ternary) {
            return $this->refactorTernary($node->expr, null);
        }
        return $this->refactorDirectCoalesce($node);
    }
    private function refactorArrowFunctionReturn(ArrowFunction $arrowFunction): ?Closure
    {
        if (!$arrowFunction->expr instanceof Throw_) {
            return null;
        }
        $stmts = [new Expression($arrowFunction->expr)];
        return $this->anonymousFunctionFactory->create($arrowFunction->params, $stmts, $arrowFunction->returnType, $arrowFunction->static);
    }
    /**
     * @return If_|Expression|Stmt[]|null
     */
    private function refactorAssign(Assign $assign)
    {
        if (!$this->hasThrowInAssignExpr($assign)) {
            return null;
        }
        if ($assign->expr instanceof Coalesce) {
            return $this->refactorCoalesce($assign->expr, $assign);
        }
        if ($assign->expr instanceof Throw_) {
            return new Expression($assign->expr);
        }
        if ($assign->expr instanceof Ternary) {
            return $this->refactorTernary($assign->expr, $assign);
        }
        return null;
    }
    /**
     * @return If_|Stmt[]|null
     */
    private function refactorTernary(Ternary $ternary, ?Assign $assign, bool $onArg = \false)
    {
        if ($ternary->if instanceof Throw_) {
            $else = $ternary->if;
        } elseif ($ternary->else instanceof Throw_) {
            $else = $ternary->else;
        } else {
            return null;
        }
        $inversedTernaryExpr = $ternary->if instanceof Throw_ ? $ternary->cond : $this->binaryOpManipulator->inverseNode($ternary->cond);
        $if = new If_($inversedTernaryExpr, ['stmts' => [new Expression($else)]]);
        if (!$assign instanceof Assign && $onArg) {
            $tempVar = $this->namedVariableFactory->createVariable('arg', $ternary);
            $assign = new Assign($tempVar, $ternary->if ?? $ternary->cond);
            $inversedTernaryExpr = $this->binaryOpManipulator->inverseNode($tempVar);
            $if = new If_($inversedTernaryExpr, ['stmts' => [new Expression($else)]]);
            return [new Expression($assign), $if];
        }
        if (!$assign instanceof Assign) {
            return $if;
        }
        $assign->expr = $ternary->if ?? $ternary->cond;
        return [$if, new Expression($assign)];
    }
    /**
     * @return If_|Stmt[]|null
     */
    private function refactorCoalesce(Coalesce $coalesce, ?Assign $assign)
    {
        if (!$coalesce->right instanceof Throw_) {
            $rightCoalesce = $coalesce->right;
            $leftCoalesce = $coalesce->left;
            while ($rightCoalesce instanceof Coalesce) {
                $leftCoalesce = new Coalesce($leftCoalesce, $rightCoalesce->left);
                $rightCoalesce = $rightCoalesce->right;
            }
            if ($rightCoalesce instanceof Throw_) {
                $coalesce = new Coalesce($leftCoalesce, $rightCoalesce);
                return $this->processCoalesce($coalesce, $assign, \true);
            }
            return null;
        }
        return $this->processCoalesce($coalesce, $assign);
    }
    /**
     * @return If_|Stmt[]|null
     */
    private function processCoalesce(Coalesce $coalesce, ?Assign $assign, bool $assignEarly = \false)
    {
        if (!$this->coalesceAnalyzer->hasIssetableLeft($coalesce)) {
            return null;
        }
        $condExpr = $this->createCondExpr($coalesce);
        $if = new If_($condExpr, ['stmts' => [new Expression($coalesce->right)]]);
        if (!$assign instanceof Assign) {
            return $if;
        }
        $assign->expr = $coalesce->left;
        if ($assignEarly && $if->cond instanceof Identical) {
            $expression = new Expression(new Assign($assign->var, $if->cond->left));
            $if->cond->left = $assign->var;
            return [$expression, $if];
        }
        return [$if, new Expression($assign)];
    }
    private function hasThrowInAssignExpr(Assign $assign): bool
    {
        return (bool) $this->betterNodeFinder->findFirst($assign->expr, static fn(Node $node): bool => $node instanceof Throw_);
    }
    /**
     * @return Node[]|null
     */
    private function refactorReturn(Return_ $return): ?array
    {
        $throw = $this->betterNodeFinder->findFirstInstanceOf($return, Throw_::class);
        if (!$throw instanceof Throw_) {
            return null;
        }
        if ($return->expr instanceof Coalesce) {
            $coalesce = $return->expr;
            if (!$coalesce->right instanceof Throw_) {
                return null;
            }
            $if = $this->createIf($coalesce, $coalesce->right);
            return [$if, new Return_($coalesce->left)];
        }
        if ($return->expr instanceof Throw_) {
            return [new Expression($return->expr)];
        }
        if ($return->expr instanceof Ternary) {
            $if = $this->refactorTernary($return->expr, null);
            if (!$if instanceof If_) {
                return null;
            }
            $returnExpr = $return->expr->if instanceof Throw_ ? $return->expr->else : $return->expr->if ?? $return->expr->cond;
            return [$if, new Return_($returnExpr)];
        }
        return null;
    }
    private function createIf(Coalesce $coalesce, Throw_ $throw): If_
    {
        $conditionalExpr = $coalesce->left;
        if ($conditionalExpr instanceof Variable || $conditionalExpr instanceof ArrayDimFetch || $conditionalExpr instanceof PropertyFetch) {
            $booleanNot = new BooleanNot(new Isset_([$conditionalExpr]));
        } else {
            $booleanNot = new Identical($conditionalExpr, new ConstFetch(new Name('null')));
        }
        return new If_($booleanNot, ['stmts' => [new Expression($throw)]]);
    }
    /**
     * @return \PhpParser\Node\Expr\BooleanNot|\PhpParser\Node\Expr\BinaryOp\Identical
     */
    private function createCondExpr(Coalesce $coalesce)
    {
        if ($coalesce->left instanceof Variable || $coalesce->left instanceof ArrayDimFetch) {
            return new BooleanNot(new Isset_([$coalesce->left]));
        }
        return new Identical($coalesce->left, $this->nodeFactory->createNull());
    }
    /**
     * @return Stmt[]|null
     */
    private function refactorDirectCoalesce(Expression $expression): ?array
    {
        /** @var Coalesce[] $coalesces */
        $coalesces = $this->betterNodeFinder->findInstanceOf($expression, Coalesce::class);
        foreach ($coalesces as $coalesce) {
            if (!$coalesce->right instanceof Throw_) {
                continue;
            }
            // add condition if above
            $throwExpr = $coalesce->right;
            $if = new If_(new Identical($coalesce->left, new ConstFetch(new Name('null'))), ['stmts' => [new Expression($throwExpr)]]);
            // replace coalsese with left :)
            $this->traverseNodesWithCallable($expression, static function (Node $node): ?Expr {
                if (!$node instanceof Coalesce) {
                    return null;
                }
                return $node->left;
            });
            return [$if, $expression];
        }
        return null;
    }
}
