Goit

// Copyright (C) 2023, Jakob Wakeling

// All rights reserved.

package cron

import (

	"fmt"

	"time"

	"github.com/Jamozed/Goit/src/util"

)

type Schedule struct{ Month, Day, Weekday, Hour, Minute, Second int64 }

var (

	Immediate = Schedule{-1, -1, -1, -1, -1, -1}

	Yearly    = Schedule{1, 1, -1, 0, 0, 0}

	Monthly   = Schedule{-1, 1, -1, 0, 0, 0}

	Weekly    = Schedule{-1, -1, 1, 0, 0, 0}

	Daily     = Schedule{-1, -1, -1, 0, 0, 0}

	Hourly    = Schedule{-1, -1, -1, -1, 0, 0}

	Minutely  = Schedule{-1, -1, -1, -1, -1, 0}

)

func (s Schedule) Next(t time.Time) time.Time {

	t = t.Add(1 * time.Second)

	added := false

wrap:

	for s.Month != -1 && int64(t.Month()) != s.Month {

		if !added {

			t = time.Date(t.Year(), t.Month(), 1, 0, 0, 0, 0, t.Location())

			added = true

		}

		t = t.AddDate(0, 1, 0)

		if t.Month() == time.January {

			goto wrap

		}

	}

	for !((s.Day == -1 || int64(t.Day()) == s.Day) && (s.Weekday == -1 || int64(t.Weekday()) == s.Weekday)) {

		if !added {

			t = time.Date(t.Year(), t.Month(), t.Day(), 0, 0, 0, 0, t.Location())

			added = true

		}

		t = t.AddDate(0, 0, 1)

		if t.Day() == 1 {

			goto wrap

		}

	}

	for s.Hour != -1 && int64(t.Hour()) != s.Hour {

		if !added {

			t = time.Date(t.Year(), t.Month(), t.Day(), t.Hour(), 0, 0, 0, t.Location())

			added = true

		}

		t = t.Add(1 * time.Hour)

		if t.Hour() == 0 {

			goto wrap

		}

	}

	for s.Minute != -1 && int64(t.Minute()) != s.Minute {

		if !added {

			t = t.Truncate(time.Minute)

			added = true

		}

		t = t.Add(1 * time.Minute)

		if t.Minute() == 0 {

			goto wrap

		}

	}

	for s.Second != -1 && int64(t.Second()) != s.Second {

		if !added {

			t = t.Truncate(time.Second)

			added = true

		}

		t = t.Add(1 * time.Second)

		if t.Second() == 0 {

			goto wrap

		}

	}

	return t

}

func (s Schedule) IsImmediate() bool {

	return s == Immediate

}

func (s Schedule) String() string {

	if s.IsImmediate() {

		return "immediate"

	}

	return fmt.Sprintf(

		"%s %s %s %s %s %s",

		util.If(s.Month == -1, "*", fmt.Sprint(s.Month)),

		util.If(s.Day == -1, "*", fmt.Sprint(s.Day)),

		util.If(s.Weekday == -1, "*", fmt.Sprint(s.Weekday)),

		util.If(s.Hour == -1, "*", fmt.Sprint(s.Hour)),

		util.If(s.Minute == -1, "*", fmt.Sprint(s.Minute)),

		util.If(s.Second == -1, "*", fmt.Sprint(s.Second)),

	)

}