/*
 *  eXist Open Source Native XML Database
 *  Copyright (C) 2001-04 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 program; if not, write to the Free Software
 *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *  
 *  $Id$
 */
package org.exist.xmldb;

import org.exist.TestUtils;
import org.exist.util.FileUtils;
import org.exist.util.MimeTable;
import org.exist.util.MimeType;
import org.exist.util.SingleInstanceConfiguration;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.xmldb.api.DatabaseManager;
import org.xmldb.api.base.*;
import org.xmldb.api.modules.CollectionManagementService;
import org.xmldb.api.modules.XMLResource;

import java.io.OutputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.List;

import static java.nio.charset.StandardCharsets.UTF_8;

/**
 * @author wolf
 *
 */
public class MultiDBTest {

//    public static void main(String[] args) {
//        TestRunner.run(MultiDBTest.class);
//    }
    
    private final static int INSTANCE_COUNT = 5;
    
    private final static String CONFIG =
        "<exist>" +
        "   <db-connection database=\"native\" files=\".\" cacheSize=\"32M\">" +
        "       <pool min=\"1\" max=\"5\" sync-period=\"120000\"/>" +
        "   </db-connection>" +
        "</exist>";

    @Test
    public void store()
       throws Exception
    {
        for (int i = 0; i < INSTANCE_COUNT; i++) {
            Collection root = DatabaseManager.getCollection("xmldb:test" + i + "://" + XmldbURI.ROOT_COLLECTION);
            Collection test = root.getChildCollection("test");
            if (test == null) {
                CollectionManagementService service = (CollectionManagementService)
                    root.getService("CollectionManagementService", "1.0");
                test = service.createCollection("test");
            }

            final Path samples = TestUtils.shakespeareSamples();
            final List<Path> files = FileUtils.list(samples);
            final MimeTable mimeTab = MimeTable.getInstance();
            for (final Path file : files) {
                final MimeType mime = mimeTab.getContentTypeFor(FileUtils.fileName(file));
                if(mime != null && mime.isXMLType()) {
                    loadFile(test, file.toAbsolutePath().toString());
                }
            }

            doQuery(test, "//SPEECH[SPEAKER='HAMLET']");
        }
    }
    
    protected static void loadFile(Collection collection, String path) throws XMLDBException {
        // create new XMLResource; an id will be assigned to the new resource
        XMLResource document = (XMLResource)
            collection.createResource(path.substring(path.lastIndexOf(java.io.File.separatorChar)),
                "XMLResource");
        document.setContent(Paths.get(path));
        collection.storeResource(document);
    }
    
    private static void doQuery(Collection collection, String query) throws XMLDBException {
        XQueryService service = (XQueryService)
            collection.getService("XQueryService", "1.0");
        ResourceSet result = service.query(query);
        for(ResourceIterator i = result.getIterator(); i.hasMoreResources(); ) {
            @SuppressWarnings("unused")
            String content = i.nextResource().getContent().toString();
        }
    }

    @Before
    public void setUp() throws Exception {
        Path homeDir = SingleInstanceConfiguration.getPath().map(Path::getParent).orElse(Paths.get("."));

        Path testDir = homeDir.resolve("test").resolve("temp");
        Files.createDirectories(testDir);

       // initialize database drivers
       Class<?> cl = Class.forName("org.exist.xmldb.DatabaseImpl");
       for (int i = 0; i < INSTANCE_COUNT; i++) {
          Path dir = testDir.resolve("db" + i);
          Files.createDirectories(dir);
          Path conf = dir.resolve("conf.xml");
          try(final OutputStream os = Files.newOutputStream(conf)) {
              os.write(CONFIG.getBytes(UTF_8));
          }

          Database database = (Database) cl.newInstance();
          database.setProperty("create-database", "true");
          database.setProperty("configuration", conf.toAbsolutePath().toString());
          database.setProperty("database-id", "test" + i);
          DatabaseManager.registerDatabase(database);
       }
    }

    @After
    public void tearDown()
       throws Exception
    {
        Runtime rt = Runtime.getRuntime();
        long free = rt.freeMemory() / 1024;
        long total = rt.totalMemory() / 1024;
        for (int i = 0; i < INSTANCE_COUNT; i++) {
            Collection root = DatabaseManager.getCollection("xmldb:test" + i + "://" + XmldbURI.ROOT_COLLECTION, "admin", "");
            CollectionManagementService service = (CollectionManagementService)
                root.getService("CollectionManagementService", "1.0");
            service.removeCollection("test");

            DatabaseInstanceManager mgr = (DatabaseInstanceManager)
                root.getService("DatabaseInstanceManager", "1.0");
            mgr.shutdown();
        }
    }
}