/*
    * Copyright 2007-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.ksb.messaging.dao.impl;
  
  import java.util.List;
  
  import javax.persistence.EntityManager;
  import javax.persistence.PersistenceContext;
  import javax.persistence.Query;
  import javax.xml.namespace.QName;
  
  import org.kuali.rice.ksb.messaging.FlattenedServiceDefinition;
  import org.kuali.rice.ksb.messaging.ServiceInfo;
  import org.kuali.rice.ksb.messaging.dao.ServiceInfoDAO;
  
  
  public class ServiceInfoDAOJpaImpl implements ServiceInfoDAO {
  	
  	@PersistenceContext
  	EntityManager entityManager;
  	
      public void addEntry(ServiceInfo entry) {
          if(entry.getMessageEntryId() == null) {
          	entityManager.persist(entry.getSerializedServiceNamespace());
          	entry.setFlattenedServiceDefinitionId(entry.getSerializedServiceNamespace().getFlattenedServiceDefinitionId());
              entityManager.persist(entry);
          }
          else {
          	entityManager.merge(entry.getSerializedServiceNamespace());
              entityManager.merge(entry);
          }
      }
  
      @SuppressWarnings("unchecked")
      public List<ServiceInfo> fetchAll() {
      	Query query = entityManager.createNamedQuery("ServiceInfo.FetchAll");
      	return (List<ServiceInfo>) query.getResultList();
      }
  
      @SuppressWarnings("unchecked")
      public List<ServiceInfo> fetchAllActive() {
      	Query query = entityManager.createNamedQuery("ServiceInfo.FetchAllActive");
      	return (List<ServiceInfo>) query.getResultList();
      }
  
      @SuppressWarnings("unchecked")
      public List<ServiceInfo> fetchActiveByNameLocalPart(String localPart) {
      	Query query = entityManager.createNamedQuery("ServiceInfo.FetchActiveByName");
      	query.setParameter("serviceName", "{%}" + localPart);
      	return (List<ServiceInfo>) query.getResultList();    	
      }
      
      @SuppressWarnings("unchecked")
      public List<ServiceInfo> fetchActiveByName(QName serviceName) {
      	Query query = entityManager.createNamedQuery("ServiceInfo.FetchActiveByName");
      	query.setParameter("serviceName", serviceName.toString());
      	return (List<ServiceInfo>) query.getResultList();    	
      }
      
      @SuppressWarnings("unchecked")
      public List<ServiceInfo> fetchActiveByNamespace(String serviceNamespace) {
      	Query query = entityManager.createNamedQuery("ServiceInfo.FetchActiveByName");
      	query.setParameter("serviceNamespace", "{"+serviceNamespace+"}%");
      	return (List<ServiceInfo>) query.getResultList();    	
      }
      
      @SuppressWarnings("unchecked")
      public List<ServiceInfo> findLocallyPublishedServices(String ipNumber, String serviceNamespace) {
      	Query query = entityManager.createNamedQuery("ServiceInfo.FindLocallyPublishedServices");
      	query.setParameter("serverIp", ipNumber);
      	query.setParameter("serviceNamespace", serviceNamespace);
      	return (List<ServiceInfo>) query.getResultList();
      }
  
      public void removeEntry(ServiceInfo entry) {
      	//Query query = entityManager.createNamedQuery("ServiceInfo.DeleteByEntry");
      	//query.setParameter("messageEntryId", entry.getMessageEntryId());
  		//query.executeUpdate();
      	//entityManager.remove(entityManager.getReference(ServiceInfo.class, entry.getMessageEntryId()));
      	entityManager.remove(entityManager.find(FlattenedServiceDefinition.class, entry.getFlattenedServiceDefinitionId()));
      	entityManager.remove(entityManager.find(ServiceInfo.class, entry.getMessageEntryId()));
      }
  
      public ServiceInfo findServiceInfo(Long serviceInfoId) {
      	return entityManager.find(ServiceInfo.class, serviceInfoId);
      }
 
     public void removeLocallyPublishedServices(String ipNumber, String serviceNamespace) {
     	//Query query = entityManager.createNamedQuery("ServiceInfo.DeleteLocallyPublishedServices");
     	//query.setParameter("serverIp", ipNumber);
     	//query.setParameter("serviceNamespace", serviceNamespace);
 		//query.executeUpdate();
     	List<ServiceInfo> localServices = findLocallyPublishedServices(ipNumber, serviceNamespace);
     	for (ServiceInfo localService : localServices) {
     		entityManager.remove(entityManager.find(FlattenedServiceDefinition.class, localService.getFlattenedServiceDefinitionId()));
     		entityManager.remove(entityManager.find(ServiceInfo.class, localService.getMessageEntryId()));
     	}
     }
 
 	public EntityManager getEntityManager() {
         return this.entityManager;
     }
 
     public void setEntityManager(EntityManager entityManager) {
         this.entityManager = entityManager;
     }
 
 	public FlattenedServiceDefinition findFlattenedServiceDefinition(Long flattenedServiceDefinitionId) {
 		return entityManager.find(FlattenedServiceDefinition.class, flattenedServiceDefinitionId);
 	}
 }