James Moger
2012-06-18 67d4f89b0cddb3de05c20e08c20f1bea714c2a9e
src/com/gitblit/FileUserService.java
@@ -20,6 +20,7 @@
import java.io.IOException;
import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
@@ -32,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;
@@ -69,6 +71,49 @@
    */
   @Override
   public void setup(IStoredSettings settings) {
   }
   /**
    * 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;
   }
   /**
@@ -155,6 +200,15 @@
   }
   /**
    * Logout a user.
    *
    * @param user
    */
   @Override
   public void logout(UserModel user) {
   }
   /**
    * Retrieve the user object for the specified username.
    * 
    * @param username
@@ -163,11 +217,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++) {
@@ -217,7 +271,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);
@@ -231,7 +285,9 @@
         }
         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);
@@ -239,8 +295,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 +390,23 @@
         }
         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;
   }
@@ -368,6 +441,7 @@
      } catch (Throwable t) {
         logger.error(MessageFormat.format("Failed to get usernames for role {0}!", role), t);
      }
      Collections.sort(list);
      return list;
   }
@@ -604,8 +678,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();
@@ -618,20 +693,32 @@
               TeamModel team = new TeamModel(username.substring(1));
               List<String> repositories = new ArrayList<String>();
               List<String> users = new ArrayList<String>();
               List<String> mailingLists = new ArrayList<String>();
               List<String> preReceive = new ArrayList<String>();
               List<String> postReceive = new ArrayList<String>();
               for (String role : roles) {
                  if (role.charAt(0) == '!') {
                     users.add(role.substring(1));
                  } else if (role.charAt(0) == '&') {
                     mailingLists.add(role.substring(1));
                  } else if (role.charAt(0) == '^') {
                     preReceive.add(role.substring(1));
                  } else if (role.charAt(0) == '%') {
                     postReceive.add(role.substring(1));
                  } else {
                     repositories.add(role);
                  }
               }
               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());
            }
         }
      }
@@ -652,6 +739,21 @@
   @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;
   }
@@ -687,6 +789,7 @@
      } catch (Throwable t) {
         logger.error(MessageFormat.format("Failed to get teamnames for role {0}!", role), t);
      }
      Collections.sort(list);
      return list;
   }
@@ -742,7 +845,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];
@@ -823,14 +926,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(',');
         }
      }
      if (!ArrayUtils.isEmpty(model.mailingLists)) {
         for (String address : model.mailingLists) {
            sb.append('&');
            sb.append(address);
            sb.append(',');
         }
      }
      if (!ArrayUtils.isEmpty(model.preReceiveScripts)) {
         for (String script : model.preReceiveScripts) {
            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);