James Moger
2012-10-02 1f52c8b5f123b97fc631465479bc2855c12b9ee3
src/com/gitblit/GitBlit.java
@@ -22,6 +22,8 @@
import java.io.InputStream;
import java.io.InputStreamReader;
import java.lang.reflect.Field;
import java.net.URI;
import java.net.URISyntaxException;
import java.text.MessageFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
@@ -30,6 +32,7 @@
import java.util.Collections;
import java.util.Date;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.List;
@@ -79,6 +82,7 @@
import com.gitblit.models.FederationModel;
import com.gitblit.models.FederationProposal;
import com.gitblit.models.FederationSet;
import com.gitblit.models.ForkModel;
import com.gitblit.models.Metric;
import com.gitblit.models.ProjectModel;
import com.gitblit.models.RepositoryModel;
@@ -747,6 +751,14 @@
   private void addToCachedRepositoryList(String name, RepositoryModel model) {
      if (settings.getBoolean(Keys.git.cacheRepositoryList, true)) {
         repositoryListCache.put(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);
            }
         }
      }
   }
   
@@ -754,12 +766,13 @@
    * Removes the repository from the list of cached repositories.
    * 
    * @param name
    * @return the model being removed
    */
   private void removeFromCachedRepositoryList(String name) {
   private RepositoryModel removeFromCachedRepositoryList(String name) {
      if (StringUtils.isEmpty(name)) {
         return;
         return null;
      }
      repositoryListCache.remove(name);
      return repositoryListCache.remove(name);
   }
   /**
@@ -988,7 +1001,7 @@
         if (model == null) {
            return null;
         }
         addToCachedRepositoryList(repositoryName, model);
         addToCachedRepositoryList(repositoryName, model);
         return model;
      }
      
@@ -1034,7 +1047,7 @@
    * @return project config map
    */
   private Map<String, ProjectModel> getProjectConfigs() {
      if (projectConfigs.isOutdated()) {
      if (projectCache.isEmpty() || projectConfigs.isOutdated()) {
         
         try {
            projectConfigs.load();
@@ -1077,9 +1090,10 @@
    * Returns a list of project models for the user.
    * 
    * @param user
    * @param includeUsers
    * @return list of projects that are accessible to the user
    */
   public List<ProjectModel> getProjectModels(UserModel user) {
   public List<ProjectModel> getProjectModels(UserModel user, boolean includeUsers) {
      Map<String, ProjectModel> configs = getProjectConfigs();
      // per-user project lists, this accounts for security and visibility
@@ -1104,10 +1118,25 @@
      }
      
      // sort projects, root project first
      List<ProjectModel> projects = new ArrayList<ProjectModel>(map.values());
      Collections.sort(projects);
      projects.remove(map.get(""));
      projects.add(0, map.get(""));
      List<ProjectModel> projects;
      if (includeUsers) {
         // all projects
         projects = new ArrayList<ProjectModel>(map.values());
         Collections.sort(projects);
         projects.remove(map.get(""));
         projects.add(0, map.get(""));
      } else {
         // all non-user projects
         projects = new ArrayList<ProjectModel>();
         ProjectModel root = map.remove("");
         for (ProjectModel model : map.values()) {
            if (!model.isUserProject()) {
               projects.add(model);
            }
         }
         Collections.sort(projects);
         projects.add(0, root);
      }
      return projects;
   }
   
@@ -1119,7 +1148,7 @@
    * @return a project model, or null if it does not exist
    */
   public ProjectModel getProjectModel(String name, UserModel user) {
      for (ProjectModel project : getProjectModels(user)) {
      for (ProjectModel project : getProjectModels(user, true)) {
         if (project.name.equalsIgnoreCase(name)) {
            return project;
         }
@@ -1137,15 +1166,37 @@
      Map<String, ProjectModel> configs = getProjectConfigs();
      ProjectModel project = configs.get(name.toLowerCase());
      if (project == null) {
         return null;
      }
      // clone the object
      project = DeepCopier.copy(project);
      String folder = name.toLowerCase() + "/";
      for (String repository : getRepositoryList()) {
         if (repository.toLowerCase().startsWith(folder)) {
            project.addRepository(repository);
         project = new ProjectModel(name);
         if (name.length() > 0 && name.charAt(0) == '~') {
            UserModel user = getUserModel(name.substring(1));
            if (user != null) {
               project.title = user.getDisplayName();
               project.description = "personal repositories";
            }
         }
      } else {
         // clone the object
         project = DeepCopier.copy(project);
      }
      if (StringUtils.isEmpty(name)) {
         // get root repositories
         for (String repository : getRepositoryList()) {
            if (repository.indexOf('/') == -1) {
               project.addRepository(repository);
            }
         }
      } else {
         // get repositories in subfolder
         String folder = name.toLowerCase() + "/";
         for (String repository : getRepositoryList()) {
            if (repository.toLowerCase().startsWith(folder)) {
               project.addRepository(repository);
            }
         }
      }
      if (project.repositories.size() == 0) {
         // no repositories == no project
         return null;
      }
      return project;
   }
@@ -1189,18 +1240,26 @@
      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('/'));
      }
      
      StoredConfig config = r.getConfig();
      boolean hasOrigin = !StringUtils.isEmpty(config.getString("remote", "origin", "url"));
      if (config != null) {
         model.description = getConfig(config, "description", "");
         model.owner = getConfig(config, "owner", "");
         model.useTickets = getConfig(config, "useTickets", false);
         model.useDocs = getConfig(config, "useDocs", false);
         model.allowForks = getConfig(config, "allowForks", true);
         model.accessRestriction = AccessRestrictionType.fromName(getConfig(config,
               "accessRestriction", settings.getString(Keys.git.defaultAccessRestriction, null)));
         model.authorizationControl = AuthorizationControl.fromName(getConfig(config,
               "authorizationControl", settings.getString(Keys.git.defaultAuthorizationControl, null)));
         model.showRemoteBranches = getConfig(config, "showRemoteBranches", false);
         model.showRemoteBranches = getConfig(config, "showRemoteBranches", hasOrigin);
         model.isFrozen = getConfig(config, "isFrozen", false);
         model.showReadme = getConfig(config, "showReadme", false);
         model.skipSizeCalculation = getConfig(config, "skipSizeCalculation", false);
@@ -1229,6 +1288,23 @@
      model.HEAD = JGitUtils.getHEADRef(r);
      model.availableRefs = JGitUtils.getAvailableHeadTargets(r);
      r.close();
      if (model.origin != null && model.origin.startsWith("file://")) {
         // repository was cloned locally... perhaps as a fork
         try {
            File folder = new File(new URI(model.origin));
            String originRepo = com.gitblit.utils.FileUtils.getRelativePath(getRepositoriesFolder(), folder);
            if (!StringUtils.isEmpty(originRepo)) {
               // ensure origin still exists
               File repoFolder = new File(getRepositoriesFolder(), originRepo);
               if (repoFolder.exists()) {
                  model.originRepository = originRepo;
               }
            }
         } catch (URISyntaxException e) {
            logger.error("Failed to determine fork for " + model, e);
         }
      }
      return model;
   }
   
@@ -1250,6 +1326,113 @@
      }
      r.close();
      return true;
   }
   /**
    * Determines if the specified user has a fork of the specified origin
    * repository.
    *
    * @param username
    * @param origin
    * @return true the if the user has a fork
    */
   public boolean hasFork(String username, String origin) {
      return getFork(username, origin) != null;
   }
   /**
    * Gets the name of a user's fork of the specified origin
    * repository.
    *
    * @param username
    * @param origin
    * @return the name of the user's fork, null otherwise
    */
   public String getFork(String username, String origin) {
      String userProject = "~" + username.toLowerCase();
      if (settings.getBoolean(Keys.git.cacheRepositoryList, true)) {
         String userPath = userProject + "/";
         // collect all origin nodes in fork network
         Set<String> roots = new HashSet<String>();
         roots.add(origin);
         RepositoryModel originModel = repositoryListCache.get(origin);
         while (originModel != null) {
            if (!ArrayUtils.isEmpty(originModel.forks)) {
               for (String fork : originModel.forks) {
                  if (!fork.startsWith(userPath)) {
                     roots.add(fork);
                  }
               }
            }
            if (originModel.originRepository != null) {
               roots.add(originModel.originRepository);
               originModel = repositoryListCache.get(originModel.originRepository);
            } else {
               // break
               originModel = null;
            }
         }
         for (String repository : repositoryListCache.keySet()) {
            if (repository.toLowerCase().startsWith(userPath)) {
               RepositoryModel model = repositoryListCache.get(repository);
               if (!StringUtils.isEmpty(model.originRepository)) {
                  if (roots.contains(model.originRepository)) {
                     // user has a fork in this graph
                     return model.name;
                  }
               }
            }
         }
      } else {
         // not caching
         ProjectModel project = getProjectModel(userProject);
         for (String repository : project.repositories) {
            if (repository.toLowerCase().startsWith(userProject)) {
               RepositoryModel model = repositoryListCache.get(repository);
               if (model.originRepository.equalsIgnoreCase(origin)) {
                  // user has a fork
                  return model.name;
               }
            }
         }
      }
      // user does not have a fork
      return null;
   }
   /**
    * Returns the fork network for a repository by traversing up the fork graph
    * to discover the root and then down through all children of the root node.
    *
    * @param repository
    * @return a ForkModel
    */
   public ForkModel getForkNetwork(String repository) {
      if (settings.getBoolean(Keys.git.cacheRepositoryList, true)) {
         // find the root
         RepositoryModel model = repositoryListCache.get(repository);
         while (model.originRepository != null) {
            model = repositoryListCache.get(model.originRepository);
         }
         ForkModel root = getForkModel(model.name);
         return root;
      }
      return null;
   }
   private ForkModel getForkModel(String repository) {
      RepositoryModel model = repositoryListCache.get(repository);
      ForkModel fork = new ForkModel(model.originRepository, model.name);
      if (!ArrayUtils.isEmpty(model.forks)) {
         for (String aFork : model.forks) {
            ForkModel fm = getForkModel(aFork);
            fork.forks.add(fm);
         }
      }
      return fork;
   }
   /**
@@ -1425,9 +1608,27 @@
                     "Failed to rename repository permissions ''{0}'' to ''{1}''.",
                     repositoryName, repository.name));
            }
            // rename fork origins in their configs
            if (!ArrayUtils.isEmpty(repository.forks)) {
               for (String fork : repository.forks) {
                  Repository rf = getRepository(fork);
                  try {
                     StoredConfig config = rf.getConfig();
                     String origin = config.getString("remote", "origin", "url");
                     origin = origin.replace(repositoryName, repository.name);
                     config.setString("remote", "origin", "url", origin);
                     config.save();
                  } catch (Exception e) {
                     logger.error("Failed to update repository fork config for " + fork, e);
                  }
                  rf.close();
               }
            }
            // clear the cache
            clearRepositoryMetadataCache(repositoryName);
            repository.resetDisplayName();
         }
         // load repository
@@ -1483,6 +1684,7 @@
      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.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, "showRemoteBranches", repository.showRemoteBranches);
@@ -1558,7 +1760,11 @@
         closeRepository(repositoryName);
         // clear the repository cache
         clearRepositoryMetadataCache(repositoryName);
         removeFromCachedRepositoryList(repositoryName);
         RepositoryModel model = removeFromCachedRepositoryList(repositoryName);
         if (!ArrayUtils.isEmpty(model.forks)) {
            resetRepositoryListCache();
         }
         File folder = new File(repositoriesFolder, repositoryName);
         if (folder.exists() && folder.isDirectory()) {
@@ -2423,4 +2629,53 @@
      scheduledExecutor.shutdownNow();
      luceneExecutor.close();
   }
   /**
    * Creates a personal fork of the specified repository. The clone is view
    * restricted by default and the owner of the source repository is given
    * access to the clone.
    *
    * @param repository
    * @param user
    * @return the repository model of the fork, if successful
    * @throws GitBlitException
    */
   public RepositoryModel fork(RepositoryModel repository, UserModel user) throws GitBlitException {
      String cloneName = MessageFormat.format("~{0}/{1}.git", user.username, StringUtils.stripDotGit(StringUtils.getLastPathElement(repository.name)));
      String fromUrl = MessageFormat.format("file://{0}/{1}", repositoriesFolder.getAbsolutePath(), repository.name);
      // clone the repository
      try {
         JGitUtils.cloneRepository(repositoriesFolder, cloneName, fromUrl, true, null);
      } catch (Exception e) {
         throw new GitBlitException(e);
      }
      // create a Gitblit repository model for the clone
      RepositoryModel cloneModel = repository.cloneAs(cloneName);
      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);
            }
         }
         // inherit origin's access lists
         List<String> users = getRepositoryUsers(repository);
         setRepositoryUsers(cloneModel, users);
         List<String> teams = getRepositoryTeams(repository);
         setRepositoryTeams(cloneModel, teams);
      }
      // add this clone to the cached model
      addToCachedRepositoryList(cloneModel.name, cloneModel);
      return cloneModel;
   }
}