James Moger
2012-03-27 f5432e015d173f2665f7ca34735057e7b2b25d16
src/com/gitblit/GitBlit.java
@@ -23,16 +23,19 @@
import java.io.InputStreamReader;
import java.lang.reflect.Field;
import java.text.MessageFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.Date;
import java.util.HashMap;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import java.util.TimeZone;
import java.util.TreeSet;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.Executors;
@@ -70,6 +73,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;
@@ -103,7 +107,7 @@
public class GitBlit implements ServletContextListener {
   private static GitBlit gitblit;
   private final Logger logger = LoggerFactory.getLogger(GitBlit.class);
   private final ScheduledExecutorService scheduledExecutor = Executors.newScheduledThreadPool(5);
@@ -123,8 +127,6 @@
   private File repositoriesFolder;
   private boolean exportAll = true;
   private IUserService userService;
   private IStoredSettings settings;
@@ -134,6 +136,10 @@
   private ServerStatus serverStatus;
   private MailExecutor mailExecutor;
   private LuceneExecutor luceneExecutor;
   private TimeZone timezone;
   public GitBlit() {
      if (gitblit == null) {
@@ -162,6 +168,24 @@
   public static boolean isGO() {
      return self().settings instanceof FileSettings;
   }
   /**
    * Returns the preferred timezone for the Gitblit instance.
    *
    * @return a timezone
    */
   public static TimeZone getTimezone() {
      if (self().timezone == null) {
         String tzid = getString("web.timezone", null);
         if (StringUtils.isEmpty(tzid)) {
            self().timezone = TimeZone.getDefault();
            return self().timezone;
         }
         self().timezone = TimeZone.getTimeZone(tzid);
      }
      return self().timezone;
   }
   /**
    * Returns the boolean value for the specified key. If the key does not
@@ -642,7 +666,8 @@
    * @return list of all repositories
    */
   public List<String> getRepositoryList() {
      return JGitUtils.getRepositoryList(repositoriesFolder, exportAll,
      return JGitUtils.getRepositoryList(repositoriesFolder,
            settings.getBoolean(Keys.git.onlyAccessBareRepositories, false),
            settings.getBoolean(Keys.git.searchRepositoriesSubfolders, true));
   }
@@ -653,21 +678,38 @@
    * @return repository or null
    */
   public Repository getRepository(String repositoryName) {
      return getRepository(repositoryName, true);
   }
   /**
    * Returns the JGit repository for the specified name.
    *
    * @param repositoryName
    * @param logError
    * @return repository or null
    */
   public Repository getRepository(String repositoryName, boolean logError) {
      Repository r = null;
      try {
         r = repositoryResolver.open(null, repositoryName);
      } catch (RepositoryNotFoundException e) {
         r = null;
         logger.error("GitBlit.getRepository(String) failed to find "
               + new File(repositoriesFolder, repositoryName).getAbsolutePath());
         if (logError) {
            logger.error("GitBlit.getRepository(String) failed to find "
                  + new File(repositoriesFolder, repositoryName).getAbsolutePath());
         }
      } catch (ServiceNotAuthorizedException e) {
         r = null;
         logger.error("GitBlit.getRepository(String) failed to find "
               + new File(repositoriesFolder, repositoryName).getAbsolutePath(), e);
         if (logError) {
            logger.error("GitBlit.getRepository(String) failed to find "
                  + new File(repositoriesFolder, repositoryName).getAbsolutePath(), e);
         }
      } catch (ServiceNotEnabledException e) {
         r = null;
         logger.error("GitBlit.getRepository(String) failed to find "
               + new File(repositoriesFolder, repositoryName).getAbsolutePath(), e);
         if (logError) {
            logger.error("GitBlit.getRepository(String) failed to find "
                  + new File(repositoriesFolder, repositoryName).getAbsolutePath(), e);
         }
      }
      return r;
   }
@@ -742,7 +784,8 @@
      RepositoryModel model = new RepositoryModel();
      model.name = repositoryName;
      model.hasCommits = JGitUtils.hasCommits(r);
      model.lastChange = JGitUtils.getLastChange(r, null);
      model.lastChange = JGitUtils.getLastChange(r);
      model.isBare = r.isBare();
      StoredConfig config = JGitUtils.readConfig(r);
      if (config != null) {
         model.description = getConfig(config, "description", "");
@@ -768,7 +811,11 @@
               "gitblit", null, "postReceiveScript")));
         model.mailingLists = new ArrayList<String>(Arrays.asList(config.getStringList(
               "gitblit", null, "mailingList")));
         model.indexedBranches = new ArrayList<String>(Arrays.asList(config.getStringList(
               "gitblit", null, "indexBranch")));
      }
      model.HEAD = JGitUtils.getHEADRef(r);
      model.availableRefs = JGitUtils.getAvailableHeadTargets(r);
      r.close();
      return model;
   }
@@ -825,6 +872,9 @@
            repository.close();
         }
      }
      // close any open index writer/searcher in the Lucene executor
      luceneExecutor.close(repositoryName);
   }
   /**
@@ -841,7 +891,7 @@
      if (repositoryMetricsCache.hasCurrent(model.name, model.lastChange)) {
         return new ArrayList<Metric>(repositoryMetricsCache.getObject(model.name));
      }
      List<Metric> metrics = MetricUtils.getDateMetrics(repository, null, true, null);
      List<Metric> metrics = MetricUtils.getDateMetrics(repository, null, true, null, getTimezone());
      repositoryMetricsCache.updateObject(model.name, model.lastChange, metrics);
      return new ArrayList<Metric>(metrics);
   }
@@ -964,6 +1014,18 @@
      // update settings
      if (r != null) {
         updateConfiguration(r, repository);
         // only update symbolic head if it changes
         String currentRef = JGitUtils.getHEADRef(r);
         if (!StringUtils.isEmpty(repository.HEAD) && !repository.HEAD.equals(currentRef)) {
            logger.info(MessageFormat.format("Relinking {0} HEAD from {1} to {2}",
                  repository.name, currentRef, repository.HEAD));
            if (JGitUtils.setHEADtoRef(r, repository.HEAD)) {
               // clear the cache
               clearRepositoryCache(repository.name);
            }
         }
         // close the repository object
         r.close();
      }
   }
@@ -1001,6 +1063,9 @@
      }
      if (!ArrayUtils.isEmpty(repository.mailingLists)) {
         config.setStringList("gitblit", null, "mailingList", repository.mailingLists);
      }
      if (!ArrayUtils.isEmpty(repository.indexedBranches)) {
         config.setStringList("gitblit", null, "indexBranch", repository.indexedBranches);
      }
      try {
         config.save();
@@ -1590,6 +1655,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.
@@ -1735,29 +1814,42 @@
      this.settings = settings;
      repositoriesFolder = getRepositoriesFolder();
      logger.info("Git repositories folder " + repositoriesFolder.getAbsolutePath());
      repositoryResolver = new FileResolver<Void>(repositoriesFolder, exportAll);
      repositoryResolver = new FileResolver<Void>(repositoriesFolder, true);
      logTimezone("JVM", TimeZone.getDefault());
      logTimezone(Constants.NAME, getTimezone());
      serverStatus = new ServerStatus(isGO());
      String realm = settings.getString(Keys.realm.userService, "users.properties");
      IUserService loginService = null;
      try {
         // check to see if this "file" is a login service class
         Class<?> realmClass = Class.forName(realm);
         if (IUserService.class.isAssignableFrom(realmClass)) {
            loginService = (IUserService) realmClass.newInstance();
         }
         loginService = (IUserService) realmClass.newInstance();
      } catch (Throwable t) {
         loginService = new GitblitUserService();
      }
      setUserService(loginService);
      mailExecutor = new MailExecutor(settings);
      if (mailExecutor.isReady()) {
         logger.info("Mail executor is scheduled to process the message queue every 2 minutes.");
         scheduledExecutor.scheduleAtFixedRate(mailExecutor, 1, 2, TimeUnit.MINUTES);
      } else {
         logger.warn("Mail server is not properly configured.  Mail services 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();
      }
      }
   }
   private void logTimezone(String type, TimeZone zone) {
      SimpleDateFormat df = new SimpleDateFormat("z Z");
      df.setTimeZone(zone);
      String offset = df.format(new Date());
      logger.info(type + " timezone is " + zone.getID() + " (" + offset + ")");
   }
   /**
@@ -1776,14 +1868,18 @@
         WebXmlSettings webxmlSettings = new WebXmlSettings(context);
         // 0.7.0 web.properties in the deployed war folder
         File overrideFile = new File(context.getRealPath("/WEB-INF/web.properties"));
         if (overrideFile.exists()) {
            webxmlSettings.applyOverrides(overrideFile);
         String webProps = context.getRealPath("/WEB-INF/web.properties");
         if (!StringUtils.isEmpty(webProps)) {
            File overrideFile = new File(webProps);
            if (overrideFile.exists()) {
               webxmlSettings.applyOverrides(overrideFile);
            }
         }
         // 0.8.0 gitblit.properties file located outside the deployed war
         // folder lie, for example, on RedHat OpenShift.
         overrideFile = getFileOrFolder("gitblit.properties");
         File overrideFile = getFileOrFolder("gitblit.properties");
         if (!overrideFile.getPath().equals("gitblit.properties")) {
            webxmlSettings.applyOverrides(overrideFile);
         }
@@ -1816,5 +1912,6 @@
   public void contextDestroyed(ServletContextEvent contextEvent) {
      logger.info("Gitblit context destroyed by servlet container.");
      scheduledExecutor.shutdownNow();
      luceneExecutor.close();
   }
}