James Moger
2012-09-10 fabe060d3a435f116128851f828e35c2af5fde67
src/com/gitblit/wicket/pages/EditRepositoryPage.java
@@ -21,36 +21,45 @@
import java.util.Collections;
import java.util.HashSet;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.apache.wicket.PageParameters;
import org.apache.wicket.behavior.SimpleAttributeModifier;
import org.apache.wicket.extensions.markup.html.form.palette.Palette;
import org.apache.wicket.markup.html.WebMarkupContainer;
import org.apache.wicket.markup.html.basic.Label;
import org.apache.wicket.markup.html.form.Button;
import org.apache.wicket.markup.html.form.CheckBox;
import org.apache.wicket.markup.html.form.ChoiceRenderer;
import org.apache.wicket.markup.html.form.DropDownChoice;
import org.apache.wicket.markup.html.form.Form;
import org.apache.wicket.markup.html.form.IChoiceRenderer;
import org.apache.wicket.markup.html.form.Radio;
import org.apache.wicket.markup.html.form.RadioGroup;
import org.apache.wicket.markup.html.form.TextField;
import org.apache.wicket.markup.html.list.ListItem;
import org.apache.wicket.markup.html.list.ListView;
import org.apache.wicket.model.CompoundPropertyModel;
import org.apache.wicket.model.IModel;
import org.apache.wicket.model.Model;
import org.apache.wicket.model.util.CollectionModel;
import org.apache.wicket.model.util.ListModel;
import com.gitblit.Constants;
import com.gitblit.Constants.AccessRestrictionType;
import com.gitblit.Constants.AuthorizationControl;
import com.gitblit.Constants.FederationStrategy;
import com.gitblit.GitBlit;
import com.gitblit.GitBlitException;
import com.gitblit.Keys;
import com.gitblit.models.RefModel;
import com.gitblit.models.RepositoryModel;
import com.gitblit.models.UserModel;
import com.gitblit.utils.ArrayUtils;
import com.gitblit.utils.StringUtils;
import com.gitblit.wicket.GitBlitWebSession;
import com.gitblit.wicket.StringChoiceRenderer;
import com.gitblit.wicket.WicketUtils;
import com.gitblit.wicket.panels.BulletListPanel;
@@ -66,7 +75,12 @@
      // create constructor
      super();
      isCreate = true;
      setupPage(new RepositoryModel());
      RepositoryModel model = new RepositoryModel();
      String restriction = GitBlit.getString(Keys.git.defaultAccessRestriction, null);
      model.accessRestriction = AccessRestrictionType.fromName(restriction);
      String authorization = GitBlit.getString(Keys.git.defaultAuthorizationControl, null);
      model.authorizationControl = AuthorizationControl.fromName(authorization);
      setupPage(model);
   }
   public EditRepositoryPage(PageParameters params) {
@@ -82,6 +96,7 @@
      // ensure this user can create or edit this repository
      checkPermissions(repositoryModel);
      List<String> indexedBranches = new ArrayList<String>();
      List<String> federationSets = new ArrayList<String>();
      List<String> repositoryUsers = new ArrayList<String>();
      List<String> repositoryTeams = new ArrayList<String>();
@@ -98,24 +113,37 @@
            Collections.sort(repositoryUsers);
         }
         federationSets.addAll(repositoryModel.federationSets);
         if (!ArrayUtils.isEmpty(repositoryModel.indexedBranches)) {
            indexedBranches.addAll(repositoryModel.indexedBranches);
         }
      }
      final String oldName = repositoryModel.name;
      // users palette
      final Palette<String> usersPalette = new Palette<String>("users", new ListModel<String>(
            repositoryUsers), new CollectionModel<String>(GitBlit.self().getAllUsernames()),
            new ChoiceRenderer<String>("", ""), 10, false);
            new StringChoiceRenderer(), 10, false);
      // teams palette
      final Palette<String> teamsPalette = new Palette<String>("teams", new ListModel<String>(
            repositoryTeams), new CollectionModel<String>(GitBlit.self().getAllTeamnames()),
            new ChoiceRenderer<String>("", ""), 5, false);
            new StringChoiceRenderer(), 8, false);
      // indexed local branches palette
      List<String> allLocalBranches = new ArrayList<String>();
      allLocalBranches.add(Constants.DEFAULT_BRANCH);
      allLocalBranches.addAll(repositoryModel.getLocalBranches());
      boolean luceneEnabled = GitBlit.getBoolean(Keys.web.allowLuceneIndexing, true);
      final Palette<String> indexedBranchesPalette = new Palette<String>("indexedBranches", new ListModel<String>(
            indexedBranches), new CollectionModel<String>(allLocalBranches),
            new StringChoiceRenderer(), 8, false);
      indexedBranchesPalette.setEnabled(luceneEnabled);
      // federation sets palette
      List<String> sets = GitBlit.getStrings(Keys.federation.sets);
      final Palette<String> federationSetsPalette = new Palette<String>("federationSets",
            new ListModel<String>(federationSets), new CollectionModel<String>(sets),
            new ChoiceRenderer<String>("", ""), 5, false);
            new StringChoiceRenderer(), 8, false);
      // pre-receive palette
      if (!ArrayUtils.isEmpty(repositoryModel.preReceiveScripts)) {
@@ -124,7 +152,7 @@
      final Palette<String> preReceivePalette = new Palette<String>("preReceiveScripts",
            new ListModel<String>(preReceiveScripts), new CollectionModel<String>(GitBlit
                  .self().getPreReceiveScriptsUnused(repositoryModel)),
            new ChoiceRenderer<String>("", ""), 12, true);
            new StringChoiceRenderer(), 12, true);
      // post-receive palette
      if (!ArrayUtils.isEmpty(repositoryModel.postReceiveScripts)) {
@@ -133,7 +161,29 @@
      final Palette<String> postReceivePalette = new Palette<String>("postReceiveScripts",
            new ListModel<String>(postReceiveScripts), new CollectionModel<String>(GitBlit
                  .self().getPostReceiveScriptsUnused(repositoryModel)),
            new ChoiceRenderer<String>("", ""), 12, true);
            new StringChoiceRenderer(), 12, true);
      // custom fields
      final Map<String, String> customFieldsMap = GitBlit.getMap(Keys.groovy.customFields);
      List<String> customKeys = new ArrayList<String>(customFieldsMap.keySet());
      final ListView<String> customFieldsListView = new ListView<String>("customFieldsListView", customKeys) {
         private static final long serialVersionUID = 1L;
         @Override
         protected void populateItem(ListItem<String> item) {
            String key = item.getModelObject();
            item.add(new Label("customFieldLabel", customFieldsMap.get(key)));
            String value = "";
            if (repositoryModel.customFields != null && repositoryModel.customFields.containsKey(key)) {
               value = repositoryModel.customFields.get(key);
            }
            TextField<String> field = new TextField<String>("customFieldValue", new Model<String>(value));
            item.add(field);
         }
      };
      customFieldsListView.setReuseItems(true);
      CompoundPropertyModel<RepositoryModel> model = new CompoundPropertyModel<RepositoryModel>(
            repositoryModel);
@@ -146,7 +196,7 @@
            try {
               // confirm a repository name was entered
               if (StringUtils.isEmpty(repositoryModel.name)) {
                  error("Please set repository name!");
                  error(getString("gb.pleaseSetRepositoryName"));
                  return;
               }
@@ -157,35 +207,38 @@
               // prohibit folder paths
               if (repositoryModel.name.startsWith("/")) {
                  error("Leading root folder references (/) are prohibited.");
                  error(getString("gb.illegalLeadingSlash"));
                  return;
               }
               if (repositoryModel.name.startsWith("../")) {
                  error("Relative folder references (../) are prohibited.");
                  error(getString("gb.illegalRelativeSlash"));
                  return;
               }
               if (repositoryModel.name.contains("/../")) {
                  error("Relative folder references (../) are prohibited.");
                  error(getString("gb.illegalRelativeSlash"));
                  return;
               }
               if (repositoryModel.name.endsWith("/")) {
                  repositoryModel.name = repositoryModel.name.substring(0, repositoryModel.name.length() - 1);
               }
               // confirm valid characters in repository name
               Character c = StringUtils.findInvalidCharacter(repositoryModel.name);
               if (c != null) {
                  error(MessageFormat.format("Illegal character ''{0}'' in repository name!",
                  error(MessageFormat.format(getString("gb.illegalCharacterRepositoryName"),
                        c));
                  return;
               }
               // confirm access restriction selection
               if (repositoryModel.accessRestriction == null) {
                  error("Please select access restriction!");
                  error(getString("gb.selectAccessRestriction"));
                  return;
               }
               // confirm federation strategy selection
               if (repositoryModel.federationStrategy == null) {
                  error("Please select federation strategy!");
                  error(getString("gb.selectFederationStrategy"));
                  return;
               }
@@ -211,6 +264,14 @@
                  repositoryModel.mailingLists = new ArrayList<String>(list);
               }
               // indexed branches
               List<String> indexedBranches = new ArrayList<String>();
               Iterator<String> branches = indexedBranchesPalette.getSelectedChoices();
               while (branches.hasNext()) {
                  indexedBranches.add(branches.next());
               }
               repositoryModel.indexedBranches = indexedBranches;
               // pre-receive scripts
               List<String> preReceiveScripts = new ArrayList<String>();
               Iterator<String> pres = preReceivePalette.getSelectedChoices();
@@ -226,7 +287,19 @@
                  postReceiveScripts.add(post.next());
               }
               repositoryModel.postReceiveScripts = postReceiveScripts;
               // custom fields
               repositoryModel.customFields = new LinkedHashMap<String, String>();
               for (int i = 0; i < customFieldsListView.size(); i++) {
                  ListItem<String> child = (ListItem<String>) customFieldsListView.get(i);
                  String key = child.getModelObject();
                  TextField<String> field = (TextField<String>) child.get("customFieldValue");
                  String value = field.getValue();
                  repositoryModel.customFields.put(key, value);
               }
               // save the repository
               GitBlit.self().updateRepositoryModel(oldName, repositoryModel, isCreate);
@@ -262,6 +335,9 @@
         }
      };
      // do not let the browser pre-populate these fields
      form.add(new SimpleAttributeModifier("autocomplete", "off"));
      // field names reflective match RepositoryModel fields
      form.add(new TextField<String>("name").setEnabled(isCreate || isAdmin));
      form.add(new TextField<String>("description"));
@@ -272,9 +348,13 @@
      form.add(new CheckBox("isFrozen"));
      // TODO enable origin definition
      form.add(new TextField<String>("origin").setEnabled(false/* isCreate */));
      // enable alteration of the default branch after clone
      form.add(new DropDownChoice<RefModel>("defaultHead", repositoryModel.availableHeads,
            new RefModelRenderer()).setEnabled(GitBlitWebSession.get().canAdmin()));
      // allow relinking HEAD to a branch or tag other than master on edit repository
      List<String> availableRefs = new ArrayList<String>();
      if (!ArrayUtils.isEmpty(repositoryModel.availableRefs)) {
         availableRefs.addAll(repositoryModel.availableRefs);
      }
      form.add(new DropDownChoice<String>("HEAD", availableRefs).setEnabled(availableRefs.size() > 0));
      // federation strategies - remove ORIGIN choice if this repository has
      // no origin.
@@ -286,7 +366,7 @@
      form.add(new DropDownChoice<FederationStrategy>("federationStrategy", federationStrategies,
            new FederationTypeRenderer()));
      form.add(new CheckBox("useTickets"));
      form.add(new CheckBox("useDocs"));
      form.add(new CheckBox("useDocs"));
      form.add(new CheckBox("showRemoteBranches"));
      form.add(new CheckBox("showReadme"));
      form.add(new CheckBox("skipSizeCalculation"));
@@ -294,15 +374,28 @@
      mailingLists = new Model<String>(ArrayUtils.isEmpty(repositoryModel.mailingLists) ? ""
            : StringUtils.flattenStrings(repositoryModel.mailingLists, " "));
      form.add(new TextField<String>("mailingLists", mailingLists));
      form.add(indexedBranchesPalette);
      RadioGroup<AuthorizationControl> group = new RadioGroup<AuthorizationControl>("authorizationControl");
      Radio<AuthorizationControl> allowAuthenticated = new Radio<AuthorizationControl>("allowAuthenticated", new Model<AuthorizationControl>(AuthorizationControl.AUTHENTICATED));
      Radio<AuthorizationControl> allowNamed = new Radio<AuthorizationControl>("allowNamed", new Model<AuthorizationControl>(AuthorizationControl.NAMED));
      group.add(allowAuthenticated);
      group.add(allowNamed);
      form.add(group);
      form.add(usersPalette);
      form.add(teamsPalette);
      form.add(federationSetsPalette);
      form.add(preReceivePalette);
      form.add(new BulletListPanel("inheritedPreReceive", "inherited", GitBlit.self()
      form.add(new BulletListPanel("inheritedPreReceive", getString("gb.inherited"), GitBlit.self()
            .getPreReceiveScriptsInherited(repositoryModel)));
      form.add(postReceivePalette);
      form.add(new BulletListPanel("inheritedPostReceive", "inherited", GitBlit.self()
      form.add(new BulletListPanel("inheritedPostReceive", getString("gb.inherited"), GitBlit.self()
            .getPostReceiveScriptsInherited(repositoryModel)));
      WebMarkupContainer customFieldsSection = new WebMarkupContainer("customFieldsSection");
      customFieldsSection.add(customFieldsListView);
      form.add(customFieldsSection.setVisible(!GitBlit.getString(Keys.groovy.customFields, "").isEmpty()));
      form.add(new Button("save"));
      Button cancel = new Button("cancel") {
@@ -337,13 +430,13 @@
         if (authenticateAdmin) {
            if (user == null) {
               // No Login Available
               error("Administration requires a login", true);
               error(getString("gb.errorAdminLoginRequired"), true);
            }
            if (isCreate) {
               // Create Repository
               if (!user.canAdmin) {
                  // Only Administrators May Create
                  error("Only an administrator may create a repository", true);
                  error(getString("gb.errorOnlyAdminMayCreateRepository"), true);
               }
            } else {
               // Edit Repository
@@ -354,29 +447,14 @@
               } else {
                  if (!model.owner.equalsIgnoreCase(user.username)) {
                     // User is not an Admin nor Owner
                     error("Only an administrator or the owner may edit a repository", true);
                     error(getString("gb.errorOnlyAdminOrOwnerMayEditRepository"), true);
                  }
               }
            }
         }
      } else {
         // No Administration Permitted
         error("Administration is disabled", true);
      }
   }
   private class RefModelRenderer implements IChoiceRenderer<RefModel> {
      private static final long serialVersionUID = 1L;
      @Override
      public String getDisplayValue(RefModel type) {
         return type.displayName;
      }
      @Override
      public String getIdValue(RefModel type, int index) {
         return type.getName();
         error(getString("gb.errorAdministrationDisabled"), true);
      }
   }