Byte-snake-engine/constrictor.go

package rules

type ConstrictorRuleset struct {
	StandardRuleset
}

func (r *ConstrictorRuleset) Name() string { return GameTypeConstrictor }

func (r *ConstrictorRuleset) ModifyInitialBoardState(initialBoardState *BoardState) (*BoardState, error) {
	initialBoardState, err := r.StandardRuleset.ModifyInitialBoardState(initialBoardState)
	if err != nil {
		return nil, err
	}

	r.removeFood(initialBoardState)

	err = r.applyConstrictorRules(initialBoardState)
	if err != nil {
		return nil, err
	}

	return initialBoardState, nil
}

func (r *ConstrictorRuleset) CreateNextBoardState(prevState *BoardState, moves []SnakeMove) (*BoardState, error) {
	nextState, err := r.StandardRuleset.CreateNextBoardState(prevState, moves)
	if err != nil {
		return nil, err
	}

	r.removeFood(nextState)

	err = r.applyConstrictorRules(nextState)
	if err != nil {
		return nil, err
	}

	return nextState, nil
}

func (r *ConstrictorRuleset) removeFood(b *BoardState) {
	_, _ = r.callStageFunc(RemoveFoodConstrictor, b, []SnakeMove{})
}

func RemoveFoodConstrictor(b *BoardState, settings Settings, moves []SnakeMove) (bool, error) {
	// Remove all food from the board
	b.Food = []Point{}

	return false, nil
}

func (r *ConstrictorRuleset) applyConstrictorRules(b *BoardState) error {
	_, err := r.callStageFunc(GrowSnakesConstrictor, b, []SnakeMove{})
	return err
}

func GrowSnakesConstrictor(b *BoardState, settings Settings, moves []SnakeMove) (bool, error) {
	// Set all snakes to max health and ensure they grow next turn
	for i := 0; i < len(b.Snakes); i++ {
		b.Snakes[i].Health = SnakeMaxHealth

		tail := b.Snakes[i].Body[len(b.Snakes[i].Body)-1]
		subTail := b.Snakes[i].Body[len(b.Snakes[i].Body)-2]
		if tail != subTail {
			growSnake(&b.Snakes[i])
		}
	}

	return false, nil
}