1
0
mirror of https://github.com/danog/psalm.git synced 2024-12-13 17:57:37 +01:00
psalm/examples/plugins/ClassUnqualifier.php
2022-12-18 13:20:31 -06:00

60 lines
2.0 KiB
PHP

<?php
namespace Psalm\Example\Plugin;
use Psalm\FileManipulation;
use Psalm\Internal\Type\TypeTokenizer;
use Psalm\Plugin\EventHandler\AfterClassLikeExistenceCheckInterface;
use Psalm\Plugin\EventHandler\Event\AfterClassLikeExistenceCheckEvent;
use function array_map;
use function implode;
use function strpos;
use function strtolower;
class ClassUnqualifier implements AfterClassLikeExistenceCheckInterface
{
public static function afterClassLikeExistenceCheck(
AfterClassLikeExistenceCheckEvent $event
): void {
$fq_class_name = $event->getFqClassName();
$code_location = $event->getCodeLocation();
$statements_source = $event->getStatementsSource();
$file_replacements = $event->getFileReplacements();
$candidate_type = $code_location->getSelectedText();
$aliases = $statements_source->getAliasedClassesFlipped();
if ($statements_source->getFilePath() !== $code_location->file_path) {
return;
}
if (strpos($candidate_type, '\\' . $fq_class_name) !== false) {
$type_tokens = TypeTokenizer::tokenize($candidate_type, false);
foreach ($type_tokens as &$type_token) {
if ($type_token[0] === ('\\' . $fq_class_name)
&& isset($aliases[strtolower($fq_class_name)])
) {
$type_token[0] = $aliases[strtolower($fq_class_name)];
}
}
unset($type_token);
$new_candidate_type = implode(
'',
array_map(
fn($f) => $f[0],
$type_tokens,
),
);
if ($new_candidate_type !== $candidate_type) {
$bounds = $code_location->getSelectionBounds();
$file_replacements[] = new FileManipulation($bounds[0], $bounds[1], $new_candidate_type);
}
$event->setFileReplacements($file_replacements);
}
}
}