mirror of
https://gitflic.ru/project/openide/openide.git
synced 2026-04-19 04:51:24 +07:00
94ae100455
Otherwise, some typeshed directory could be considered as a package and name would be resolved to it even if the corresponding python package is not installed (PyTypeShed.maySearchForStubInRoot is ignored in this case). GitOrigin-RevId: c453b28655b710f092c9ece3d868a0ee5141c0dd
51 lines
1.9 KiB
Java
51 lines
1.9 KiB
Java
// Copyright 2000-2021 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
|
|
package com.jetbrains.python.psi.impl;
|
|
|
|
import com.intellij.openapi.module.Module;
|
|
import com.intellij.openapi.module.ModuleUtilCore;
|
|
import com.intellij.openapi.projectRoots.JavaSdkType;
|
|
import com.intellij.openapi.projectRoots.Sdk;
|
|
import com.intellij.openapi.roots.ModuleRootManager;
|
|
import com.intellij.psi.JavaPsiFacade;
|
|
import com.intellij.psi.PsiClass;
|
|
import com.intellij.psi.PsiElement;
|
|
import com.intellij.psi.PsiPackage;
|
|
import com.intellij.psi.util.QualifiedName;
|
|
import com.jetbrains.python.psi.resolve.PyQualifiedNameResolveContext;
|
|
import org.jetbrains.annotations.NotNull;
|
|
import org.jetbrains.annotations.Nullable;
|
|
|
|
|
|
public class PyJavaImportResolver implements PyImportResolver {
|
|
@Override
|
|
@Nullable
|
|
public PsiElement resolveImportReference(@NotNull QualifiedName name, @NotNull PyQualifiedNameResolveContext context, boolean withRoots) {
|
|
String fqn = name.toString();
|
|
final JavaPsiFacade psiFacade = JavaPsiFacade.getInstance(context.getProject());
|
|
final PsiPackage aPackage = psiFacade.findPackage(fqn);
|
|
if (aPackage != null && isUnderJvmModule(aPackage)) {
|
|
return aPackage;
|
|
}
|
|
|
|
Module module = context.getModule();
|
|
if (module != null) {
|
|
final PsiClass aClass = psiFacade.findClass(fqn, module.getModuleWithDependenciesAndLibrariesScope(false));
|
|
if (aClass != null) return aClass;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
private static boolean isUnderJvmModule(@NotNull PsiPackage psiPackage) {
|
|
final Module module = ModuleUtilCore.findModuleForPsiElement(psiPackage);
|
|
|
|
if (module != null) {
|
|
final Sdk sdk = ModuleRootManager.getInstance(module).getSdk();
|
|
if (sdk != null && sdk.getSdkType() instanceof JavaSdkType) {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|