package main

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

func main() {

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

	var p1 int = 0
	var p2 int = 0

	h := len(lines)
	w := len(lines[0])

	for y := 0; y < h; y++ {
		for x := 0; x < w; x++ {

			if lines[y][x] == 'X' {
				// check right
				if x <= len(lines[y])-4 {
					r := string([]byte{lines[y][x+1], lines[y][x+2], lines[y][x+3]})
					if r == "MAS" {
						p1++
					}
				}

				// check left
				if x >= 3 {
					l := string([]byte{lines[y][x-1], lines[y][x-2], lines[y][x-3]})
					if l == "MAS" {
						p1++
					}
				}

				// check if up possible
				if y >= 3 {
					u := string([]byte{lines[y-1][x], lines[y-2][x], lines[y-3][x]})
					if u == "MAS" {
						p1++
					}
					if x >= 3 {
						ul := string([]byte{lines[y-1][x-1], lines[y-2][x-2], lines[y-3][x-3]})
						if ul == "MAS" {
							p1++
						}
					}
					if x < w-3 {
						ur := string([]byte{lines[y-1][x+1], lines[y-2][x+2], lines[y-3][x+3]})
						if ur == "MAS" {
							p1++
						}
					}

				}

				// check down
				if y <= h-4 {
					d := string([]byte{lines[y+1][x], lines[y+2][x], lines[y+3][x]})
					if d == "MAS" {
						p1++
					}
					if x >= 3 {
						dl := string([]byte{lines[y+1][x-1], lines[y+2][x-2], lines[y+3][x-3]})
						if dl == "MAS" {
							p1++
						}
					}
					if x < w-3 {
						dr := string([]byte{lines[y+1][x+1], lines[y+2][x+2], lines[y+3][x+3]})
						if dr == "MAS" {
							p1++
						}
					}
				}
			}

			if lines[y][x] == 'A' {
				if y >= 1 && y < h-1 && x >= 1 && x < w-1 {
					check := string([]byte{lines[y-1][x-1], lines[y-1][x+1], lines[y+1][x-1], lines[y+1][x+1]})
					if check == "MSMS" {
						p2++
					}
					if check == "SMSM" {
						p2++
					}
					if check == "MMSS" {
						p2++
					}
					if check == "SSMM" {
						p2++
					}
				}
			}
		}
	}

	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
}