alsutton/enterprisepasswordsafe

package com.enterprisepasswordsafe.engine.passwords;

import com.enterprisepasswordsafe.database.*;
import com.enterprisepasswordsafe.engine.accesscontrol.PasswordPermission;

import java.io.UnsupportedEncodingException;
import java.security.GeneralSecurityException;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;

public class PasswordPermissionApplier {

    private final GroupAccessControlDAO groupAccessControlDAO;
    private final GroupDAO groupDAO;
    private final HierarchyNodePermissionDAO hierarchyNodePermissionDAO;
    private final UserAccessControlDAO userAccessControlDAO;
    private final UserDAO userDAO;

    public PasswordPermissionApplier() {
        groupAccessControlDAO = GroupAccessControlDAO.getInstance();
        groupDAO = GroupDAO.getInstance();
        hierarchyNodePermissionDAO = new HierarchyNodePermissionDAO();
        userAccessControlDAO = UserAccessControlDAO.getInstance();
        userDAO = UserDAO.getInstance();
    }

    public void setDefaultPermissions(final Password newPassword, final String parentNodeId, final Group adminGroup)
            throws UnsupportedEncodingException, SQLException, GeneralSecurityException {
        Map<String,PasswordPermission> uPerms = new HashMap<>();
        Map<String,PasswordPermission> gPerms = new HashMap<>();
        hierarchyNodePermissionDAO.getCombinedDefaultPermissionsForNode(parentNodeId, uPerms, gPerms);

        setUserDefaults(uPerms, adminGroup, newPassword);
        setGroupDefaults(gPerms, adminGroup, newPassword);

    }

    private void setUserDefaults(Map<String,PasswordPermission> uPerms, Group adminGroup, Password newPassword)
            throws GeneralSecurityException, UnsupportedEncodingException, SQLException {

        for(Map.Entry<String,PasswordPermission> thisEntry : uPerms.entrySet()) {
            String userId = thisEntry.getKey();
            User theUser = userDAO.getByIdDecrypted(userId, adminGroup);
            if( theUser != null ) {
                userAccessControlDAO.create(theUser, newPassword, thisEntry.getValue());
            }
        }
    }

    private void setGroupDefaults(Map<String,PasswordPermission> gPerms, Group adminGroup, Password newPassword)
            throws GeneralSecurityException, UnsupportedEncodingException, SQLException {
        final User adminUser = userDAO.getAdminUser(adminGroup);
        for(Map.Entry<String,PasswordPermission> thisEntry : gPerms.entrySet()) {
            final String groupId = thisEntry.getKey();
            Group theGroup = groupDAO.getByIdDecrypted(groupId, adminUser);
            if( theGroup != null ) {
                groupAccessControlDAO.create(theGroup, newPassword, thisEntry.getValue());
            }
        }
    }
}