php-language-server/src/Index/ProjectIndex.php

<?php
declare(strict_types = 1);

namespace LanguageServer\Index;

/**
 * A project index manages the source and dependency indexes
 */
class ProjectIndex extends AbstractAggregateIndex
{
    /**
     * The index for dependencies
     *
     * @var DependenciesIndex
     */
    private $dependenciesIndex;

    /**
     * The Index for the project source
     *
     * @var Index
     */
    private $sourceIndex;

    public function __construct(Index $sourceIndex, DependenciesIndex $dependenciesIndex, \stdClass $composerJson = null)
    {
        $this->sourceIndex = $sourceIndex;
        $this->dependenciesIndex = $dependenciesIndex;
        $this->composerJson = $composerJson;
        parent::__construct();
    }

    /**
     * @return ReadableIndex[]
     */
    protected function getIndexes(): array
    {
        return [$this->sourceIndex, $this->dependenciesIndex];
    }

    /**
     * @param string $uri
     * @return Index
     */
    public function getIndexForUri(string $uri): Index
    {
        if (\LanguageServer\uriInVendorDir($this->composerJson, $uri, $matches)) {
            $packageName = $matches[1];
            return $this->dependenciesIndex->getDependencyIndex($packageName);
        }
        return $this->sourceIndex;
    }
}