private static PsiFile findFileToHighlight(@NotNull Project project,
                                             @Nullable VirtualFile virtualFile) {
    if (virtualFile == null || !virtualFile.isValid()) {
      return null;
    }

    FileManagerImpl fileManagerImpl = (FileManagerImpl)PsiManagerEx.getInstanceEx(project).getFileManager();
    PsiFile psiFile = fileManagerImpl.getFastCachedPsiFile(virtualFile);

    if (psiFile instanceof PsiCompiledFile compiled) {
      return compiled.getDecompiledPsiFile();
    }

    return psiFile;
  }
private static PsiFile findFileToHighlight(@NotNull Project project,
                                             @Nullable VirtualFile virtualFile)