package org.embulk.input.zendesk.utils;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.datatype.jdk8.Jdk8Module;

import org.embulk.EmbulkTestRuntime;
import org.embulk.config.ConfigLoader;
import org.embulk.config.ConfigSource;
import org.embulk.config.ModelManager;
import org.embulk.spi.ExecSession;
import org.embulk.test.EmbulkTests;
import org.junit.Assert;

import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Field;

import static junit.framework.TestCase.fail;

public final class ZendeskTestHelper
{
    private ZendeskTestHelper() {}

    private static ObjectMapper mapper;
    private static final ConfigLoader configLoader;

    static {
        mapper = new ObjectMapper();
        mapper.registerModule(new Jdk8Module());
        configLoader = new ConfigLoader(new ModelManager(null, mapper));
    }

    public static JsonNode getJsonFromFile(String fileName)
    {
        try {
            return mapper.readTree(EmbulkTests.readResource(fileName));
        }
        catch (IOException e) {
            fail("Fail to load file from " + fileName);
        }
        return null;
    }

    public static ConfigSource getConfigSource(String file)
    {
        ConfigSource configSource = null;
        InputStream is = null;
        try {
            is = ZendeskTestHelper.class.getResourceAsStream("/config/" + file);
            configSource = configLoader.fromYaml(is);
        }
        catch (IOException ex) {
            Assert.fail(ex.getMessage());
        }
        finally {
            if (is != null) {
                try {
                    is.close();
                }
                catch (IOException ex) { }
            }
        }
        return configSource;
    }

    public static void setPreviewMode(final EmbulkTestRuntime runtime, final boolean isPreview)
    {
        // A small hack to make the plugin executed in preview mode so
        try {
            final Field previewField = ExecSession.class.getDeclaredField("preview");
            previewField.setAccessible(true);
            previewField.set(runtime.getExec(), isPreview);
        }
        catch (NoSuchFieldException | IllegalAccessException e) {
            Assert.fail(e.getMessage());
        }
    }

    public static void setPreviewMode(final ZendeskPluginTestRuntime runtime, final boolean isPreview)
    {
        // A small hack to make the plugin executed in preview mode so
        try {
            final Field previewField = ExecSession.class.getDeclaredField("preview");
            previewField.setAccessible(true);
            previewField.set(runtime.getExec(), isPreview);
        }
        catch (NoSuchFieldException | IllegalAccessException e) {
            Assert.fail(e.getMessage());
        }
    }
}