package main

import (
	"bufio"
	"fmt"
	"os"
)

func main() {
	fmt.Println(part1(readData("example.txt")), 3)
	fmt.Println(part1(readData("data.txt")), 3)
}

type Data struct {
	keys  [][5]int
	locks [][5]int
}

func fits(k, l [5]int) bool {
	for i := range k {
		if k[i]+l[i] > 5 {
			return false
		}
	}
	return true
}

func part1(data Data) (count int) {
	for _, k := range data.keys {
		for _, l := range data.locks {
			if fits(k, l) {
				count++
			}
		}
	}
	return
}

func readData(fileName string) (data Data) {
	data = Data{}
	fp, _ := os.Open(fileName)
	scanner := bufio.NewScanner(fp)

	for scanner.Scan() {
		isKey := scanner.Text()[0] == '.'
		code := [5]int{0, 0, 0, 0, 0}
		for i := 0; scanner.Scan() && i < 5; i++ {
			line := scanner.Text()
			for j := range 5 {
				if line[j] == '#' {
					code[j]++
				}
			}
		}
		scanner.Scan()
		if isKey {
			data.keys = append(data.keys, code)
		} else {
			data.locks = append(data.locks, code)
		}
	}

	return
}