// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.container.core.config.testutil;

import com.google.inject.Guice;
import com.yahoo.component.AbstractComponent;
import com.yahoo.component.provider.ComponentRegistry;
import com.yahoo.config.subscription.ConfigSourceSet;
import com.yahoo.container.Container;
import com.yahoo.container.di.CloudSubscriberFactory;
import com.yahoo.container.di.ComponentDeconstructor;
import com.yahoo.container.core.config.HandlersConfigurerDi;
import com.yahoo.jdisc.handler.RequestHandler;
import com.yahoo.osgi.MockOsgi;

import java.io.File;
import java.io.IOException;
import java.util.LinkedHashSet;
import java.util.Random;
import java.util.Set;

/**
 * Class for testing HandlersConfigurer.
 * Not for public use.
 *
 * If possible, please avoid using this class and HandlersConfigurer in your tests
 * @author Tony Vaagenes
 * @author gjoranv
 *
*/
public class HandlersConfigurerTestWrapper {
    private ConfigSourceSet configSources =
            new ConfigSourceSet(this.getClass().getSimpleName() + ": " + new Random().nextLong());
    private HandlersConfigurerDi configurer;

    // TODO: Remove once tests use ConfigSet rather than dir:
    private final static String testFiles[] = {
            "components.cfg",
            "handlers.cfg",
            "bundles.cfg",
            "string.cfg",
            "int.cfg",
            "renderers.cfg",
            "diagnostics.cfg",
            "qr-templates.cfg",
            "documentmanager.cfg",
            "schemamapping.cfg",
            "chains.cfg",
            "container-mbus.cfg",
            "container-mbus.cfg",
            "specialtokens.cfg",
            "documentdb-info.cfg",
            "qr-search.cfg",
            "query-profiles.cfg"
    };
    private final Set<File> createdFiles = new LinkedHashSet<>();
    private int lastGeneration = 0;
    private final Container container;

    private void createFiles(String configId) {
        if (configId.startsWith("dir:")) {
            try {
                System.setProperty("config.id", configId);
                String dirName = configId.substring(4);
                for (String file : testFiles) {
                    createIfNotExists(dirName, file);
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    // TODO: Remove once tests use ConfigSet rather than dir:
    private void createIfNotExists(String dir, String file) throws IOException {
        final File f = new File(dir + "/" + file);
        if (f.createNewFile()) {
            createdFiles.add(f);
        }
    }

    public HandlersConfigurerTestWrapper(String configId) {
        this(Container.get(), configId);
    }

    public HandlersConfigurerTestWrapper(Container container, String configId) {
        createFiles(configId);
        MockOsgi mockOsgi = new MockOsgi();
        container.setOsgi(mockOsgi);
        ComponentDeconstructor testDeconstructor = getTestDeconstructor();
        configurer = new HandlersConfigurerDi(
                new CloudSubscriberFactory(configSources),
                container,
                configId,
                testDeconstructor,
                Guice.createInjector(),
                mockOsgi);
        this.container = container;
    }

    private ComponentDeconstructor getTestDeconstructor() {
        return new ComponentDeconstructor() {
            @Override
            public void deconstruct(Object component) {
                if (component instanceof AbstractComponent) {
                    AbstractComponent abstractComponent = (AbstractComponent) component;
                    if (abstractComponent.isDeconstructable())
                        ((AbstractComponent) component).deconstruct();
            }
        }};
    }

    public void reloadConfig() {
        configurer.reloadConfig(++lastGeneration);
        configurer.getNewComponentGraph(Guice.createInjector(), false);
    }

    public void shutdown() {
        // TODO: Remove once tests use ConfigSet rather than dir:
        for (File f : createdFiles) {
            f.delete();
        }
    }

    public ComponentRegistry<RequestHandler> getRequestHandlerRegistry() {
        return container.getRequestHandlerRegistry();
    }

}

@@ -188,21 +188,97 @@
 .inject.
-Guice
+AbstractModule;%0Aimport com.google.inject.Guice;%0Aimport com.google.inject.Injector
 ;%0Aimport
@@ -676,24 +676,115 @@
 estHandler;%0A
+import com.yahoo.language.Linguistics;%0Aimport com.yahoo.language.simple.SimpleLinguistics;%0A
 import com.y
@@ -3461,28 +3461,21 @@
         
-G
+g
 uice
-.create
 Injector
@@ -4178,20 +4178,13 @@
 aph(
-G
+g
 uice
-.create
 Inje
@@ -4451,24 +4451,24 @@
 egistry() %7B%0A
-
         retu
@@ -4516,10 +4516,336 @@
 %0A    %7D%0A%0A
+    private static Injector guiceInjector() %7B%0A        return Guice.createInjector(new AbstractModule() %7B%0A            @Override%0A            protected void configure() %7B%0A                // Needed by e.g. SearchHandler%0A                bind(Linguistics.class).toInstance(new SimpleLinguistics());%0A            %7D%0A        %7D);%0A    %7D%0A%0A
 %7D%0A

Add binding for Linguistics to HandlersConfigurerTestWrapper.