psalm/examples/plugins/ClassUnqualifier.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(
                    static 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);
        }
    }
}