package main

import (
	"bytes"
	"errors"
	"fmt"

	helpers "git.bullercodeworks.com/brian/adventofcode/helpers"
)

type Vault struct {
	maxX, maxY int
	vault      map[helpers.Coordinate]byte
	keys       map[helpers.Coordinate]byte
	doors      map[helpers.Coordinate]byte
}

func NewVault(file string) *Vault {
	inp := helpers.FileToBytes(file)
	wrk := bytes.Split(inp, []byte{'\n'})

	v := Vault{
		vault: make(map[helpers.Coordinate]byte),
		keys:  make(map[helpers.Coordinate]byte),
		doors: make(map[helpers.Coordinate]byte),
	}
	v.maxY = len(wrk)
	v.maxX = len(wrk[0])
	for y, yv := range wrk {
		for x, xv := range yv {
			v.vault[*helpers.NewCoordinate(x, y)] = xv
			if isKey(xv) {
				v.keys[*helpers.NewCoordinate(x, y)] = xv
			} else if isDoor(xv) {
				v.doors[*helpers.NewCoordinate(x, y)] = xv
			}
		}
	}
	return &v
}

func (v *Vault) findKey(door byte) (helpers.Coordinate, error) {
	var r helpers.Coordinate
	if !isDoor(door) {
		return r, errors.New("Invalid Door")
	}
	for k := range v.keys {
		if v.keys[k] == doorToKey(door) {
			return k, nil
		}
	}
	return r, errors.New("No key for that door")
}

func (v *Vault) findDoor(key byte) (helpers.Coordinate, error) {
	var r helpers.Coordinate
	if !isKey(key) {
		return r, errors.New("Invalid Key")
	}
	for k := range v.doors {
		if v.doors[k] == keyToDoor(key) {
			return k, nil
		}
	}
	return r, errors.New("No door for that key")
}

func (v *Vault) Print() {
	for y := 0; y < v.maxY; y++ {
		for x := 0; x < v.maxX; x++ {
			fmt.Print(string(v.vault[*helpers.NewCoordinate(x, y)]))
		}
		fmt.Println()
	}
}