/*
 *  Tanaguru - Automated webpage assessment
 *  Copyright (C) 2008-2015  Tanaguru.org
 * 
 *  This file is part of Tanaguru.
 * 
 *  Tanaguru is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU Affero General Public License as
 *  published by the Free Software Foundation, either version 3 of the
 *  License, or (at your option) any later version.
 * 
 *  This program 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 Affero General Public License for more details.
 * 
 *  You should have received a copy of the GNU Affero General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 * 
 *  Contact us by mail: tanaguru AT tanaguru DOT org
 */
package org.tanaguru.service.command;

import java.util.List;
import java.util.Map;
import java.util.Set;
import javax.persistence.PersistenceException;
import org.apache.log4j.Logger;
import org.tanaguru.entity.audit.AuditStatus;
import org.tanaguru.entity.audit.Content;
import org.tanaguru.entity.parameterization.Parameter;
import org.tanaguru.entity.service.audit.AuditDataService;
import org.tanaguru.entity.subject.Page;
import org.tanaguru.entity.subject.Site;
import org.tanaguru.entity.subject.WebResource;
import org.tanaguru.service.ContentLoaderService;
import org.tanaguru.util.FileNaming;

/**
 *
 * @author jkowalczyk
 */
public class UploadAuditCommandImpl extends AuditCommandImpl {

    private static final Logger LOGGER = Logger.getLogger(UploadAuditCommandImpl.class);
    
    /**
     * The map that contains the files to test identified by 
     */
    private Map<String, String> fileMap;

    /**
     * The contentLoaderService
     */
    private ContentLoaderService contentLoaderService;
    public ContentLoaderService getContentLoaderService() {
        return contentLoaderService;
    }

    public void setContentLoaderService(ContentLoaderService contentLoaderService) {
        this.contentLoaderService = contentLoaderService;
    }
    
    /**
     * 
     * @param fileMap
     * @param paramSet
     * @param auditDataService 
     */
    public UploadAuditCommandImpl(
            Map<String, String> fileMap, 
            Set<Parameter> paramSet,
            AuditDataService auditDataService) {
        super(paramSet, auditDataService);
        
        this.fileMap = fileMap;
    }
    
    /**
     * 
     * @param fileMap
     * @param paramSet
     * @param auditDataService 
     * @param w3cValidatorPath 
     * @param java8Path
     */
    public UploadAuditCommandImpl(
            Map<String, String> fileMap, 
            Set<Parameter> paramSet,
            AuditDataService auditDataService,
            String w3cValidatorPath,
            String java8Path) {
        super(paramSet, auditDataService, w3cValidatorPath, java8Path);
        
        this.fileMap = fileMap;
    }

    @Override
    public void init() {
        super.init();
        setStatusToAudit(AuditStatus.CONTENT_LOADING);
    }

    @Override
    public void loadContent() {
        if (LOGGER.isInfoEnabled()) {
            LOGGER.info("Loading files content " + fileMap);
        }
        if (!getAudit().getStatus().equals(AuditStatus.CONTENT_LOADING) || fileMap.isEmpty()) {
            LOGGER.warn(
                    new StringBuilder("Audit Status is ")
                    .append(getAudit().getStatus())
                    .append(" while ")
                    .append(AuditStatus.CONTENT_LOADING)
                    .append(" was required ").toString());
            setStatusToAudit(AuditStatus.ERROR);
            return;
        }

        createWebResources();
        //call the load content service to convert files into SSP and link it
        //to the appropriate webResource
        List<Content> contentList = contentLoaderService.loadContent(getAudit().getSubject(), fileMap);

        for (Content content : contentList) {

            content.setAudit(getAudit());
            try {
                getContentDataService().saveOrUpdate(content);
            } catch (PersistenceException pe) {
                getAudit().setStatus(AuditStatus.ERROR);
                break;
            }
        }

        setStatusToAudit(AuditStatus.CONTENT_ADAPTING);
        if (LOGGER.isInfoEnabled()) {
            LOGGER.info(fileMap +" has been loaded");
        }
    }
    
    private void createWebResources() {
        WebResource webResource;
        if (fileMap.size() > 1) {
            webResource = getWebResourceDataService().createSite(
                    FileNaming.addProtocolToUrl(fileMap.keySet().iterator().next()));
            getWebResourceDataService().saveOrUpdate(webResource);
            for (String pageUrl : fileMap.keySet()) {
                Page page = getWebResourceDataService().createPage(pageUrl);
                ((Site) webResource).addChild(page);
                getWebResourceDataService().saveOrUpdate(page);
            }
        } else {
            webResource = getWebResourceDataService().
                    createPage(fileMap.keySet().iterator().next());
        }
        // the webresource needs to be persisted a second time because of the
        // relation with the audit
        webResource.setAudit(getAudit());
        getWebResourceDataService().saveOrUpdate(webResource);
        getAudit().setSubject(webResource);
    }

}