uci.go

package main

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

func setPosition(pos *Position, tokens []string) {
	if tokens[1] == "startpos" {
		pos.load(STARTING_POS)

		if len(tokens) == 2 {
			return
		}
	}

	if tokens[1] == "fen" {
		fen := ""
		for i := 2; i < len(tokens); i++ {
			fen += tokens[i] + " "
		}

		pos.load(fen)
	}

	for i := 0; i < len(tokens); i++ {
		token := tokens[i]
		if token == "moves" {
			for j := i + 1; j < len(tokens); j++ {
				handleUciMove(pos, tokens[j])
			}

			break
		}
	}
}

func handleMove(pos *Position, tokens []string) {
	moveStr := strings.ToLower(tokens[1])
	handleUciMove(pos, moveStr)
}

func handleUciMove(pos *Position, moveStr string) {
	moves, nrMoves := generateMoves(*pos, false)

	// Find the move by comparing move strings
	for i := 0; i < nrMoves; i++ {
		move := moves[i]
		if !pos.isMoveLegal(move, false) {
			continue
		}

		if moveStr == move.toStr() {
			pos.makeMove(move)
			return
		}
	}

	fmt.Printf("Invalid move: %s", moveStr)
}

func uciLoop(pos *Position) {
	for {
		in := bufio.NewReader(os.Stdin)
		line, _ := in.ReadString('\n')

		var nltokens = strings.Split(line, "\n")
		var tokens = strings.Split(nltokens[0], " ")
		exit := false

		switch tokens[0] {
		case "uci":
			fmt.Printf("id name Axii id author qantata\nuciok")
		case "position":
			setPosition(pos, tokens)
		case "showpos":
			pos.pretty()
		case "go":
			searchGo(*pos)
		case "ucinewgame":
			pos.load(STARTING_POS)
		case "move":
			handleMove(pos, tokens)
		case "isready":
			fmt.Println("readyok")
		case "exit":
			exit = true
		case "quit":
			exit = true
		default:
			fmt.Printf("Unknown token %s\n", tokens[0])
		}

		fmt.Println()

		if exit {
			fmt.Println("Goodbye.")
			break
		}
	}
}