/*******************************************************************************
 * Copyright (c) 2007, 2010 Association for Decentralized Information Management
 * in Industry THTH ry.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     VTT Technical Research Centre of Finland - initial API and implementation
 *******************************************************************************/
package org.simantics.db.impl.query;

import org.simantics.databoard.binding.Binding;
import org.simantics.databoard.serialization.Serializer;
import org.simantics.databoard.util.URIStringUtils;
import org.simantics.db.common.WriteBindings;
import org.simantics.db.common.exception.DebugException;
import org.simantics.db.exception.DatabaseException;
import org.simantics.db.impl.graph.ReadGraphImpl;
import org.simantics.db.impl.procedure.InternalProcedure;

import gnu.trove.map.hash.TObjectIntHashMap;

final public class NamespaceIndex extends StringQuery<InternalProcedure<TObjectIntHashMap<String>>> {
	
    NamespaceIndex(final String id) {
        super(id);
    }
    
 	@Override
 	final public void removeEntry(QueryProcessor provider) {
 		provider.cache.remove(this);
 	}
 	
	final private void index(ReadGraphImpl graph, final QueryProcessor provider, int root, final InternalProcedure<TObjectIntHashMap<String>> procedure) throws DatabaseException {
		
		if(root == 0) {
            add2(graph, null);
            procedure.execute(graph, null);
//            System.err.println("NamespaceIndex[" + id + "]->null");
            return;
		}

		final int consistsOf = provider.getConsistsOf();
		final int hasName = provider.getHasName();
        
        final TObjectIntHashMap<String> result = new TObjectIntHashMap<String>();
        
        QueryCache.runnerObjects(graph, root, consistsOf, graph.parent, null, new SyncIntProcedure() {
        	
        	@Override
			public void run(ReadGraphImpl graph) throws DatabaseException {
        		
        		if(isPending()) { 
                    add2(graph, result);
                    procedure.execute(graph, result);
//                    System.err.println("NamespaceIndex[" + id + "]->" + result.size());
        		} else {
        			procedure.exception(graph, (Throwable)statusOrException);
        		}
				
			}

			@Override
			public void finished(ReadGraphImpl graph) throws DatabaseException {
				dec(graph);
			}

        	@Override
        	public void execute(ReadGraphImpl graph, final int obj) throws DatabaseException {
        		
        		//System.out.println(id + " => " + obj);

        		inc();
        		
        		QueryCache.runnerObjects(graph, obj, hasName, graph.parent, null, new IntProcedure() {
                	
                	@Override
                	public void execute(ReadGraphImpl graph, int i) throws DatabaseException {

                		inc();

                		QueryCache.runnerValueQuery(graph, i, NamespaceIndex.this, null, new InternalProcedure<byte[]>() {
                			
                			@Override
                			public void execute(ReadGraphImpl graph, byte[] value) throws DatabaseException {
                				
                				if(value != null) {

	                                try {

	                                	Binding b = WriteBindings.STRING;
	                                    Serializer serializer = b.serializer();
	                                    final String part = (String)serializer.deserialize(value);
	
	                                    synchronized(result) {
	                                        Object previous = result.put(URIStringUtils.escape(part), obj);
	                                        // TODO: this is not the most elegant solution
	                                        if(previous != null) previous = "";
	                                    }
	                                    
	                	        	} catch (Throwable e) {
	                                    if(DebugException.DEBUG) new DebugException(e).printStackTrace();
	                                }
	                	        	
                				}
                				
                	        	dec(graph);
                	        	
                			}
                			
                			@Override
                			public void exception(ReadGraphImpl graph, Throwable t) throws DatabaseException {
        	 					except(t);
                				dec(graph);
                            }

                		});
                		
                	}
                	
                	@Override
                	public void finished(ReadGraphImpl graph) throws DatabaseException {
                		dec(graph);
                	}
        			
        			@Override
        			public void exception(ReadGraphImpl graph, Throwable t) throws DatabaseException {
	 					except(t);
                        dec(graph);
                    }

                });

        	}
        	
        });
        
    }

    @Override
 	public Object compute(ReadGraphImpl graph, final InternalProcedure<TObjectIntHashMap<String>> procedure) throws DatabaseException {
    	
    	QueryProcessor processor = graph.processor;
    	
//    	System.err.println("NamespaceIndex " + id);
    	
        if("http://".equals(id) || "http:/".equals(id)) {
            index(graph, processor, processor.getRootLibrary(), procedure);
        } else {
            final String[] parts = URIStringUtils.splitURI(id);
            if(parts != null) {
                QueryCache.runnerNamespaceIndex(graph, parts[0], this, null, new InternalProcedure<TObjectIntHashMap<String>>() {
    
                    @Override
                    public void execute(ReadGraphImpl graph, TObjectIntHashMap<String> index) throws DatabaseException {
    
                        if(index != null) {
                            index(graph, processor, index.get(parts[1]), procedure);
                        } else {
                            add2(graph, null);
                            procedure.execute(graph, null);
//                            System.err.println("NamespaceIndex[" + id + "]->null");
                        }
                        
                    }
    
                    @Override
                    public void exception(ReadGraphImpl graph, Throwable t) throws DatabaseException {
                        if(DebugException.DEBUG) new DebugException(t).printStackTrace();
                        except(t);
                        procedure.exception(graph, t);
                    }
    
                });
            } else {
                add2(graph, null);
                procedure.execute(graph, null);
//                System.err.println("NamespaceIndex[" + id + "]->null");
            }

        }
        
        return getResult();
        
    }

    @Override
    public String toString() {
    	return "NamespaceIndex[" + id + "]";
    }

    synchronized private void add(TObjectIntHashMap<String> result) {
    	
    	throw new Error("Not possible!");
    	
    }

    private void add2(ReadGraphImpl graph, TObjectIntHashMap<String> result) {
    	
    	if(!isPending()) {
        	new Exception(""+hashCode()).printStackTrace();
    	}
    	
    	assert(isPending());

//        ArrayList<InternalProcedure<TObjectIntHashMap<String>>> p = null;

        synchronized(this) {

            setResult(result);
        	setReady();
//            p = procs;
//            procs = null; 
        
        }
        
//        if(p != null) {
//        
//	        for(InternalProcedure<TObjectIntHashMap<String>> proc : p) proc.execute(graph, result);
//	        
//        }
    	
    }
    
    @Override
    public Object performFromCache(ReadGraphImpl graph, InternalProcedure<TObjectIntHashMap<String>> procedure) throws DatabaseException {
        
        assert(isReady());
        
    	if(handleException(graph, procedure)) return (Throwable)statusOrException;
        
    	TObjectIntHashMap<String> result = (TObjectIntHashMap<String>)getResult();
    	
        procedure.execute(graph, result);
        
        return result;
        
    }
    
    @Override
    public void recompute(ReadGraphImpl graph) throws DatabaseException {
        
        compute(graph, new InternalProcedure<TObjectIntHashMap<String>>() {

            @Override
            public void execute(ReadGraphImpl graph, TObjectIntHashMap<String> result) {
            }
			
			@Override
			public void exception(ReadGraphImpl graph, Throwable t) {
                if(DebugException.DEBUG) new DebugException(t).printStackTrace();
				throw new Error("Error in recompute.", t);
            }

        });
        
    }
    
}