/*
 *  eXist Open Source Native XML Database
 *  Copyright (C) 2001-2015 The eXist Project
 *  http://exist-db.org
 *
 *  This program 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
 *  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 Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public
 *  License along with this library; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */
package org.exist.performance.actions;

import org.exist.util.*;
import org.w3c.dom.Element;
import org.exist.EXistException;
import org.exist.xmldb.EXistResource;
import org.exist.performance.Connection;
import org.exist.performance.Runner;
import org.exist.performance.AbstractAction;
import org.xmldb.api.base.XMLDBException;
import org.xmldb.api.base.Collection;
import org.xmldb.api.base.Resource;
import org.xmldb.api.modules.CollectionManagementService;

import java.io.IOException;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.List;
import java.util.StringTokenizer;
import java.util.Observer;
import java.util.Observable;

public class StoreFromFile extends AbstractAction {

    private String collectionPath;
    private String dir;
    private String includes;
    private String mimeType = "application/xml";
    private boolean overwrite = true;

    public void configure(Runner runner, Action parent, Element config) throws EXistException {
        super.configure(runner, parent, config);
        if (!config.hasAttribute("collection"))
            throw new EXistException(StoreFromFile.class.getName() + " requires an attribute 'collection'");
        collectionPath = config.getAttribute("collection");
        if (!config.hasAttribute("dir"))
            throw new EXistException(StoreFromFile.class.getName() + " requires an attribute 'dir'");
        dir = config.getAttribute("dir");
        includes = config.getAttribute("includes");
        if (config.hasAttribute("mime-type"))
            mimeType = config.getAttribute("mime-type");
        overwrite = getBooleanValue(config, "overwrite", true);
    }

    @Override
    public void execute(Connection connection) throws XMLDBException, EXistException {
        Collection collection = connection.getCollection(collectionPath);
        if (collection == null)
            throw new EXistException("collection " + collectionPath + " not found");
        ProgressObserver observer = new ProgressObserver();
        if (collection instanceof Observable)
            ((Observable)collection).addObserver(observer);
        String resourceType = getResourceType();
        final Path baseDir = Paths.get(dir);
        try {
            final List<Path> files = DirectoryScanner.scanDir(baseDir, includes);
            Collection col = collection;
            String relDir, prevDir = null;
            for (final Path file : files) {
                String relPath = file.toString().substring(baseDir.toString().length());
                int p = relPath.lastIndexOf(java.io.File.separatorChar);
                relDir = relPath.substring(0, p);
                relDir = relDir.replace(java.io.File.separatorChar, '/');
                if (prevDir == null || (!relDir.equals(prevDir))) {
                    col = makeColl(collection, relDir);
                    if (col instanceof Observable)
                        ((Observable) col).addObserver(observer);
                    prevDir = relDir;
                }
                if (col.getResource(FileUtils.fileName(file)) == null || overwrite) {
                    //TODO  : these probably need to be encoded and check mime via MimeTable
                    Resource resource =
                            col.createResource(FileUtils.fileName(file), resourceType);
                    resource.setContent(file);
                    ((EXistResource) resource).setMimeType(mimeType);
                    LOG.debug("Storing " + col.getName() + "/" + resource.getId());
                    col.storeResource(resource);
                }
            }
        } catch(final IOException ioe) {
            throw new EXistException(ioe);
        }
    }

    private String getResourceType() {
        String resourceType = "XMLResource";
        MimeType mime = MimeTable.getInstance().getContentType(mimeType);
        if (mime != null)
            resourceType = mime.isXMLType() ? "XMLResource" : "BinaryResource";
        return resourceType;
    }

    private Collection makeColl(Collection parentColl, String relPath)
    throws XMLDBException {
        CollectionManagementService mgtService;
        Collection current = parentColl, c;
        String token;
        StringTokenizer tok = new StringTokenizer(relPath, "/");
        while (tok.hasMoreTokens()) {
            token = tok.nextToken();
            c = current.getChildCollection(token);
            if (c == null) {
                mgtService = (CollectionManagementService) current.getService("CollectionManagementService", "1.0");
                current = mgtService.createCollection(token);
            } else
                current = c;
        }
        return current;
    }

    private class ProgressObserver implements Observer {

        private long timestamp = System.currentTimeMillis();

        public void update(Observable o, Object arg) {
            if (System.currentTimeMillis() - timestamp > 20000) {
                ProgressIndicator ind = (ProgressIndicator) arg;
                if (!(o instanceof org.exist.storage.ElementIndex)) {
                    LOG.debug("Stored: " + (int)((ind.getValue() / ind.getMax()) * 100) + " %");
                }
                timestamp = System.currentTimeMillis();
            }
        }
    }
}