//    @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);
    }
  }