/*
 * See the NOTICE file distributed with this work for additional
 * information regarding copyright ownership.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */
package org.xwiki.officeimporter.internal.splitter;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.inject.Inject;
import javax.inject.Named;
import javax.inject.Singleton;

import org.xwiki.bridge.DocumentAccessBridge;
import org.xwiki.component.annotation.Component;
import org.xwiki.component.manager.ComponentManager;
import org.xwiki.model.EntityType;
import org.xwiki.model.reference.DocumentReference;
import org.xwiki.model.reference.DocumentReferenceResolver;
import org.xwiki.model.reference.EntityReferenceSerializer;
import org.xwiki.officeimporter.OfficeImporterException;
import org.xwiki.officeimporter.document.XDOMOfficeDocument;
import org.xwiki.officeimporter.splitter.TargetDocumentDescriptor;
import org.xwiki.officeimporter.splitter.XDOMOfficeDocumentSplitter;
import org.xwiki.refactoring.WikiDocument;
import org.xwiki.refactoring.splitter.DocumentSplitter;
import org.xwiki.refactoring.splitter.criterion.HeadingLevelSplittingCriterion;
import org.xwiki.refactoring.splitter.criterion.SplittingCriterion;
import org.xwiki.refactoring.splitter.criterion.naming.NamingCriterion;
import org.xwiki.rendering.block.Block.Axes;
import org.xwiki.rendering.block.IdBlock;
import org.xwiki.rendering.block.LinkBlock;
import org.xwiki.rendering.block.match.ClassBlockMatcher;
import org.xwiki.rendering.renderer.BlockRenderer;

/**
 * Default implementation of {@link XDOMOfficeDocumentSplitter}.
 * 
 * @version $Id: 39088f430147ef2de5beac63d0f5f4cd9ccec7d7 $
 * @since 2.1M1
 */
@Component
@Singleton
public class DefaultXDOMOfficeDocumentSplitter implements XDOMOfficeDocumentSplitter
{
    /**
     * Document access bridge.
     */
    @Inject
    private DocumentAccessBridge docBridge;

    /**
     * Plain text renderer used for rendering heading names.
     */
    @Inject
    @Named("plain/1.0")
    private BlockRenderer plainTextRenderer;

    /**
     * Document name serializer used for serializing document names into strings.
     */
    @Inject
    @Named("compactwiki")
    private EntityReferenceSerializer<String> entityReferenceSerializer;

    /**
     * Required for converting string document names to {@link org.xwiki.model.reference.DocumentReference} instances.
     */
    @Inject
    @Named("currentmixed")
    private DocumentReferenceResolver<String> currentMixedDocumentReferenceResolver;

    /**
     * The {@link DocumentSplitter} used for splitting wiki documents.
     */
    @Inject
    private DocumentSplitter documentSplitter;

    /**
     * Used by {@link org.xwiki.officeimporter.splitter.TargetDocumentDescriptor}.
     */
    @Inject
    private ComponentManager componentManager;

    @Override
    public Map<TargetDocumentDescriptor, XDOMOfficeDocument> split(XDOMOfficeDocument officeDocument,
        int[] headingLevelsToSplit, String namingCriterionHint, DocumentReference baseDocumentReference)
        throws OfficeImporterException
    {
        // TODO: This code needs to be refactored along with the xwiki-refactoring module code.
        String strBaseDoc = this.entityReferenceSerializer.serialize(baseDocumentReference);
        Map<TargetDocumentDescriptor, XDOMOfficeDocument> result =
            new HashMap<TargetDocumentDescriptor, XDOMOfficeDocument>();

        // Create splitting and naming criterion for refactoring.
        SplittingCriterion splittingCriterion = new HeadingLevelSplittingCriterion(headingLevelsToSplit);
        NamingCriterion namingCriterion = DocumentSplitterUtils.getNamingCriterion(
            namingCriterionHint, strBaseDoc, this.docBridge, this.plainTextRenderer);

        // Create the root document required by refactoring module.
        WikiDocument rootDoc = new WikiDocument(strBaseDoc, officeDocument.getContentDocument(), null);
		Map<String, String> anchorMap = new HashMap<String, String> ();
        List<WikiDocument> documents = this.documentSplitter.split(rootDoc, splittingCriterion, namingCriterion);
		WikiDocument mainDoc = null;
        for (WikiDocument doc : documents) {
            // Initialize a target page descriptor.
            DocumentReference targetReference = this.currentMixedDocumentReferenceResolver.resolve(doc.getFullName());
            TargetDocumentDescriptor targetDocumentDescriptor =
                new TargetDocumentDescriptor(targetReference, this.componentManager);
            if (doc.getParent() != null) { // A split document
                DocumentReference targetParent =
                    this.currentMixedDocumentReferenceResolver.resolve(doc.getParent().getFullName());
                targetDocumentDescriptor.setParentReference(targetParent);
                try {
                	// Keep a reference to all id blocks (anchors) so that we can adjust the mainDoc afterwards
                	// The id anchor is going to look something like
                	// == {{id name="__RefHeading___Toc341868357"/}}6.6.Networking ==

					List<IdBlock> idblocks = doc.getXdom().getBlocks(
							new ClassBlockMatcher(IdBlock.class),
							Axes.DESCENDANT);
					for (IdBlock ib : idblocks) {
						// System.out.println("Anchor found " + ib.getName());
						// Put a pound in front, so that it will equal the string for the link in the mainDoc below
						String anchorName = "#" + ib.getName();
						// Store in the anchorName with the new page name to change the mainDoc's link below
						anchorMap.put(anchorName, doc.getFullName()
								+ anchorName);

					}
                } catch(Exception e){
                	e.printStackTrace();
                }
    			

            } else {
				System.out.println("First document's parent is null " + doc + " and the targetDocumentDescriptor " + targetDocumentDescriptor);
				mainDoc = doc;
				
			}

            // Rewire artifacts.
            Map<String, byte[]> artifacts = DocumentSplitterUtils.relocateArtifacts(doc, officeDocument);

            // Create the resulting XDOMOfficeDocument.
            XDOMOfficeDocument splitDocument = new XDOMOfficeDocument(doc.getXdom(), artifacts, this.componentManager);
            result.put(targetDocumentDescriptor, splitDocument);
            
        }
        // mainDoc has links that point to non-existent anchors on its page, since they were split into
        // sub documents DJ 
        List<LinkBlock> linkBlocks =
                mainDoc.getXdom().getBlocks(new ClassBlockMatcher(LinkBlock.class), Axes.DESCENDANT);
        if (linkBlocks != null && linkBlocks.size() > 0){
            for (LinkBlock linkBlock : linkBlocks) {
            	if (linkBlock != null && linkBlock.getReference() != null){
            	String refer = linkBlock.getReference().getReference();
            	// in the example above, the wiki text is going to look like this in the appendix
            	// **6.6.** **Networking [[13>>#__RefHeading___Toc341868357]]**
            	//So I need to change it to 
            	// **6.6.** **Networking [[13>>Main.6-Application System Architecture-6#__RefHeading___Toc341868357]]**
					if (anchorMap != null) {
						for (String anchor : anchorMap.keySet()) {
							if (anchor != null && anchor.equals(refer)) {
								linkBlock.getReference().setReference(
										anchorMap.get(anchor));
								break;
							}
						}
					}
      	}
            }
        }
	
        return result;
    }
}