@Nullable
  @Override
  public VirtualFile getVirtualFile(@NotNull Object o) {
    if (o instanceof PsiElement element) {
      final PsiFile file = element.getContainingFile();
      return file != null ? file.getVirtualFile() : null;
    }
    return null;
  }
@Nullable
  @Override
  public VirtualFile getVirtualFile(@NotNull Object o)