psalm-plugin-phpunit/Plugin.php

<?php

namespace Psalm\PhpUnitPlugin;

use Composer\Semver\Comparator;
use Composer\Semver\VersionParser;
use PackageVersions\Versions;
use SimpleXMLElement;
use Psalm\Plugin\PluginEntryPointInterface;
use Psalm\Plugin\RegistrationInterface;

/** @psalm-suppress UnusedClass */
class Plugin implements PluginEntryPointInterface
{
    /** @return void */
    public function __invoke(RegistrationInterface $psalm, SimpleXMLElement $config = null)
    {
        $psalm->addStubFile(__DIR__ . '/stubs/Assert.php');
        if ($this->packageVersionIs('phpunit/phpunit', '>=', '7.5')) {
            $psalm->addStubFile(__DIR__ . '/stubs/Assert_75.php');
        }
        $psalm->addStubFile(__DIR__ . '/stubs/TestCase.php');
        $psalm->addStubFile(__DIR__ . '/stubs/MockBuilder.php');
        $psalm->addStubFile(__DIR__ . '/stubs/InvocationMocker.php');
        $psalm->addStubFile(__DIR__ . '/stubs/Prophecy.php');

        class_exists(Hooks\TestCaseHandler::class, true);
        $psalm->registerHooksFromClass(Hooks\TestCaseHandler::class);
    }

    private function packageVersionIs(string $package, string $op, string $ref): bool
    {
        $currentVersion = (string) explode('@', Versions::getVersion($package))[0];

        $parser = new VersionParser();

        $currentVersion = $parser->normalize($currentVersion);
        $ref = $parser->normalize($ref);

        return Comparator::compare($currentVersion, $op, $ref);
    }
}