James Moger
2012-09-10 fabe060d3a435f116128851f828e35c2af5fde67
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;
@@ -72,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
@@ -88,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;
   }
   /**
@@ -155,6 +203,15 @@
   }
   /**
    * Logout a user.
    *
    * @param user
    */
   @Override
   public void logout(UserModel user) {
   }
   /**
    * Retrieve the user object for the specified username.
    * 
    * @param username
@@ -163,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++) {
@@ -217,7 +274,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 +288,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 +298,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 +393,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 +444,7 @@
      } catch (Throwable t) {
         logger.error(MessageFormat.format("Failed to get usernames for role {0}!", role), t);
      }
      Collections.sort(list);
      return list;
   }
@@ -604,8 +681,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();
@@ -619,11 +697,17 @@
               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));
                     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);
                  }
@@ -631,11 +715,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());
            }
         }
      }
@@ -656,6 +742,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;
   }
@@ -691,6 +792,7 @@
      } catch (Throwable t) {
         logger.error(MessageFormat.format("Failed to get teamnames for role {0}!", role), t);
      }
      Collections.sort(list);
      return list;
   }
@@ -746,7 +848,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];
@@ -827,19 +929,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(',');
         }
      }
      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);