most-specific-period/main.go

package main

import (
	"bufio"
	"flag"
	"fmt"
	"os"
	"strings"
	"time"

	"github.com/taigrr/most-specific-period/msp"
)

type Period struct {
	EndTime    time.Time
	StartTime  time.Time
	Identifier string
}

func (p Period) GetEndTime() time.Time {
	return p.EndTime
}

func (p Period) GetStartTime() time.Time {
	return p.StartTime
}

func (p Period) GetIdentifier() string {
	return p.Identifier
}

func init() {
	flag.Usage = func() {
		fmt.Fprintf(os.Stderr, "Usage of %s:\n", os.Args[0])
		flag.PrintDefaults()
	}
}

func warnMessage() {
	fmt.Print("Please type your date formats as follows, hit return between each field (RFC 3339), and hit Control+D to signal you are complete: \nIdentifier: id\nStartTime: 2019-10-12T07:20:50.52Z\nEndTime: 2019-10-12T07:20:50.52Z\n")
}

func helpMessage() {
	fmt.Print("\nmost-specific-period [-h][-d]\n\nGenerates a timeline of periods and will provide a most specific period if available.\n\n-h\tShows this help menu\n-d\tProvide an RFC 3339 time to provide an alternate point for calculating MSP.")
}

func main() {
	var start time.Time
	help := flag.Bool("h", false, "displays help command")
	userDate := flag.String("d", "", "use a custom date to calculate MSP")
	flag.Parse()
	if *help {
		helpMessage()
		os.Exit(0)
	}

	if userDate != nil && *userDate != "" {
		t, err := time.Parse(time.RFC3339, *userDate)
		if err != nil {
			fmt.Println("Please enter the date using the YYYY-MM-DDT00:00:00.00Z")
			os.Exit(1)
		}
		start = t
	} else {
		start = time.Now()
	}
	terminal := false
	fi, _ := os.Stdin.Stat()
	if (fi.Mode() & os.ModeCharDevice) == 0 {
		// this is a file being read in, no need to print the prompt just yet
	} else {
		// this is a terminal, let's help the user out
		terminal = true
		warnMessage()
	}
	s := bufio.NewScanner(os.Stdin)
	count := 1

	if terminal {
		fmt.Print("Identifier: ")
	}

	periods := []msp.Period{}
	currentPeriod := Period{}
	for s.Scan() {
		input := s.Text()
		input = strings.TrimSpace(input)
		if input == "" {
			continue
		}
		if count%3 == 0 {
			t, err := time.Parse(time.RFC3339, input)
			if err != nil {
				fmt.Printf("ERROR: Invalid timestamp: %v", t)
				os.Exit(1)
			}
			currentPeriod.EndTime = t

			periods = append(periods, currentPeriod)
			if terminal {
				fmt.Print("Identifier: ")
			}
		}
		if count%3 == 1 {
			currentPeriod = Period{Identifier: s.Text()}
			if terminal {
				fmt.Print("StartTime: ")
			}
		}
		if count%3 == 2 {
			t, err := time.Parse(time.RFC3339, input)
			if err != nil {
				fmt.Printf("ERROR: Invalid timestamp: %v", t)
				os.Exit(1)
			}
			currentPeriod.StartTime = t

			if terminal {
				fmt.Print("EndTime: ")
			}
		}
		count++
	}

	vals := msp.GenerateTimeline(periods...)
	fmt.Print("\nTimeline of changeovers:\n")
	for _, val := range vals {
		fmt.Println(val)
	}
	m, err := msp.MostSpecificPeriod(start, periods...)
	if err != nil {
		fmt.Printf("No significant period found\n")
		os.Exit(1)
	}
	if terminal {
		fmt.Printf("\nThe MSP from the list was: ")
	}
	fmt.Printf("%s\n", m)
}