swift-chess/Sources/Engine/Board.swift

enum Event {
	case kingInCheck(_ by: Piece)
	case piecePinned(from: Piece, on: Piece)
}
protocol EventDelegate {
	func notify(_ event: Event)
	func movePiece(_ piece: Piece, to dst: Square.Position)
}
public class Board: CustomStringConvertible, EventDelegate {
	public typealias Grid = [[Square]]

	private enum UnicodeBar: String, CustomStringConvertible {
		case VerticalBar = "│"
		case TopHorizontalLine = "┌───┬───┬───┬───┬───┬───┬───┬───┐"
		case MiddleHorizontalLine = "  ├───┼───┼───┼───┼───┼───┼───┼───┤"
		case BottomHorizonLine = "  └───┴───┴───┴───┴───┴───┴───┴───┘"

		var description: String {
			return self.rawValue
		}
	}

	func notify(_ event: Event) {

	}

	func movePiece(_ piece: Piece, to dst: Square.Position) {

	}

	public func getSquareInfo(on pos: Square.Position) -> Square? {
		self[pos]
	}

	private var squares = [Square]()
	private var board: Grid {
		var board = Grid()
		var rank = -1
		for i in 0...63 {
			if i % 8 == 0 {
				board.append([])
				rank += 1
			}
			board[rank].append(squares[i])
		}
		return board
	}

	public private(set) var fen: Fen

	public func setBoard() throws {
		var file: Int8 = 1
		var rank: Int8 = 8
		var index: Int8 = 0
		var b: [Square] = squares
		for c in fen.placement {
			let r = 8 - Int(rank)
			if c == "/" {
				if file != 9 {
					throw Fen.FenError.NotAppropriateLength(
						n: file, column: index)
				}
				rank -= 1
				file = 0
			} else if c.isWholeNumber, let n = Int8(String(c)) {
				if n < 1 {
					throw Fen.FenError.NumberTooSmall(n: n, column: index)
				} else if n > 8 {
					throw Fen.FenError.NumberTooBig(n: n, column: index)
				}

				let f = file
				for i in f..<(f + n) {
					file += 1
					b[8 * r + Int(i)].piece = nil
				}
				file -= 1
			} else if c.isASCII {
				switch Kind[c] {
				case .some(let (k, c)):
					let piece: Piece =
						switch k {
						case .Pawn:
							Pawn(with: c, on: .init(rank: rank, file: file))
						case .Knight:
							Knight(with: c, on: .init(rank: rank, file: file))
						case .Bishop:
							Bishop(with: c, on: .init(rank: rank, file: file))
						case .Rook:
							Rook(with: c, on: .init(rank: rank, file: file))
						case .Queen:
							Queen(with: c, on: .init(rank: rank, file: file))
						case .King:
							King(with: c, on: .init(rank: rank, file: file))
						}
					piece.delegate = self
					b[8 * r + Int(file) - 1].piece = piece
				case .none:
					throw Fen.FenError.InvalidCharacter(
						c: String(c), column: index)
				}
			}

			file += 1
			index += 1
		}
		#if DEBUG
			print(b)
		#endif
		squares = b
	}

	public required init(
		fen: Fen =
			.init(
				fen: "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1")
	) {
		var rank: Int8 = 8
		self.fen = fen
		for i in 0...63 {
			let index = Int8(i)
			let square = Square(
				position: .init(rank: rank, file: (index % 8) + 1),
				index: i,
				color: index % 2 != rank % 2 ? .Black : .White)
			squares.append(square)
			if (index + 1) % 8 == 0 {
				rank -= 1
			}
		}
		// TODO: Handle better
		do {
			try setBoard()
		} catch Fen.FenError.NotAppropriateLength(let n, let column) {
			fatalError("Not appropriate length: \(n) on \(column)")
		} catch {

		}
	}

	public var description: String {
		var boardString =
			"    A   B   C   D   E   F   G   H \n  \(UnicodeBar.TopHorizontalLine)\n"
		var _rank: UInt8 = 8
		board.forEach { rank in
			boardString += String(_rank) + " \(UnicodeBar.VerticalBar)"
			rank.forEach { square in
				let p = square.piece?.unicodeRepresentation ?? " "
				boardString += " \(p) \(UnicodeBar.VerticalBar)"
			}
			boardString += "\n"
			_rank -= 1
			if _rank > 0 {
				boardString += "\(UnicodeBar.MiddleHorizontalLine)\n"
			}
		}
		boardString += "\(UnicodeBar.BottomHorizonLine)"
		return boardString
	}

	internal subscript(pos: Square.Position) -> Square? {
		guard let i = pos.index else {
			return nil
		}
		if i > squares.count {
			return nil
		}
		return squares[i]
	}
}