package org.embulk.output.kintone;

import org.embulk.config.ConfigDiff;
import org.embulk.config.ConfigSource;
import org.embulk.config.TaskReport;
import org.embulk.config.TaskSource;
import org.embulk.spi.Exec;
import org.embulk.spi.OutputPlugin;
import org.embulk.spi.Schema;
import org.embulk.spi.TransactionalPageOutput;

import java.util.Collection;
import java.util.List;

public class KintoneOutputPlugin
        implements OutputPlugin
{
    @Override
    public ConfigDiff transaction(ConfigSource config,
                                  Schema schema, int taskCount,
                                  OutputPlugin.Control control)
    {
        PluginTask task = config.loadConfig(PluginTask.class);

        control.run(task.dump());
        return Exec.newConfigDiff();
    }

    @Override
    public ConfigDiff resume(TaskSource taskSource,
                             Schema schema, int taskCount,
                             OutputPlugin.Control control)
    {
        throw new UnsupportedOperationException("kintone output plugin does not support resuming");
    }

    @Override
    public void cleanup(TaskSource taskSource,
                        Schema schema, int taskCount,
                        List<TaskReport> successTaskReports)
    {
    }

    @Override
    public TransactionalPageOutput open(TaskSource taskSource, Schema schema, int taskIndex)
    {
        PluginTask task = taskSource.loadTask(PluginTask.class);
        Collection<KintoneColumnOption> options = task.getColumnOptions().values();

        KintoneMode mode = KintoneMode.getKintoneModeByValue(task.getMode());
        switch (mode) {
            case INSERT:
                for (KintoneColumnOption option : options) {
                    if (option.getUpdateKey()) {
                        throw new IllegalArgumentException(
                                "when mode is insert, require no update_key.");
                    }
                }
                break;
            case UPDATE:
            case UPSERT:
                boolean hasUpdateKey = false;
                for (KintoneColumnOption option : options) {
                    if (option.getUpdateKey()) {
                        if (hasUpdateKey) {
                            throw new IllegalArgumentException(
                                    "when mode is update and upsert, only one column can have an update_key.");
                        }
                        hasUpdateKey = true;
                    }
                }
                if (!hasUpdateKey) {
                    throw new IllegalArgumentException(
                            "when mode is update and upsert, require update_key.");
                }
                break;
            default:
                throw new IllegalArgumentException(String.format(
                        "Unknown mode '%s'",
                        task.getMode()));
        }
        return new KintonePageOutput(task, schema);
    }
}