oglimmer/ggo

package de.oglimmer.ggo.ui.shortlife;

import java.io.Serializable;
import java.util.Map;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import de.oglimmer.ggo.logic.Player;
import lombok.NonNull;
import lombok.RequiredArgsConstructor;
import lombok.ToString;

/**
 * Transforms the server-side model into client-side model
 */
@RequiredArgsConstructor
@ToString
public class UIBoardStateProvider implements Serializable {

    private static final long serialVersionUID = 1L;

    @NonNull
    private Player forPlayer;

    public Map<String, UIField> getCorToFields() {
        return forPlayer.getGame().getBoard().getFields().stream().map(f -> new UIField(f, forPlayer))
                .collect(Collectors.toMap(UIField::getId, f -> f));
    }

    public Map<String, UIUnit> getIdToUnits() {
        Stream<UIUnit> units = forPlayer.getGame().getBoard().getFields().stream().filter(f -> f.getUnit() != null)
                .map(f -> new UIUnit(f.getUnit(), f.getPos().x, f.getPos().y, forPlayer));
        Stream<UIUnit> structures = forPlayer.getGame().getBoard().getFields().stream()
                .filter(f -> f.getStructure() != null).map(f -> new UIUnit(f.getStructure(),
                        f.getStructure().getPlayer().getSide().toString(), f.getPos().x, f.getPos().y));
        return Stream.concat(units, structures).collect(Collectors.toMap(UIUnit::getId, u -> u));
    }

    public Map<String, UIHandItem> getIdToHanditems() {
        return forPlayer.getUnitInHand().stream().map(u -> new UIHandItem(u, forPlayer))
                .collect(Collectors.toMap(UIHandItem::getId, hi -> hi));
    }

    public Map<String, UIButton> getIdToButtons() {
        return forPlayer.getGame().getCurrentPhase().getButtons(forPlayer).stream()
                .collect(Collectors.toMap(UIButton::getId, b -> b));
    }

    public Boolean isShowCoordinates() {
        return forPlayer.getGame().getCurrentPhase().isShowCoordinates();
    }
}