package main

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

func contains(s []int, e int) bool {
	for _, a := range s {
		if a == e {
			return true
		}
	}
	return false
}

func check(nums []int, maps map[int][]int) bool {
	correct := true
	for i := 0; i < len(nums); i++ {
		var checking = nums[i]
		for j := 0; j < len(nums); j++ {

			if !contains(maps[checking], nums[j]) {
				if i < j {
					correct = false
				}
			}
		}
	}

	return correct
}

func main() {

	lines, err := parse()
	if err != nil {
		fmt.Println("Error:", err)
	}

	var p1 int = 0
	var p2 int = 0

	maps := make(map[int][]int)

	for y := 0; y < len(lines); y++ {
		if lines[y] != "" && lines[y][2] == '|' {
			split := strings.Split(lines[y], "|")
			l, err := strconv.Atoi(split[0])
			if err != nil {
				panic(err)
			}
			r, err := strconv.Atoi(split[1])
			if err != nil {
				panic(err)
			}
			maps[l] = append(maps[l], r)
		}

		if lines[y] != "" && lines[y][2] == ',' {
			numsstr := strings.Split(lines[y], ",")
			var nums []int

			for i := 0; i < len(numsstr); i++ {
				x, err := strconv.Atoi(numsstr[i])
				if err != nil {
					panic(err)
				}
				nums = append(nums, x)
			}

			if check(nums, maps) {
				p1 += nums[len(nums)/2]
			}
		}
	}

	fmt.Println("p1: ", p1)
	fmt.Println("p2: ", p2)
}

func parse() ([]string, error) {
	if len(os.Args) < 2 {
		return nil, fmt.Errorf("no file provided")
	}
	filePath := os.Args[1]
	data, err := os.ReadFile(filePath)
	if err != nil {
		return nil, fmt.Errorf("error opening file: %v", err)
	}
	chunks := strings.Split(string(data), "\n")
	var nonEmpthyChunks = []string{}

	for _, chunk := range chunks {
		if chunk != "" {
			nonEmpthyChunks = append(nonEmpthyChunks, chunk)
		}
	}
	return nonEmpthyChunks, nil
}