package org.embulk.filter.timestamp_format;

import org.embulk.config.ConfigException;
import org.embulk.filter.timestamp_format.cast.DoubleCast;
import org.embulk.filter.timestamp_format.cast.LongCast;
import org.embulk.filter.timestamp_format.cast.StringCast;
import org.embulk.filter.timestamp_format.TimestampFormatFilterPlugin.ColumnConfig;
import org.embulk.filter.timestamp_format.TimestampFormatFilterPlugin.PluginTask;
import org.embulk.spi.Exec;
import org.embulk.spi.type.DoubleType;
import org.embulk.spi.type.LongType;
import org.embulk.spi.type.StringType;
import org.embulk.spi.type.Type;
import org.msgpack.value.FloatValue;
import org.msgpack.value.IntegerValue;
import org.msgpack.value.StringValue;
import org.msgpack.value.Value;
import org.msgpack.value.ValueFactory;

import org.slf4j.Logger;

import java.util.HashMap;

class JsonCaster
{
    private static final Logger logger = Exec.getLogger(TimestampFormatFilterPlugin.class);
    private final PluginTask task;
    private final HashMap<String, TimestampParser> timestampParserMap;
    private final HashMap<String, TimestampFormatter> timestampFormatterMap;
    private final HashMap<String, TimestampUnit> fromTimestampUnitMap;
    private final HashMap<String, TimestampUnit> toTimestampUnitMap;

    JsonCaster(PluginTask task,
               HashMap<String, TimestampParser> timestampParserMap,
               HashMap<String, TimestampFormatter> timestampFormatterMap,
               HashMap<String, TimestampUnit> fromTimestampUnitMap,
               HashMap<String, TimestampUnit> toTimestampUnitMap)
    {
        this.task = task;
        this.timestampParserMap = timestampParserMap;
        this.timestampFormatterMap = timestampFormatterMap;
        this.fromTimestampUnitMap = fromTimestampUnitMap;
        this.toTimestampUnitMap = toTimestampUnitMap;
    }

    public Value fromLong(ColumnConfig columnConfig, IntegerValue value)
    {
        Type outputType = columnConfig.getType();
        TimestampUnit fromUnit = fromTimestampUnitMap.get(columnConfig.getName());
        if (outputType instanceof StringType) {
            TimestampFormatter formatter = timestampFormatterMap.get(columnConfig.getName());
            return ValueFactory.newString(LongCast.asString(value.asLong(), fromUnit, formatter));
        }
        else if (outputType instanceof LongType) {
            TimestampUnit toUnit = toTimestampUnitMap.get(columnConfig.getName());
            return ValueFactory.newInteger(LongCast.asLong(value.asLong(), fromUnit, toUnit));
        }
        else if (outputType instanceof DoubleType) {
            TimestampUnit toUnit = toTimestampUnitMap.get(columnConfig.getName());
            return ValueFactory.newFloat(LongCast.asDouble(value.asLong(), fromUnit, toUnit));
        }
        else {
            assert false;
            throw new RuntimeException();
        }
    }

    public Value fromDouble(ColumnConfig columnConfig, FloatValue value)
    {
        Type outputType = columnConfig.getType();
        TimestampUnit fromUnit = fromTimestampUnitMap.get(columnConfig.getName());
        if (outputType instanceof StringType) {
            TimestampFormatter formatter = timestampFormatterMap.get(columnConfig.getName());
            return ValueFactory.newString(DoubleCast.asString(value.toDouble(), fromUnit, formatter));
        }
        else if (outputType instanceof LongType) {
            TimestampUnit toUnit = toTimestampUnitMap.get(columnConfig.getName());
            return ValueFactory.newInteger(DoubleCast.asLong(value.toDouble(), fromUnit, toUnit));
        }
        else if (outputType instanceof DoubleType) {
            TimestampUnit toUnit = toTimestampUnitMap.get(columnConfig.getName());
            return ValueFactory.newFloat(DoubleCast.asDouble(value.toDouble(), fromUnit, toUnit));
        }
        else {
            assert false;
            throw new RuntimeException();
        }
    }

    public Value fromString(ColumnConfig columnConfig, StringValue value)
    {
        Type outputType = columnConfig.getType();
        TimestampParser parser = timestampParserMap.get(columnConfig.getName());
        if (outputType instanceof StringType) {
            TimestampFormatter formatter = timestampFormatterMap.get(columnConfig.getName());
            return ValueFactory.newString(StringCast.asString(value.asString(), parser, formatter));
        }
        else if (outputType instanceof LongType) {
            TimestampUnit toUnit = toTimestampUnitMap.get(columnConfig.getName());
            return ValueFactory.newInteger(StringCast.asLong(value.asString(), parser, toUnit));
        }
        else if (outputType instanceof DoubleType) {
            TimestampUnit toUnit = toTimestampUnitMap.get(columnConfig.getName());
            return ValueFactory.newFloat(StringCast.asDouble(value.asString(), parser, toUnit));
        }
        else {
            assert false;
            throw new RuntimeException();
        }
    }
}