package game

import "log"

func chooseCommand(s State, teamID int) command {
	t := s.Teams[teamID]
	h := ActiveBot(t)
	if collide(h.Position.Pos+1, h.Position.Lane, s) != nil {
		if h.Position.Lane <= t.Lane && h.Position.Lane < NumLanes-1 {
			return left
		}
		return right
	}

	var nextBot *Bot
	for i, b := range t.Bots {
		if b.ID == h.ID+1 {
			nextBot = &t.Bots[i]
			break
		}
	}

	if nextBot != nil {
		if h.Position.Lane != nextBot.Position.Lane {
			if abs(nextBot.Position.Pos-h.Position.Pos) < h.v {
				return slowDown
			}
		}
	}

	return speedUp
}

func smartChooseCommand(s State, teamID int) command {
	bestCmd, bestN := speedUp, 0

	log.Printf("team %d base score: %d", teamID, score(s, teamID))
	for _, cmd := range []command{speedUp, slowDown, left, right} {
		ss := doCommand(cmd, s, teamID)
		n := score(ss, teamID)
		log.Printf("team %d score %s: %d", teamID, cmd, n)
		if n > bestN {
			bestCmd, bestN = cmd, n
		}
	}

	return bestCmd
}

func score(s State, teamID int) int {
	t := s.Teams[teamID]
	b := ActiveBot(t)
	if b == nil {
		return 0
	}
	return b.Position.Pos
}