James Moger
2011-12-22 e6935876b97a63bae2ec087b4fc390c832aef155
src/com/gitblit/GitBlit.java
@@ -25,11 +25,14 @@
import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
@@ -69,6 +72,7 @@
import com.gitblit.models.ServerSettings;
import com.gitblit.models.ServerStatus;
import com.gitblit.models.SettingModel;
import com.gitblit.models.TeamModel;
import com.gitblit.models.UserModel;
import com.gitblit.utils.ByteFormat;
import com.gitblit.utils.FederationUtils;
@@ -244,6 +248,64 @@
    */
   public static boolean isDebugMode() {
      return self().settings.getBoolean(Keys.web.debugMode, false);
   }
   /**
    * Returns the file object for the specified configuration key.
    *
    * @return the file
    */
   public static File getFileOrFolder(String key, String defaultFileOrFolder) {
      String fileOrFolder = GitBlit.getString(key, defaultFileOrFolder);
      return getFileOrFolder(fileOrFolder);
   }
   /**
    * Returns the file object which may have it's base-path determined by
    * environment variables for running on a cloud hosting service. All Gitblit
    * file or folder retrievals are (at least initially) funneled through this
    * method so it is the correct point to globally override/alter filesystem
    * access based on environment or some other indicator.
    *
    * @return the file
    */
   public static File getFileOrFolder(String fileOrFolder) {
      String openShift = System.getenv("OPENSHIFT_DATA_DIR");
      if (!StringUtils.isEmpty(openShift)) {
         // running on RedHat OpenShift
         return new File(openShift, fileOrFolder);
      }
      return new File(fileOrFolder);
   }
   /**
    * Returns the path of the repositories folder. This method checks to see if
    * Gitblit is running on a cloud service and may return an adjusted path.
    *
    * @return the repositories folder path
    */
   public static File getRepositoriesFolder() {
      return getFileOrFolder(Keys.git.repositoriesFolder, "git");
   }
   /**
    * Returns the path of the proposals folder. This method checks to see if
    * Gitblit is running on a cloud service and may return an adjusted path.
    *
    * @return the proposals folder path
    */
   public static File getProposalsFolder() {
      return getFileOrFolder(Keys.federation.proposalsFolder, "proposals");
   }
   /**
    * Returns the path of the Groovy folder. This method checks to see if
    * Gitblit is running on a cloud service and may return an adjusted path.
    *
    * @return the Groovy scripts folder path
    */
   public static File getGroovyScriptsFolder() {
      return getFileOrFolder(Keys.groovy.scriptsFolder, "groovy");
   }
   /**
@@ -463,6 +525,85 @@
   }
   /**
    * Returns the list of available teams that a user or repository may be
    * assigned to.
    *
    * @return the list of teams
    */
   public List<String> getAllTeamnames() {
      List<String> teams = new ArrayList<String>(userService.getAllTeamNames());
      Collections.sort(teams);
      return teams;
   }
   /**
    * Returns the TeamModel object for the specified name.
    *
    * @param teamname
    * @return a TeamModel object or null
    */
   public TeamModel getTeamModel(String teamname) {
      return userService.getTeamModel(teamname);
   }
   /**
    * Returns the list of all teams who are allowed to bypass the access
    * restriction placed on the specified repository.
    *
    * @see IUserService.getTeamnamesForRepositoryRole(String)
    * @param repository
    * @return list of all teamnames that can bypass the access restriction
    */
   public List<String> getRepositoryTeams(RepositoryModel repository) {
      return userService.getTeamnamesForRepositoryRole(repository.name);
   }
   /**
    * Sets the list of all uses who are allowed to bypass the access
    * restriction placed on the specified repository.
    *
    * @see IUserService.setTeamnamesForRepositoryRole(String, List<String>)
    * @param repository
    * @param teamnames
    * @return true if successful
    */
   public boolean setRepositoryTeams(RepositoryModel repository, List<String> repositoryTeams) {
      return userService.setTeamnamesForRepositoryRole(repository.name, repositoryTeams);
   }
   /**
    * Updates the TeamModel object for the specified name.
    *
    * @param teamname
    * @param team
    * @param isCreate
    */
   public void updateTeamModel(String teamname, TeamModel team, boolean isCreate)
         throws GitBlitException {
      if (!teamname.equalsIgnoreCase(team.name)) {
         if (userService.getTeamModel(team.name) != null) {
            throw new GitBlitException(MessageFormat.format(
                  "Failed to rename ''{0}'' because ''{1}'' already exists.", teamname,
                  team.name));
         }
      }
      if (!userService.updateTeamModel(teamname, team)) {
         throw new GitBlitException(isCreate ? "Failed to add team!" : "Failed to update team!");
      }
   }
   /**
    * Delete the team object with the specified teamname
    *
    * @see IUserService.deleteTeam(String)
    * @param teamname
    * @return true if successful
    */
   public boolean deleteTeam(String teamname) {
      return userService.deleteTeam(teamname);
   }
   /**
    * Clears all the cached data for the specified repository.
    * 
    * @param repositoryName
@@ -555,7 +696,7 @@
         return null;
      }
      if (model.accessRestriction.atLeast(AccessRestrictionType.VIEW)) {
         if (user != null && user.canAccessRepository(model.name)) {
         if (user != null && user.canAccessRepository(model)) {
            return model;
         }
         return null;
@@ -599,6 +740,12 @@
               "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")));
         model.postReceiveScripts = new ArrayList<String>(Arrays.asList(config.getStringList(
               "gitblit", null, "postReceiveScript")));
         model.mailingLists = new ArrayList<String>(Arrays.asList(config.getStringList(
               "gitblit", null, "mailingList")));
      }
      r.close();
      return model;
@@ -818,6 +965,16 @@
      config.setString("gitblit", null, "federationStrategy",
            repository.federationStrategy.name());
      config.setBoolean("gitblit", null, "isFederated", repository.isFederated);
      if (repository.preReceiveScripts != null) {
         config.setStringList("gitblit", null, "preReceiveScript", repository.preReceiveScripts);
      }
      if (repository.postReceiveScripts != null) {
         config.setStringList("gitblit", null, "postReceiveScript",
               repository.postReceiveScripts);
      }
      if (repository.mailingLists != null) {
         config.setStringList("gitblit", null, "mailingList", repository.mailingLists);
      }
      try {
         config.save();
      } catch (IOException e) {
@@ -1067,6 +1224,7 @@
      case PULL_REPOSITORIES:
         return token.equals(all) || token.equals(unr) || token.equals(jur);
      case PULL_USERS:
      case PULL_TEAMS:
         return token.equals(all) || token.equals(unr);
      case PULL_SETTINGS:
         return token.equals(all);
@@ -1120,8 +1278,7 @@
      try {
         // make the proposals folder
         File proposalsFolder = new File(getString(Keys.federation.proposalsFolder, "proposals")
               .trim());
         File proposalsFolder = getProposalsFolder();
         proposalsFolder.mkdirs();
         // cache json to a file
@@ -1153,7 +1310,7 @@
    */
   public List<FederationProposal> getPendingFederationProposals() {
      List<FederationProposal> list = new ArrayList<FederationProposal>();
      File folder = new File(getString(Keys.federation.proposalsFolder, "proposals").trim());
      File folder = getProposalsFolder();
      if (folder.exists()) {
         File[] files = folder.listFiles(new FileFilter() {
            @Override
@@ -1276,9 +1433,51 @@
    * @return true if the proposal was deleted
    */
   public boolean deletePendingFederationProposal(FederationProposal proposal) {
      File folder = new File(getString(Keys.federation.proposalsFolder, "proposals").trim());
      File folder = getProposalsFolder();
      File file = new File(folder, proposal.token + Constants.PROPOSAL_EXT);
      return file.delete();
   }
   /**
    * Returns the list of all available Groovy push hook scripts that are not
    * already specified globally for all repositories. Script files must have
    * .groovy extension
    *
    * @return list of available hook scripts
    */
   public List<String> getAvailableScripts() {
      File groovyFolder = getGroovyScriptsFolder();
      File[] files = groovyFolder.listFiles(new FileFilter() {
         @Override
         public boolean accept(File pathname) {
            return pathname.isFile() && pathname.getName().endsWith(".groovy");
         }
      });
      Set<String> globals = new HashSet<String>();
      String[] keys = { Keys.groovy.preReceiveScripts, Keys.groovy.postReceiveScripts };
      for (String key : keys) {
         for (String script : getStrings(key)) {
            if (script.endsWith(".groovy")) {
               globals.add(script.substring(0, script.lastIndexOf('.')));
            } else {
               globals.add(script);
            }
         }
      }
      // create list of available scripts by excluding scripts that are
      // globally specified
      List<String> scripts = new ArrayList<String>();
      if (files != null) {
         for (File file : files) {
            String script = file.getName().substring(0, file.getName().lastIndexOf('.'));
            if (!globals.contains(script)) {
               scripts.add(script);
            }
         }
      }
      return scripts;
   }
   /**
@@ -1287,9 +1486,40 @@
    * @param subject
    * @param message
    */
   public void notifyAdministrators(String subject, String message) {
   public void sendMailToAdministrators(String subject, String message) {
      try {
         Message mail = mailExecutor.createMessageForAdministrators();
         if (mail != null) {
            mail.setSubject(subject);
            mail.setText(message);
            mailExecutor.queue(mail);
         }
      } catch (MessagingException e) {
         logger.error("Messaging error", e);
      }
   }
   /**
    * Notify users by email of something.
    *
    * @param subject
    * @param message
    * @param toAddresses
    */
   public void sendMail(String subject, String message, Collection<String> toAddresses) {
      this.sendMail(subject, message, toAddresses.toArray(new String[0]));
   }
   /**
    * Notify users by email of something.
    *
    * @param subject
    * @param message
    * @param toAddresses
    */
   public void sendMail(String subject, String message, String... toAddresses) {
      try {
         Message mail = mailExecutor.createMessage(toAddresses);
         if (mail != null) {
            mail.setSubject(subject);
            mail.setText(message);
@@ -1313,6 +1543,7 @@
            setting.currentValue = settings.getString(key, "");
         }
      }
      settingsModel.pushScripts = getAvailableScripts();
      return settingsModel;
   }
@@ -1391,7 +1622,7 @@
   public void configureContext(IStoredSettings settings, boolean startFederation) {
      logger.info("Reading configuration from " + settings.toString());
      this.settings = settings;
      repositoriesFolder = new File(settings.getString(Keys.git.repositoriesFolder, "git"));
      repositoriesFolder = getRepositoriesFolder();
      logger.info("Git repositories folder " + repositoriesFolder.getAbsolutePath());
      repositoryResolver = new FileResolver<Void>(repositoriesFolder, exportAll);
      serverStatus = new ServerStatus(isGO());
@@ -1404,18 +1635,7 @@
            loginService = (IUserService) realmClass.newInstance();
         }
      } catch (Throwable t) {
         // not a login service class or class could not be instantiated.
         // try to use default file login service
         File realmFile = new File(realm);
         if (!realmFile.exists()) {
            try {
               realmFile.createNewFile();
            } catch (IOException x) {
               logger.error(
                     MessageFormat.format("COULD NOT CREATE REALM FILE {0}!", realmFile), x);
            }
         }
         loginService = new FileUserService(realmFile);
         loginService = new GitblitUserService();
      }
      setUserService(loginService);
      mailExecutor = new MailExecutor(settings);
@@ -1441,7 +1661,21 @@
      settingsModel = loadSettingModels();
      if (settings == null) {
         // Gitblit WAR is running in a servlet container
         WebXmlSettings webxmlSettings = new WebXmlSettings(contextEvent.getServletContext());
         ServletContext context = contextEvent.getServletContext();
         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);
         }
         // 0.8.0 gitblit.properties file located outside the deployed war
         // folder lie, for example, on RedHat OpenShift.
         overrideFile = getFileOrFolder("gitblit.properties");
         if (!overrideFile.getPath().equals("gitblit.properties")) {
            webxmlSettings.applyOverrides(overrideFile);
         }
         configureContext(webxmlSettings, true);
      }