/*
 * Copyright (C) 2005-2010 Alfresco Software Limited.
 *
 * This file is part of Alfresco
 *
 * Alfresco 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 3 of the License, or
 * (at your option) any later version.
 *
 * Alfresco 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 Alfresco. If not, see <http://www.gnu.org/licenses/>.
 */

package org.alfresco.repo.rendition.executer;

import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.TreeSet;

import org.alfresco.model.ContentModel;
import org.alfresco.repo.model.filefolder.FileFolderServiceImpl;
import org.alfresco.service.cmr.model.FileFolderService;
import org.alfresco.service.cmr.model.FileInfo;
import org.alfresco.service.cmr.repository.ContentService;
import org.alfresco.service.cmr.repository.ContentWriter;
import org.alfresco.service.cmr.repository.NodeRef;
import org.alfresco.service.cmr.repository.NodeService;
import org.alfresco.service.cmr.repository.StoreRef;
import org.alfresco.service.cmr.search.ResultSet;
import org.alfresco.service.cmr.search.SearchService;
import org.alfresco.test_category.BaseSpringTestsCategory;
import org.alfresco.util.BaseAlfrescoSpringTest;
import org.alfresco.util.GUID;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.junit.experimental.categories.Category;
import org.w3c.dom.Document;
import org.w3c.dom.NodeList;

/**
 * @author Brian
 *
 */
@Category(BaseSpringTestsCategory.class)
public class XSLTFunctionsTest extends BaseAlfrescoSpringTest
{
    private final static Log log = LogFactory.getLog(XSLTFunctionsTest.class);
    private XSLTFunctions xsltFunctions;
    private SearchService searchService;
    private NodeRef companyHome;
    private FileFolderService fileFolderService;

    /* (non-Javadoc)
     * @see org.alfresco.util.BaseAlfrescoSpringTest#onSetUpInTransaction()
     */
    @SuppressWarnings("deprecation")
    @Override
    protected void onSetUpInTransaction() throws Exception
    {
        super.onSetUpInTransaction();
        this.searchService = (SearchService) this.applicationContext.getBean("SearchService");
        this.xsltFunctions = (XSLTFunctions) this.applicationContext.getBean("xsltFunctions");
        this.nodeService = (NodeService) this.applicationContext.getBean("NodeService");
        this.contentService = (ContentService) this.applicationContext.getBean("ContentService");
        this.fileFolderService = (FileFolderService) this.applicationContext.getBean("FileFolderService");
        ResultSet rs = searchService.query(StoreRef.STORE_REF_WORKSPACE_SPACESSTORE, SearchService.LANGUAGE_XPATH, "/app:company_home");
        this.companyHome = rs.getNodeRef(0);
    }

    public void testSimplestParseXMLDocument()
    {
        FileInfo file = createXmlFile(companyHome);
        try
        {
            Document doc = xsltFunctions.parseXMLDocument(companyHome, file.getName());
            NodeList foodNodes = doc.getElementsByTagName("food");
            assertEquals(10, foodNodes.getLength());
        }
        catch (Exception ex)
        {
            log.error("Error!", ex);
            fail(ex.getMessage());
        }
    }

    public void testPathParseXMLDocument()
    {
        String path = "path/to/xml/files";
        List<String> pathElements = Arrays.asList(path.split("/"));
        FileInfo folder = FileFolderServiceImpl.makeFolders(fileFolderService, companyHome, pathElements, ContentModel.TYPE_FOLDER);
        FileInfo file = createXmlFile(folder.getNodeRef());
        
        try
        {
            Document doc = xsltFunctions.parseXMLDocument(companyHome, path + "/" + file.getName());
            NodeList foodNodes = doc.getElementsByTagName("food");
            assertEquals(10, foodNodes.getLength());
        }
        catch (Exception ex)
        {
            log.error("Error!", ex);
            fail(ex.getMessage());
        }
    }
    
    public void testParseXMLDocuments()
    {
        String path = "path/to/xml/files";
        List<String> pathElements = Arrays.asList(path.split("/"));
        FileInfo folder = FileFolderServiceImpl.makeFolders(fileFolderService, companyHome, pathElements, ContentModel.TYPE_FOLDER);
        FileInfo file1 = createXmlFile(folder.getNodeRef());
        FileInfo file2 = createXmlFile(folder.getNodeRef());
        FileInfo file3 = createXmlFile(folder.getNodeRef());
        FileInfo file4 = createXmlFile(folder.getNodeRef());
        FileInfo file5 = createXmlFile(folder.getNodeRef());
        
        try
        {
            Map<String,Document> xmlFileMap = xsltFunctions.parseXMLDocuments("cm:content", 
                    companyHome, "/" + path);
            assertEquals(5, xmlFileMap.size());
            Set<String> names = new TreeSet<String>(Arrays.asList(new String[] {file1.getName(), file2.getName(), 
                    file3.getName(), file4.getName(), file5.getName()}));
            names.removeAll(xmlFileMap.keySet());
            assertEquals(0, names.size());

            NodeList foodNodes = xmlFileMap.get(file3.getName()).getElementsByTagName("food");
            assertEquals(10, foodNodes.getLength());
        }
        catch (Exception ex)
        {
            log.error("Error!", ex);
            fail(ex.getMessage());
        }
    }
    
    /**
     * 
     */
    private FileInfo createXmlFile(NodeRef folder)
    {
        String name = GUID.generate();
        FileInfo testXmlFile = fileFolderService.create(folder, name + ".xml", ContentModel.TYPE_CONTENT);
        ContentWriter writer = contentService.getWriter(testXmlFile.getNodeRef(), ContentModel.PROP_CONTENT, true);
        writer.setMimetype("text/xml");
        writer.setEncoding("UTF-8");
        writer.putContent(sampleXML);
        return testXmlFile;
    }
    

    private String sampleXML = "<?xml version=\"1.0\"?>" +
    "<nutrition>" +

    "<daily-values>" +
        "<total-fat units=\"g\">65</total-fat>" +
        "<saturated-fat units=\"g\">20</saturated-fat>" +
        "<cholesterol units=\"mg\">300</cholesterol>" +
        "<sodium units=\"mg\">2400</sodium>" +
        "<carb units=\"g\">300</carb>" +
        "<fiber units=\"g\">25</fiber>" +
        "<protein units=\"g\">50</protein>" +
    "</daily-values>" +

    "<food>" +
        "<name>Avocado Dip</name>" +
        "<mfr>Sunnydale</mfr>" +
        "<serving units=\"g\">29</serving>" +
        "<calories total=\"110\" fat=\"100\"/>" +
        "<total-fat>11</total-fat>" +
        "<saturated-fat>3</saturated-fat>" +
        "<cholesterol>5</cholesterol>" +
        "<sodium>210</sodium>" +
        "<carb>2</carb>" +
        "<fiber>0</fiber>" +
        "<protein>1</protein>" +
        "<vitamins>" +
            "<a>0</a>" +
            "<c>0</c>" +
        "</vitamins>" +
        "<minerals>" +
            "<ca>0</ca>" +
            "<fe>0</fe>" +
        "</minerals>" +
    "</food>" +

    "<food>" +
        "<name>Bagels, New York Style </name>" +
        "<mfr>Thompson</mfr>" +
        "<serving units=\"g\">104</serving>" +
        "<calories total=\"300\" fat=\"35\"/>" +
        "<total-fat>4</total-fat>" +
        "<saturated-fat>1</saturated-fat>" +
        "<cholesterol>0</cholesterol>" +
        "<sodium>510</sodium>" +
        "<carb>54</carb>" +
        "<fiber>3</fiber>" +
        "<protein>11</protein>" +
        "<vitamins>" +
            "<a>0</a>" +
            "<c>0</c>" +
        "</vitamins>" +
        "<minerals>" +
            "<ca>8</ca>" +
            "<fe>20</fe>" +
        "</minerals>" +
    "</food>" +

    "<food>" +
        "<name>Beef Frankfurter, Quarter Pound </name>" +
        "<mfr>Armitage</mfr>" +
        "<serving units=\"g\">115</serving>" +
        "<calories total=\"370\" fat=\"290\"/>" +
        "<total-fat>32</total-fat>" +
        "<saturated-fat>15</saturated-fat>" +
        "<cholesterol>65</cholesterol>" +
        "<sodium>1100</sodium>" +
        "<carb>8</carb>" +
        "<fiber>0</fiber>" +
        "<protein>13</protein>" +
        "<vitamins>" +
            "<a>0</a>" +
            "<c>2</c>" +
        "</vitamins>" +
        "<minerals>" +
            "<ca>1</ca>" +
            "<fe>6</fe>" +
        "</minerals>" +
    "</food>" +

    "<food>" +
        "<name>Chicken Pot Pie</name>" +
        "<mfr>Lakeson</mfr>" +
        "<serving units=\"g\">198</serving>" +
        "<calories total=\"410\" fat=\"200\"/>" +
        "<total-fat>22</total-fat>" +
        "<saturated-fat>9</saturated-fat>" +
        "<cholesterol>25</cholesterol>" +
        "<sodium>810</sodium>" +
        "<carb>42</carb>" +
        "<fiber>2</fiber>" +
        "<protein>10</protein>" +
        "<vitamins>" +
            "<a>20</a>" +
            "<c>2</c>" +
        "</vitamins>" +
        "<minerals>" +
            "<ca>2</ca>" +
            "<fe>10</fe>" +
        "</minerals>" +
    "</food>" +

    "<food>" +
        "<name>Cole Slaw</name>" +
        "<mfr>Fresh Quick</mfr>" +
        "<serving units=\" cup\">1.5</serving>" +
        "<calories total=\"20\" fat=\"0\"/>" +
        "<total-fat>0</total-fat>" +
        "<saturated-fat>0</saturated-fat>" +
        "<cholesterol>0</cholesterol>" +
        "<sodium>15</sodium>" +
        "<carb>5</carb>" +
        "<fiber>2</fiber>" +
        "<protein>1</protein>" +
        "<vitamins>" +
            "<a>30</a>" +
            "<c>45</c>" +
        "</vitamins>" +
        "<minerals>" +
            "<ca>4</ca>" +
            "<fe>2</fe>" +
        "</minerals>" +
    "</food>" +

    "<food>" +
        "<name>Eggs</name>" +
        "<mfr>Goodpath</mfr>" +
        "<serving units=\"g\">50</serving>" +
        "<calories total=\"70\" fat=\"40\"/>" +
        "<total-fat>4.5</total-fat>" +
        "<saturated-fat>1.5</saturated-fat>" +
        "<cholesterol>215</cholesterol>" +
        "<sodium>65</sodium>" +
        "<carb>1</carb>" +
        "<fiber>0</fiber>" +
        "<protein>6</protein>" +
        "<vitamins>" +
            "<a>6</a>" +
            "<c>0</c>" +
        "</vitamins>" +
        "<minerals>" +
            "<ca>2</ca>" +
            "<fe>4</fe>" +
        "</minerals>" +
    "</food>" +

    "<food>" +
        "<name>Hazelnut Spread</name>" +
        "<mfr>Ferreira</mfr>" +
        "<serving units=\"tbsp\">2</serving>" +
        "<calories total=\"200\" fat=\"90\"/>" +
        "<total-fat>10</total-fat>" +
        "<saturated-fat>2</saturated-fat>" +
        "<cholesterol>0</cholesterol>" +
        "<sodium>20</sodium>" +
        "<carb>23</carb>" +
        "<fiber>2</fiber>" +
        "<protein>3</protein>" +
        "<vitamins>" +
            "<a>0</a>" +
            "<c>0</c>" +
        "</vitamins>" +
        "<minerals>" +
            "<ca>6</ca>" +
            "<fe>4</fe>" +
        "</minerals>" +
    "</food>" +

    "<food>" +
        "<name>Potato Chips</name>" +
        "<mfr>Lees</mfr>" +
        "<serving units=\"g\">28</serving>" +
        "<calories total=\"150\" fat=\"90\"/>" +
        "<total-fat>10</total-fat>" +
        "<saturated-fat>3</saturated-fat>" +
        "<cholesterol>0</cholesterol>" +
        "<sodium>180</sodium>" +
        "<carb>15</carb>" +
        "<fiber>1</fiber>" +
        "<protein>2</protein>" +
        "<vitamins>" +
            "<a>0</a>" +
            "<c>10</c>" +
        "</vitamins>" +
        "<minerals>" +
            "<ca>0</ca>" +
            "<fe>0</fe>" +
        "</minerals>" +
    "</food>" +

    "<food>" +
        "<name>Soy Patties, Grilled</name>" +
        "<mfr>Gardenproducts</mfr>" +
        "<serving units=\"g\">96</serving>" +
        "<calories total=\"160\" fat=\"45\"/>" +
        "<total-fat>5</total-fat>" +
        "<saturated-fat>0</saturated-fat>" +
        "<cholesterol>0</cholesterol>" +
        "<sodium>420</sodium>" +
        "<carb>10</carb>" +
        "<fiber>4</fiber>" +
        "<protein>9</protein>" +
        "<vitamins>" +
            "<a>0</a>" +
            "<c>0</c>" +
        "</vitamins>" +
        "<minerals>" +
            "<ca>0</ca>" +
            "<fe>0</fe>" +
        "</minerals>" +
    "</food>" +

    "<food>" +
        "<name>Truffles, Dark Chocolate</name>" +
        "<mfr>Lyndon's</mfr>" +
        "<serving units=\"g\">39</serving>" +
        "<calories total=\"220\" fat=\"170\"/>" +
        "<total-fat>19</total-fat>" +
        "<saturated-fat>14</saturated-fat>" +
        "<cholesterol>25</cholesterol>" +
        "<sodium>10</sodium>" +
        "<carb>16</carb>" +
        "<fiber>1</fiber>" +
        "<protein>1</protein>" +
        "<vitamins>" +
            "<a>0</a>" +
            "<c>0</c>" +
        "</vitamins>" +
        "<minerals>" +
            "<ca>0</ca>" +
            "<fe>0</fe>" +
        "</minerals>" +
    "</food>" +

    "</nutrition>";
}