EDIT

// $Id$

/**

 * Copyright (C) 2015 EDIT

 * European Distributed Institute of Taxonomy

 * http://www.e-taxonomy.eu

 *

 * The contents of this file are subject to the Mozilla Public License Version 1.1

 * See LICENSE.TXT at the top of this package for the full license terms.

 */

package eu.etaxonomy.taxeditor.remoting.cache;

import java.lang.reflect.Field;

import java.util.ArrayList;

import java.util.Collection;

import java.util.HashMap;

import java.util.HashSet;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import java.util.Set;

import net.sf.ehcache.Cache;

import net.sf.ehcache.Element;

import org.apache.log4j.Logger;

import org.hibernate.collection.spi.PersistentCollection;

import org.hibernate.proxy.HibernateProxy;

import org.hibernate.proxy.LazyInitializer;

import org.springframework.util.ReflectionUtils;

import eu.etaxonomy.cdm.api.cache.CdmCacher;

import eu.etaxonomy.cdm.model.common.CdmBase;

/**

 * @author cmathew

 * @date 9 Feb 2015

 *

 */

public class EntityCacherDebugResult {

    private static final Logger logger = Logger.getLogger(EntityCacherDebugResult.class);

    private Map<CdmEntityInfo, CdmEntityInfo> duplicateCdmEntityMap;

    private List<CdmEntityInfo> notInCacheList;

    private CdmTransientEntityCacher cacher;

    private List<CdmEntityInfo> rootElements;

    StringBuilder debugOutput = new StringBuilder();

    public EntityCacherDebugResult() {

    }

    public <T extends CdmBase> EntityCacherDebugResult(CdmTransientEntityCacher cacher, List<T> rootEntities) {

        this.cacher = cacher;

        init();

        if(rootEntities != null && !rootEntities.isEmpty()) {

            for(CdmBase rootEntity : rootEntities) {

                debug(rootEntity, true);

                String out = toString(duplicateCdmEntityMap, notInCacheList, rootEntity);

                System.out.println(out);

                debugOutput.append(out);

                clear();

            }

        }

    }

    private void init() {

        duplicateCdmEntityMap = new HashMap<CdmEntityInfo, CdmEntityInfo>();

        notInCacheList = new ArrayList<CdmEntityInfo>();

        rootElements = new ArrayList<CdmEntityInfo>();

    }

    private void clear() {

        duplicateCdmEntityMap.clear();

        notInCacheList.clear();

    }

    public void addDuplicateEntity(CdmEntityInfo cei, CdmEntityInfo cachedCei) {

        duplicateCdmEntityMap.put(cei, cachedCei);

    }

    public void addEntityNotInCache(CdmEntityInfo cei) {

        notInCacheList.add(cei);

    }

    public List<CdmEntityInfo> getRootElements() {

        return rootElements;

    }

    private void print(Map<CdmEntityInfo, CdmEntityInfo> duplicateCdmEntityMap,

            List<CdmEntityInfo> notInCacheList,

            CdmBase rootEntity) {

        System.out.println(toString(duplicateCdmEntityMap, notInCacheList, rootEntity));

    }

    @Override

    public String toString() {

        return debugOutput.toString();

    }

    private String toString(Map<CdmEntityInfo, CdmEntityInfo> duplicateCdmEntityMap,

            List<CdmEntityInfo> notInCacheList,

            CdmBase rootEntity) {

        StringBuilder sb = new StringBuilder();

        sb.append(System.getProperty("line.separator"));

        sb.append("<<< Root Entity " + rootEntity.getUserFriendlyTypeName() + " with id " + rootEntity.getId() + " >>>");

        sb.append(System.getProperty("line.separator"));

        if(duplicateCdmEntityMap.isEmpty()) {

            sb.append("No Duplicate CDM Entities.");

        } else {

            sb.append("Duplicate CDM Entities,");

            for (Map.Entry<CdmEntityInfo, CdmEntityInfo> entry : duplicateCdmEntityMap.entrySet())

            {

                sb.append(System.getProperty("line.separator"));

                CdmEntityInfo cei = entry.getKey();

                CdmBase cb = (CdmBase) cei.getObject();

                Object cbParent = cei.getParent().getObject();

                sb.append(" - " + cei.getField().getName() + ":" + cb.getUserFriendlyTypeName() + "/" + cb.getId() +

                        " in entity " + cbParent.getClass().getCanonicalName());

                if(cbParent instanceof CdmBase) {

                    sb.append(" with id : " + ((CdmBase)cbParent).getId());

                }

                sb.append(System.getProperty("line.separator"));

                sb.append("  -- entity belongs to cache(s) : " + getCachesContainingEntity(cb));

                sb.append(System.getProperty("line.separator"));

                CdmEntityInfo dupCei = entry.getValue();

                CdmBase dupCb = (CdmBase) dupCei.getObject();

                Object dupCbParent = dupCei.getParent().getObject();

                sb.append(" - " + dupCei.getField().getName() + ":" + dupCb.getUserFriendlyTypeName() + "/" + dupCb.getId() +

                        " in entity " + dupCbParent.getClass().getCanonicalName());

                if(dupCbParent instanceof CdmBase) {

                    sb.append(" with id : " + ((CdmBase)dupCbParent).getId());

                }

                sb.append(System.getProperty("line.separator"));

                sb.append("  -- entity belongs to cache(s) : " + getCachesContainingEntity(dupCb));

                sb.append(System.getProperty("line.separator"));

                sb.append("-----------");

            }

        }

        sb.append(System.getProperty("line.separator"));

        sb.append(System.getProperty("line.separator"));

        if(notInCacheList.isEmpty()) {

            sb.append("No Entities found which are not in Cache.");

        } else {

            sb.append("Not In Cache Entities,");

            for(CdmEntityInfo cei : notInCacheList) {

                CdmBase cb = (CdmBase) cei.getObject();

                Object cbParent = cei.getParent().getObject();

                sb.append(System.getProperty("line.separator"));

                sb.append(" - "  + cei.getField().getName() + ":" + cb.getUserFriendlyTypeName() + "/" + cb.getId());

                if(cbParent instanceof CdmBase) {

                    sb.append(" of entity " + ((CdmBase)cbParent).getUserFriendlyTypeName());

                } else {

                    sb.append(" of entity " + cbParent.getClass().getName());

                }

            }

        }

        sb.append(System.getProperty("line.separator"));

        return sb.toString();

    }

    private String getCachesContainingEntity(CdmBase cdmEntity) {

        Cache defaultCache = CdmRemoteCacheManager.getInstance().getDefaultCacheManager().getCache(CdmCacher.DEFAULT_CACHE_NAME);

        String caches = "";

        Element dce = defaultCache.get(cdmEntity.getUuid());

        if(dce != null && dce.getObjectValue() == cdmEntity) {

            caches = "{DC}";

        }

        Object cte = cacher.getFromCache(CdmTransientEntityCacher.generateKey(cdmEntity));

        if(cte != null && cte == cdmEntity) {

            caches += "{TC}";

        }

        return caches;

    }

    private void debug(CdmBase cdmEntity, boolean recursive) {

        if(cdmEntity == null) {

            return;

        }

        logger.info("---- starting recursive debug for cdm entity " + cdmEntity.getClass().getName() + " with id " + cdmEntity.getId());

        Set<CdmEntityInfo> alreadyVisitedEntities = new HashSet<CdmEntityInfo>();

        CdmEntityInfo cei = new CdmEntityInfo(cdmEntity);

        debugRecursive(cdmEntity, alreadyVisitedEntities, cei);

        rootElements.add(cei);

        alreadyVisitedEntities.clear();

        logger.info("---- ending recursive debug for cdm entity " + cdmEntity.getClass().getName() + " with id " + cdmEntity.getId() + "\n");

    }

    private <T extends Object> void debugRecursive(T obj,

            Set<CdmEntityInfo> alreadyVisitedEntities,

            CdmEntityInfo cei) {

        if(obj == null) {

            return;

        }

        if(obj instanceof CdmBase) {

            debugRecursive((CdmBase)obj, alreadyVisitedEntities, cei);

        } else if (obj instanceof Map) {

            debug((Map<T,T>)obj, alreadyVisitedEntities, cei);

        } else if (obj instanceof Collection) {

            debug((Collection<T>)obj, alreadyVisitedEntities, cei);

        }

        logger.info("No caching yet for type " + obj.getClass().getName());

    }

    private <T extends Object> void debug(Map<T,T> map,

            Set<CdmEntityInfo> alreadyVisitedEntities,

            CdmEntityInfo cei) {

        if(map == null || map.isEmpty()) {

            return;

        }

        int originalMapSize = map.size();

        Iterator<Map.Entry<T,T>> iter = map.entrySet().iterator();

        int i=0;

        while ( iter.hasNext() ) {

            Map.Entry<T,T> e = iter.next();

            CdmEntityInfo childCei = new CdmEntityInfo(e);

            cei.addChild(childCei);

            debugRecursive(e.getKey(), alreadyVisitedEntities, childCei);

            debugRecursive(e.getValue(), alreadyVisitedEntities, childCei);

        }

    }

    private <T extends Object> void debug(Collection<T> collection,

            Set<CdmEntityInfo> alreadyVisitedEntities,

            CdmEntityInfo cei) {

        int length = collection.size();

        Object[] result = new Object[length];

        Iterator<T> collectionItr = collection.iterator();

        while(collectionItr.hasNext()) {

            Object obj = collectionItr.next();

            CdmEntityInfo childCei = new CdmEntityInfo(obj);

            cei.addChild(childCei);

            debugRecursive(obj, alreadyVisitedEntities, childCei);

        }

    }

    private void debugRecursive(CdmBase cdmEntity,

            Set<CdmEntityInfo> alreadyVisitedEntities,

            CdmEntityInfo cei) {

        // we want to recursive through the cdmEntity (and not the cachedCdmEntity)

        // since there could be new or deleted objects in the cdmEntity sub-graph

        // start by getting the fields from the cdm entity

        String className = cdmEntity.getClass().getName();

        CdmModelFieldPropertyFromClass cmgmfc = cacher.getFromCdmlibModelCache(className);

        if(cmgmfc != null) {

            alreadyVisitedEntities.add(cei);

            List<String> fields = cmgmfc.getFields();

            for(String field : fields) {

                // retrieve the actual object corresponding to the field.

                // this object will be either a CdmBase or a Collection / Map

                // with CdmBase as the generic type

                CdmEntityInfo childCei = getDebugCdmBaseTypeFieldValue(cdmEntity, field, alreadyVisitedEntities, cei);

                if(!childCei.isProxy()) {

                    Object object = childCei.getObject();

                    if(object != null && object instanceof CdmBase) {

                        CdmBase cdmEntityInSubGraph = (CdmBase)object;

                        if(!containsIdenticalCdmEntity(alreadyVisitedEntities, cdmEntityInSubGraph)) {

                            logger.info("recursive debugging object of type " + cdmEntityInSubGraph.getClass().getName() + " with id " + cdmEntityInSubGraph.getId());

                            debugRecursive(cdmEntityInSubGraph, alreadyVisitedEntities, childCei);

                        } else {

                            logger.info("object of type " + cdmEntityInSubGraph.getClass().getName() + " with id " + cdmEntityInSubGraph.getId() + " already visited");

                        }

                    }

                }

            }

        } else {

            throw new CdmClientCacheException("CdmEntity with class " + cdmEntity.getClass().getName() + " is not found in the cdmlib model cache. " +

                    "The cache may be corrupted or not in sync with the latest model version" );

        }

    }

    private CdmEntityInfo getDebugCdmBaseTypeFieldValue(CdmBase cdmEntity,

            String fieldName,

            Set<CdmEntityInfo> alreadyVisitedEntities,

            CdmEntityInfo cei) {

        CdmEntityInfo childCei = null;

        Class<?> clazz = cdmEntity.getClass();

        try {

            // this call will search in the provided class as well as

            // the super classes until it finds the field

            Field field = ReflectionUtils.findField(clazz, fieldName);

            if(field == null) {

                throw new CdmClientCacheException("Field '" + fieldName

                        + "' not found when searching in class '" + clazz.getName() + "' and its supercalsses");

            }

            field.setAccessible(true);

            Object o = field.get(cdmEntity);

            CdmBase cdmEntityInSubGraph = null;

            boolean isHibernateProxy = false;

            boolean isPersistentCollection = false;

            childCei = new CdmEntityInfo(o);

            cei.addChild(childCei);

            childCei.setField(field);

            if(o != null) {

                if(o instanceof HibernateProxy) {

                    LazyInitializer hli = ((HibernateProxy)o).getHibernateLazyInitializer();

                    if(!hli.isUninitialized()) {

                        o = hli.getImplementation();

                    } else {

                        isHibernateProxy = true;

                    }

                }

                if(o instanceof PersistentCollection) {

                    PersistentCollection pc = ((PersistentCollection)o);

                    if(pc.wasInitialized()) {

                        o = ProxyUtils.getObject(pc);

                    } else {

                        isPersistentCollection = true;

                    }

                }

                childCei.setObject(o);

                childCei.setProxy(isHibernateProxy || isPersistentCollection);

                if(!isHibernateProxy && !isPersistentCollection) {

                    if(CdmBase.class.isAssignableFrom(o.getClass())) {

                        logger.info("found initialised cdm entity '" + fieldName + "' in object of type " + clazz.getName() + " with id " + cdmEntity.getId());

                        cdmEntityInSubGraph  = (CdmBase)o;

                        //logger.info("  - found duplicate entity at " + fieldName + "' in object of type " + clazz.getName() + " with id " + cdmEntity.getId());

                        CdmEntityInfo dupCei = getDuplicate(alreadyVisitedEntities, cdmEntityInSubGraph);

                        if(dupCei != null) {

                            addDuplicateEntity(childCei, dupCei);

                        }

                        CdmBase cachedCdmEntityInSubGraph = cacher.getFromCache(cdmEntityInSubGraph);

                        // the only exception to updating the field to the latest value

                        // is the case where the field has been already initialised, cached and

                        // is not the same as the one in the cache, in which case we set the value

                        // of the field to the one found in the cache

                        if(cachedCdmEntityInSubGraph == null) {

                            // found a cdm entity which is not in cache - need to record this

                            //logger.info("  - found entity not in cache " + fieldName + "' in object of type " + clazz.getName() + " with id " + cdmEntity.getId());

                            addEntityNotInCache(childCei);

                        }

                    } else if(o instanceof Map) {

                        debugRecursive((Map)o, alreadyVisitedEntities, childCei);

                    } else if(o instanceof Collection) {

                        debugRecursive((Collection)o, alreadyVisitedEntities, childCei);

                    }

                }

            }

            // we return the original cdm entity in the sub graph because we

            // want to continue to recurse on the input cdm entity graph

            // and not the one in the cache

            return childCei;

        } catch (SecurityException e) {

            throw new CdmClientCacheException(e);

        } catch (IllegalArgumentException e) {

            throw new CdmClientCacheException(e);

        } catch (IllegalAccessException e) {

            throw new CdmClientCacheException(e);

        }

    }

    private CdmEntityInfo getDuplicate(Set<CdmEntityInfo> alreadyVisitedEntities, Object objectToCompare) {

        if(objectToCompare != null ) {

            for(CdmEntityInfo cei: alreadyVisitedEntities) {

                if(objectToCompare.equals(cei.getObject()) && objectToCompare != cei.getObject()) {

                    return cei;

                }

            }

        }

        return null;

    }

    private boolean containsIdenticalCdmEntity(Set<CdmEntityInfo> ceiSet, Object objectToCompare) {

        if(objectToCompare != null) {

            for(CdmEntityInfo cei : ceiSet) {

                if(cei.getObject() == objectToCompare) {

                    return true;

                }

            }

        }

        return false;

    }

    public class CdmEntityInfo {

        private Object object;

        private CdmEntityInfo parent;

        private List<CdmEntityInfo> children;

        private Field field;

        private String label;

        private boolean isProxy;

        public CdmEntityInfo(Object object) {

            this.object = object;

            isProxy = false;

            children = new ArrayList<CdmEntityInfo>();

        }

        public CdmEntityInfo getParent() {

            return parent;

        }

        public void setParent(CdmEntityInfo parent) {

            this.parent = parent;

        }

        public List<CdmEntityInfo> getChildren() {

            return children;

        }

        public void setChildren(List<CdmEntityInfo> children) {

            this.children = children;

        }

        public void addChild(CdmEntityInfo cei) {

            this.children.add(cei);

            cei.setParent(this);

        }

        public Field getField() {

            return field;

        }

        public void setField(Field field) {

            this.field = field;

        }

        public String getLabel() {

            String label;

            String fieldName = "";

            if(field != null) {

                fieldName = field.getName();

            }

            if(object != null) {

                String className = object.getClass().getName();

                if(object instanceof HibernateProxy) {

                    LazyInitializer hli = ((HibernateProxy)object).getHibernateLazyInitializer();

                    if(hli.isUninitialized()) {

                        className = "HibernateProxy";

                    } else {

                        className = "InitialisedHibernateProxy";

                    }

                    label = "[" + className + "] " + fieldName;

                } else if(object instanceof PersistentCollection) {

                    PersistentCollection pc = ((PersistentCollection)object);

                    if(!pc.wasInitialized()) {

                        className = "PersistentCollection";

                    } else {

                        className = "InitialisedPersistentCollection";

                    }

                    label = "[" + className + "] " + fieldName;

                } else if(object instanceof Collection) {

                    label = "[" + className + "] " + fieldName + " : " + String.valueOf(((Collection)object).size());

                } else if(object instanceof Map) {

                    label = "[" + className + "] " + fieldName + " : " + String.valueOf(((Map)object).size());

                } else if(object instanceof CdmBase) {

                    label = getCachesContainingEntity((CdmBase)object) +  "[" + className + ",id" + ((CdmBase)object).getId() + "] " + fieldName + " : " + object.toString();

                } else {

                    label = "[" + className + "] " + fieldName + " : " + object.toString();

                }

            } else {

                label = "[NULL] " + fieldName;

            }

            return label;

        }

        public void setLabel(String label) {

            this.label = label;

        }

        public Object getObject() {

            return object;

        }

        public void setObject(Object object) {

            this.object = object;

        }

        public boolean isProxy() {

            return isProxy;

        }

        public void setProxy(boolean isProxy) {

            this.isProxy = isProxy;

        }

    }

}