/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 *  contributor license agreements.  See the NOTICE file distributed with
 *  this work for additional information regarding copyright ownership.
 *  The ASF licenses this file to You under the Apache License, Version 2.0
 *  (the "License"); you may not use this file except in compliance with
 *  the License.  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 */
package org.apache.lenya.cms.publication;

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

import org.apache.lenya.ac.impl.AbstractAccessControlTest;
import org.apache.lenya.cms.site.NodeSet;
import org.apache.lenya.cms.site.SiteException;
import org.apache.lenya.cms.site.SiteNode;
import org.apache.lenya.cms.site.SiteStructure;
import org.apache.lenya.cms.site.SiteUtil;
import org.apache.lenya.util.StringUtil;

/**
 * Document manager test.
 */
public class DocumentManagerTest extends AbstractAccessControlTest {

    /**
     * Do the test.
     * @throws Exception
     */
    public void testDocumentManager() throws Exception {
        DocumentManager docManager = null;
        try {
            docManager = (DocumentManager) getManager().lookup(DocumentManager.ROLE);
            doTestMoveAll(docManager, "/doctypes", "/tutorial/doctypes");
            doTestMoveAll(docManager, "/tutorial/doctypes", "/doctypes");
            doTestCopyAll(docManager, "/doctypes", "/tutorial/doctypes");

            String areaName1 = "authoring";
            String areaName2 = "live";
            String path1 = "/tutorial";
            String path2 = "/doctypes";

            doTestCopyToArea(docManager, areaName1, areaName2, path1, path2);

        } finally {
            if (docManager != null) {
                getManager().release(docManager);
            }
        }
    }

    protected void doTestCopyToArea(DocumentManager docManager, String sourceAreaName,
            String destAreaName, String path1, String path2) throws PublicationException,
            SiteException {
        
        DocumentFactory factory = getFactory();
        Publication pub = factory.getPublication("test");
        SiteStructure sourceArea = pub.getArea(sourceAreaName).getSite();
        SiteStructure destArea = pub.getArea(destAreaName).getSite();

        if (destArea.contains(path1)) {
            destArea.getNode(path1).delete();
        }
        if (destArea.contains(path2)) {
            destArea.getNode(path2).delete();
        }
        
        assertFalse(destArea.contains(path1));
        assertFalse(destArea.contains(path2));
        
        // copy second node first to test correct ordering
        doTestCopyToArea(docManager, path2, sourceAreaName, destAreaName);
        doTestCopyToArea(docManager, path1, sourceAreaName, destAreaName);

        List sourceNodes = Arrays.asList(sourceArea.getNodes());

        SiteNode authoringNode1 = sourceArea.getNode(path1);
        assertTrue(sourceNodes.contains(authoringNode1));
        int sourcePos1 = sourceNodes.indexOf(authoringNode1);

        SiteNode sourceNode2 = sourceArea.getNode(path2);
        int sourcePos2 = sourceNodes.indexOf(sourceNode2);

        assertTrue(sourcePos1 < sourcePos2);

        assertTrue(destArea.contains(path1));
        List liveNodes = Arrays.asList(destArea.getNodes());
        SiteNode liveNode1 = destArea.getNode(path1);
        assertTrue(liveNodes.contains(liveNode1));
        int livePos1 = liveNodes.indexOf(liveNode1);

        SiteNode liveNode2 = destArea.getNode(path2);
        int livePos2 = liveNodes.indexOf(liveNode2);

        assertTrue(livePos1 < livePos2);
    }

    protected void doTestCopyToArea(DocumentManager docManager, String path, String areaName1,
            String areaName2) throws PublicationException {
        DocumentFactory factory = getFactory();
        Publication pub = factory.getPublication("test");
        Area area1 = pub.getArea(areaName1);
        Document doc = area1.getSite().getNode(path).getLink("en").getDocument();
        docManager.copyToArea(doc, areaName2);
    }

    protected void doTestCopyAll(DocumentManager docManager, String sourcePath, String targetPath)
            throws SiteException, DocumentException, PublicationException {
        DocumentFactory factory = getFactory();
        Publication pub = factory.getPublication("test");
        Area authoring = pub.getArea("authoring");

        SiteNode sourceNode = authoring.getSite().getNode(sourcePath);
        
        NodeSet nodes = SiteUtil.getSubSite(getManager(), sourceNode);
        Document[] docs = nodes.getDocuments();
        Map doc2path = new HashMap();

        String sourceBase = sourcePath.substring(0, sourcePath.lastIndexOf("/"));
        String targetBase = targetPath.substring(0, targetPath.lastIndexOf("/"));

        for (int i = 0; i < docs.length; i++) {
            doc2path.put(docs[i], docs[i].getPath().substring(sourceBase.length()));
        }

        docManager.copyAll(authoring, sourcePath, authoring, targetPath);
        for (int i = 0; i < docs.length; i++) {
            assertTrue(docs[i].hasLink());
            String oldPath = (String) doc2path.get(docs[i]);
            String newPath = targetBase + oldPath;
            assertTrue(authoring.getSite().contains(newPath));
            SiteNode newNode = authoring.getSite().getNode(newPath);
            Document newDoc = newNode.getLink(docs[i].getLanguage()).getDocument();
            assertEquals(newDoc.getContentLength(), docs[i].getContentLength());
            assertFalse(newDoc.getUUID().equals(docs[i].getUUID()));
        }
        
        String[] sourceNames = getChildNames(sourceNode);
        SiteNode targetNode = authoring.getSite().getNode(targetPath);
        String[] targetNames = getChildNames(targetNode);
        assertEquals(StringUtil.join(sourceNames, ","), StringUtil.join(targetNames, ","));
    }

	protected String[] getChildNames(SiteNode node) {
		SiteNode[] sourceChildren = node.getChildren();
        String[] names = new String[sourceChildren.length];
        for (int i = 0; i < names.length; i++) {
        	names[i] = sourceChildren[i].getName();
        }
		return names;
	}

    protected void doTestMoveAll(DocumentManager docManager, String sourcePath, String targetPath)
            throws SiteException, DocumentException, PublicationException {
        DocumentFactory factory = getFactory();
        Publication pub = factory.getPublication("test");
        Area authoring = pub.getArea("authoring");

        SiteNode sourceNode = authoring.getSite().getNode(sourcePath);
        String[] sourceNames = getChildNames(sourceNode);

        NodeSet nodes = SiteUtil.getSubSite(getManager(), sourceNode);
        Document[] docs = nodes.getDocuments();
        Map doc2path = new HashMap();

        String sourceBase = sourcePath.substring(0, sourcePath.lastIndexOf("/"));
        String targetBase = targetPath.substring(0, targetPath.lastIndexOf("/"));

        for (int i = 0; i < docs.length; i++) {
            doc2path.put(docs[i], docs[i].getPath().substring(sourceBase.length()));
        }

        docManager.moveAll(authoring, sourcePath, authoring, targetPath);
        for (int i = 0; i < docs.length; i++) {
            assertTrue(docs[i].hasLink());
            String oldPath = (String) doc2path.get(docs[i]);
            String newPath = docs[i].getPath();
            assertEquals(targetBase + oldPath, newPath);
        }
        
        SiteNode targetNode = authoring.getSite().getNode(targetPath);
        String[] targetNames = getChildNames(targetNode);
        assertEquals(StringUtil.join(sourceNames, ","), StringUtil.join(targetNames, ","));
        
    }
}