void (@NotNull Map<String, Object> context, Project project, Module module, @NotNull final PsiFile file) { if (file instanceof PsiClassOwner) { final FileInfo info = new FileInfo(file) { @Override public String getClassName() { if (file instanceof PsiJavaFile) { return ((PsiJavaFile)file).getClasses()[0].getName(); } else { return super.getClassName(); } } @Override public String getQualifiedClassName() { if (file instanceof PsiJavaFile) { return ((PsiJavaFile)file).getClasses()[0].getQualifiedName(); } else { return super.getQualifiedClassName(); } } }; context.put("file", info); } }