Author: shane.bryzak(a)jboss.com Date: 2010-07-21 19:49:42 -0400 (Wed, 21 Jul 2010) New Revision: 13466 Modified: modules/security/trunk/impl/src/main/java/org/jboss/seam/security/management/picketlink/JpaIdentityStore.java Log: implement updateCredential() Modified: modules/security/trunk/impl/src/main/java/org/jboss/seam/security/management/picketlink/JpaIdentityStore.java =================================================================== --- modules/security/trunk/impl/src/main/java/org/jboss/seam/security/management/picketlink/JpaIdentityStore.java 2010-07-21 23:13:57 UTC (rev 13465) +++ modules/security/trunk/impl/src/main/java/org/jboss/seam/security/management/picketlink/JpaIdentityStore.java 2010-07-21 23:49:42 UTC (rev 13466) @@ -1744,10 +1744,88 @@ IdentityObject identityObject, IdentityObjectCredential credential) throws IdentityException { - + EntityManager em = getEntityManager(ctx); - // TODO Auto-generated method stub - System.out.println("*** Invoked unimplemented method updateCredential()"); + Property<Object> credentialValue = modelProperties.get(PROPERTY_CREDENTIAL_VALUE); + + if (credentialClass != null) + { + Property<Object> credentialIdentity = modelProperties.get(PROPERTY_CREDENTIAL_IDENTITY); + Property<Object> credentialType = modelProperties.get(PROPERTY_CREDENTIAL_TYPE); + Object identity = lookupIdentity(identityObject, em); + + CriteriaBuilder builder = em.getCriteriaBuilder(); + CriteriaQuery<?> criteria = builder.createQuery(credentialClass); + Root<?> root = criteria.from(credentialClass); + + List<Predicate> predicates = new ArrayList<Predicate>(); + predicates.add(builder.equal(root.get(credentialIdentity.getName()), + identity)); + + if (credentialType != null) + { + if (String.class.equals(credentialType.getJavaClass())) + { + predicates.add(builder.equal(root.get(credentialType.getName()), + credential.getType().getName())); + } + else + { + predicates.add(builder.equal(root.get(credentialType.getName()), + lookupCredentialTypeEntity(credential.getType().getName(), em))); + } + } + + criteria.where(predicates.toArray(new Predicate[0])); + + List<?> results = em.createQuery(criteria).getResultList(); + + if (results.isEmpty()) + { + // The credential doesn't exist, let's create it + try + { + Object newCredential = credentialClass.newInstance(); + credentialIdentity.setValue(newCredential, identity); + credentialValue.setValue(newCredential, credential.getValue()); + credentialType.setValue(newCredential, + lookupCredentialTypeEntity(credential.getType().getName(), em)); + + em.persist(newCredential); + } + catch (IllegalAccessException ex) + { + throw new IdentityException("Error updating credential - could " + + "not create credential instance", ex); + } + catch (InstantiationException ex) + { + throw new IdentityException("Error updating credential - could " + + "not create credential instance", ex); + } + } + else + { + // TODO there shouldn't be multiple credentials with the same type, + // but if there are, we need to deal with it somehow.. for now just use the first one + + Object result = results.get(0); + credentialValue.setValue(result, credential.getValue()); + + em.merge(result); + } + } + else + { + // The credential is stored in the identity class, update it there + + Property<Object> credentialProp = modelProperties.get(PROPERTY_CREDENTIAL_VALUE); + Object identity = lookupIdentity(identityObject, em); + + credentialProp.setValue(identity, credential.getValue()); + + em.merge(identity); + } }