mirror of https://github.com/rectorphp/rector.git
157 lines
6.4 KiB
PHP
157 lines
6.4 KiB
PHP
<?php
|
|
|
|
declare (strict_types=1);
|
|
namespace Rector\MysqlToMysqli\Rector\Assign;
|
|
|
|
use PhpParser\Node;
|
|
use PhpParser\Node\Arg;
|
|
use PhpParser\Node\Expr\ArrayDimFetch;
|
|
use PhpParser\Node\Expr\Assign;
|
|
use PhpParser\Node\Expr\FuncCall;
|
|
use PhpParser\Node\Expr\PropertyFetch;
|
|
use PhpParser\Node\Expr\StaticPropertyFetch;
|
|
use PhpParser\Node\Expr\Variable;
|
|
use PhpParser\Node\Name;
|
|
use PhpParser\Node\Scalar\LNumber;
|
|
use Rector\Core\Rector\AbstractRector;
|
|
use Rector\NodeTypeResolver\Node\AttributeKey;
|
|
use Rector\PostRector\Collector\NodesToAddCollector;
|
|
use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
|
|
use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
|
|
/**
|
|
* @changelog https://www.phpclasses.org/blog/package/9199/post/3-Smoothly-Migrate-your-PHP-Code-using-the-Old-MySQL-extension-to-MySQLi.html
|
|
* @see \Rector\Tests\MysqlToMysqli\Rector\Assign\MysqlAssignToMysqliRector\MysqlAssignToMysqliRectorTest
|
|
*/
|
|
final class MysqlAssignToMysqliRector extends AbstractRector
|
|
{
|
|
/**
|
|
* @var array<string, string>
|
|
*/
|
|
private const FIELD_TO_FIELD_DIRECT = ['mysql_field_len' => 'length', 'mysql_field_name' => 'name', 'mysql_field_table' => 'table'];
|
|
/**
|
|
* @var string
|
|
*/
|
|
private const MYSQLI_DATA_SEEK = 'mysqli_data_seek';
|
|
/**
|
|
* @readonly
|
|
* @var \Rector\PostRector\Collector\NodesToAddCollector
|
|
*/
|
|
private $nodesToAddCollector;
|
|
public function __construct(NodesToAddCollector $nodesToAddCollector)
|
|
{
|
|
$this->nodesToAddCollector = $nodesToAddCollector;
|
|
}
|
|
public function getRuleDefinition() : RuleDefinition
|
|
{
|
|
return new RuleDefinition('Converts more complex mysql functions to mysqli', [new CodeSample(<<<'CODE_SAMPLE'
|
|
$data = mysql_db_name($result, $row);
|
|
CODE_SAMPLE
|
|
, <<<'CODE_SAMPLE'
|
|
mysqli_data_seek($result, $row);
|
|
$fetch = mysql_fetch_row($result);
|
|
$data = $fetch[0];
|
|
CODE_SAMPLE
|
|
)]);
|
|
}
|
|
/**
|
|
* @return array<class-string<Node>>
|
|
*/
|
|
public function getNodeTypes() : array
|
|
{
|
|
return [Assign::class];
|
|
}
|
|
/**
|
|
* @param Assign $node
|
|
*/
|
|
public function refactor(Node $node) : ?Node
|
|
{
|
|
if (!$node->expr instanceof FuncCall) {
|
|
return null;
|
|
}
|
|
/** @var FuncCall $funcCallNode */
|
|
$funcCallNode = $node->expr;
|
|
if ($this->isName($funcCallNode, 'mysql_tablename')) {
|
|
return $this->processMysqlTableName($node, $funcCallNode);
|
|
}
|
|
if ($this->isName($funcCallNode, 'mysql_db_name')) {
|
|
return $this->processMysqlDbName($node, $funcCallNode);
|
|
}
|
|
if ($this->isName($funcCallNode, 'mysql_db_query')) {
|
|
return $this->processMysqliSelectDb($node, $funcCallNode);
|
|
}
|
|
if ($this->isName($funcCallNode, 'mysql_fetch_field')) {
|
|
return $this->processMysqlFetchField($node, $funcCallNode);
|
|
}
|
|
if ($this->isName($funcCallNode, 'mysql_result')) {
|
|
return $this->processMysqlResult($node, $funcCallNode);
|
|
}
|
|
return $this->processFieldToFieldDirect($node, $funcCallNode);
|
|
}
|
|
private function processMysqlTableName(Assign $assign, FuncCall $funcCall) : FuncCall
|
|
{
|
|
$funcCall->name = new Name(self::MYSQLI_DATA_SEEK);
|
|
$newFuncCall = new FuncCall(new Name('mysql_fetch_array'), [$funcCall->args[0]]);
|
|
$newAssignNode = new Assign($assign->var, new ArrayDimFetch($newFuncCall, new LNumber(0)));
|
|
$this->nodesToAddCollector->addNodeAfterNode($newAssignNode, $assign);
|
|
return $funcCall;
|
|
}
|
|
private function processMysqlDbName(Assign $assign, FuncCall $funcCall) : FuncCall
|
|
{
|
|
$funcCall->name = new Name(self::MYSQLI_DATA_SEEK);
|
|
$mysqlFetchRowFuncCall = new FuncCall(new Name('mysqli_fetch_row'), [$funcCall->args[0]]);
|
|
$fetchVariable = new Variable('fetch');
|
|
$newAssignNode = new Assign($fetchVariable, $mysqlFetchRowFuncCall);
|
|
$this->nodesToAddCollector->addNodeAfterNode($newAssignNode, $assign);
|
|
$newAssignNodeAfter = new Assign($assign->var, new ArrayDimFetch($fetchVariable, new LNumber(0)));
|
|
$this->nodesToAddCollector->addNodeAfterNode($newAssignNodeAfter, $assign);
|
|
return $funcCall;
|
|
}
|
|
private function processMysqliSelectDb(Assign $assign, FuncCall $funcCall) : FuncCall
|
|
{
|
|
$funcCall->name = new Name('mysqli_select_db');
|
|
$newAssignNode = new Assign($assign->var, new FuncCall(new Name('mysqli_query'), [$funcCall->args[1]]));
|
|
$this->nodesToAddCollector->addNodeAfterNode($newAssignNode, $assign);
|
|
unset($funcCall->args[1]);
|
|
return $funcCall;
|
|
}
|
|
private function processMysqlFetchField(Assign $assign, FuncCall $funcCall) : Assign
|
|
{
|
|
$funcCall->name = isset($funcCall->args[1]) ? new Name('mysqli_fetch_field_direct') : new Name('mysqli_fetch_field');
|
|
return $assign;
|
|
}
|
|
private function processMysqlResult(Assign $assign, FuncCall $funcCall) : FuncCall
|
|
{
|
|
$fetchField = null;
|
|
if (isset($funcCall->args[2]) && $funcCall->args[2] instanceof Arg) {
|
|
$fetchField = $funcCall->args[2]->value;
|
|
unset($funcCall->args[2]);
|
|
}
|
|
$funcCall->name = new Name(self::MYSQLI_DATA_SEEK);
|
|
$mysqlFetchArrayFuncCall = new FuncCall(new Name('mysqli_fetch_array'), [$funcCall->args[0]]);
|
|
$fetchVariable = new Variable('fetch');
|
|
$newAssignNode = new Assign($fetchVariable, $mysqlFetchArrayFuncCall);
|
|
$this->nodesToAddCollector->addNodeAfterNode($newAssignNode, $assign);
|
|
$newAssignNodeAfter = new Assign($assign->var, new ArrayDimFetch($fetchVariable, $fetchField ?? new LNumber(0)));
|
|
$this->nodesToAddCollector->addNodeAfterNode($newAssignNodeAfter, $assign);
|
|
return $funcCall;
|
|
}
|
|
private function processFieldToFieldDirect(Assign $assign, FuncCall $funcCall) : ?Assign
|
|
{
|
|
foreach (self::FIELD_TO_FIELD_DIRECT as $funcName => $property) {
|
|
if ($this->isName($funcCall, $funcName)) {
|
|
$parentNode = $funcCall->getAttribute(AttributeKey::PARENT_NODE);
|
|
if ($parentNode instanceof PropertyFetch) {
|
|
continue;
|
|
}
|
|
if ($parentNode instanceof StaticPropertyFetch) {
|
|
continue;
|
|
}
|
|
$funcCall->name = new Name('mysqli_fetch_field_direct');
|
|
$assign->expr = new PropertyFetch($funcCall, $property);
|
|
return $assign;
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
}
|