lane change right

game/ai.go

@@ -5,8 +5,11 @@ import "log"
 func chooseCommand(t Team, s State) command {
 	h := t.Baton.Holder
 	if collide(h.Pos+1, h.Lane, s) {
+		if h.Lane <= t.Lane {
 			return left
 		}
+		return right
+	}
 
 	var nextBot *Bot
 	for i, b := range t.Bots {

game/commands.go

@@ -0,0 +1,30 @@
+package game
+
+import "math/rand"
+
+func doCommand(cmd command, b *Bot) {
+	da := 1
+	da += rand.Intn(3) - 1
+
+	switch cmd {
+	case speedUp:
+		b.a += da
+		accelerate(b)
+	case slowDown:
+		b.a -= da
+		accelerate(b)
+	case left:
+		b.Lane++
+	case right:
+		b.Lane--
+	}
+}
+
+type command int
+
+const (
+	speedUp command = iota
+	slowDown
+	left
+	right
+)

game/game.go

@@ -2,7 +2,6 @@ package game
 
 import (
 	"log"
-	"math/rand"
 )
 
 type State struct {
@@ -26,6 +25,7 @@ func NewState() State {
 		teams = append(teams, Team{
 			Bots:  bots,
 			Baton: Baton{Holder: &bots[0]},
+			Lane:  i,
 		})
 	}
 
@@ -48,6 +48,7 @@ type Team struct {
 	Bots  []Bot
 	Baton Baton
 	won   bool
+	Lane  int
 }
 
 type Bot struct {
@@ -85,30 +86,6 @@ func UpdateState(sOld State) State {
 	return s
 }
 
-func doCommand(cmd command, b *Bot) {
-	da := 1
-	da += rand.Intn(3) - 1
-
-	switch cmd {
-	case speedUp:
-		b.a += da
-		accelerate(b)
-	case slowDown:
-		b.a -= da
-		accelerate(b)
-	case left:
-		b.Lane++
-	}
-}
-
-type command int
-
-const (
-	speedUp command = iota
-	slowDown
-	left
-)
-
 func maybePassBaton(t *Team) {
 	for i, b := range t.Bots {
 		h := t.Baton.Holder