Merge pull request #7 from plm/protect_refs_hook

James Moger

2012-02-16 91780e2e17a8020872c8da2d8941114e098ef2a4

 src/com/gitblit/GitBlit.java

@@ -75,6 +75,7 @@
import com.gitblit.models.SettingModel;

import com.gitblit.models.TeamModel;

import com.gitblit.models.UserModel;

import com.gitblit.utils.ArrayUtils;

import com.gitblit.utils.ByteFormat;

import com.gitblit.utils.FederationUtils;

import com.gitblit.utils.JGitUtils;

@@ -121,8 +122,6 @@
   private ServletContext servletContext;



   private File repositoriesFolder;



   private boolean exportAll = true;



   private IUserService userService;



@@ -451,7 +450,7 @@
      List<String> names = new ArrayList<String>(userService.getAllUsernames());

      return names;

   }

	


   /**

    * Returns the list of all users available to the login service.

    * 

@@ -545,7 +544,7 @@
      List<String> teams = new ArrayList<String>(userService.getAllTeamNames());

      return teams;

   }

	


   /**

    * Returns the list of available teams that a user or repository may be

    * assigned to.

@@ -641,7 +640,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));

   }



@@ -652,21 +652,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,6 +759,7 @@
      model.name = repositoryName;

      model.hasCommits = JGitUtils.hasCommits(r);

      model.lastChange = JGitUtils.getLastChange(r, null);

      model.isBare = r.isBare();

      StoredConfig config = JGitUtils.readConfig(r);

      if (config != null) {

         model.description = getConfig(config, "description", "");

@@ -768,6 +786,8 @@
         model.mailingLists = new ArrayList<String>(Arrays.asList(config.getStringList(

               "gitblit", null, "mailingList")));

      }

      model.HEAD = JGitUtils.getHEADRef(r);

      model.availableRefs = JGitUtils.getAvailableHeadTargets(r);

      r.close();

      return model;

   }

@@ -963,6 +983,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();

      }

   }

@@ -991,14 +1023,14 @@
      config.setString("gitblit", null, "federationStrategy",

            repository.federationStrategy.name());

      config.setBoolean("gitblit", null, "isFederated", repository.isFederated);

      if (repository.preReceiveScripts != null) {

      if (!ArrayUtils.isEmpty(repository.preReceiveScripts)) {

         config.setStringList("gitblit", null, "preReceiveScript", repository.preReceiveScripts);

      }

      if (repository.postReceiveScripts != null) {

      if (!ArrayUtils.isEmpty(repository.postReceiveScripts)) {

         config.setStringList("gitblit", null, "postReceiveScript",

               repository.postReceiveScripts);

      }

      if (repository.mailingLists != null) {

      if (!ArrayUtils.isEmpty(repository.mailingLists)) {

         config.setStringList("gitblit", null, "mailingList", repository.mailingLists);

      }

      try {

@@ -1734,16 +1766,14 @@
      this.settings = settings;

      repositoriesFolder = getRepositoriesFolder();

      logger.info("Git repositories folder " + repositoriesFolder.getAbsolutePath());

      repositoryResolver = new FileResolver<Void>(repositoriesFolder, exportAll);

      repositoryResolver = new FileResolver<Void>(repositoriesFolder, true);

      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();

      }

@@ -1787,6 +1817,21 @@
            webxmlSettings.applyOverrides(overrideFile);

         }

         configureContext(webxmlSettings, true);



         // Copy the included scripts to the configured groovy folder

         File localScripts = getFileOrFolder(Keys.groovy.scriptsFolder, "groovy");

         if (!localScripts.exists()) {

            File includedScripts = new File(context.getRealPath("/WEB-INF/groovy"));

            if (!includedScripts.equals(localScripts)) {

               try {

                  com.gitblit.utils.FileUtils.copy(localScripts, includedScripts.listFiles());

               } catch (IOException e) {

                  logger.error(MessageFormat.format(

                        "Failed to copy included Groovy scripts from {0} to {1}",

                        includedScripts, localScripts));

               }

            }

         }

      }



      serverStatus.servletContainer = servletContext.getServerInfo();