setup checkmate

js/goals/setup-checkmate.ts

@@ -0,0 +1,21 @@
+import { Game } from 'chess-fetcher'
+import { TrophyCheckResult } from '../types/types'
+import { fenToPosition } from '../utils/fen-to-position'
+
+export function originalSetupCheckmate(game: Game, fen: string): TrophyCheckResult {
+    if (game.result.via !== 'checkmate') {
+        return []
+    }
+
+    let position = fenToPosition(fen)
+
+    if (position.startsWith('rnbqkbnr..K') || position.endsWith('k.....RNBQKBNR')) {
+        return [
+            {
+                color: game.result.winner === 'white' ? 'w' : 'b',
+            },
+        ]
+    }
+
+    return []
+}

tests/setup-checkmate.test.ts

@@ -0,0 +1,22 @@
+import { Game } from 'chess-fetcher'
+import { describe, expect, test } from 'vitest'
+import { originalSetupCheckmate } from '../js/goals/setup-checkmate'
+
+describe('original setup checkmate', () => {
+    test.each([
+        ['white', { color: 'w' }, '8/8/8/8/8/8/2k5/RNBQKBNR b KQ - 0 1'],
+        ['black', { color: 'b' }, 'rnbqkbnr/2K5/8/8/8/8/8/8 w kq - 0 1'],
+    ])('test FEN: %p %p', (winner, expected, fen) => {
+        expect(
+            originalSetupCheckmate(
+                {
+                    result: {
+                        via: 'checkmate',
+                        winner: winner,
+                    },
+                } as Game,
+                fen
+            )
+        ).toStrictEqual([expected])
+    })
+})