/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package org.apache.cassandra.cql3;

import java.time.Instant;
import java.time.LocalDateTime;
import java.time.OffsetDateTime;
import java.time.ZoneOffset;
import java.time.format.DateTimeFormatter;

import org.junit.Assert;
import org.junit.Test;

import org.apache.cassandra.exceptions.InvalidRequestException;
import org.apache.cassandra.serializers.TimeSerializer;

import static org.apache.cassandra.cql3.Duration.NANOS_PER_HOUR;
import static org.apache.cassandra.cql3.Duration.NANOS_PER_MICRO;
import static org.apache.cassandra.cql3.Duration.NANOS_PER_MILLI;
import static org.apache.cassandra.cql3.Duration.NANOS_PER_MINUTE;
import static org.apache.cassandra.cql3.Duration.NANOS_PER_SECOND;
import static org.junit.Assert.assertEquals;

public class DurationTest
{
    @Test
    public void testFromStringWithStandardPattern()
    {
        assertEquals(Duration.newInstance(14, 0, 0), Duration.from("1y2mo"));
        assertEquals(Duration.newInstance(-14, 0, 0), Duration.from("-1y2mo"));
        assertEquals(Duration.newInstance(14, 0, 0), Duration.from("1Y2MO"));
        assertEquals(Duration.newInstance(0, 14, 0), Duration.from("2w"));
        assertEquals(Duration.newInstance(0, 2, 10 * NANOS_PER_HOUR), Duration.from("2d10h"));
        assertEquals(Duration.newInstance(0, 2, 0), Duration.from("2d"));
        assertEquals(Duration.newInstance(0, 0, 30 * NANOS_PER_HOUR), Duration.from("30h"));
        assertEquals(Duration.newInstance(0, 0, 30 * NANOS_PER_HOUR + 20 * NANOS_PER_MINUTE), Duration.from("30h20m"));
        assertEquals(Duration.newInstance(0, 0, 20 * NANOS_PER_MINUTE), Duration.from("20m"));
        assertEquals(Duration.newInstance(0, 0, 56 * NANOS_PER_SECOND), Duration.from("56s"));
        assertEquals(Duration.newInstance(0, 0, 567 * NANOS_PER_MILLI), Duration.from("567ms"));
        assertEquals(Duration.newInstance(0, 0, 1950 * NANOS_PER_MICRO), Duration.from("1950us"));
        assertEquals(Duration.newInstance(0, 0, 1950 * NANOS_PER_MICRO), Duration.from("1950µs"));
        assertEquals(Duration.newInstance(0, 0, 1950000), Duration.from("1950000ns"));
        assertEquals(Duration.newInstance(0, 0, 1950000), Duration.from("1950000NS"));
        assertEquals(Duration.newInstance(0, 0, -1950000), Duration.from("-1950000ns"));
        assertEquals(Duration.newInstance(15, 0, 130 * NANOS_PER_MINUTE), Duration.from("1y3mo2h10m"));
    }

    @Test
    public void testFromStringWithIso8601Pattern()
    {
        assertEquals(Duration.newInstance(12, 2, 0), Duration.from("P1Y2D"));
        assertEquals(Duration.newInstance(14, 0, 0), Duration.from("P1Y2M"));
        assertEquals(Duration.newInstance(0, 14, 0), Duration.from("P2W"));
        assertEquals(Duration.newInstance(12, 0, 2 * NANOS_PER_HOUR), Duration.from("P1YT2H"));
        assertEquals(Duration.newInstance(-14, 0, 0), Duration.from("-P1Y2M"));
        assertEquals(Duration.newInstance(0, 2, 0), Duration.from("P2D"));
        assertEquals(Duration.newInstance(0, 0, 30 * NANOS_PER_HOUR), Duration.from("PT30H"));
        assertEquals(Duration.newInstance(0, 0, 30 * NANOS_PER_HOUR + 20 * NANOS_PER_MINUTE), Duration.from("PT30H20M"));
        assertEquals(Duration.newInstance(0, 0, 20 * NANOS_PER_MINUTE), Duration.from("PT20M"));
        assertEquals(Duration.newInstance(0, 0, 56 * NANOS_PER_SECOND), Duration.from("PT56S"));
        assertEquals(Duration.newInstance(15, 0, 130 * NANOS_PER_MINUTE), Duration.from("P1Y3MT2H10M"));
    }

    @Test
    public void testFromStringWithIso8601AlternativePattern()
    {
        assertEquals(Duration.newInstance(12, 2, 0), Duration.from("P0001-00-02T00:00:00"));
        assertEquals(Duration.newInstance(14, 0, 0), Duration.from("P0001-02-00T00:00:00"));
        assertEquals(Duration.newInstance(12, 0, 2 * NANOS_PER_HOUR), Duration.from("P0001-00-00T02:00:00"));
        assertEquals(Duration.newInstance(-14, 0, 0), Duration.from("-P0001-02-00T00:00:00"));
        assertEquals(Duration.newInstance(0, 2, 0), Duration.from("P0000-00-02T00:00:00"));
        assertEquals(Duration.newInstance(0, 0, 30 * NANOS_PER_HOUR), Duration.from("P0000-00-00T30:00:00"));
        assertEquals(Duration.newInstance(0, 0, 30 * NANOS_PER_HOUR + 20 * NANOS_PER_MINUTE), Duration.from("P0000-00-00T30:20:00"));
        assertEquals(Duration.newInstance(0, 0, 20 * NANOS_PER_MINUTE), Duration.from("P0000-00-00T00:20:00"));
        assertEquals(Duration.newInstance(0, 0, 56 * NANOS_PER_SECOND), Duration.from("P0000-00-00T00:00:56"));
        assertEquals(Duration.newInstance(15, 0, 130 * NANOS_PER_MINUTE), Duration.from("P0001-03-00T02:10:00"));
    }

    @Test
    public void testInvalidDurations()
    {
        assertInvalidDuration(Long.MAX_VALUE + "d", "Invalid duration. The total number of days must be less or equal to 2147483647");
        assertInvalidDuration("2µ", "Unable to convert '2µ' to a duration");
        assertInvalidDuration("-2µ", "Unable to convert '2µ' to a duration");
        assertInvalidDuration("12.5s", "Unable to convert '12.5s' to a duration");
        assertInvalidDuration("2m12.5s", "Unable to convert '2m12.5s' to a duration");
        assertInvalidDuration("2m-12s", "Unable to convert '2m-12s' to a duration");
        assertInvalidDuration("12s3s", "Invalid duration. The seconds are specified multiple times");
        assertInvalidDuration("12s3m", "Invalid duration. The seconds should be after minutes");
        assertInvalidDuration("1Y3M4D", "Invalid duration. The minutes should be after days");
        assertInvalidDuration("P2Y3W", "Unable to convert 'P2Y3W' to a duration");
        assertInvalidDuration("P0002-00-20", "Unable to convert 'P0002-00-20' to a duration");
    }

    @Test
    public void testAddTo()
    {
        assertTimeEquals("2016-09-21T00:00:00.000", Duration.from("0m").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-21T00:00:00.000", Duration.from("10us").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-21T00:10:00.000", Duration.from("10m").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-21T01:30:00.000", Duration.from("90m").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-21T02:10:00.000", Duration.from("2h10m").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-23T00:10:00.000", Duration.from("2d10m").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-24T01:00:00.000", Duration.from("2d25h").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-10-21T00:00:00.000", Duration.from("1mo").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2017-11-21T00:00:00.000", Duration.from("14mo").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2017-02-28T00:00:00.000", Duration.from("12mo").addTo(toMillis("2016-02-29T00:00:00")));
    }

    @Test
    public void testAddToWithNegativeDurations()
    {
        assertTimeEquals("2016-09-21T00:00:00.000", Duration.from("-0m").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-21T00:00:00.000", Duration.from("-10us").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-20T23:50:00.000", Duration.from("-10m").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-20T22:30:00.000", Duration.from("-90m").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-20T21:50:00.000", Duration.from("-2h10m").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-18T23:50:00.000", Duration.from("-2d10m").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-17T23:00:00.000", Duration.from("-2d25h").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-08-21T00:00:00.000", Duration.from("-1mo").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2015-07-21T00:00:00.000", Duration.from("-14mo").addTo(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2015-02-28T00:00:00.000", Duration.from("-12mo").addTo(toMillis("2016-02-29T00:00:00")));
    }

    @Test
    public void testSubstractFrom()
    {
        assertTimeEquals("2016-09-21T00:00:00.000", Duration.from("0m").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-21T00:00:00.000", Duration.from("10us").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-20T23:50:00.000", Duration.from("10m").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-20T22:30:00.000", Duration.from("90m").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-20T21:50:00.000", Duration.from("2h10m").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-18T23:50:00.000", Duration.from("2d10m").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-17T23:00:00.000", Duration.from("2d25h").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-08-21T00:00:00.000", Duration.from("1mo").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2015-07-21T00:00:00.000", Duration.from("14mo").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2015-02-28T00:00:00.000", Duration.from("12mo").substractFrom(toMillis("2016-02-29T00:00:00")));
    }

    @Test
    public void testSubstractWithNegativeDurations()
    {
        assertTimeEquals("2016-09-21T00:00:00.000", Duration.from("-0m").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-21T00:00:00.000", Duration.from("-10us").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-21T00:10:00.000", Duration.from("-10m").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-21T01:30:00.000", Duration.from("-90m").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-21T02:10:00.000", Duration.from("-2h10m").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-23T00:10:00.000", Duration.from("-2d10m").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-09-24T01:00:00.000", Duration.from("-2d25h").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2016-10-21T00:00:00.000", Duration.from("-1mo").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2017-11-21T00:00:00.000", Duration.from("-14mo").substractFrom(toMillis("2016-09-21T00:00:00")));
        assertTimeEquals("2017-02-28T00:00:00.000", Duration.from("-12mo").substractFrom(toMillis("2016-02-29T00:00:00")));
    }

    @Test
    public void testFloorTime()
    {
        long time = floorTime("16:12:00", "2h");
        Duration result = Duration.newInstance(0, 0, time);
        Duration expected = Duration.from("16h");
        assertEquals(expected, result);
    }

    @Test
    public void testInvalidFloorTimestamp()
    {
        try
        {
            floorTimestamp("2016-09-27T16:12:00", "2h", "2017-09-01T00:00:00");
            Assert.fail();
        }
        catch (InvalidRequestException e)
        {
            assertEquals("The floor function starting time is greater than the provided time", e.getMessage());
        }

        try
        {
            floorTimestamp("2016-09-27T16:12:00", "-2h", "2016-09-27T00:00:00");
            Assert.fail();
        }
        catch (InvalidRequestException e)
        {
            assertEquals("Negative durations are not supported by the floor function", e.getMessage());
        }
    }

    @Test
    public void testFloorTimestampWithDurationInHours()
    {
        // Test floor with a timestamp equals to the start time
        long result = floorTimestamp("2016-09-27T16:12:00", "2h", "2016-09-27T16:12:00");
        assertTimeEquals("2016-09-27T16:12:00.000", result);

        // Test floor with a duration equals to zero
        result = floorTimestamp("2016-09-27T18:12:00", "0h", "2016-09-27T16:12:00");
        assertTimeEquals("2016-09-27T18:12:00.000", result);

        // Test floor with a timestamp exactly equals to the start time + (1 x duration)
        result = floorTimestamp("2016-09-27T18:12:00", "2h", "2016-09-27T16:12:00");
        assertTimeEquals("2016-09-27T18:12:00.000", result);

        // Test floor with a timestamp in the first bucket
        result = floorTimestamp("2016-09-27T16:13:00", "2h", "2016-09-27T16:12:00");
        assertTimeEquals("2016-09-27T16:12:00.000", result);

        // Test floor with a timestamp in another bucket
        result = floorTimestamp("2016-09-27T16:12:00", "2h", "2016-09-27T00:00:00");
        assertTimeEquals("2016-09-27T16:00:00.000", result);
    }

    @Test
    public void testFloorTimestampWithDurationInDays()
    {
        // Test floor with a start time at the beginning of the month
        long result = floorTimestamp("2016-09-27T16:12:00", "2d", "2016-09-01T00:00:00");
        assertTimeEquals("2016-09-27T00:00:00.000", result);

        // Test floor with a start time in the previous month
        result = floorTimestamp("2016-09-27T16:12:00", "2d", "2016-08-01T00:00:00");
        assertTimeEquals("2016-09-26T00:00:00.000", result);
    }

    @Test
    public void testFloorTimestampWithDurationInDaysAndHours()
    {
        long result = floorTimestamp("2016-09-27T16:12:00", "2d12h", "2016-09-01T00:00:00");
        assertTimeEquals("2016-09-26T00:00:00.000", result);
    }

    @Test
    public void testFloorTimestampWithDurationInMonths()
    {
        // Test floor with a timestamp equals to the start time
        long result = floorTimestamp("2016-09-01T00:00:00", "2mo", "2016-09-01T00:00:00");
        assertTimeEquals("2016-09-01T00:00:00.000", result);

        // Test floor with a timestamp in the first bucket
        result = floorTimestamp("2016-09-27T16:12:00", "2mo", "2016-09-01T00:00:00");
        assertTimeEquals("2016-09-01T00:00:00.000", result);

        // Test floor with a start time at the beginning of the year (LEAP YEAR)
        result = floorTimestamp("2016-09-27T16:12:00", "1mo", "2016-01-01T00:00:00");
        assertTimeEquals("2016-09-01T00:00:00.000", result);

        // Test floor with a start time at the beginning of the previous year
        result = floorTimestamp("2016-09-27T16:12:00", "2mo", "2015-01-01T00:00:00");
        assertTimeEquals("2016-09-01T00:00:00.000", result);

        // Test floor with a start time in the previous year
        result = floorTimestamp("2016-09-27T16:12:00", "2mo", "2015-02-02T00:00:00");
        assertTimeEquals("2016-08-02T00:00:00.000", result);

    }

    @Test
    public void testFloorTimestampWithDurationInMonthsAndDays()
    {
        long result = floorTimestamp("2016-09-27T16:12:00", "2mo2d", "2016-01-01T00:00:00");
        assertTimeEquals("2016-09-09T00:00:00.000", result);

        result = floorTimestamp("2016-09-27T16:12:00", "2mo5d", "2016-01-01T00:00:00");
        assertTimeEquals("2016-09-21T00:00:00.000", result);

        // Test floor with a timestamp in the first bucket
        result = floorTimestamp("2016-09-04T16:12:00", "2mo5d", "2016-07-01T00:00:00");
        assertTimeEquals("2016-07-01T00:00:00.000", result);

        // Test floor with a timestamp in a bucket starting on the last day of the month
        result = floorTimestamp("2016-09-27T16:12:00", "2mo10d", "2016-01-01T00:00:00");
        assertTimeEquals("2016-07-31T00:00:00.000", result);

        // Test floor with a timestamp in a bucket starting on the first day of the month
        result = floorTimestamp("2016-09-27T16:12:00", "2mo12d", "2016-01-01T00:00:00");
        assertTimeEquals("2016-08-06T00:00:00.000", result);

        // Test leap years
        result = floorTimestamp("2016-04-27T16:12:00", "1mo30d", "2016-01-01T00:00:00");
        assertTimeEquals("2016-03-02T00:00:00.000", result);

        result = floorTimestamp("2015-04-27T16:12:00", "1mo30d", "2015-01-01T00:00:00");
        assertTimeEquals("2015-03-03T00:00:00.000", result);
    }

    @Test
    public void testFloorTimestampWithDurationSmallerThanPrecision()
    {
        long result = floorTimestamp("2016-09-27T18:14:00", "5us", "2016-09-27T16:12:00");
        assertTimeEquals("2016-09-27T18:14:00.000", result);

        result = floorTimestamp("2016-09-27T18:14:00", "1h5us", "2016-09-27T16:12:00");
        assertTimeEquals("2016-09-27T18:12:00.000", result);
    }

    @Test
    public void testFloorTimestampWithLeapSecond()
    {
        long result = floorTimestamp("2016-07-02T00:00:00", "2m", "2016-06-30T23:58:00");
        assertTimeEquals("2016-07-02T00:00:00.000", result);
    }

    @Test
    public void testFloorTimestampWithComplexDuration()
    {
        long result = floorTimestamp("2016-07-02T00:00:00", "2mo2d8h", "2016-01-01T00:00:00");
        assertTimeEquals("2016-05-05T16:00:00.000", result);
    }

    @Test
    public void testInvalidFloorTime()
    {
        try
        {
            floorTime("16:12:00", "2d");
            Assert.fail();
        }
        catch (InvalidRequestException e)
        {
            assertEquals("For time values, the floor can only be computed for durations smaller that a day", e.getMessage());
        }

        try
        {
            floorTime("16:12:00", "25h");
            Assert.fail();
        }
        catch (InvalidRequestException e)
        {
            assertEquals("For time values, the floor can only be computed for durations smaller that a day", e.getMessage());
        }

        try
        {
            floorTime("16:12:00", "-2h");
            Assert.fail();
        }
        catch (InvalidRequestException e)
        {
            assertEquals("Negative durations are not supported by the floor function", e.getMessage());
        }
    }

    private static long toMillis(String timeAsString)
    {
        OffsetDateTime dateTime = LocalDateTime.parse(timeAsString, DateTimeFormatter.ofPattern("yyyy-MM-dd'T'HH:mm:ss"))
                                               .atOffset(ZoneOffset.UTC);

        return Instant.from(dateTime).toEpochMilli();
    }

    private static String fromMillis(long timeInMillis)
    {
        return Instant.ofEpochMilli(timeInMillis)
                      .atOffset(ZoneOffset.UTC)
                      .format(DateTimeFormatter.ofPattern("yyyy-MM-dd'T'HH:mm:ss.SSS"));
    }

    private static void assertTimeEquals(String expected, long actualTimeInMillis)
    {
        assertEquals(expected, fromMillis(actualTimeInMillis));
    }

    private static long floorTimestamp(String time, String duration, String startingTime)
    {
        return Duration.floorTimestamp(toMillis(time), Duration.from(duration), toMillis(startingTime));
    }

    private static long floorTime(String time, String duration)
    {
        return Duration.floorTime(timeInNanos(time), Duration.from(duration));
    }

    private static long timeInNanos(String timeAsString)
    {
        return TimeSerializer.timeStringToLong(timeAsString);
    }

    private static void assertInvalidDuration(String duration, String expectedErrorMessage)
    {
        try
        {
            Duration.from(duration);
            Assert.fail();
        }
        catch (InvalidRequestException e)
        {
            assertEquals(expectedErrorMessage, e.getMessage());
        }
    }
}
