GoSungrow/cmd/cmd_cron.go

package cmd

import (
	"GoSungrow/Only"
	"fmt"
	"github.com/go-co-op/gocron"
	"github.com/spf13/cobra"
	"os"
	"strings"
	"time"
)


type CronStruct struct {
	Scheduler *gocron.Scheduler
	Job *gocron.Job
}
var Cron CronStruct


func AttachCmdCron(cmd *cobra.Command) *cobra.Command {
	// ******************************************************************************** //
	var cmdCron = &cobra.Command{
		Use:                   "cron",
		Aliases:               []string{""},
		Short:                 fmt.Sprintf("Run a command via schedule."),
		Long:                  fmt.Sprintf("Run a command via schedule."),
		DisableFlagParsing:    false,
		DisableFlagsInUseLine: false,
		PreRunE:               Cmd.ProcessArgs,
		Run:                   cmdCronFunc,
		Args:                  cobra.MinimumNArgs(1),
	}
	cmd.AddCommand(cmdCron)
	cmdCron.Example = PrintExamples(cmdCron, "run 00 18 . . . sync default", "run 42 02 04 . . list all")


	// ******************************************************************************** //
	var cmdCronRun = &cobra.Command{
		Use:                   "run <minute> <hour> <month day> <month> <week day>  <command ...>",
		Aliases:               []string{""},
		Short:                 fmt.Sprintf("Run scheduled a command."),
		Long:                  fmt.Sprintf("Run scheduled a command."),
		DisableFlagParsing:    false,
		DisableFlagsInUseLine: false,
		PreRunE:               Cmd.ProcessArgs,
		Run:                   cmdCronRunFunc,
		Args:                  cobra.MinimumNArgs(6),
	}
	cmdCron.AddCommand(cmdCronRun)
	cmdCronRun.Example = PrintExamples(cmdCronRun, "00 18 . . . sync default", "42 02 04 . . list all", "./1 . . . . list all")

	// ******************************************************************************** //
	var cmdConfigRead = &cobra.Command{}
	cmdCron.AddCommand(cmdConfigRead)
	cmdConfigRead.Example = PrintExamples(cmdConfigRead, "")

	// ******************************************************************************** //
	var cmdCronAdd = &cobra.Command{
		Use:                   "add",
		Aliases:               []string{""},
		Short:                 fmt.Sprintf("Add scheduled a command."),
		Long:                  fmt.Sprintf("Add scheduled a command."),
		DisableFlagParsing:    false,
		DisableFlagsInUseLine: false,
		PreRunE:               Cmd.ProcessArgs,
		Run:                   cmdCronAddFunc,
		Args:                  cobra.MinimumNArgs(1),
	}
	cmdCron.AddCommand(cmdCronAdd)
	cmdCronAdd.Example = PrintExamples(cmdCronAdd, "add")

	// ******************************************************************************** //
	var cmdCronRemove = &cobra.Command{
		Use:                   "del",
		Aliases:               []string{"remove"},
		Short:                 fmt.Sprintf("Remove a scheduled command."),
		Long:                  fmt.Sprintf("Remove a scheduled command."),
		DisableFlagParsing:    false,
		DisableFlagsInUseLine: false,
		PreRunE:               Cmd.ProcessArgs,
		Run:                   cmdCronRemoveFunc,
		Args:                  cobra.MinimumNArgs(1),
	}
	cmdCron.AddCommand(cmdCronRemove)
	cmdCronRemove.Example = PrintExamples(cmdCronRemove, "del")

	// ******************************************************************************** //
	var cmdCronList = &cobra.Command{
		Use:                   "list",
		Aliases:               []string{""},
		Short:                 fmt.Sprintf("List scheduled commands."),
		Long:                  fmt.Sprintf("List scheduled commands."),
		DisableFlagParsing:    false,
		DisableFlagsInUseLine: false,
		PreRunE:               Cmd.ProcessArgs,
		Run:                   cmdCronListFunc,
		Args:                  cobra.MinimumNArgs(1),
	}
	cmdCron.AddCommand(cmdCronList)
	cmdCronList.Example = PrintExamples(cmdCronList, "list")

	return cmdCron
}

func cmdCronFunc(cmd *cobra.Command, args []string) {
	for range Only.Once {
		if len(args) == 0 {
			Cmd.Error = cmd.Help()
			break
		}
	}
}

func cmdCronRunFunc(_ *cobra.Command, args []string) {
	for range Only.Once {
		// */1 * * * * /dir/command args args
		cronString := strings.Join(args[0:5], " ")
		cronString = strings.ReplaceAll(cronString, ".", "*")
		ResetArgs(args[5:]...)

		Cron.Scheduler = gocron.NewScheduler(time.UTC)
		Cron.Scheduler = Cron.Scheduler.Cron(cronString)
		Cron.Scheduler = Cron.Scheduler.SingletonMode()

		Cron.Job, Cmd.Error = Cron.Scheduler.Do(ReExecute)
		if Cmd.Error != nil {
			break
		}

		LogPrint("Created job schedule using '%s'\n", cronString)
		LogPrint("Job command '%s'\n", strings.Join(os.Args, " "))

		Cron.Scheduler.StartBlocking()
		if Cmd.Error != nil {
			break
		}
	}
}

func cmdCronAddFunc(_ *cobra.Command, _ []string) {
	for range Only.Once {
		fmt.Println("Not yet implemented.")

		// var msg string
		// switch {
		// 	case args[0] == "":
		// 		fallthrough
		// 	case args[0] == "default":
		// 		//u, _ := user.Current()
		// 		//msg = fmt.Sprintf("Regular sync by %s", u.ApiUsername)
		// 	default:
		// 		msg = args[0]
		// }
		//
		// args = args[1:]
		//
		// //Cmd.Error = Cmd.CronAdd(msg, args...)
		// if Cmd.Error != nil {
		// 	break
		// }
	}
}

func cmdCronRemoveFunc(_ *cobra.Command, _ []string) {
	for range Only.Once {
		fmt.Println("Not yet implemented.")

		// var msg string
		// switch {
		// 	case args[0] == "":
		// 		fallthrough
		// 	case args[0] == "default":
		// 		//u, _ := user.Current()
		// 		//msg = fmt.Sprintf("Regular sync by %s", u.ApiUsername)
		// 	default:
		// 		msg = args[0]
		// }
		//
		// args = args[1:]
		//
		// //Cmd.Error = Cmd.CronAdd(msg, args...)
		// if Cmd.Error != nil {
		// 	break
		// }
	}
}

func cmdCronListFunc(_ *cobra.Command, _ []string) {
	for range Only.Once {
		fmt.Println("Not yet implemented.")

		// var msg string
		// 	switch {
		// 		case args[0] == "":
		// 			fallthrough
		// 		case args[0] == "default":
		// 			//u, _ := user.Current()
		// 			//msg = fmt.Sprintf("Regular sync by %s", u.ApiUsername)
		// 		default:
		// 			msg = args[0]
		// }
		//
		// args = args[1:]
		//
		// Cmd.Error = Cmd.CronList(msg, args...)
		// if Cmd.Error != nil {
		// 	break
		// }
	}
}


func timeStamp() string {
	return time.Now().UTC().Format(time.UnixDate) + " : "
}
func LogPrint(format string, args ...interface{}) {
	format = timeStamp() + format
	fmt.Printf(format, args...)
}

func LogPrintDate(format string, args ...interface{}) {
	fmt.Printf("%s ", TimeNow())
	fmt.Printf(format, args...)
	// fmt.Println()
}

func TimeNow() string {
	return time.Now().Format("2006-01-02 15:04:05")
}

func ReExecute() error {
	for range Only.Once {
		LogPrint("Running scheduled command '%s'\n", strings.Join(os.Args, " "))
		// LogPrint("Last run '%s'\n", Cron.Job.LastRun().Format(time.UnixDate))
		LogPrint("Next run '%s'\n", Cron.Job.ScheduledTime().Format(time.UnixDate))
		LogPrint("Run count '%d'\n", Cron.Job.RunCount())

		Cmd.Error = rootCmd.Execute()
		if Cmd.Error != nil {
			LogPrint("ERROR: %s\n", Cmd.Error)
			break
		}
	}

	return Cmd.Error
}

func ResetArgs(args ...string) {
	for range Only.Once {
		// fmt.Printf("oldArgs: %v\n", os.Args)
		newArgs := []string{os.Args[0]}
		newArgs = append(newArgs, args...)
		os.Args = newArgs
		// fmt.Printf("newArgs: %v\n", os.Args)
	}
}