James Moger
2012-07-11 a31cf95ac0787eb559cb78f48c52bc6b79e970d8
src/com/gitblit/GitBlit.java
@@ -30,6 +30,7 @@
import java.util.Collections;
import java.util.Date;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
@@ -55,6 +56,8 @@
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.lib.RepositoryCache.FileKey;
import org.eclipse.jgit.lib.StoredConfig;
import org.eclipse.jgit.storage.file.WindowCache;
import org.eclipse.jgit.storage.file.WindowCacheConfig;
import org.eclipse.jgit.transport.resolver.FileResolver;
import org.eclipse.jgit.transport.resolver.RepositoryResolver;
import org.eclipse.jgit.transport.resolver.ServiceNotAuthorizedException;
@@ -73,6 +76,7 @@
import com.gitblit.models.FederationSet;
import com.gitblit.models.Metric;
import com.gitblit.models.RepositoryModel;
import com.gitblit.models.SearchResult;
import com.gitblit.models.ServerSettings;
import com.gitblit.models.ServerStatus;
import com.gitblit.models.SettingModel;
@@ -184,6 +188,16 @@
      }
      return self().timezone;
   }
   /**
    * Returns the user-defined blob encodings.
    *
    * @return an array of encodings, may be empty
    */
   public static String [] getEncodings() {
      return getStrings(Keys.web.blobEncodings).toArray(new String[0]);
   }
   /**
    * Returns the boolean value for the specified key. If the key does not
@@ -250,6 +264,17 @@
    */
   public static List<String> getStrings(String key) {
      return self().settings.getStrings(key);
   }
   /**
    * Returns a map of space-separated key-value pairs from the specified key.
    *
    * @see IStoredSettings.getStrings(String key)
    * @param name
    * @return map of string, string
    */
   public static Map<String, String> getMap(String key) {
      return self().settings.getMap(key);
   }
   /**
@@ -375,6 +400,38 @@
      this.userService = userService;
      this.userService.setup(settings);
   }
   /**
    *
    * @return true if the user service supports credential changes
    */
   public boolean supportsCredentialChanges() {
      return userService.supportsCredentialChanges();
   }
   /**
    *
    * @return true if the user service supports display name changes
    */
   public boolean supportsDisplayNameChanges() {
      return userService.supportsDisplayNameChanges();
   }
   /**
    *
    * @return true if the user service supports email address changes
    */
   public boolean supportsEmailAddressChanges() {
      return userService.supportsEmailAddressChanges();
   }
   /**
    *
    * @return true if the user service supports team membership changes
    */
   public boolean supportsTeamMembershipChanges() {
      return userService.supportsTeamMembershipChanges();
   }
   /**
    * Authenticate a user based on a username and password.
@@ -455,13 +512,31 @@
            userCookie = new Cookie(Constants.NAME, "");
         } else {
            // set cookie for login
            char[] cookie = userService.getCookie(user);
            userCookie = new Cookie(Constants.NAME, new String(cookie));
            userCookie.setMaxAge(Integer.MAX_VALUE);
            String cookie = userService.getCookie(user);
            if (StringUtils.isEmpty(cookie)) {
               // create empty cookie
               userCookie = new Cookie(Constants.NAME, "");
            } else {
               // create real cookie
               userCookie = new Cookie(Constants.NAME, cookie);
               userCookie.setMaxAge(Integer.MAX_VALUE);
            }
         }
         userCookie.setPath("/");
         response.addCookie(userCookie);
      }
   }
   /**
    * Logout a user.
    *
    * @param user
    */
   public void logout(UserModel user) {
      if (userService == null) {
         return;
      }
      userService.logout(user);
   }
   /**
@@ -791,7 +866,7 @@
         model.useTickets = getConfig(config, "useTickets", false);
         model.useDocs = getConfig(config, "useDocs", false);
         model.accessRestriction = AccessRestrictionType.fromName(getConfig(config,
               "accessRestriction", null));
               "accessRestriction", settings.getString(Keys.git.defaultAccessRestriction, null)));
         model.showRemoteBranches = getConfig(config, "showRemoteBranches", false);
         model.isFrozen = getConfig(config, "isFrozen", false);
         model.showReadme = getConfig(config, "showReadme", false);
@@ -800,15 +875,23 @@
         model.federationStrategy = FederationStrategy.fromName(getConfig(config,
               "federationStrategy", null));
         model.federationSets = new ArrayList<String>(Arrays.asList(config.getStringList(
               "gitblit", null, "federationSets")));
               Constants.CONFIG_GITBLIT, null, "federationSets")));
         model.isFederated = getConfig(config, "isFederated", false);
         model.origin = config.getString("remote", "origin", "url");
         model.preReceiveScripts = new ArrayList<String>(Arrays.asList(config.getStringList(
               "gitblit", null, "preReceiveScript")));
               Constants.CONFIG_GITBLIT, null, "preReceiveScript")));
         model.postReceiveScripts = new ArrayList<String>(Arrays.asList(config.getStringList(
               "gitblit", null, "postReceiveScript")));
               Constants.CONFIG_GITBLIT, null, "postReceiveScript")));
         model.mailingLists = new ArrayList<String>(Arrays.asList(config.getStringList(
               "gitblit", null, "mailingList")));
               Constants.CONFIG_GITBLIT, null, "mailingList")));
         model.indexedBranches = new ArrayList<String>(Arrays.asList(config.getStringList(
               Constants.CONFIG_GITBLIT, null, "indexBranch")));
         // Custom defined properties
         model.customFields = new LinkedHashMap<String, String>();
         for (String aProperty : config.getNames(Constants.CONFIG_GITBLIT, Constants.CONFIG_CUSTOM_FIELDS)) {
            model.customFields.put(aProperty, config.getString(Constants.CONFIG_GITBLIT, Constants.CONFIG_CUSTOM_FIELDS, aProperty));
         }
      }
      model.HEAD = JGitUtils.getHEADRef(r);
      model.availableRefs = JGitUtils.getAvailableHeadTargets(r);
@@ -868,6 +951,9 @@
            repository.close();
         }
      }
      // close any open index writer/searcher in the Lucene executor
      luceneExecutor.close(repositoryName);
   }
   /**
@@ -899,7 +985,7 @@
    * @return field value or defaultValue
    */
   private String getConfig(StoredConfig config, String field, String defaultValue) {
      String value = config.getString("gitblit", null, field);
      String value = config.getString(Constants.CONFIG_GITBLIT, null, field);
      if (StringUtils.isEmpty(value)) {
         return defaultValue;
      }
@@ -907,7 +993,7 @@
   }
   /**
    * Returns the gitblit boolean vlaue for the specified key. If key is not
    * Returns the gitblit boolean value for the specified key. If key is not
    * set, returns defaultValue.
    * 
    * @param config
@@ -916,7 +1002,7 @@
    * @return field value or defaultValue
    */
   private boolean getConfig(StoredConfig config, String field, boolean defaultValue) {
      return config.getBoolean("gitblit", field, defaultValue);
      return config.getBoolean(Constants.CONFIG_GITBLIT, field, defaultValue);
   }
   /**
@@ -1033,34 +1119,58 @@
    */
   public void updateConfiguration(Repository r, RepositoryModel repository) {
      StoredConfig config = JGitUtils.readConfig(r);
      config.setString("gitblit", null, "description", repository.description);
      config.setString("gitblit", null, "owner", repository.owner);
      config.setBoolean("gitblit", null, "useTickets", repository.useTickets);
      config.setBoolean("gitblit", null, "useDocs", repository.useDocs);
      config.setString("gitblit", null, "accessRestriction", repository.accessRestriction.name());
      config.setBoolean("gitblit", null, "showRemoteBranches", repository.showRemoteBranches);
      config.setBoolean("gitblit", null, "isFrozen", repository.isFrozen);
      config.setBoolean("gitblit", null, "showReadme", repository.showReadme);
      config.setBoolean("gitblit", null, "skipSizeCalculation", repository.skipSizeCalculation);
      config.setBoolean("gitblit", null, "skipSummaryMetrics", repository.skipSummaryMetrics);
      config.setStringList("gitblit", null, "federationSets", repository.federationSets);
      config.setString("gitblit", null, "federationStrategy",
      config.setString(Constants.CONFIG_GITBLIT, null, "description", repository.description);
      config.setString(Constants.CONFIG_GITBLIT, null, "owner", repository.owner);
      config.setBoolean(Constants.CONFIG_GITBLIT, null, "useTickets", repository.useTickets);
      config.setBoolean(Constants.CONFIG_GITBLIT, null, "useDocs", repository.useDocs);
      config.setString(Constants.CONFIG_GITBLIT, null, "accessRestriction", repository.accessRestriction.name());
      config.setBoolean(Constants.CONFIG_GITBLIT, null, "showRemoteBranches", repository.showRemoteBranches);
      config.setBoolean(Constants.CONFIG_GITBLIT, null, "isFrozen", repository.isFrozen);
      config.setBoolean(Constants.CONFIG_GITBLIT, null, "showReadme", repository.showReadme);
      config.setBoolean(Constants.CONFIG_GITBLIT, null, "skipSizeCalculation", repository.skipSizeCalculation);
      config.setBoolean(Constants.CONFIG_GITBLIT, null, "skipSummaryMetrics", repository.skipSummaryMetrics);
      config.setString(Constants.CONFIG_GITBLIT, null, "federationStrategy",
            repository.federationStrategy.name());
      config.setBoolean("gitblit", null, "isFederated", repository.isFederated);
      if (!ArrayUtils.isEmpty(repository.preReceiveScripts)) {
         config.setStringList("gitblit", null, "preReceiveScript", repository.preReceiveScripts);
      config.setBoolean(Constants.CONFIG_GITBLIT, null, "isFederated", repository.isFederated);
      updateList(config, "federationSets", repository.federationSets);
      updateList(config, "preReceiveScript", repository.preReceiveScripts);
      updateList(config, "postReceiveScript", repository.postReceiveScripts);
      updateList(config, "mailingList", repository.mailingLists);
      updateList(config, "indexBranch", repository.indexedBranches);
      // User Defined Properties
      if (repository.customFields != null) {
         if (repository.customFields.size() == 0) {
            // clear section
            config.unsetSection(Constants.CONFIG_GITBLIT, Constants.CONFIG_CUSTOM_FIELDS);
         } else {
            for (Entry<String, String> property : repository.customFields.entrySet()) {
               // set field
               String key = property.getKey();
               String value = property.getValue();
               config.setString(Constants.CONFIG_GITBLIT, Constants.CONFIG_CUSTOM_FIELDS, key, value);
            }
         }
      }
      if (!ArrayUtils.isEmpty(repository.postReceiveScripts)) {
         config.setStringList("gitblit", null, "postReceiveScript",
               repository.postReceiveScripts);
      }
      if (!ArrayUtils.isEmpty(repository.mailingLists)) {
         config.setStringList("gitblit", null, "mailingList", repository.mailingLists);
      }
      try {
         config.save();
      } catch (IOException e) {
         logger.error("Failed to save repository config!", e);
      }
   }
   private void updateList(StoredConfig config, String field, List<String> list) {
      // a null list is skipped, not cleared
      // this is for RPC administration where an older manager might be used
      if (list == null) {
         return;
      }
      if (ArrayUtils.isEmpty(list)) {
         config.unset(Constants.CONFIG_GITBLIT, null, field);
      } else {
         config.setStringList(Constants.CONFIG_GITBLIT, null, field, list);
      }
   }
@@ -1645,6 +1755,20 @@
      }
      return scripts;
   }
   /**
    * Search the specified repositories using the Lucene query.
    *
    * @param query
    * @param page
    * @param pageSize
    * @param repositories
    * @return
    */
   public List<SearchResult> search(String query, int page, int pageSize, List<String> repositories) {
      List<SearchResult> srs = luceneExecutor.search(query, page, pageSize, repositories);
      return srs;
   }
   /**
    * Notify the administrators by email.
@@ -1697,15 +1821,6 @@
   }
   /**
    * Update the Lucene index of a repository.
    *
    * @param repository
    */
   public void updateLuceneIndex(RepositoryModel repository) {
      luceneExecutor.queue(repository);
   }
   /**
    * Returns the descriptions/comments of the Gitblit config settings.
    * 
    * @return SettingsModel
@@ -1714,9 +1829,13 @@
      // ensure that the current values are updated in the setting models
      for (String key : settings.getAllKeys(null)) {
         SettingModel setting = settingsModel.get(key);
         if (setting != null) {
            setting.currentValue = settings.getString(key, "");
         if (setting == null) {
            // unreferenced setting, create a setting model
            setting = new SettingModel();
            setting.name = key;
            settingsModel.add(setting);
         }
         setting.currentValue = settings.getString(key, "");
      }
      settingsModel.pushScripts = getAllScripts();
      return settingsModel;
@@ -1731,6 +1850,10 @@
    */
   private ServerSettings loadSettingModels() {
      ServerSettings settingsModel = new ServerSettings();
      settingsModel.supportsCredentialChanges = userService.supportsCredentialChanges();
      settingsModel.supportsDisplayNameChanges = userService.supportsDisplayNameChanges();
      settingsModel.supportsEmailAddressChanges = userService.supportsEmailAddressChanges();
      settingsModel.supportsTeamMembershipChanges = userService.supportsTeamMembershipChanges();
      try {
         // Read bundled Gitblit properties to extract setting descriptions.
         // This copy is pristine and only used for populating the setting
@@ -1822,16 +1945,34 @@
      } else {
         logger.warn("Mail server is not properly configured.  Mail services disabled.");
      }
      luceneExecutor = new LuceneExecutor(settings);
      if (luceneExecutor.isReady()) {
         logger.info("Lucene executor is scheduled to process the repository queue every 10 minutes.");
         scheduledExecutor.scheduleAtFixedRate(luceneExecutor, 1, 10, TimeUnit.MINUTES);
      } else {
         logger.warn("Lucene executor is disabled.");
      }
      luceneExecutor = new LuceneExecutor(settings, repositoriesFolder);
      logger.info("Lucene executor is scheduled to process indexed branches every 2 minutes.");
      scheduledExecutor.scheduleAtFixedRate(luceneExecutor, 1, 2, TimeUnit.MINUTES);
      if (startFederation) {
         configureFederation();
      }
      }
      // Configure JGit
      WindowCacheConfig cfg = new WindowCacheConfig();
      cfg.setPackedGitWindowSize(settings.getFilesize(Keys.git.packedGitWindowSize, cfg.getPackedGitWindowSize()));
      cfg.setPackedGitLimit(settings.getFilesize(Keys.git.packedGitLimit, cfg.getPackedGitLimit()));
      cfg.setDeltaBaseCacheLimit(settings.getFilesize(Keys.git.deltaBaseCacheLimit, cfg.getDeltaBaseCacheLimit()));
      cfg.setPackedGitOpenFiles(settings.getFilesize(Keys.git.packedGitOpenFiles, cfg.getPackedGitOpenFiles()));
      cfg.setStreamFileThreshold(settings.getFilesize(Keys.git.streamFileThreshold, cfg.getStreamFileThreshold()));
      cfg.setPackedGitMMAP(settings.getBoolean(Keys.git.packedGitMmap, cfg.isPackedGitMMAP()));
      try {
         WindowCache.reconfigure(cfg);
         logger.debug(MessageFormat.format("{0} = {1,number,0}", Keys.git.packedGitWindowSize, cfg.getPackedGitWindowSize()));
         logger.debug(MessageFormat.format("{0} = {1,number,0}", Keys.git.packedGitLimit, cfg.getPackedGitLimit()));
         logger.debug(MessageFormat.format("{0} = {1,number,0}", Keys.git.deltaBaseCacheLimit, cfg.getDeltaBaseCacheLimit()));
         logger.debug(MessageFormat.format("{0} = {1,number,0}", Keys.git.packedGitOpenFiles, cfg.getPackedGitOpenFiles()));
         logger.debug(MessageFormat.format("{0} = {1,number,0}", Keys.git.streamFileThreshold, cfg.getStreamFileThreshold()));
         logger.debug(MessageFormat.format("{0} = {1}", Keys.git.packedGitMmap, cfg.isPackedGitMMAP()));
      } catch (IllegalArgumentException e) {
         logger.error("Failed to configure JGit parameters!", e);
      }
   }
   
   private void logTimezone(String type, TimeZone zone) {
@@ -1850,7 +1991,6 @@
   @Override
   public void contextInitialized(ServletContextEvent contextEvent) {
      servletContext = contextEvent.getServletContext();
      settingsModel = loadSettingModels();
      if (settings == null) {
         // Gitblit WAR is running in a servlet container
         ServletContext context = contextEvent.getServletContext();
@@ -1889,7 +2029,8 @@
            }
         }
      }
      settingsModel = loadSettingModels();
      serverStatus.servletContainer = servletContext.getServerInfo();
   }