/*

  * Copyright 2008 The Kuali Foundation

  *

  * Licensed under the Educational Community License, Version 2.0 (the "License");

  * you may not use this file except in compliance with the License.

  * You may obtain a copy of the License at

  *

  * http://www.opensource.org/licenses/ecl2.php

  *

  * Unless required by applicable law or agreed to in writing, software

  * distributed under the License is distributed on an "AS IS" BASIS,

  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  * See the License for the specific language governing permissions and

  * limitations under the License.

  */

 package org.kuali.rice.kim.dao.impl;

 import java.util.ArrayList;

 import java.util.Collection;

 import java.util.List;

 import org.apache.ojb.broker.query.Criteria;

 import org.apache.ojb.broker.query.Query;

 import org.apache.ojb.broker.query.QueryFactory;

 import org.kuali.rice.kim.bo.role.KimResponsibility;

 import org.kuali.rice.kim.bo.role.impl.RoleResponsibilityActionImpl;

 import org.kuali.rice.kim.bo.role.impl.RoleResponsibilityImpl;

 import org.kuali.rice.kim.dao.KimResponsibilityDao;

 import org.kuali.rice.kim.util.KIMPropertyConstants;

 import org.kuali.rice.kim.util.KimConstants;

 import org.kuali.rice.kns.dao.impl.PlatformAwareDaoBaseOjb;

 import org.kuali.rice.kns.util.KNSPropertyConstants;

 /**

  * This is a description of what this class does - kellerj don't forget to fill this in. 

  * 

  * @author Kuali Rice Team (rice.collab@kuali.org)

  *

  */

 public class KimResponsibilityDaoOjb extends PlatformAwareDaoBaseOjb implements KimResponsibilityDao {

         /**

          * @see org.kuali.rice.kim.dao.KimResponsibilityDao#getRoleIdsForResponsibilities(java.util.Collection)

          */

         @SuppressWarnings("unchecked")

         public List<String> getRoleIdsForResponsibilities(Collection<? extends KimResponsibility> responsibilities) {

                 List<String> responsibilityIds = new ArrayList<String>( responsibilities.size() );

                 for ( KimResponsibility kp : responsibilities ) {

                         responsibilityIds.add( kp.getResponsibilityId() );

                 }

                 Criteria c = new Criteria();

                 c.addIn( KimConstants.PrimaryKeyConstants.RESPONSIBILITY_ID, responsibilityIds );

                 c.addEqualTo( KNSPropertyConstants.ACTIVE, true );

                 Query query = QueryFactory.newQuery( RoleResponsibilityImpl.class, c, true );

                 Collection<RoleResponsibilityImpl> coll = getPersistenceBrokerTemplate().getCollectionByQuery(query);

                 List<String> roleIds = new ArrayList<String>( coll.size() );

                 for ( RoleResponsibilityImpl rp : coll ) {

                         roleIds.add( rp.getRoleId() );

                 }

                 return roleIds;

         }

         /**

          * @see org.kuali.rice.kim.dao.KimResponsibilityDao#getRoleIdsForResponsibility(org.kuali.rice.kim.bo.role.impl.KimResponsibilityImpl)

          */

         @SuppressWarnings("unchecked")

         public List<String> getRoleIdsForResponsibility( KimResponsibility responsibility) {

                 Criteria c = new Criteria();

                 c.addEqualTo( KimConstants.PrimaryKeyConstants.RESPONSIBILITY_ID, responsibility.getResponsibilityId() );

                 c.addEqualTo( KNSPropertyConstants.ACTIVE, true );

                 Query query = QueryFactory.newQuery( RoleResponsibilityImpl.class, c, true );

                 Collection<RoleResponsibilityImpl> coll = getPersistenceBrokerTemplate().getCollectionByQuery(query);

                 List<String> roleIds = new ArrayList<String>( coll.size() );

                 for ( RoleResponsibilityImpl rp : coll ) {

                         roleIds.add( rp.getRoleId() );

                 }

                 return roleIds;

         }

         /**

          * @see org.kuali.rice.kim.dao.KimResponsibilityDao#getResponsibilityAction(String, String)

          */

         @SuppressWarnings("unchecked")

         public RoleResponsibilityActionImpl getResponsibilityAction(String roleId, String responsibilityId, String roleMemberId ) {

                 if ( roleMemberId == null || responsibilityId == null ) {

                         return null;

                 }

                 Criteria c = new Criteria();

                 c.addEqualTo( "roleResponsibility.responsibilityId", responsibilityId );

                 c.addEqualTo( "roleResponsibility.roleId", roleId );

                 c.addEqualTo( "roleResponsibility.active", true );

                 Criteria idCriteria = new Criteria();

                 idCriteria.addEqualTo( KIMPropertyConstants.RoleMember.ROLE_MEMBER_ID, roleMemberId );

                 // also handle when roleMemberId is "*" in table

                 Criteria starCrit = new Criteria();

                 starCrit.addEqualTo( KIMPropertyConstants.RoleMember.ROLE_MEMBER_ID, "*" );

                 idCriteria.addOrCriteria( starCrit );

                 c.addAndCriteria( idCriteria );

                 Query query = QueryFactory.newQuery( RoleResponsibilityActionImpl.class, c );

                 Collection<RoleResponsibilityActionImpl> coll = getPersistenceBrokerTemplate().getCollectionByQuery(query);

                 RoleResponsibilityActionImpl result = null;

                 if ( coll.size() == 0 ) {

                         // ok, an exact match failed, attempt to pull where the role_rsp_id is "*"

                         // but, at this point, we would require an exact match on the role member ID

                         c = new Criteria();

                         c.addEqualTo( "roleResponsibilityId", "*" );

                         c.addEqualTo( KIMPropertyConstants.RoleMember.ROLE_MEMBER_ID, roleMemberId );

                         query = QueryFactory.newQuery( RoleResponsibilityActionImpl.class, c );

                         coll = getPersistenceBrokerTemplate().getCollectionByQuery(query);

                         if ( coll.size() != 0 ) {

                                 result = coll.iterator().next();

                         }

                 } else {

                         result = coll.iterator().next();

                 }

                 return result;

         }

 }