James Moger
2011-12-22 e6935876b97a63bae2ec087b4fc390c832aef155
src/com/gitblit/models/UserModel.java
@@ -20,6 +20,8 @@
import java.util.HashSet;
import java.util.Set;
import com.gitblit.utils.StringUtils;
/**
 * UserModel is a serializable model class that represents a user and the user's
 * restricted repository memberships. Instances of UserModels are also used as
@@ -38,19 +40,74 @@
   public boolean canAdmin;
   public boolean excludeFromFederation;
   public final Set<String> repositories = new HashSet<String>();
   public final Set<TeamModel> teams = new HashSet<TeamModel>();
   public UserModel(String username) {
      this.username = username;
   }
   /**
    * This method does not take into consideration Ownership where the
    * administrator has not explicitly granted access to the owner.
    *
    * @param repositoryName
    * @return
    */
   @Deprecated
   public boolean canAccessRepository(String repositoryName) {
      return canAdmin || repositories.contains(repositoryName.toLowerCase());
      return canAdmin || repositories.contains(repositoryName.toLowerCase())
            || hasTeamAccess(repositoryName);
   }
   public boolean canAccessRepository(RepositoryModel repository) {
      boolean isOwner = !StringUtils.isEmpty(repository.owner)
            && repository.owner.equals(username);
      return canAdmin || isOwner || repositories.contains(repository.name.toLowerCase())
            || hasTeamAccess(repository.name);
   }
   public boolean hasTeamAccess(String repositoryName) {
      for (TeamModel team : teams) {
         if (team.hasRepository(repositoryName)) {
            return true;
         }
      }
      return false;
   }
   public boolean hasRepository(String name) {
      return repositories.contains(name.toLowerCase());
   }
   public void addRepository(String name) {
      repositories.add(name.toLowerCase());
   }
   public void removeRepository(String name) {
      repositories.remove(name.toLowerCase());
   }
   public boolean isTeamMember(String teamname) {
      for (TeamModel team : teams) {
         if (team.name.equalsIgnoreCase(teamname)) {
            return true;
         }
      }
      return false;
   }
   public TeamModel getTeam(String teamname) {
      if (teams == null) {
         return null;
      }
      for (TeamModel team : teams) {
         if (team.name.equalsIgnoreCase(teamname)) {
            return team;
         }
      }
      return null;
   }
   @Override
   public String getName() {
      return username;