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.IdentityHashMap;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import javassist.util.proxy.ProxyFactory;

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.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;

    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);

                print();

                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() {

        System.out.println(toString());

    }

    @Override

    public String toString() {

        StringBuilder sb = new StringBuilder();

        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();

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

                CdmEntityInfo dupCei = entry.getValue();

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

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

                sb.append(" - entity : " + cb.getUserFriendlyTypeName() + "/" + cb.getId() +

                        " as field " + cei.getField().getName() +

                        " of entity " + cbParent.getUserFriendlyTypeName() + "/" + cbParent.getId());

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

                sb.append(" - duplicate entity : " + dupCb.getUserFriendlyTypeName() + "/" + dupCb.getId() +

                        " as field " + dupCei.getField().getName() +

                        " of entity " + dupCbParent.getUserFriendlyTypeName() + "/" + dupCbParent.getId());

                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();

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

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

                sb.append(" - entity : " + cb.getUserFriendlyTypeName() + "/" + cb.getId() +

                        " as field " + cei.getField().getName() +

                        " of entity" + cbParent.getUserFriendlyTypeName() + "/" + cbParent.getId());

            }

        }

        return sb.toString();

    }

    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());

        IdentityHashMap<Object, CdmEntityInfo> alreadyVisitedEntities = new IdentityHashMap<Object, 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,

            IdentityHashMap<Object, 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,

            IdentityHashMap<Object, 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,

            IdentityHashMap<Object, 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,

            IdentityHashMap<Object, 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.put(cdmEntity, 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);

                CdmBase cdmEntityInSubGraph = (CdmBase)childCei.getObject();

                if(cdmEntityInSubGraph != null) {

                    if(!alreadyVisitedEntities.keySet().contains(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,

            IdentityHashMap<Object, 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);

            if(o != null) {

                if(o instanceof HibernateProxy) {

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

                    if(!hli.isUninitialized()) {

                        o = hli.getImplementation();

                    }

                }

                if(o instanceof PersistentCollection) {

                    PersistentCollection pc = ((PersistentCollection)o);

                    if(pc.wasInitialized()) {

                        o = ProxyUtils.getObject(pc);

                    }

                }

            }

            childCei = new CdmEntityInfo(o);

            cei.addChild(childCei);

            childCei.setField(field);

            CdmBase cdmEntityInSubGraph = null;

            if(o != null

                    && !ProxyFactory.isProxyClass(o.getClass())

                    && !(o instanceof PersistentCollection) ) {

                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;

                    if(alreadyVisitedEntities.keySet().contains(cdmEntityInSubGraph)) {

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

                        CdmEntityInfo dupCei = alreadyVisitedEntities.get(cdmEntityInSubGraph);

                        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);

        }

    }

    public class CdmEntityInfo {

        private final Object object;

        private CdmEntityInfo parent;

        private List<CdmEntityInfo> children;

        private Field field;

        private String label;

        public CdmEntityInfo(Object object) {

            this.object = object;

            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) {

                if(object instanceof HibernateProxy) {

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

                    if(hli.isUninitialized()) {

                        label = "[HibernateProxy] " + fieldName;

                    } else {

                        label = fieldName + "Object is a HibernateProxy, but initialised";

                    }

                } else if(object instanceof PersistentCollection) {

                    PersistentCollection pc = ((PersistentCollection)object);

                    if(!pc.wasInitialized()) {

                        label = "[PersistentCollection] " + fieldName;

                    } else {

                        label = fieldName + "Object is a PersistentCollection, but initialised";

                    }

                } else if(object instanceof Collection) {

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

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

                } else if(object instanceof Map) {

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

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

                } else {

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

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

                }

            } else {

                label = "[NULL] " + fieldName;

            }

            return label;

        }

        public void setLabel(String label) {

            this.label = label;

        }

        public Object getObject() {

            return object;

        }

    }

}