org.jetbrains.kotlin.resolve.lazy.declarations.FileBasedDeclarationProviderFactory Maven / Gradle / Ivy
/*
* Copyright 2010-2015 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jetbrains.kotlin.resolve.lazy.declarations;
import com.google.common.collect.LinkedHashMultimap;
import com.google.common.collect.Multimap;
import com.google.common.collect.Sets;
import kotlin.collections.CollectionsKt;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.jetbrains.kotlin.name.FqName;
import org.jetbrains.kotlin.psi.KtFile;
import org.jetbrains.kotlin.resolve.lazy.data.KtClassLikeInfo;
import org.jetbrains.kotlin.storage.NotNullLazyValue;
import org.jetbrains.kotlin.storage.StorageManager;
import java.util.Collection;
import java.util.Set;
public class FileBasedDeclarationProviderFactory extends AbstractDeclarationProviderFactory {
private static class Index {
private final Multimap filesByPackage = LinkedHashMultimap.create();
private final Set declaredPackages = Sets.newHashSet();
}
private final StorageManager storageManager;
private final NotNullLazyValue index;
public FileBasedDeclarationProviderFactory(@NotNull StorageManager storageManager, @NotNull Collection files) {
super(storageManager);
this.storageManager = storageManager;
this.index = storageManager.createLazyValue(() -> computeFilesByPackage(files));
}
@NotNull
private static Index computeFilesByPackage(@NotNull Collection files) {
Index index = new Index();
for (KtFile file : files) {
FqName packageFqName = file.getPackageFqName();
addMeAndParentPackages(index, packageFqName);
index.filesByPackage.put(packageFqName, file);
}
return index;
}
private static void addMeAndParentPackages(@NotNull Index index, @NotNull FqName name) {
index.declaredPackages.add(name);
if (!name.isRoot()) {
addMeAndParentPackages(index, name.parent());
}
}
private boolean isPackageDeclaredExplicitly(@NotNull FqName packageFqName) {
return index.invoke().declaredPackages.contains(packageFqName);
}
/*package*/ Collection getAllDeclaredSubPackagesOf(@NotNull FqName parent) {
return CollectionsKt.filter(index.invoke().declaredPackages, fqName -> !fqName.isRoot() && fqName.parent().equals(parent));
}
@Nullable
@Override
protected PackageMemberDeclarationProvider createPackageMemberDeclarationProvider(@NotNull FqName packageFqName) {
if (isPackageDeclaredExplicitly(packageFqName)) {
return new FileBasedPackageMemberDeclarationProvider(
storageManager, packageFqName, this, index.invoke().filesByPackage.get(packageFqName));
}
return null;
}
@NotNull
@Override
public ClassMemberDeclarationProvider getClassMemberDeclarationProvider(@NotNull KtClassLikeInfo classLikeInfo) {
if (!index.invoke().filesByPackage.containsKey(classLikeInfo.getContainingPackageFqName())) {
throw new IllegalStateException("This factory doesn't know about this class: " + classLikeInfo);
}
return new PsiBasedClassMemberDeclarationProvider(storageManager, classLikeInfo);
}
}