From 332e93c9e8b2ef756ac2b2ec1055262046f988fa Mon Sep 17 00:00:00 2001 From: James Moger <james.moger@gitblit.com> Date: Fri, 30 Mar 2012 20:01:03 -0400 Subject: [PATCH] Merge pull request #10 from jonnybbb/master --- src/com/gitblit/wicket/pages/RootPage.java | 220 ++++++++++++++++++++++++++++++++++++++++-------------- 1 files changed, 161 insertions(+), 59 deletions(-) diff --git a/src/com/gitblit/wicket/pages/RootPage.java b/src/com/gitblit/wicket/pages/RootPage.java index 0278470..deb2f4b 100644 --- a/src/com/gitblit/wicket/pages/RootPage.java +++ b/src/com/gitblit/wicket/pages/RootPage.java @@ -17,14 +17,21 @@ import java.text.MessageFormat; import java.util.ArrayList; +import java.util.Arrays; +import java.util.Calendar; +import java.util.Collections; +import java.util.Date; +import java.util.HashMap; +import java.util.HashSet; import java.util.LinkedHashSet; import java.util.List; +import java.util.Map; import java.util.Set; +import java.util.concurrent.atomic.AtomicInteger; import java.util.regex.Pattern; import org.apache.wicket.PageParameters; import org.apache.wicket.markup.html.form.PasswordTextField; -import org.apache.wicket.markup.html.form.StatelessForm; import org.apache.wicket.markup.html.form.TextField; import org.apache.wicket.model.IModel; import org.apache.wicket.model.Model; @@ -40,6 +47,7 @@ import com.gitblit.wicket.GitBlitWebSession; import com.gitblit.wicket.PageRegistration; import com.gitblit.wicket.PageRegistration.DropDownMenuItem; +import com.gitblit.wicket.SessionlessForm; import com.gitblit.wicket.WicketUtils; import com.gitblit.wicket.panels.NavigationPanel; @@ -70,8 +78,8 @@ boolean authenticateView = GitBlit.getBoolean(Keys.web.authenticateViewPages, false); boolean authenticateAdmin = GitBlit.getBoolean(Keys.web.authenticateAdminPages, true); boolean allowAdmin = GitBlit.getBoolean(Keys.web.allowAdministration, true); - - if (authenticateAdmin) { + + if (authenticateAdmin) { showAdmin = allowAdmin && GitBlitWebSession.get().canAdmin(); // authentication requires state and session setStatelessHint(false); @@ -90,8 +98,12 @@ // navigation links List<PageRegistration> pages = new ArrayList<PageRegistration>(); - pages.add(new PageRegistration("gb.repositories", RepositoriesPage.class)); - pages.add(new PageRegistration("gb.activity", ActivityPage.class)); + pages.add(new PageRegistration("gb.repositories", RepositoriesPage.class, + getRootPageParameters())); + pages.add(new PageRegistration("gb.activity", ActivityPage.class, getRootPageParameters())); + if (GitBlit.getBoolean(Keys.web.allowLuceneIndexing, true)) { + pages.add(new PageRegistration("gb.search", LuceneSearchPage.class)); + } if (showAdmin) { pages.add(new PageRegistration("gb.users", UsersPage.class)); } @@ -107,7 +119,7 @@ add(navPanel); // login form - StatelessForm<Void> loginForm = new StatelessForm<Void>("loginForm") { + SessionlessForm<Void> loginForm = new SessionlessForm<Void>("loginForm", getClass(), getPageParameters()) { private static final long serialVersionUID = 1L; @@ -118,11 +130,11 @@ UserModel user = GitBlit.self().authenticate(username, password); if (user == null) { - error("Invalid username or password!"); + error(getString("gb.invalidUsernameOrPassword")); } else if (user.username.equals(Constants.FEDERATION_USER)) { // disallow the federation user from logging in via the // web ui - error("Invalid username or password!"); + error(getString("gb.invalidUsernameOrPassword")); user = null; } else { loginUser(user); @@ -136,7 +148,7 @@ WicketUtils.setInputPlaceholder(pwField, getString("gb.password")); loginForm.add(pwField); add(loginForm); - + if (authenticateView || authenticateAdmin) { loginForm.setVisible(!GitBlitWebSession.get().isLoggedIn()); } else { @@ -150,9 +162,9 @@ } else if (showAdmin) { int pendingProposals = GitBlit.self().getPendingFederationProposals().size(); if (pendingProposals == 1) { - info("There is 1 federation proposal awaiting review."); + info(getString("gb.OneProposalToReview")); } else if (pendingProposals > 1) { - info(MessageFormat.format("There are {0} federation proposals awaiting review.", + info(MessageFormat.format(getString("gb.nFederationProposalsToReview"), pendingProposals)); } } @@ -160,10 +172,36 @@ super.setupPage(repositoryName, pageName); } + private PageParameters getRootPageParameters() { + if (reusePageParameters()) { + PageParameters pp = getPageParameters(); + if (pp != null) { + PageParameters params = new PageParameters(pp); + // remove named repository parameter + params.remove("r"); + + // remove days back parameter if it is the default value + if (params.containsKey("db") + && params.getInt("db") == GitBlit.getInteger(Keys.web.activityDuration, 14)) { + params.remove("db"); + } + return params; + } + } + return null; + } + + protected boolean reusePageParameters() { + return false; + } + private void loginUser(UserModel user) { if (user != null) { // Set the user into the session - GitBlitWebSession.get().setUser(user); + GitBlitWebSession session = GitBlitWebSession.get(); + // issue 62: fix session fixation vulnerability + session.replaceSession(); + session.setUser(user); // Set Cookie if (GitBlit.getBoolean(Keys.web.allowCookieAuthentication, false)) { @@ -172,8 +210,16 @@ } if (!continueToOriginalDestination()) { - // Redirect to home page - setResponsePage(getApplication().getHomePage()); + PageParameters params = getPageParameters(); + if (params == null) { + // redirect to this page + setResponsePage(getClass()); + } else { + // Strip username and password and redirect to this page + params.remove("username"); + params.remove("password"); + setResponsePage(getClass(), params); + } } } } @@ -182,25 +228,41 @@ } - protected List<DropDownMenuItem> getFilterMenuItems() { + protected List<DropDownMenuItem> getRepositoryFilterItems(PageParameters params) { final UserModel user = GitBlitWebSession.get().getUser(); Set<DropDownMenuItem> filters = new LinkedHashSet<DropDownMenuItem>(); + List<RepositoryModel> repositories = GitBlit.self().getRepositoryModels(user); // accessible repositories by federation set - for (RepositoryModel repository : GitBlit.self().getRepositoryModels(user)) { + Map<String, AtomicInteger> setMap = new HashMap<String, AtomicInteger>(); + for (RepositoryModel repository : repositories) { for (String set : repository.federationSets) { - filters.add(new DropDownMenuItem(set, "set", set)); + String key = set.toLowerCase(); + if (setMap.containsKey(key)) { + setMap.get(key).incrementAndGet(); + } else { + setMap.put(key, new AtomicInteger(1)); + } } } - if (filters.size() > 0) { + if (setMap.size() > 0) { + List<String> sets = new ArrayList<String>(setMap.keySet()); + Collections.sort(sets); + for (String set : sets) { + filters.add(new DropDownMenuItem(MessageFormat.format("{0} ({1})", set, + setMap.get(set).get()), "set", set, params)); + } // divider filters.add(new DropDownMenuItem()); } // user's team memberships if (user != null && user.teams.size() > 0) { - for (TeamModel team : user.teams) { - filters.add(new DropDownMenuItem(team.name, "team", team.name)); + List<TeamModel> teams = new ArrayList<TeamModel>(user.teams); + Collections.sort(teams); + for (TeamModel team : teams) { + filters.add(new DropDownMenuItem(MessageFormat.format("{0} ({1})", team.name, + team.repositories.size()), "team", team.name, params)); } // divider filters.add(new DropDownMenuItem()); @@ -209,20 +271,38 @@ // custom filters String customFilters = GitBlit.getString(Keys.web.customFilters, null); if (!StringUtils.isEmpty(customFilters)) { + boolean addedExpression = false; List<String> expressions = StringUtils.getStringsFromValue(customFilters, "!!!"); for (String expression : expressions) { - filters.add(new DropDownMenuItem(null, "x", expression)); + if (!StringUtils.isEmpty(expression)) { + addedExpression = true; + filters.add(new DropDownMenuItem(null, "x", expression, params)); + } + } + // if we added any custom expressions, add a divider + if (addedExpression) { + filters.add(new DropDownMenuItem()); } } - - if (filters.size() > 0) { - // if we have any filters, add the divider - filters.add(new DropDownMenuItem()); - // add All Repositories - filters.add(new DropDownMenuItem("All Repositories", null, null)); - } - return new ArrayList<DropDownMenuItem>(filters); + } + + protected List<DropDownMenuItem> getTimeFilterItems(PageParameters params) { + // days back choices - additive parameters + int daysBack = GitBlit.getInteger(Keys.web.activityDuration, 14); + if (daysBack < 1) { + daysBack = 14; + } + List<DropDownMenuItem> items = new ArrayList<DropDownMenuItem>(); + Set<Integer> choicesSet = new HashSet<Integer>(Arrays.asList(daysBack, 14, 28, 60, 90, 180)); + List<Integer> choices = new ArrayList<Integer>(choicesSet); + Collections.sort(choices); + for (Integer db : choices) { + String txt = "last " + db + (db.intValue() > 1 ? " days" : "day"); + items.add(new DropDownMenuItem(txt, "db", db.toString(), params)); + } + items.add(new DropDownMenuItem()); + return items; } protected List<RepositoryModel> getRepositories(PageParameters params) { @@ -231,74 +311,96 @@ return GitBlit.self().getRepositoryModels(user); } + boolean hasParameter = false; String repositoryName = WicketUtils.getRepositoryName(params); String set = WicketUtils.getSet(params); String regex = WicketUtils.getRegEx(params); String team = WicketUtils.getTeam(params); + int daysBack = params.getInt("db", 0); - List<RepositoryModel> models = null; + List<RepositoryModel> availableModels = GitBlit.self().getRepositoryModels(user); + Set<RepositoryModel> models = new HashSet<RepositoryModel>(); if (!StringUtils.isEmpty(repositoryName)) { // try named repository - models = new ArrayList<RepositoryModel>(); - RepositoryModel model = GitBlit.self().getRepositoryModel(repositoryName); - if (user.canAccessRepository(model)) { - models.add(model); + hasParameter = true; + for (RepositoryModel model : availableModels) { + if (model.name.equalsIgnoreCase(repositoryName)) { + models.add(model); + break; + } } - } - - // get all user accessible repositories - if (models == null) { - models = GitBlit.self().getRepositoryModels(user); } if (!StringUtils.isEmpty(regex)) { // filter the repositories by the regex - List<RepositoryModel> accessible = GitBlit.self().getRepositoryModels(user); - List<RepositoryModel> matchingModels = new ArrayList<RepositoryModel>(); + hasParameter = true; Pattern pattern = Pattern.compile(regex); - for (RepositoryModel aModel : accessible) { - if (pattern.matcher(aModel.name).find()) { - matchingModels.add(aModel); + for (RepositoryModel model : availableModels) { + if (pattern.matcher(model.name).find()) { + models.add(model); } } - models = matchingModels; - } else if (!StringUtils.isEmpty(set)) { + } + + if (!StringUtils.isEmpty(set)) { // filter the repositories by the specified sets + hasParameter = true; List<String> sets = StringUtils.getStringsFromValue(set, ","); - List<RepositoryModel> matchingModels = new ArrayList<RepositoryModel>(); - for (RepositoryModel model : models) { + for (RepositoryModel model : availableModels) { for (String curr : sets) { if (model.federationSets.contains(curr)) { - matchingModels.add(model); + models.add(model); } } } - models = matchingModels; - } else if (!StringUtils.isEmpty(team)) { + } + + if (!StringUtils.isEmpty(team)) { // filter the repositories by the specified teams + hasParameter = true; List<String> teams = StringUtils.getStringsFromValue(team, ","); // need TeamModels first List<TeamModel> teamModels = new ArrayList<TeamModel>(); for (String name : teams) { - TeamModel model = GitBlit.self().getTeamModel(name); - if (model != null) { - teamModels.add(model); + TeamModel teamModel = GitBlit.self().getTeamModel(name); + if (teamModel != null) { + teamModels.add(teamModel); } } // brute-force our way through finding the matching models - List<RepositoryModel> matchingModels = new ArrayList<RepositoryModel>(); - for (RepositoryModel repositoryModel : models) { + for (RepositoryModel repositoryModel : availableModels) { for (TeamModel teamModel : teamModels) { if (teamModel.hasRepository(repositoryModel.name)) { - matchingModels.add(repositoryModel); + models.add(repositoryModel); } } } - models = matchingModels; } - return models; + + if (!hasParameter) { + models.addAll(availableModels); + } + + // time-filter the list + if (daysBack > 0) { + Calendar cal = Calendar.getInstance(); + cal.set(Calendar.HOUR_OF_DAY, 0); + cal.set(Calendar.MINUTE, 0); + cal.set(Calendar.SECOND, 0); + cal.set(Calendar.MILLISECOND, 0); + cal.add(Calendar.DATE, -1 * daysBack); + Date threshold = cal.getTime(); + Set<RepositoryModel> timeFiltered = new HashSet<RepositoryModel>(); + for (RepositoryModel model : models) { + if (model.lastChange.after(threshold)) { + timeFiltered.add(model); + } + } + models = timeFiltered; + } + return new ArrayList<RepositoryModel>(models); } } -- Gitblit v1.9.1