package main

import (
	"fmt"
	"strings"
	"time"

	intcode "git.bullercodeworks.com/brian/adventofcode/2019/intcode-processor"
	helpers "git.bullercodeworks.com/brian/adventofcode/helpers"
)

var beam map[string]int
var maxY int

func main() {
	pt := helpers.GetArgNumber(1)
	prog := intcode.ReadIntCodeFile("input")
	beam = make(map[string]int)
	if pt == "1" {
		part1(prog)
	} else {
		part2(prog)
	}
}

func part1(prog []int) {
	var count int
	var lastRowStart int
	for y := 0; y < 50; y++ {
		var foundFirst bool
		fmt.Print(strings.Repeat(" ", lastRowStart))
		for x := lastRowStart; x < 50; x++ {
			out := runCodeAt(prog, x, y)
			fmt.Print(out)
			beam[c(x, y)] = out
			if out == 1 {
				if !foundFirst {
					lastRowStart = x
				}
				foundFirst = true
				count++
			} else if foundFirst {
				break
			}
		}
		fmt.Println()
		maxY = y
	}
	fmt.Println("Total:", count)
}

func part2(prog []int) {
	shipSize := 100
	// Guesses
	//x, y := 0, 1000
	var x, square int
	bot, _ := 1034, 1038
	for square != shipSize {
		bot++
		fmt.Print("Calculating ", bot, "... ")
		square, x = maxSquare(prog, bot, bot/2)
		fmt.Println(square)
	}
	fmt.Println(x, (bot+1)-100)
}

func part2Slow(prog []int) {
	shipSize := 100
	var x, y int
	retX, retY := -1, -1
	var lastRowStart int
	for y = maxY; true; y++ {
		var foundFirst bool
		fmt.Print(strings.Repeat(" ", lastRowStart))
		for x = lastRowStart; x < y; x++ {
			out := runCodeAt(prog, x, y)
			fmt.Print(out)
			beam[c(x, y)] = out
			if out == 1 {
				if !foundFirst {
					lastRowStart = x
				}
				foundFirst = true
			} else if foundFirst {
				break
			}
		}
		square, _ := maxSquare(prog, y, lastRowStart)
		fmt.Println(":", square)
		// Every row, check what the maximum square is
		if square >= shipSize {
			retY, retX = y, lastRowStart
			break
		}
	}
	fmt.Println(retX, ",", retY)
}

// maxSquare returns the square size and the bottom-left x position it starts at
func maxSquare(prog []int, row, firstX int) (int, int) {
	for ; firstX < row; firstX++ {
		// Find the first X
		if runCodeAt(prog, firstX, row) == 0 {
			continue
		}

		var square int
		for d := 0; true; d++ {
			if runCodeAt(prog, firstX+d, row-d) == 0 {
				return square, firstX
			}
			square++
		}
	}
	return 0, firstX
}

func runCodeAt(prog []int, x, y int) int {
	// If we've already cached this pos, just return it
	if v, ok := beam[c(x, y)]; ok {
		return v
	}

	var res int
	p := intcode.NewProgram(prog)
	go func() {
		sendInputs(p, []int{x, y})
		res = getOutput(p)
	}()
	p.Run()

	beam[c(x, y)] = res
	return res
}

func sendInputs(p *intcode.Program, inp []int) {
	for k := range inp {
		for !p.NeedsInput() {
			time.Sleep(1)
		}
		p.Input(inp[k])
	}
}

func getOutput(p *intcode.Program) int {
	for !p.NeedsOutput() {
		time.Sleep(1)
	}
	return p.Output()
}

func c(x, y int) string {
	return fmt.Sprintf("[%d, %d]", x, y)
}