package com.trendmicro.grid.acl.ds.jpa;
   
   import com.trendmicro.grid.acl.ds.PackageFamilyRepository;
   import com.trendmicro.grid.acl.ds.jpa.entities.JpaPackageFamily;
   import com.trendmicro.grid.acl.ds.jpa.entities.JpaVendor;
   import com.trendmicro.grid.acl.l0.datatypes.NameListPage;
   import com.trendmicro.grid.acl.l0.datatypes.PackageFamily;
   import com.trendmicro.grid.acl.metadata.Metadata;
   import org.springframework.stereotype.Repository;
  import org.springframework.transaction.annotation.Transactional;
  
  import javax.annotation.Resource;
  import javax.persistence.EntityManager;
  import javax.persistence.PersistenceContext;
  import java.util.List;
  
  import static com.trendmicro.grid.acl.ds.jpa.util.JpaUtils.toNameListPage;
  import static net.sf.tinyjee.util.Assert.assertNotNull;
  
  /**
   * Implements VendorRepository using JPA.
   *
   * @author Juergen_Kellerer, 2010-06-11
   * @version 1.0
   */
  @Repository
  @Transactional(readOnly = true)
  public class JpaPackageFamilyRepository implements PackageFamilyRepository {
  
  	private static int pageSize = 1000;
  
  	public static int getPageSize() {
  		return pageSize;
  	}
  
  	public static void setPageSize(int pageSize) {
  		JpaPackageFamilyRepository.pageSize = pageSize;
  	}
  
  	@PersistenceContext(unitName = "CoreDB")
  	EntityManager em;
  
  	@Resource
  	JpaVendorRepository vendorRepository;
  
  	/**
  	 * {@inheritDoc}
  	 */
  	@Transactional(readOnly = false)
  	public JpaPackageFamily create(String vendorName, String basename, String displayName, Metadata metadata) {
  		JpaPackageFamily family = getByName(basename);
  		if (family == null)
  			family = create(vendorRepository.create(vendorName, vendorName, null), basename, displayName, metadata);
  		return family;
  	}
  
  	@Transactional(readOnly = false)
  	public JpaPackageFamily create(JpaVendor vendor, String basename, String displayName, Metadata metadata) {
  		JpaPackageFamily family = new JpaPackageFamily(vendor, basename, displayName, metadata);
  		em.persist(family);
  		return family;
  	}
  
  	/**
  	 * {@inheritDoc}
  	 */
  	@Transactional(readOnly = false)
  	public void update(PackageFamily packageFamily) {
  		assertNotNull("packageFamily", packageFamily);
  
  		JpaPackageFamily f = getByName(packageFamily.getBasename());
  
  		if (f == null) {
  			throw new IllegalArgumentException("Failed updating package family. " +
  					"No entry exists with name '" + packageFamily.getBasename() + "'");
  		}
  
  		f.setDisplayName(packageFamily.getDisplayName());
  		f.setMetadata(packageFamily.getMetadata());
  	}
  
  	/**
  	 * {@inheritDoc}
  	 */
  	public JpaPackageFamily getByName(String basename) {
  		List<JpaPackageFamily> result = em.createNamedQuery("PackageFamily.SelectByName", JpaPackageFamily.class).
  				setParameter("basename", basename).getResultList();
  		return result.isEmpty() ? null : result.get(0);
  	}
  
  	/**
  	 * Returns a lightweight reference used to satisfy foreign key constraints.
  	 *
  	 * @param basename the name of the package family.
  	 * @return a lightweight reference used to satisfy foreign key constraints.
  	 */
  	public JpaPackageFamily getReferenceByName(String basename) {
  		List<Integer> result = em.createNamedQuery("PackageFamily.SelectPrimaryKeyByName", Integer.class).
  				setParameter("basename", basename).getResultList();
 		return result.isEmpty() ? null : em.getReference(JpaPackageFamily.class, result.get(0));
 	}
 
 	/**
 	 * {@inheritDoc}
 	 */
 	public NameListPage listNames(int pageNumber) {
 		return toNameListPage(em.createNamedQuery("PackageFamily.SelectNames", String.class), pageNumber, pageSize);
 	}
 
 	/**
 	 * {@inheritDoc}
 	 */
 	public NameListPage listNamesByVendor(String vendorName, int pageNumber) {
 		return toNameListPage(em.createNamedQuery("PackageFamily.SelectNamesByVendor", String.class).
 				setParameter("vendorName", vendorName), pageNumber, pageSize);
 	}
 }