# frozen-string-literal: true
module Bioshogi
  module SoldierWalker
    extend self

    # soldier が移動可能な手をすべて取得する
    #
    # アルゴリズム
    #
    #     １       １
    #   +---+    +---+
    #   | ・|一  | ・|一
    #   | ・|二  | ・|二
    #   | 香|三  | 杏|三
    #   +---+    +---+
    #
    #   ▲１三香の場合
    #   A. "１二" に移動してみて、その状態でさらに動けるなら "１二香" をストアしつつ、成れるなら成ってさらに動けるなら "１二杏" もストア
    #   B. "１一" に移動してみて、その状態でさらに動けるなら "１一杏" をストアしつつ、成れるなら成ってさらに動けるなら "１三杏" をストア
    #
    #   ▲１三杏 の場合
    #   C. "１二" に移動してみて、その状態でさらに動けるなら "１二杏" をストア。動けないので成れるなら成って→成れない
    #   D. "１一" に移動してみて → 移動できない
    #
    #   となるので成っているかどうかにかかわらず B の方法でやればいい
    #
    def call(container, soldier, options = {})
      Enumerator.new do |yielder|
        soldier.all_vectors.each do |vector|
          place = soldier.place
          loop do
            place = place.vector_add(vector)

            # 盤外に出たら終わり
            unless place
              break
            end

            captured_soldier = container.board.surface[place]

            # 自分の駒に衝突したら終わり
            if captured_soldier && captured_soldier.location == soldier.location
              break
            end

            # 空または相手駒の升には行ける(取れる)
            piece_store(container, soldier, place, captured_soldier, yielder, options)

            # 相手駒があるのでこれ以上は進めない
            if captured_soldier
              break
            end

            # いっぽだけベクトルならそれで終わり
            if vector.kind_of?(V::OnceVector)
              break
            end
          end
        end
      end
    end

    private

    # place の場所は空なので player の soldier を place に置けそうだ
    # でも place に置いてそれ以上動けなかったら反則になるので
    # 1. それ以上動けるなら置く
    # 2. 成れるなら成ってみて、それ以上動けるなら置く
    def piece_store(container, origin_soldier, place, captured_soldier, yielder, options)
      if options[:king_captured_only]
        if captured_soldier && captured_soldier.piece.key == :king
        else
          return
        end
      end

      # 死に駒にならないのであれば有効
      soldier = origin_soldier.merge(place: place)

      # 成れるなら成る
      if origin_soldier.tsugini_nareru_on?(soldier.place)
        move_hand = Hand::Move.create(soldier: soldier.merge(promoted: true), origin_soldier: origin_soldier, captured_soldier: captured_soldier)
        success = piece_store_core(container, move_hand, yielder, options)
        if success
          if options[:promoted_only]
            # 成と不成の両方がある(かもしれない)場合は成の方だけ生成する
            # これは本当はよくない
            # 暫定的な対処
            # 探索があまりに重いのでこうしている
            return
          end
        end
      end

      if soldier.alive?
        # すでに成っている(または成らない)手を生成
        move_hand = Hand::Move.create(soldier: soldier, origin_soldier: origin_soldier, captured_soldier: captured_soldier)
        piece_store_core(container, move_hand, yielder, options)
      end
    end

    def piece_store_core(container, move_hand, yielder, options)
      # 自玉に王手がかかる手は除外するか？
      if options[:legal_only]
        unless move_hand.legal_hand?(container)
          return
        end
      end

      # 相手に王手がかかる手だけにする
      if options[:mate_only]
        unless move_hand.mate_hand?(container)
          return
        end
      end

      yielder << move_hand

      move_hand
    end
  end
end