package d2

import (
	"log"
	"strconv"
	"strings"
)

type position struct {
	x int64
	y int64
}

func D2(input string) int64 {
	position := position{}
	directions := strings.Split(strings.TrimSpace(input), "\n")

	for _, direction := range directions {
		directionAndValue := strings.Split(strings.TrimSpace(direction), " ")

		if len(directionAndValue) != 2 {
			log.Fatal("line has wrong number of elements: ", direction)
		}

		value, err := strconv.Atoi(directionAndValue[1])

		if err != nil {
			log.Fatal(err)
		}

		switch directionAndValue[0] {
		case "forward":
			position.x += int64(value)
		case "down":
			position.y += int64(value)
		case "up":
			position.y -= int64(value)
		}
	}

	return position.x * position.y
}

func D2P2(input string) int {
	directions := strings.Split(strings.TrimSpace(input), "\n")
	x := 0
	y := 0
	aim := 0

	for _, direction := range directions {
		directionAndValue := strings.Split(strings.TrimSpace(direction), " ")

		if len(directionAndValue) != 2 {
			log.Fatal("line has wrong number of elements: ", direction)
		}

		value, err := strconv.Atoi(directionAndValue[1])

		if err != nil {
			log.Fatal(err)
		}

		switch directionAndValue[0] {
		case "forward":
			y += aim * value
			x += value
		case "down":
			aim += value
		case "up":
			aim -= value
		}
	}

	return x * y
}