// @Override protected void buildDialog( Project project, PsiDirectory directory, CreateFileFromTemplateDialog.Builder builder) { PsiFile childs[] = directory.getFiles(); Set<String> packages = new HashSet<String>(); for (PsiFile child : childs) { if (child instanceof GoFile) { GoFile goFile = (GoFile) child; if (!goFile.getPackage().isMainPackage()) { packages.add(goFile.getPackage().getPackageName()); } } } builder.addKind("New file", GoIcons.GO_ICON_16x16, "single"); for (String packageName : packages) { builder.addKind( "New file in library: " + packageName, GoIcons.GO_ICON_16x16, "lib." + packageName); } }