main.go

package main

import (
	"fmt"
	"os"
	"strconv"

	"github.com/mitchellh/go-wordwrap"
	log "github.com/sirupsen/logrus"
	"github.com/yarsiemanym/advent-of-code-2022/common"
	"github.com/yarsiemanym/advent-of-code-2022/day00"
	"github.com/yarsiemanym/advent-of-code-2022/day01"
	"github.com/yarsiemanym/advent-of-code-2022/day02"
	"github.com/yarsiemanym/advent-of-code-2022/day03"
	"github.com/yarsiemanym/advent-of-code-2022/day04"
	"github.com/yarsiemanym/advent-of-code-2022/day05"
	"github.com/yarsiemanym/advent-of-code-2022/day06"
	"github.com/yarsiemanym/advent-of-code-2022/day07"
	"github.com/yarsiemanym/advent-of-code-2022/day08"
	"github.com/yarsiemanym/advent-of-code-2022/day09"
	"github.com/yarsiemanym/advent-of-code-2022/day10"
	"github.com/yarsiemanym/advent-of-code-2022/day11"
	"github.com/yarsiemanym/advent-of-code-2022/day12"
	"github.com/yarsiemanym/advent-of-code-2022/day13"
	"github.com/yarsiemanym/advent-of-code-2022/day14"
	"github.com/yarsiemanym/advent-of-code-2022/day15"
	"github.com/yarsiemanym/advent-of-code-2022/day16"
	"github.com/yarsiemanym/advent-of-code-2022/day17"
	"github.com/yarsiemanym/advent-of-code-2022/day18"
	"github.com/yarsiemanym/advent-of-code-2022/day19"
	"github.com/yarsiemanym/advent-of-code-2022/day20"
	"github.com/yarsiemanym/advent-of-code-2022/vt100"
)

func main() {
	common.InitLogging()
	common.InitSession()
	checkForHelpCommand()

	log.Info("Begin execution.")
	puzzle := setupPuzzle()
	answer := puzzle.Solve()

	vt100.Printf("%d Day %d\n", []interface{}{answer.Year, answer.Day}, vt100.GreenForegroundAttribute)

	vt100.Print("Part 1 Answer: ", vt100.DimAttribute)
	vt100.Println(answer.Part1, vt100.YellowForegroundAttribute)

	vt100.Print("Part 2 Answer: ", vt100.DimAttribute)
	vt100.Println(answer.Part2, vt100.YellowForegroundAttribute)

	vt100.Print("Total Execution Time: ", vt100.DimAttribute)
	vt100.Println(answer.ExecutionTime.String(), vt100.RedForegroundAttribute)

	log.Info("End execution.")
}

func checkForHelpCommand() {
	arg1 := os.Args[1]

	if arg1 == "-h" || arg1 == "--help" || arg1 == "help" {
		printUsage()
		os.Exit(0)
	}
}

func printUsage() {
	limit := uint(100)
	description1 := "Run the solution for the puzzle from specified day of Advent of Code 2022. If a local copy of your puzzle input does not exist, it will attempt to automatically download it using your session token. When complete, the answers to parts 1 and 2 will be printed to the terminal."
	description2 := "Day 0 is a special day with a mock-puzzle to exercise the application before the first real puzzle unlocks."

	fmt.Println(wordwrap.WrapString(description1, limit))
	fmt.Println("")
	fmt.Println(wordwrap.WrapString(description2, limit))
	fmt.Println("")
	fmt.Println("Usage:")
	fmt.Println("")
	fmt.Println("\tadvent-of-code-2022 <day>")
	fmt.Println("")
	fmt.Println("Parameters:")
	fmt.Println("")
	fmt.Println("\tday\t\t\tRun the solution for the specified day, i.e. 0-25.")
	fmt.Println("")
	fmt.Println("Environment Variables:")
	fmt.Println("")
	fmt.Println("\tAOC_SESSION_TOKEN\tSet your Advent of Code session token.")
	fmt.Println("\tAOC_LOG_LEVEL\t\tSet the log level. Defaults to \"warn\" if not set.")
	fmt.Println("")
}

func setupPuzzle() common.Puzzle {
	log.Debug("Setting up puzzle.")
	day := sanitizeDayArg(os.Args[1])

	puzzle := common.Puzzle{
		Year: 2022,
		Day:  day,
	}

	if len(os.Args) >= 3 {
		puzzle.InputFile = os.Args[2]
	}

	switch puzzle.Day {
	case 0:
		puzzle.SetSolution(day00.Solve)
	case 1:
		puzzle.SetSolution(day01.Solve)
	case 2:
		puzzle.SetSolution(day02.Solve)
	case 3:
		puzzle.SetSolution(day03.Solve)
	case 4:
		puzzle.SetSolution(day04.Solve)
	case 5:
		puzzle.SetSolution(day05.Solve)
	case 6:
		puzzle.SetSolution(day06.Solve)
	case 7:
		puzzle.SetSolution(day07.Solve)
	case 8:
		puzzle.SetSolution(day08.Solve)
	case 9:
		puzzle.SetSolution(day09.Solve)
	case 10:
		puzzle.SetSolution(day10.Solve)
	case 11:
		puzzle.SetSolution(day11.Solve)
	case 12:
		puzzle.SetSolution(day12.Solve)
	case 13:
		puzzle.SetSolution(day13.Solve)
	case 14:
		puzzle.SetSolution(day14.Solve)
	case 15:
		puzzle.SetSolution(day15.Solve)
	case 16:
		puzzle.SetSolution(day16.Solve)
	case 17:
		puzzle.SetSolution(day17.Solve)
	case 18:
		puzzle.SetSolution(day18.Solve)
	case 19:
		puzzle.SetSolution(day19.Solve)
	case 20:
		puzzle.SetSolution(day20.Solve)
	default:
		log.Fatalf("Day %d has no solution yet.", puzzle.Day)
	}

	return puzzle
}

func sanitizeDayArg(arg string) int {
	log.Debug("Sanitizing day argument.")
	log.Tracef("arg = \"%s\"", arg)

	day, err := strconv.Atoi(arg)

	if err != nil {
		log.Fatalf("\"%s\" is not an integer.", arg)
	}

	if day < 0 || day > 25 {
		log.Fatalf("%d is not between 0 and 25.", day)
	}

	log.Tracef("day = %d", day)
	return day
}