James Moger
2012-10-18 13417cf9c6eec555b51da49742e47939d2f5715b
src/com/gitblit/models/UserModel.java
@@ -17,8 +17,11 @@
import java.io.Serializable;
import java.security.Principal;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
@@ -26,6 +29,7 @@
import com.gitblit.Constants.AccessRestrictionType;
import com.gitblit.Constants.AuthorizationControl;
import com.gitblit.Constants.Unused;
import com.gitblit.utils.ArrayUtils;
import com.gitblit.utils.StringUtils;
/**
@@ -80,7 +84,7 @@
    */
   @Deprecated
   public boolean canAccessRepository(String repositoryName) {
      return canAdmin || repositories.contains(repositoryName.toLowerCase())
      return canAdmin() || repositories.contains(repositoryName.toLowerCase())
            || hasTeamAccess(repositoryName);
   }
@@ -90,7 +94,7 @@
      boolean isOwner = !StringUtils.isEmpty(repository.owner)
            && repository.owner.equals(username);
      boolean allowAuthenticated = isAuthenticated && AuthorizationControl.AUTHENTICATED.equals(repository.authorizationControl);
      return canAdmin || isOwner || repositories.contains(repository.name.toLowerCase())
      return canAdmin() || isOwner || repositories.contains(repository.name.toLowerCase())
            || hasTeamAccess(repository.name) || allowAuthenticated;
   }
@@ -124,6 +128,21 @@
   }
   
   /**
    * Returns a list of repository permissions for this user exclusive of
    * permissions inherited from team memberships.
    *
    * @return the user's list of permissions
    */
   public List<RepositoryAccessPermission> getRepositoryPermissions() {
      List<RepositoryAccessPermission> list = new ArrayList<RepositoryAccessPermission>();
      for (Map.Entry<String, AccessPermission> entry : permissions.entrySet()) {
         list.add(new RepositoryAccessPermission(entry.getKey(), entry.getValue()));
      }
      Collections.sort(list);
      return list;
   }
   /**
    * Returns true if the user has any type of specified access permission for
    * this repository.
    * 
@@ -132,7 +151,21 @@
    */
   public boolean hasRepositoryPermission(String name) {
      String repository = AccessPermission.repositoryFromRole(name).toLowerCase();
      return permissions.containsKey(repository) || repositories.contains(repository);
      if (permissions.containsKey(repository)) {
         // exact repository permission specified
         return true;
      } else {
         // search for regex permission match
         for (String key : permissions.keySet()) {
            if (name.matches(key)) {
               AccessPermission p = permissions.get(key);
               if (p != null) {
                  return true;
               }
            }
         }
      }
      return false;
   }
   
   /**
@@ -163,7 +196,7 @@
   }
   public AccessPermission getRepositoryPermission(RepositoryModel repository) {
      if (canAdmin || repository.isOwner(username) || repository.isUsersPersonalRepository(username)) {
      if (canAdmin() || repository.isOwner(username) || repository.isUsersPersonalRepository(username)) {
         return AccessPermission.REWIND;
      }
      if (AuthorizationControl.AUTHENTICATED.equals(repository.authorizationControl) && isAuthenticated) {
@@ -175,9 +208,20 @@
      // and the permissions of teams of which the user belongs
      AccessPermission permission = AccessPermission.NONE;
      if (permissions.containsKey(repository.name.toLowerCase())) {
         // exact repository permission specified
         AccessPermission p = permissions.get(repository.name.toLowerCase());
         if (p != null) {
            permission = p;
         }
      } else {
         // search for regex permission match
         for (String key : permissions.keySet()) {
            if (repository.name.matches(key)) {
               AccessPermission p = permissions.get(key);
               if (p != null) {
                  permission = p;
               }
            }
         }
      }
      
@@ -191,7 +235,7 @@
      return permission;
   }
   
   private boolean canAccess(RepositoryModel repository, AccessRestrictionType ifRestriction, AccessPermission requirePermission) {
   protected boolean canAccess(RepositoryModel repository, AccessRestrictionType ifRestriction, AccessPermission requirePermission) {
      if (repository.accessRestriction.atLeast(ifRestriction)) {
         AccessPermission permission = getRepositoryPermission(repository);
         return permission.atLeast(requirePermission);
@@ -240,24 +284,84 @@
         // can not fork your own repository
         return false;
      }
      if (canAdmin || repository.isOwner(username)) {
      if (canAdmin() || repository.isOwner(username)) {
         return true;
      }
      if (!repository.allowForks) {
         return false;
      }
      if (!isAuthenticated || !canFork) {
      if (!isAuthenticated || !canFork()) {
         return false;
      }
      return canClone(repository);
   }
   
   public boolean canDelete(RepositoryModel model) {
      return canAdmin || model.isUsersPersonalRepository(username);
      return canAdmin() || model.isUsersPersonalRepository(username);
   }
   
   public boolean canEdit(RepositoryModel model) {
      return canAdmin || model.isUsersPersonalRepository(username) || model.isOwner(username);
      return canAdmin() || model.isUsersPersonalRepository(username) || model.isOwner(username);
   }
   /**
    * This returns true if the user has fork privileges or the user has fork
    * privileges because of a team membership.
    *
    * @return true if the user can fork
    */
   public boolean canFork() {
      if (canFork) {
         return true;
      }
      if (!ArrayUtils.isEmpty(teams)) {
         for (TeamModel team : teams) {
            if (team.canFork) {
               return true;
            }
         }
      }
      return false;
   }
   /**
    * This returns true if the user has admin privileges or the user has admin
    * privileges because of a team membership.
    *
    * @return true if the user can admin
    */
   public boolean canAdmin() {
      if (canAdmin) {
         return true;
      }
      if (!ArrayUtils.isEmpty(teams)) {
         for (TeamModel team : teams) {
            if (team.canAdmin) {
               return true;
            }
         }
      }
      return false;
   }
   /**
    * This returns true if the user has create privileges or the user has create
    * privileges because of a team membership.
    *
    * @return true if the user can admin
    */
   public boolean canCreate() {
      if (canCreate) {
         return true;
      }
      if (!ArrayUtils.isEmpty(teams)) {
         for (TeamModel team : teams) {
            if (team.canCreate) {
               return true;
            }
         }
      }
      return false;
   }
   public boolean isTeamMember(String teamname) {
@@ -319,4 +423,37 @@
   public int compareTo(UserModel o) {
      return username.compareTo(o.username);
   }
   /**
    * Returns true if the name/email pair match this user account.
    *
    * @param name
    * @param email
    * @return true, if the name and email address match this account
    */
   public boolean is(String name, String email) {
      // at a minimum a usename or display name must be supplied
      if (StringUtils.isEmpty(name)) {
         return false;
      }
      boolean nameVerified = name.equalsIgnoreCase(username) || name.equalsIgnoreCase(getDisplayName());
      boolean emailVerified = false;
      if (StringUtils.isEmpty(emailAddress)) {
         // user account has not specified an email address
         // rely on username/displayname verification
         emailVerified = true;
      } else {
         // user account has specified an email address
         // require email address verification
         if (!StringUtils.isEmpty(email)) {
            emailVerified = email.equalsIgnoreCase(emailAddress);
         }
      }
      return nameVerified && emailVerified;
   }
   public boolean hasBranchPermission(String repositoryName, String branch) {
      // Default UserModel doesn't implement branch-level security. Other Realms (i.e. Gerrit) may override this method.
      return hasRepositoryPermission(repositoryName);
   }
}