package main

import (
	"bufio"
	"fmt"
	"os"
	"strconv"
	"strings"
)

func NewData(s string) Data {
	f := strings.Fields(s)
	a, err := strconv.Atoi(f[1])
	if err != nil {
		return Data{}
	}
	return Data{
		Direction: f[0],
		Amount:    a,
	}
}
func GetData(s string) []Data {
	var data []Data
	r, err := os.Open(s)
	if err != nil {
		return data
	}
	fileScanner := bufio.NewScanner(r)
	fileScanner.Split(bufio.ScanLines)
	for fileScanner.Scan() {
		data = append(data, NewData(fileScanner.Text()))
	}
	return data
}

type Data struct {
	Direction string
	Amount    int
}

func main() {
	data := GetData("./data")
	result := Part1(data)
	fmt.Printf("Result part 1: %d\n", result)
	result = Part2(data)
	fmt.Printf("Result part 1: %d\n", result)
}

type Point struct {
	X   int
	Y   int
	Aim int
}

func (p Point) Multiply() int {
	return p.Y * p.X
}

func Part1(data []Data) int {
	var point Point
	for _, d := range data {
		switch d.Direction {
		case "forward":
			point.X += d.Amount
		case "down":
			point.Y += d.Amount
		case "up":
			point.Y -= d.Amount
		}
	}
	return point.Multiply()
}

func Part2(data []Data) int {
	var point Point
	for _, d := range data {
		switch d.Direction {
		case "forward":
			point.X += d.Amount
			point.Y += point.Aim * d.Amount
		case "down":
			point.Aim += d.Amount
		case "up":
			point.Aim -= d.Amount
		}
	}
	return point.Multiply()
}