James Moger
2012-10-13 c658df9e87d65b08d5482cf04489cb0532ff83dd
src/com/gitblit/GitBlit.java
@@ -57,23 +57,19 @@
import javax.servlet.http.Cookie;
import org.apache.wicket.protocol.http.WebResponse;
import org.eclipse.jgit.errors.RepositoryNotFoundException;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.lib.RepositoryCache;
import org.eclipse.jgit.lib.RepositoryCache.FileKey;
import org.eclipse.jgit.lib.StoredConfig;
import org.eclipse.jgit.storage.file.FileBasedConfig;
import org.eclipse.jgit.storage.file.WindowCache;
import org.eclipse.jgit.storage.file.WindowCacheConfig;
import org.eclipse.jgit.transport.ServiceMayNotContinueException;
import org.eclipse.jgit.transport.resolver.FileResolver;
import org.eclipse.jgit.transport.resolver.RepositoryResolver;
import org.eclipse.jgit.transport.resolver.ServiceNotAuthorizedException;
import org.eclipse.jgit.transport.resolver.ServiceNotEnabledException;
import org.eclipse.jgit.util.FS;
import org.eclipse.jgit.util.FileUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.gitblit.Constants.AccessPermission;
import com.gitblit.Constants.AccessRestrictionType;
import com.gitblit.Constants.AuthorizationControl;
import com.gitblit.Constants.FederationRequest;
@@ -141,8 +137,6 @@
   private final Map<String, ProjectModel> projectCache = new ConcurrentHashMap<String, ProjectModel>();
   
   private final AtomicReference<String> repositoryListSettingsChecksum = new AtomicReference<String>("");
   private RepositoryResolver<Void> repositoryResolver;
   private ServletContext servletContext;
@@ -625,6 +619,7 @@
    * @param usernames
    * @return true if successful
    */
   @Deprecated
   public boolean setRepositoryUsers(RepositoryModel repository, List<String> repositoryUsers) {
      return userService.setUsernamesForRepositoryRole(repository.name, repositoryUsers);
   }
@@ -646,6 +641,22 @@
            throw new GitBlitException(MessageFormat.format(
                  "Failed to rename ''{0}'' because ''{1}'' already exists.", username,
                  user.username));
         }
         // rename repositories and owner fields for all repositories
         for (RepositoryModel model : getRepositoryModels(user)) {
            if (model.isUsersPersonalRepository(username)) {
               // personal repository
               model.owner = user.username;
               String oldRepositoryName = model.name;
               model.name = "~" + user.username + model.name.substring(model.projectPath.length());
               model.projectPath = "~" + user.username;
               updateRepositoryModel(oldRepositoryName, model, false);
            } else if (model.isOwner(username)) {
               // common/shared repo
               model.owner = user.username;
               updateRepositoryModel(model.name, model, false);
            }
         }
      }
      if (!userService.updateUserModel(username, user)) {
@@ -706,6 +717,7 @@
    * @param teamnames
    * @return true if successful
    */
   @Deprecated
   public boolean setRepositoryTeams(RepositoryModel repository, List<String> repositoryTeams) {
      return userService.setTeamnamesForRepositoryRole(repository.name, repositoryTeams);
   }
@@ -746,17 +758,17 @@
    * Adds the repository to the list of cached repositories if Gitblit is
    * configured to cache the repository list.
    * 
    * @param name
    * @param model
    */
   private void addToCachedRepositoryList(String name, RepositoryModel model) {
   private void addToCachedRepositoryList(RepositoryModel model) {
      if (settings.getBoolean(Keys.git.cacheRepositoryList, true)) {
         repositoryListCache.put(name, model);
         repositoryListCache.put(model.name, model);
         
         // update the fork origin repository with this repository clone
         if (!StringUtils.isEmpty(model.originRepository)) {
            if (repositoryListCache.containsKey(model.originRepository)) {
               RepositoryModel origin = repositoryListCache.get(model.originRepository);
               origin.addFork(name);
               origin.addFork(model.name);
            }
         }
      }
@@ -895,32 +907,18 @@
    * @return repository or null
    */
   public Repository getRepository(String repositoryName, boolean logError) {
      File dir = FileKey.resolve(new File(repositoriesFolder, repositoryName), FS.DETECTED);
      if (dir == null)
         return null;
      Repository r = null;
      try {
         r = repositoryResolver.open(null, repositoryName);
      } catch (RepositoryNotFoundException e) {
         r = null;
         FileKey key = FileKey.exact(dir, FS.DETECTED);
         r = RepositoryCache.open(key, true);
      } catch (IOException e) {
         if (logError) {
            logger.error("GitBlit.getRepository(String) failed to find "
                  + new File(repositoriesFolder, repositoryName).getAbsolutePath());
         }
      } catch (ServiceNotAuthorizedException e) {
         r = null;
         if (logError) {
            logger.error("GitBlit.getRepository(String) failed to find "
                  + new File(repositoriesFolder, repositoryName).getAbsolutePath(), e);
         }
      } catch (ServiceNotEnabledException e) {
         r = null;
         if (logError) {
            logger.error("GitBlit.getRepository(String) failed to find "
                  + new File(repositoriesFolder, repositoryName).getAbsolutePath(), e);
         }
      } catch (ServiceMayNotContinueException e) {
         r = null;
         if (logError) {
            logger.error("GitBlit.getRepository(String) failed to find "
                  + new File(repositoriesFolder, repositoryName).getAbsolutePath(), e);
         }
      }
      return r;
@@ -978,14 +976,13 @@
      if (model == null) {
         return null;
      }
      if (model.accessRestriction.atLeast(AccessRestrictionType.VIEW)) {
         if (user != null && user.canAccessRepository(model)) {
            return model;
         }
         return null;
      } else {
      if (user == null) {
         user = UserModel.ANONYMOUS;
      }
      if (user.canView(model)) {
         return model;
      }
      return null;
   }
   /**
@@ -1001,7 +998,7 @@
         if (model == null) {
            return null;
         }
         addToCachedRepositoryList(repositoryName, model);
         addToCachedRepositoryList(model);
         return model;
      }
      
@@ -1023,7 +1020,7 @@
         logger.info(MessageFormat.format("Config for \"{0}\" has changed. Reloading model and updating cache.", repositoryName));
         model = loadRepositoryModel(repositoryName);
         removeFromCachedRepositoryList(repositoryName);
         addToCachedRepositoryList(repositoryName, model);
         addToCachedRepositoryList(model);
      } else {
         // update a few repository parameters 
         if (!model.hasCommits) {
@@ -1236,15 +1233,16 @@
         return null;
      }
      RepositoryModel model = new RepositoryModel();
      model.name = repositoryName;
      model.isBare = r.isBare();
      File basePath = getFileOrFolder(Keys.git.repositoriesFolder, "git");
      if (model.isBare) {
         model.name = com.gitblit.utils.FileUtils.getRelativePath(basePath, r.getDirectory());
      } else {
         model.name = com.gitblit.utils.FileUtils.getRelativePath(basePath, r.getDirectory().getParentFile());
      }
      model.hasCommits = JGitUtils.hasCommits(r);
      model.lastChange = JGitUtils.getLastChange(r);
      model.isBare = r.isBare();
      if (repositoryName.indexOf('/') == -1) {
         model.projectPath = "";
      } else {
         model.projectPath = repositoryName.substring(0, repositoryName.indexOf('/'));
      }
      model.projectPath = StringUtils.getFirstPathElement(repositoryName);
      
      StoredConfig config = r.getConfig();
      boolean hasOrigin = !StringUtils.isEmpty(config.getString("remote", "origin", "url"));
@@ -1259,6 +1257,7 @@
               "accessRestriction", settings.getString(Keys.git.defaultAccessRestriction, null)));
         model.authorizationControl = AuthorizationControl.fromName(getConfig(config,
               "authorizationControl", settings.getString(Keys.git.defaultAuthorizationControl, null)));
         model.verifyCommitter = getConfig(config, "verifyCommitter", false);
         model.showRemoteBranches = getConfig(config, "showRemoteBranches", hasOrigin);
         model.isFrozen = getConfig(config, "isFrozen", false);
         model.showReadme = getConfig(config, "showReadme", false);
@@ -1270,6 +1269,9 @@
               Constants.CONFIG_GITBLIT, null, "federationSets")));
         model.isFederated = getConfig(config, "isFederated", false);
         model.origin = config.getString("remote", "origin", "url");
         if (model.origin != null) {
            model.origin = model.origin.replace('\\', '/');
         }
         model.preReceiveScripts = new ArrayList<String>(Arrays.asList(config.getStringList(
               Constants.CONFIG_GITBLIT, null, "preReceiveScript")));
         model.postReceiveScripts = new ArrayList<String>(Arrays.asList(config.getStringList(
@@ -1462,6 +1464,11 @@
    */
   private void closeRepository(String repositoryName) {
      Repository repository = getRepository(repositoryName);
      if (repository == null) {
         return;
      }
      RepositoryCache.close(repository);
      // assume 2 uses in case reflection fails
      int uses = 2;
      try {
@@ -1558,6 +1565,13 @@
   public void updateRepositoryModel(String repositoryName, RepositoryModel repository,
         boolean isCreate) throws GitBlitException {
      Repository r = null;
      String projectPath = StringUtils.getFirstPathElement(repository.name);
      if (!StringUtils.isEmpty(projectPath)) {
         if (projectPath.equalsIgnoreCase(getString(Keys.web.repositoryRootGroupName, "main"))) {
            // strip leading group name
            repository.name = repository.name.substring(projectPath.length() + 1);
         }
      }
      if (isCreate) {
         // ensure created repository name ends with .git
         if (!repository.name.toLowerCase().endsWith(org.eclipse.jgit.lib.Constants.DOT_GIT_EXT)) {
@@ -1625,6 +1639,14 @@
                  rf.close();
               }
            }
            // remove this repository from any origin model's fork list
            if (!StringUtils.isEmpty(repository.originRepository)) {
               RepositoryModel origin = repositoryListCache.get(repository.originRepository);
               if (origin != null && !ArrayUtils.isEmpty(origin.forks)) {
                  origin.forks.remove(repositoryName);
               }
            }
            // clear the cache
            clearRepositoryMetadataCache(repositoryName);
@@ -1633,17 +1655,7 @@
         // load repository
         logger.info("edit repository " + repository.name);
         try {
            r = repositoryResolver.open(null, repository.name);
         } catch (RepositoryNotFoundException e) {
            logger.error("Repository not found", e);
         } catch (ServiceNotAuthorizedException e) {
            logger.error("Service not authorized", e);
         } catch (ServiceNotEnabledException e) {
            logger.error("Service not enabled", e);
         } catch (ServiceMayNotContinueException e) {
            logger.error("Service may not continue", e);
         }
         r = getRepository(repository.name);
      }
      // update settings
@@ -1667,7 +1679,7 @@
      // update repository cache
      removeFromCachedRepositoryList(repositoryName);
      // model will actually be replaced on next load because config is stale
      addToCachedRepositoryList(repository.name, repository);
      addToCachedRepositoryList(repository);
   }
   
   /**
@@ -1687,6 +1699,7 @@
      config.setBoolean(Constants.CONFIG_GITBLIT, null, "allowForks", repository.allowForks);
      config.setString(Constants.CONFIG_GITBLIT, null, "accessRestriction", repository.accessRestriction.name());
      config.setString(Constants.CONFIG_GITBLIT, null, "authorizationControl", repository.authorizationControl.name());
      config.setBoolean(Constants.CONFIG_GITBLIT, null, "verifyCommitter", repository.verifyCommitter);
      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);
@@ -1762,7 +1775,7 @@
         clearRepositoryMetadataCache(repositoryName);
         
         RepositoryModel model = removeFromCachedRepositoryList(repositoryName);
         if (!ArrayUtils.isEmpty(model.forks)) {
         if (model != null && !ArrayUtils.isEmpty(model.forks)) {
            resetRepositoryListCache();
         }
@@ -2491,7 +2504,6 @@
      this.settings = settings;
      repositoriesFolder = getRepositoriesFolder();
      logger.info("Git repositories folder " + repositoriesFolder.getAbsolutePath());
      repositoryResolver = new FileResolver<Void>(repositoriesFolder, true);
      // calculate repository list settings checksum for future config changes
      repositoryListSettingsChecksum.set(getRepositoryListSettingsChecksum());
@@ -2653,29 +2665,49 @@
      // create a Gitblit repository model for the clone
      RepositoryModel cloneModel = repository.cloneAs(cloneName);
      // owner has REWIND/RW+ permissions
      cloneModel.owner = user.username;
      updateRepositoryModel(cloneName, cloneModel, false);
      if (AuthorizationControl.NAMED.equals(cloneModel.authorizationControl)) {
         // add the owner of the source repository to the clone's access list
         if (!StringUtils.isEmpty(repository.owner)) {
            UserModel owner = getUserModel(repository.owner);
            if (owner != null) {
               owner.repositories.add(cloneName);
               updateUserModel(owner.username, owner, false);
            }
      // add the owner of the source repository to the clone's access list
      if (!StringUtils.isEmpty(repository.owner)) {
         UserModel originOwner = getUserModel(repository.owner);
         if (originOwner != null) {
            originOwner.setRepositoryPermission(cloneName, AccessPermission.CLONE);
            updateUserModel(originOwner.username, originOwner, false);
         }
         // inherit origin's access lists
         List<String> users = getRepositoryUsers(repository);
         setRepositoryUsers(cloneModel, users);
         List<String> teams = getRepositoryTeams(repository);
         setRepositoryTeams(cloneModel, teams);
      }
      // grant origin's user list clone permission to fork
      List<String> users = getRepositoryUsers(repository);
      List<UserModel> cloneUsers = new ArrayList<UserModel>();
      for (String name : users) {
         if (!name.equalsIgnoreCase(user.username)) {
            UserModel cloneUser = getUserModel(name);
            if (cloneUser.canClone(repository)) {
               // origin user can clone origin, grant clone access to fork
               cloneUser.setRepositoryPermission(cloneName, AccessPermission.CLONE);
            }
            cloneUsers.add(cloneUser);
         }
      }
      userService.updateUserModels(cloneUsers);
      // grant origin's team list clone permission to fork
      List<String> teams = getRepositoryTeams(repository);
      List<TeamModel> cloneTeams = new ArrayList<TeamModel>();
      for (String name : teams) {
         TeamModel cloneTeam = getTeamModel(name);
         if (cloneTeam.canClone(repository)) {
            // origin team can clone origin, grant clone access to fork
            cloneTeam.setRepositoryPermission(cloneName, AccessPermission.CLONE);
         }
         cloneTeams.add(cloneTeam);
      }
      userService.updateTeamModels(cloneTeams);
      // add this clone to the cached model
      addToCachedRepositoryList(cloneModel.name, cloneModel);
      addToCachedRepositoryList(cloneModel);
      return cloneModel;
   }
}