James Moger
2012-09-29 1e1b85270f93b3bca624c99b478f3a9a23be2395
src/com/gitblit/FileUserService.java
@@ -33,6 +33,7 @@
import com.gitblit.models.TeamModel;
import com.gitblit.models.UserModel;
import com.gitblit.utils.ArrayUtils;
import com.gitblit.utils.DeepCopier;
import com.gitblit.utils.StringUtils;
@@ -73,6 +74,49 @@
   }
   /**
    * Does the user service support changes to credentials?
    *
    * @return true or false
    * @since 1.0.0
    */
   @Override
   public boolean supportsCredentialChanges() {
      return true;
   }
   /**
    * Does the user service support changes to user display name?
    *
    * @return true or false
    * @since 1.0.0
    */
   @Override
   public boolean supportsDisplayNameChanges() {
      return false;
   }
   /**
    * Does the user service support changes to user email address?
    *
    * @return true or false
    * @since 1.0.0
    */
   @Override
   public boolean supportsEmailAddressChanges() {
      return false;
   }
   /**
    * Does the user service support changes to team memberships?
    *
    * @return true or false
    * @since 1.0.0
    */
   public boolean supportsTeamMembershipChanges() {
      return true;
   }
   /**
    * Does the user service support cookie authentication?
    * 
    * @return true or false
@@ -89,13 +133,16 @@
    * @return cookie value
    */
   @Override
   public char[] getCookie(UserModel model) {
   public String getCookie(UserModel model) {
      if (!StringUtils.isEmpty(model.cookie)) {
         return model.cookie;
      }
      Properties allUsers = super.read();
      String value = allUsers.getProperty(model.username);
      String[] roles = value.split(",");
      String password = roles[0];
      String cookie = StringUtils.getSHA1(model.username + password);
      return cookie.toCharArray();
      return cookie;
   }
   /**
@@ -156,6 +203,15 @@
   }
   /**
    * Logout a user.
    *
    * @param user
    */
   @Override
   public void logout(UserModel user) {
   }
   /**
    * Retrieve the user object for the specified username.
    * 
    * @param username
@@ -164,11 +220,11 @@
   @Override
   public UserModel getUserModel(String username) {
      Properties allUsers = read();
      String userInfo = allUsers.getProperty(username);
      String userInfo = allUsers.getProperty(username.toLowerCase());
      if (userInfo == null) {
         return null;
      }
      UserModel model = new UserModel(username);
      UserModel model = new UserModel(username.toLowerCase());
      String[] userValues = userInfo.split(",");
      model.password = userValues[0];
      for (int i = 1; i < userValues.length; i++) {
@@ -178,6 +234,8 @@
            // Permissions
            if (role.equalsIgnoreCase(Constants.ADMIN_ROLE)) {
               model.canAdmin = true;
            } else if (role.equalsIgnoreCase(Constants.FORK_ROLE)) {
               model.canFork = true;
            } else if (role.equalsIgnoreCase(Constants.NOT_FEDERATED_ROLE)) {
               model.excludeFromFederation = true;
            }
@@ -218,7 +276,7 @@
    */
   @Override
   public boolean updateUserModel(String username, UserModel model) {
      try {
      try {
         Properties allUsers = read();
         UserModel oldUser = getUserModel(username);
         ArrayList<String> roles = new ArrayList<String>(model.repositories);
@@ -227,12 +285,17 @@
         if (model.canAdmin) {
            roles.add(Constants.ADMIN_ROLE);
         }
         if (model.canFork) {
            roles.add(Constants.FORK_ROLE);
         }
         if (model.excludeFromFederation) {
            roles.add(Constants.NOT_FEDERATED_ROLE);
         }
         StringBuilder sb = new StringBuilder();
         sb.append(model.password);
         if (!StringUtils.isEmpty(model.password)) {
            sb.append(model.password);
         }
         sb.append(',');
         for (String role : roles) {
            sb.append(role);
@@ -240,8 +303,8 @@
         }
         // trim trailing comma
         sb.setLength(sb.length() - 1);
         allUsers.remove(username);
         allUsers.put(model.username, sb.toString());
         allUsers.remove(username.toLowerCase());
         allUsers.put(model.username.toLowerCase(), sb.toString());
         // null check on "final" teams because JSON-sourced UserModel
         // can have a null teams object
@@ -334,6 +397,22 @@
            continue;
         }
         list.add(user);
      }
      Collections.sort(list);
      return list;
   }
   /**
    * Returns the list of all users available to the login service.
    *
    * @return list of all usernames
    */
   @Override
   public List<UserModel> getAllUsers() {
      read();
      List<UserModel> list = new ArrayList<UserModel>();
      for (String username : getAllUsernames()) {
         list.add(getUserModel(username));
      }
      Collections.sort(list);
      return list;
@@ -607,8 +686,9 @@
   @Override
   protected synchronized Properties read() {
      long lastRead = lastModified();
      boolean reload = forceReload();
      Properties allUsers = super.read();
      if (lastRead != lastModified()) {
      if (reload || (lastRead != lastModified())) {
         // reload hash cache
         cookies.clear();
         teams.clear();
@@ -640,11 +720,13 @@
               team.addRepositories(repositories);
               team.addUsers(users);
               team.addMailingLists(mailingLists);
               team.preReceiveScripts.addAll(preReceive);
               team.postReceiveScripts.addAll(postReceive);
               teams.put(team.name.toLowerCase(), team);
            } else {
               // user definition
               String password = roles[0];
               cookies.put(StringUtils.getSHA1(username + password), username);
               cookies.put(StringUtils.getSHA1(username.toLowerCase() + password), username.toLowerCase());
            }
         }
      }
@@ -665,6 +747,20 @@
   @Override
   public List<String> getAllTeamNames() {
      List<String> list = new ArrayList<String>(teams.keySet());
      Collections.sort(list);
      return list;
   }
   /**
    * Returns the list of all teams available to the login service.
    *
    * @return list of all teams
    * @since 0.8.0
    */
   @Override
   public List<TeamModel> getAllTeams() {
      List<TeamModel> list = new ArrayList<TeamModel>(teams.values());
      list = DeepCopier.copy(list);
      Collections.sort(list);
      return list;
   }
@@ -757,7 +853,7 @@
         // remove role from team
         for (String name : needsRemoveRole) {
            String team = "@" + name;
            String[] values = allUsers.getProperty(team).split(",");
            String[] values = allUsers.getProperty(team).split(",");
            StringBuilder sb = new StringBuilder();
            for (int i = 0; i < values.length; i++) {
               String value = values[i];
@@ -838,29 +934,39 @@
   private void updateTeamCache(Properties allUsers, String teamname, TeamModel model) {
      StringBuilder sb = new StringBuilder();
      for (String repository : model.repositories) {
         sb.append(repository);
         sb.append(',');
      if (!ArrayUtils.isEmpty(model.repositories)) {
         for (String repository : model.repositories) {
            sb.append(repository);
            sb.append(',');
         }
      }
      for (String user : model.users) {
         sb.append('!');
         sb.append(user);
         sb.append(',');
      if (!ArrayUtils.isEmpty(model.users)) {
         for (String user : model.users) {
            sb.append('!');
            sb.append(user);
            sb.append(',');
         }
      }
      for (String address : model.mailingLists) {
         sb.append('&');
         sb.append(address);
         sb.append(',');
      if (!ArrayUtils.isEmpty(model.mailingLists)) {
         for (String address : model.mailingLists) {
            sb.append('&');
            sb.append(address);
            sb.append(',');
         }
      }
      for (String script : model.preReceiveScripts) {
         sb.append('^');
         sb.append(script);
         sb.append(',');
      if (!ArrayUtils.isEmpty(model.preReceiveScripts)) {
         for (String script : model.preReceiveScripts) {
            sb.append('^');
            sb.append(script);
            sb.append(',');
         }
      }
      for (String script : model.postReceiveScripts) {
         sb.append('%');
         sb.append(script);
         sb.append(',');
      if (!ArrayUtils.isEmpty(model.postReceiveScripts)) {
         for (String script : model.postReceiveScripts) {
            sb.append('%');
            sb.append(script);
            sb.append(',');
         }
      }
      // trim trailing comma
      sb.setLength(sb.length() - 1);