shogiops/src/variant/variant.ts

Shogi rules and operations

import type { Result } from '@badrap/result';
import type { Setup } from '../types.js';
import { Annanshogi } from './annanshogi.js';
import { Checkshogi } from './checkshogi.js';
import { Chushogi } from './chushogi.js';
import { Kyotoshogi } from './kyotoshogi.js';
import { Minishogi } from './minishogi.js';
import type { PositionError } from './position.js';
import { Shogi } from './shogi.js';

export interface RulesTypeMap {
  standard: Shogi;
  minishogi: Minishogi;
  chushogi: Chushogi;
  annanshogi: Annanshogi;
  kyotoshogi: Kyotoshogi;
  checkshogi: Checkshogi;
}

export function defaultPosition<R extends keyof RulesTypeMap>(rules: R): RulesTypeMap[R] {
  switch (rules) {
    case 'chushogi':
      return Chushogi.default();
    case 'minishogi':
      return Minishogi.default();
    case 'annanshogi':
      return Annanshogi.default();
    case 'kyotoshogi':
      return Kyotoshogi.default();
    case 'checkshogi':
      return Checkshogi.default();
    default:
      return Shogi.default();
  }
}

export function initializePosition<R extends keyof RulesTypeMap>(
  rules: R,
  setup: Setup,
  strict: boolean,
): Result<RulesTypeMap[R], PositionError> {
  switch (rules) {
    case 'chushogi':
      return Chushogi.from(setup, strict);
    case 'minishogi':
      return Minishogi.from(setup, strict);
    case 'annanshogi':
      return Annanshogi.from(setup, strict);
    case 'kyotoshogi':
      return Kyotoshogi.from(setup, strict);
    case 'checkshogi':
      return Checkshogi.from(setup, strict);
    default:
      return Shogi.from(setup, strict);
  }
}