Laurens Vrijnsen
2013-03-22 662fc186b14b25019ecc9ae05ce0894360d44393
src/com/gitblit/GitblitUserService.java
@@ -18,13 +18,17 @@
import java.io.File;
import java.io.IOException;
import java.text.MessageFormat;
import java.util.Collection;
import java.util.List;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.gitblit.Constants.AccountType;
import com.gitblit.models.TeamModel;
import com.gitblit.models.UserModel;
import com.gitblit.utils.DeepCopier;
import com.gitblit.utils.StringUtils;
/**
 * This class wraps the default user service and is recommended as the starting
@@ -47,6 +51,8 @@
public class GitblitUserService implements IUserService {
   protected IUserService serviceImpl;
   protected final String ExternalAccount = "#externalAccount";
   private final Logger logger = LoggerFactory.getLogger(GitblitUserService.class);
@@ -55,7 +61,7 @@
   @Override
   public void setup(IStoredSettings settings) {
      File realmFile = GitBlit.getFileOrFolder(Keys.realm.userService, "users.conf");
      File realmFile = GitBlit.getFileOrFolder(Keys.realm.userService, "${baseFolder}/users.conf");
      serviceImpl = createUserService(realmFile);
      logger.info("GUS delegating to " + serviceImpl.toString());
   }
@@ -95,8 +101,8 @@
            logger.info(MessageFormat.format("Automatically creating {0} based on {1}",
                  usersConfig.getAbsolutePath(), realmFile.getAbsolutePath()));
            ConfigUserService configService = new ConfigUserService(usersConfig);
            for (String username : serviceImpl.getAllUsernames()) {
               UserModel userModel = serviceImpl.getUserModel(username);
            for (String username : service.getAllUsernames()) {
               UserModel userModel = service.getUserModel(username);
               configService.updateUserModel(userModel);
            }
         }
@@ -112,28 +118,59 @@
   }
   @Override
   public boolean supportsCredentialChanges() {
      return serviceImpl.supportsCredentialChanges();
   }
   @Override
   public boolean supportsDisplayNameChanges() {
      return serviceImpl.supportsDisplayNameChanges();
   }
   @Override
   public boolean supportsEmailAddressChanges() {
      return serviceImpl.supportsEmailAddressChanges();
   }
   @Override
   public boolean supportsTeamMembershipChanges() {
      return serviceImpl.supportsTeamMembershipChanges();
   }
   @Override
   public boolean supportsCookies() {
      return serviceImpl.supportsCookies();
   }
   @Override
   public char[] getCookie(UserModel model) {
   public String getCookie(UserModel model) {
      return serviceImpl.getCookie(model);
   }
   @Override
   public UserModel authenticate(char[] cookie) {
      return serviceImpl.authenticate(cookie);
      UserModel user = serviceImpl.authenticate(cookie);
      setAccountType(user);
      return user;
   }
   @Override
   public UserModel authenticate(String username, char[] password) {
      return serviceImpl.authenticate(username, password);
      UserModel user = serviceImpl.authenticate(username, password);
      setAccountType(user);
      return user;
   }
   @Override
   public void logout(UserModel user) {
      serviceImpl.logout(user);
   }
   @Override
   public UserModel getUserModel(String username) {
      return serviceImpl.getUserModel(username);
      UserModel user = serviceImpl.getUserModel(username);
      setAccountType(user);
      return user;
   }
   @Override
@@ -142,10 +179,39 @@
   }
   @Override
   public boolean updateUserModel(String username, UserModel model) {
      return serviceImpl.updateUserModel(username, model);
   public boolean updateUserModels(Collection<UserModel> models) {
      return serviceImpl.updateUserModels(models);
   }
   @Override
   public boolean updateUserModel(String username, UserModel model) {
      if (model.isLocalAccount() || supportsCredentialChanges()) {
         if (!model.isLocalAccount() && !supportsTeamMembershipChanges()) {
            //  teams are externally controlled - copy from original model
            UserModel existingModel = getUserModel(username);
            model = DeepCopier.copy(model);
            model.teams.clear();
            model.teams.addAll(existingModel.teams);
         }
         return serviceImpl.updateUserModel(username, model);
      }
      if (model.username.equals(username)) {
         // passwords are not persisted by the backing user service
         model.password = null;
         if (!model.isLocalAccount() && !supportsTeamMembershipChanges()) {
            //  teams are externally controlled- copy from original model
            UserModel existingModel = getUserModel(username);
            model = DeepCopier.copy(model);
            model.teams.clear();
            model.teams.addAll(existingModel.teams);
         }
         return serviceImpl.updateUserModel(username, model);
      }
      logger.error("Users can not be renamed!");
      return false;
   }
   @Override
   public boolean deleteUserModel(UserModel model) {
      return serviceImpl.deleteUserModel(model);
@@ -162,8 +228,22 @@
   }
   @Override
   public List<UserModel> getAllUsers() {
      List<UserModel> users = serviceImpl.getAllUsers();
       for (UserModel user : users) {
          setAccountType(user);
       }
      return users;
   }
   @Override
   public List<String> getAllTeamNames() {
      return serviceImpl.getAllTeamNames();
   }
   @Override
   public List<TeamModel> getAllTeams() {
      return serviceImpl.getAllTeams();
   }
   @Override
@@ -172,6 +252,7 @@
   }
   @Override
   @Deprecated
   public boolean setTeamnamesForRepositoryRole(String role, List<String> teamnames) {
      return serviceImpl.setTeamnamesForRepositoryRole(role, teamnames);
   }
@@ -187,7 +268,20 @@
   }
   @Override
   public boolean updateTeamModels(Collection<TeamModel> models) {
      return serviceImpl.updateTeamModels(models);
   }
   @Override
   public boolean updateTeamModel(String teamname, TeamModel model) {
      if (!supportsTeamMembershipChanges()) {
         // teams are externally controlled - copy from original model
         TeamModel existingModel = getTeamModel(teamname);
         model = DeepCopier.copy(model);
         model.users.clear();
         model.users.addAll(existingModel.users);
      }
      return serviceImpl.updateTeamModel(teamname, model);
   }
@@ -207,6 +301,7 @@
   }
   @Override
   @Deprecated
   public boolean setUsernamesForRepositoryRole(String role, List<String> usernames) {
      return serviceImpl.setUsernamesForRepositoryRole(role, usernames);
   }
@@ -220,4 +315,25 @@
   public boolean deleteRepositoryRole(String role) {
      return serviceImpl.deleteRepositoryRole(role);
   }
   protected boolean isLocalAccount(String username) {
      UserModel user = getUserModel(username);
      return user != null && user.isLocalAccount();
   }
   protected void setAccountType(UserModel user) {
      if (user != null) {
         if (!StringUtils.isEmpty(user.password)
               && !ExternalAccount.equalsIgnoreCase(user.password)
               && !"StoredInLDAP".equalsIgnoreCase(user.password)) {
            user.accountType = AccountType.LOCAL;
         } else {
            user.accountType = getAccountType();
         }
      }
   }
   protected AccountType getAccountType() {
      return AccountType.LOCAL;
   }
}