James Moger
2012-01-05 cb285cbfddfc0b633d6b8cdb4dc0d2bd2b8b51ef
src/com/gitblit/wicket/pages/ActivityPage.java
@@ -15,36 +15,29 @@
 */
package com.gitblit.wicket.pages;
import java.text.DateFormat;
import java.text.MessageFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.Collections;
import java.util.Date;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.apache.wicket.Application;
import org.apache.wicket.PageParameters;
import org.apache.wicket.behavior.HeaderContributor;
import org.apache.wicket.markup.html.basic.Label;
import org.eclipse.jgit.lib.Constants;
import org.eclipse.jgit.lib.ObjectId;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.revwalk.RevCommit;
import com.gitblit.GitBlit;
import com.gitblit.models.DailyActivity;
import com.gitblit.Keys;
import com.gitblit.models.Activity;
import com.gitblit.models.Metric;
import com.gitblit.models.RefModel;
import com.gitblit.models.RepositoryCommit;
import com.gitblit.models.RepositoryModel;
import com.gitblit.models.UserModel;
import com.gitblit.utils.JGitUtils;
import com.gitblit.utils.StringUtils;
import com.gitblit.utils.TimeUtils;
import com.gitblit.wicket.GitBlitWebSession;
import com.gitblit.utils.ActivityUtils;
import com.gitblit.wicket.PageRegistration;
import com.gitblit.wicket.PageRegistration.DropDownMenuItem;
import com.gitblit.wicket.PageRegistration.DropDownMenuRegistration;
import com.gitblit.wicket.WicketUtils;
import com.gitblit.wicket.charting.GoogleChart;
import com.gitblit.wicket.charting.GoogleCharts;
@@ -62,151 +55,109 @@
public class ActivityPage extends RootPage {
   public ActivityPage(PageParameters params) {
      super();
      super(params);
      setupPage("", "");
      final UserModel user = GitBlitWebSession.get().getUser();
      // parameters
      int daysBack = WicketUtils.getDaysBack(params);
      if (daysBack < 1) {
         daysBack = 14;
      }
      String set = WicketUtils.getSet(params);
      String repositoryName = WicketUtils.getRepositoryName(params);
      }
      String objectId = WicketUtils.getObject(params);
      List<RepositoryModel> models = null;
      if (!StringUtils.isEmpty(repositoryName)) {
         // named repository
         models = new ArrayList<RepositoryModel>();
         RepositoryModel model = GitBlit.self().getRepositoryModel(repositoryName);
         if (user.canAccessRepository(model)) {
            models.add(model);
      // determine repositories to view and retrieve the activity
      List<RepositoryModel> models = getRepositories(params);
      List<Activity> recentActivity = ActivityUtils.getRecentActivity(models, daysBack, objectId);
      if (recentActivity.size() == 0) {
         // no activity, skip graphs and activity panel
         add(new Label("subheader", MessageFormat.format(getString("gb.recentActivityNone"),
               daysBack)));
         add(new Label("activityPanel"));
      } else {
         // calculate total commits and total authors
         int totalCommits = 0;
         Set<String> uniqueAuthors = new HashSet<String>();
         for (Activity activity : recentActivity) {
            totalCommits += activity.commits.size();
            uniqueAuthors.addAll(activity.getAuthorMetrics().keySet());
         }
         int totalAuthors = uniqueAuthors.size();
         // add the subheader with stat numbers
         add(new Label("subheader", MessageFormat.format(getString("gb.recentActivityStats"),
               daysBack, totalCommits, totalAuthors)));
         // create the activity charts
         GoogleCharts charts = createCharts(recentActivity);
         add(new HeaderContributor(charts));
         // add activity panel
         add(new ActivityPanel("activityPanel", recentActivity));
      }
   }
   @Override
   protected boolean reusePageParameters() {
      return true;
   }
   @Override
   protected void addDropDownMenus(List<PageRegistration> pages) {
      DropDownMenuRegistration filters = new DropDownMenuRegistration("gb.filters",
            ActivityPage.class);
      PageParameters currentParameters = getPageParameters();
      int daysBack = GitBlit.getInteger(Keys.web.activityDuration, 14);
      if (currentParameters != null && !currentParameters.containsKey("db")) {
         currentParameters.put("db", daysBack);
      }
      // get all user accessible repositories
      if (models == null) {
         models = GitBlit.self().getRepositoryModels(user);
      // preserve time filter options on repository choices
      filters.menuItems.addAll(getRepositoryFilterItems(currentParameters));
      // preserve repository filter options on time choices
      filters.menuItems.addAll(getTimeFilterItems(currentParameters));
      if (filters.menuItems.size() > 0) {
         // Reset Filter
         filters.menuItems.add(new DropDownMenuItem(getString("gb.reset"), null, null));
      }
      pages.add(filters);
   }
      // filter the repositories by the specified set
      if (!StringUtils.isEmpty(set)) {
         List<String> sets = StringUtils.getStringsFromValue(set, ",");
         List<RepositoryModel> setModels = new ArrayList<RepositoryModel>();
         for (RepositoryModel model : models) {
            for (String curr : sets) {
               if (model.federationSets.contains(curr)) {
                  setModels.add(model);
               }
            }
         }
         models = setModels;
      }
      // Activity panel shows last daysBack of activity across all
      // repositories.
      Date thresholdDate = new Date(System.currentTimeMillis() - daysBack * TimeUtils.ONEDAY);
      // Build a map of DailyActivity from the available repositories for the
      // specified threshold date.
      DateFormat df = new SimpleDateFormat("yyyy-MM-dd");
      Calendar cal = Calendar.getInstance();
      Map<String, DailyActivity> activity = new HashMap<String, DailyActivity>();
      for (RepositoryModel model : models) {
         if (model.hasCommits && model.lastChange.after(thresholdDate)) {
            Repository repository = GitBlit.self().getRepository(model.name);
            List<RevCommit> commits = JGitUtils.getRevLog(repository, objectId, thresholdDate);
            Map<ObjectId, List<RefModel>> allRefs = JGitUtils.getAllRefs(repository);
            repository.close();
            // determine commit branch
            String branch = objectId;
            if (StringUtils.isEmpty(branch)) {
               List<RefModel> headRefs = allRefs.get(commits.get(0).getId());
               List<String> localBranches = new ArrayList<String>();
               for (RefModel ref : headRefs) {
                  if (ref.getName().startsWith(Constants.R_HEADS)) {
                     localBranches.add(ref.getName().substring(Constants.R_HEADS.length()));
                  }
               }
               // determine branch
               if (localBranches.size() == 1) {
                  // only one branch, choose it
                  branch = localBranches.get(0);
               } else if (localBranches.size() > 1) {
                  if (localBranches.contains("master")) {
                     // choose master
                     branch = "master";
                  } else {
                     // choose first branch
                     branch = localBranches.get(0);
                  }
               }
            }
            for (RevCommit commit : commits) {
               Date date = JGitUtils.getCommitDate(commit);
               String dateStr = df.format(date);
               if (!activity.containsKey(dateStr)) {
                  // Normalize the date to midnight
                  cal.setTime(date);
                  cal.set(Calendar.HOUR_OF_DAY, 0);
                  cal.set(Calendar.MINUTE, 0);
                  cal.set(Calendar.SECOND, 0);
                  cal.set(Calendar.MILLISECOND, 0);
                  activity.put(dateStr, new DailyActivity(cal.getTime()));
               }
               RepositoryCommit commitModel = new RepositoryCommit(model.name, branch, commit);
               commitModel.setRefs(allRefs.get(commit.getId()));
               activity.get(dateStr).commits.add(commitModel);
            }
         }
      }
   /**
    * Creates the daily activity line chart, the active repositories pie chart,
    * and the active authors pie chart
    *
    * @param recentActivity
    * @return
    */
   private GoogleCharts createCharts(List<Activity> recentActivity) {
      // activity metrics
      Map<String, Metric> dayMetrics = new HashMap<String, Metric>();
      Map<String, Metric> repositoryMetrics = new HashMap<String, Metric>();
      Map<String, Metric> authorMetrics = new HashMap<String, Metric>();
      // prepare day metrics
      cal.setTimeInMillis(System.currentTimeMillis());
      for (int i = 0; i < daysBack; i++) {
         cal.add(Calendar.DATE, -1);
         String key = df.format(cal.getTime());
         dayMetrics.put(key, new Metric(key));
      }
      // aggregate repository and author metrics
      for (Activity activity : recentActivity) {
      // calculate activity metrics
      for (Map.Entry<String, DailyActivity> entry : activity.entrySet()) {
         // day metrics
         Metric day = dayMetrics.get(entry.getKey());
         day.count = entry.getValue().commits.size();
         for (RepositoryCommit commit : entry.getValue().commits) {
            // repository metrics
            String repository = commit.repository;
            if (!repositoryMetrics.containsKey(repository)) {
               repositoryMetrics.put(repository, new Metric(repository));
            }
            repositoryMetrics.get(repository).count++;
            // author metrics
            String author = commit.getAuthorIdent().getEmailAddress().toLowerCase();
         // aggregate author metrics
         for (Map.Entry<String, Metric> entry : activity.getAuthorMetrics().entrySet()) {
            String author = entry.getKey();
            if (!authorMetrics.containsKey(author)) {
               authorMetrics.put(author, new Metric(author));
            }
            authorMetrics.get(author).count++;
            authorMetrics.get(author).count += entry.getValue().count;
         }
      }
      // sort the activity groups and their commit contents
      int totalCommits = 0;
      List<DailyActivity> recentActivity = new ArrayList<DailyActivity>(activity.values());
      for (DailyActivity daily : recentActivity) {
         Collections.sort(daily.commits);
         totalCommits += daily.commits.size();
         // aggregate repository metrics
         for (Map.Entry<String, Metric> entry : activity.getRepositoryMetrics().entrySet()) {
            String repository = entry.getKey();
            if (!repositoryMetrics.containsKey(repository)) {
               repositoryMetrics.put(repository, new Metric(repository));
            }
            repositoryMetrics.get(repository).count += entry.getValue().count;
         }
      }
      // build google charts
@@ -221,9 +172,9 @@
      // daily line chart
      GoogleChart chart = new GoogleLineChart("chartDaily", getString("gb.dailyActivity"), "day",
            getString("gb.commits"));
      df = new SimpleDateFormat("MMM dd");
      for (DailyActivity metric : recentActivity) {
         chart.addValue(df.format(metric.date), metric.commits.size());
      SimpleDateFormat df = new SimpleDateFormat("MMM dd");
      for (Activity metric : recentActivity) {
         chart.addValue(df.format(metric.startDate), metric.commits.size());
      }
      chart.setWidth(w);
      chart.setHeight(h);
@@ -249,12 +200,24 @@
      chart.setHeight(h);
      charts.addChart(chart);
      add(new HeaderContributor(charts));
      return charts;
   }
      add(new Label("subheader", MessageFormat.format(getString("gb.recentActivitySubheader"),
            daysBack, totalCommits, authorMetrics.size())));
   @Override
   protected void onBeforeRender() {
      if (GitBlit.isDebugMode()) {
         // strip Wicket tags in debug mode for jQuery DOM traversal
         Application.get().getMarkupSettings().setStripWicketTags(true);
      }
      super.onBeforeRender();
   }
      // add activity panel
      add(new ActivityPanel("activityPanel", recentActivity));
   @Override
   protected void onAfterRender() {
      if (GitBlit.isDebugMode()) {
         // restore Wicket debug tags
         Application.get().getMarkupSettings().setStripWicketTags(false);
      }
      super.onAfterRender();
   }
}