/*
 * 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.dolphinscheduler.service.cron;

import static com.cronutils.model.field.expression.FieldExpressionFactory.always;
import static com.cronutils.model.field.expression.FieldExpressionFactory.every;
import static com.cronutils.model.field.expression.FieldExpressionFactory.on;
import static com.cronutils.model.field.expression.FieldExpressionFactory.questionMark;

import org.apache.dolphinscheduler.common.enums.CycleEnum;
import org.apache.dolphinscheduler.common.utils.DateUtils;
import org.apache.dolphinscheduler.service.exceptions.CronParseException;

import java.time.ZoneId;
import java.time.ZonedDateTime;
import java.util.Date;

import org.junit.Assert;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.cronutils.builder.CronBuilder;
import com.cronutils.model.Cron;
import com.cronutils.model.CronType;
import com.cronutils.model.definition.CronDefinitionBuilder;
import com.cronutils.model.field.CronField;
import com.cronutils.model.field.CronFieldName;
import com.cronutils.model.field.expression.Always;
import com.cronutils.model.field.expression.And;
import com.cronutils.model.field.expression.Between;
import com.cronutils.model.field.expression.Every;
import com.cronutils.model.field.expression.On;
import com.cronutils.model.field.expression.QuestionMark;

/**
 * CronUtilsTest
 */
public class CronUtilsTest {

    private static final Logger logger = LoggerFactory.getLogger(CronUtilsTest.class);

    /**
     * cron as string test
     */
    @Test
    public void testCronAsString() {
        Cron cron = CronBuilder.cron(CronDefinitionBuilder.instanceDefinitionFor(CronType.QUARTZ)).withYear(always())
            .withDoW(questionMark()).withMonth(always()).withDoM(always()).withHour(always()).withMinute(every(5))
            .withSecond(on(0)).instance();
        // Obtain the string expression
        String cronAsString = cron.asString();

        // 0 */5 * * * ? *  Every five minutes(once every 5 minutes)
        Assert.assertEquals("0 */5 * * * ? *", cronAsString);
    }


    /**
     * cron parse test
     */
    @Test
    public void testCronParse() throws CronParseException {
        String strCrontab = "0 1 2 3 * ? *";

        Cron depCron = CronUtils.parse2Cron(strCrontab);
        Assert.assertEquals("0", depCron.retrieve(CronFieldName.SECOND).getExpression().asString());
        Assert.assertEquals("1", depCron.retrieve(CronFieldName.MINUTE).getExpression().asString());
        Assert.assertEquals("2", depCron.retrieve(CronFieldName.HOUR).getExpression().asString());
        Assert.assertEquals("3", depCron.retrieve(CronFieldName.DAY_OF_MONTH).getExpression().asString());
        Assert.assertEquals("*", depCron.retrieve(CronFieldName.MONTH).getExpression().asString());
        Assert.assertEquals("*", depCron.retrieve(CronFieldName.YEAR).getExpression().asString());
    }

    /**
     * schedule type test
     */
    @Test
    public void testScheduleType() throws CronParseException {
        CycleEnum cycleEnum = CronUtils.getMaxCycle(CronUtils.parse2Cron("0 */1 * * * ? *"));
        Assert.assertEquals("MINUTE", cycleEnum.name());

        CycleEnum cycleEnum2 = CronUtils.getMaxCycle("0 * * * * ? *");
        Assert.assertEquals("MINUTE", cycleEnum2.name());

        CycleEnum cycleEnum3 = CronUtils.getMiniCycle(CronUtils.parse2Cron("0 * * * * ? *"));
        Assert.assertEquals("MINUTE", cycleEnum3.name());

        CycleEnum cycleEnum4 = CronUtils.getMaxCycle(CronUtils.parse2Cron("0 0 7 * 1 ? *"));
        Assert.assertEquals("YEAR", cycleEnum4.name());
        cycleEnum4 = CronUtils.getMiniCycle(CronUtils.parse2Cron("0 0 7 * 1 ? *"));
        Assert.assertEquals("DAY", cycleEnum4.name());

        CycleEnum cycleEnum5 = CronUtils.getMaxCycle(CronUtils.parse2Cron("0 0 7 * 1/1 ? *"));
        Assert.assertEquals("MONTH", cycleEnum5.name());

        CycleEnum cycleEnum6 = CronUtils.getMaxCycle(CronUtils.parse2Cron("0 0 7 * 1-2 ? *"));
        Assert.assertEquals("YEAR", cycleEnum6.name());

        CycleEnum cycleEnum7 = CronUtils.getMaxCycle(CronUtils.parse2Cron("0 0 7 * 1,2 ? *"));
        Assert.assertEquals("YEAR", cycleEnum7.name());
    }

    /**
     * test
     */
    @Test
    public void test2() throws CronParseException {
        Cron cron1 = CronBuilder.cron(CronDefinitionBuilder.instanceDefinitionFor(CronType.QUARTZ)).withYear(always())
            .withDoW(questionMark()).withMonth(always()).withDoM(always()).withHour(always()).withMinute(every(5))
            .withSecond(on(0)).instance();
        // minute cycle
        String[] cronArayy =
            new String[] {"* * * * * ? *", "* 0 * * * ? *", "* 5 * * 3/5 ? *", "0 0 * * * ? *", "0 0 7 * 1 ? *",
                "0 0 7 * 1/1 ? *", "0 0 7 * 1-2 ? *", "0 0 7 * 1,2 ? *"};
        for (String minCrontab : cronArayy) {
            Cron cron = CronUtils.parse2Cron(minCrontab);
            CronField minField = cron.retrieve(CronFieldName.MINUTE);
            logger.info("minField instanceof Between:" + (minField.getExpression() instanceof Between));
            logger.info("minField instanceof Every:" + (minField.getExpression() instanceof Every));
            logger.info("minField instanceof Always:" + (minField.getExpression() instanceof Always));
            logger.info("minField instanceof On:" + (minField.getExpression() instanceof On));
            logger.info("minField instanceof And:" + (minField.getExpression() instanceof And));
            CronField hourField = cron.retrieve(CronFieldName.HOUR);
            logger.info("hourField instanceof Between:" + (hourField.getExpression() instanceof Between));
            logger.info("hourField instanceof Always:" + (hourField.getExpression() instanceof Always));
            logger.info("hourField instanceof Every:" + (hourField.getExpression() instanceof Every));
            logger.info("hourField instanceof On:" + (hourField.getExpression() instanceof On));
            logger.info("hourField instanceof And:" + (hourField.getExpression() instanceof And));

            CronField dayOfMonthField = cron.retrieve(CronFieldName.DAY_OF_MONTH);
            logger.info("dayOfMonthField instanceof Between:" + (dayOfMonthField.getExpression() instanceof Between));
            logger.info("dayOfMonthField instanceof Always:" + (dayOfMonthField.getExpression() instanceof Always));
            logger.info("dayOfMonthField instanceof Every:" + (dayOfMonthField.getExpression() instanceof Every));
            logger.info("dayOfMonthField instanceof On:" + (dayOfMonthField.getExpression() instanceof On));
            logger.info("dayOfMonthField instanceof And:" + (dayOfMonthField.getExpression() instanceof And));
            logger.info(
                "dayOfMonthField instanceof QuestionMark:" + (dayOfMonthField.getExpression() instanceof QuestionMark));

            CronField monthField = cron.retrieve(CronFieldName.MONTH);
            logger.info("monthField instanceof Between:" + (monthField.getExpression() instanceof Between));
            logger.info("monthField instanceof Always:" + (monthField.getExpression() instanceof Always));
            logger.info("monthField instanceof Every:" + (monthField.getExpression() instanceof Every));
            logger.info("monthField instanceof On:" + (monthField.getExpression() instanceof On));
            logger.info("monthField instanceof And:" + (monthField.getExpression() instanceof And));
            logger.info("monthField instanceof QuestionMark:" + (monthField.getExpression() instanceof QuestionMark));

            CronField dayOfWeekField = cron.retrieve(CronFieldName.DAY_OF_WEEK);
            logger.info("dayOfWeekField instanceof Between:" + (dayOfWeekField.getExpression() instanceof Between));
            logger.info("dayOfWeekField instanceof Always:" + (dayOfWeekField.getExpression() instanceof Always));
            logger.info("dayOfWeekField instanceof Every:" + (dayOfWeekField.getExpression() instanceof Every));
            logger.info("dayOfWeekField instanceof On:" + (dayOfWeekField.getExpression() instanceof On));
            logger.info("dayOfWeekField instanceof And:" + (dayOfWeekField.getExpression() instanceof And));
            logger.info(
                "dayOfWeekField instanceof QuestionMark:" + (dayOfWeekField.getExpression() instanceof QuestionMark));

            CronField yearField = cron.retrieve(CronFieldName.YEAR);
            logger.info("yearField instanceof Between:" + (yearField.getExpression() instanceof Between));
            logger.info("yearField instanceof Always:" + (yearField.getExpression() instanceof Always));
            logger.info("yearField instanceof Every:" + (yearField.getExpression() instanceof Every));
            logger.info("yearField instanceof On:" + (yearField.getExpression() instanceof On));
            logger.info("yearField instanceof And:" + (yearField.getExpression() instanceof And));
            logger.info("yearField instanceof QuestionMark:" + (yearField.getExpression() instanceof QuestionMark));

            CycleEnum cycleEnum = CronUtils.getMaxCycle(minCrontab);
            if (cycleEnum != null) {
                logger.info(cycleEnum.name());
            } else {
                logger.info("can't get scheduleType");
            }
        }
        Assert.assertTrue(true);
    }

    @Test
    public void getSelfFireDateList() throws CronParseException {
        ZonedDateTime from =
            ZonedDateTime.ofInstant(DateUtils.stringToDate("2020-01-01 00:00:00").toInstant(), ZoneId.systemDefault());
        ZonedDateTime to =
            ZonedDateTime.ofInstant(DateUtils.stringToDate("2020-01-31 00:00:00").toInstant(), ZoneId.systemDefault());
        // test date
        Assert.assertEquals(0, CronUtils.getFireDateList(to, from, "0 0 0 * * ? ").size());
        try {
            // test error cron
            // should throw exception
            CronUtils.getFireDateList(from, to, "0 0 0 * *").size();
            Assert.assertTrue(false);
        } catch (CronParseException cronParseException) {
            Assert.assertTrue(true);
        }
        // test cron
        Assert.assertEquals(30, CronUtils.getFireDateList(from, to, "0 0 0 * * ? ").size());
        // test other
        Assert.assertEquals(30, CronUtils.getFireDateList(from, to, CronUtils.parse2Cron("0 0 0 * * ? ")).size());
        Assert.assertEquals(5, CronUtils.getSelfFireDateList(from, to, CronUtils.parse2Cron("0 0 0 * * ? "), 5).size());
        from =
            ZonedDateTime.ofInstant(DateUtils.stringToDate("2020-01-01 00:02:00").toInstant(), ZoneId.systemDefault());
        to = ZonedDateTime.ofInstant(DateUtils.stringToDate("2020-01-01 00:02:00").toInstant(), ZoneId.systemDefault());
        Assert.assertEquals(1,
            CronUtils.getFireDateList(from.minusSeconds(1L), to, CronUtils.parse2Cron("0 * * * * ? ")).size());

        from =
            ZonedDateTime.ofInstant(DateUtils.stringToDate("2020-01-01 00:02:00").toInstant(), ZoneId.systemDefault());
        to = ZonedDateTime.ofInstant(DateUtils.stringToDate("2020-01-01 00:04:00").toInstant(),
            ZoneId.systemDefault());
        Assert.assertEquals(2,
            CronUtils.getFireDateList(from.minusSeconds(1L), to.minusSeconds(1L), CronUtils.parse2Cron("0 * * * * ? "))
                .size());
    }

    @Test
    public void getExpirationTime() {
        Date startTime = DateUtils.stringToDate("2020-02-07 18:30:00");
        Date expirationTime = CronUtils.getExpirationTime(startTime, CycleEnum.HOUR);
        Assert.assertEquals("2020-02-07 19:30:00", DateUtils.dateToString(expirationTime));
        expirationTime = CronUtils.getExpirationTime(startTime, CycleEnum.DAY);
        Assert.assertEquals("2020-02-07 23:59:59", DateUtils.dateToString(expirationTime));
        expirationTime = CronUtils.getExpirationTime(startTime, CycleEnum.WEEK);
        Assert.assertEquals("2020-02-07 23:59:59", DateUtils.dateToString(expirationTime));
        expirationTime = CronUtils.getExpirationTime(startTime, CycleEnum.MONTH);
        Assert.assertEquals("2020-02-07 23:59:59", DateUtils.dateToString(expirationTime));
        expirationTime = CronUtils.getExpirationTime(startTime, CycleEnum.YEAR);
        Assert.assertEquals("2020-02-07 18:30:00", DateUtils.dateToString(expirationTime));
    }
}
