main.go

package main

import (
	"flag"
	"fmt"
	"strconv"
	"strings"
	"time"

	"github.com/antonyho/crhk-recorder/pkg/dayofweek"
	"github.com/antonyho/crhk-recorder/pkg/stream/recorder"
)

func main() {
	var (
		channel   string
		startTime string
		endTime   string
		duration  time.Duration
		weekdays  string
		repeat    bool
	)
	flag.StringVar(&channel, "c", "881", "channel name in abbreviation")
	flag.StringVar(&startTime, "s", "", "start time with timezone abbreviation")
	flag.StringVar(&endTime, "e", "", "end time with timezone abbreviation")
	flag.DurationVar(&duration, "d", 0, "record duration [don't do this over 24 hours]")
	flag.StringVar(&weekdays, "w", "", "day of week on scheduled recording [comma seperated] [Sunday=0]")
	flag.BoolVar(&repeat, "r", false, "repeat recording at scheduled time on next day")
	flag.Parse()

	if duration == 0 {
		if startTime == "" && endTime == "" {
			panic("record time value must be provided")
		}
	}

	// startTime is set
	// 		duration is not set
	// 			endTime is set - stop at given endTime
	// 			endTime is not set - can't record infinitely
	// 		duration is set
	// 			endTime is not set - go with given duration
	// 			endTime is set - stop at given endTime, ignore duration
	// startTime is not set
	// 		duration is not set
	// 			endTime is set - start now and stop at endTime
	// 			endTime is not set - this condition is unreachable
	// 		duration is set
	// 			endTime is set - start now and stop at endTime
	// 			endTime is not set - start now and go with given duration

	rcdr := recorder.NewRecorder(channel)

	if startTime == "" {
		// Add a second delay to avoid skipping
		startTime = time.Now().Add(time.Second).Format("15:04:05 -0700")
	}

	if duration > time.Duration(0) {
		if endTime == "" {
			// Change Schedule() accepts parameter types and create endTime
			// With duration parameter, it will override the endTime
			start, err := time.Parse("15:04:05 -0700", startTime)
			if err != nil {
				panic(err)
			}
			endTime = start.Add(duration).Format("15:04:05 -0700")
		}
	}

	if endTime == "" {
		panic("record time cannot be infinite")
	}

	dowMask := dayofweek.New()
	if weekdays != "" {
		days := strings.Split(weekdays, ",")
		for _, day := range days {
			day = strings.TrimSpace(day)
			d, err := strconv.ParseUint(day, 10, 8)
			if err != nil {
				panic(fmt.Errorf("incorrect day of week parameter [%s]", weekdays))
			} else if d >= 0 && d <= 6 {
				dowMask.Enable(time.Weekday(d))
			} else {
				panic(fmt.Errorf("incorrect day of week parameter [%s]", weekdays))
			}
		}
	} else {
		if !repeat {
			// Just now, just once.
			dowMask.Enable(time.Now().Weekday()) // Hope you aren't starting at 23:59:59
		} // Otherwise, all weeekdays.
	}

	if err := rcdr.Schedule(startTime, endTime, *dowMask, repeat); err != nil {
		panic(err)
	}
}