/*
 * Copyright (c) 2019-2020. Developed by Hedgecode.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.hedgecode.chess.scanner.json.lichess;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import com.google.gson.annotations.Expose;
import com.google.gson.annotations.SerializedName;

import org.hedgecode.chess.scanner.format.Move;
import org.hedgecode.chess.scanner.format.lichess.Clock;
import org.hedgecode.chess.scanner.format.lichess.Game;
import org.hedgecode.chess.scanner.format.lichess.GameData;
import org.hedgecode.chess.scanner.format.lichess.Player;

/**
 * JSONGameData
 *
 * @author Dmitry Samoshin aka gotty
 */
public class JSONGameData implements GameData {

    private static final String PLAYER_WHITE = "white";
    private static final String PLAYER_BLACK = "black";

    private static final String GAME_DRAW = "draw";

    private static final Map<String, String> RESULT = new HashMap<String, String>() {
        {
            put( PLAYER_WHITE, "1-0" );
            put( PLAYER_BLACK, "0-1" );
            put( GAME_DRAW, "1/2-1/2");
        }
    };

    private static final Map<Boolean, String> RATED_GAME = new HashMap<Boolean, String>() {
        {
            put( Boolean.FALSE, "Unrated" );
            put( Boolean.TRUE,  "Rated"   );
        }
    };

    private static final String EVENT_GAME_FORMAT = "%s %s game";

    @SerializedName("game")
    private JSONGame game;

    @SerializedName("player")
    private JSONPlayer player;

    @SerializedName("opponent")
    private JSONPlayer opponent;

    @SerializedName("clock")
    private JSONClock clock;

    @SerializedName("steps")
    private JSONMove[] steps;

    @SerializedName("treeParts")
    private JSONMove[] treeParts;

    @Expose
    private Move[] moves;

    JSONGameData() {
    }

    @Override
    public Game game() {
        return game;
    }

    @Override
    public Player player() {
        return player;
    }

    @Override
    public Player opponent() {
        return opponent;
    }

    public String event() {
        return String.format(
                EVENT_GAME_FORMAT,
                RATED_GAME.get(game.rated()),
                game.speed()
        );
    }

    public Player white() {
        return player.color().equals(PLAYER_WHITE)
                ? player
                : opponent.color().equals(PLAYER_WHITE)
                ? opponent
                : null;
    }

    public Player black() {
        return player.color().equals(PLAYER_BLACK)
                ? player
                : opponent.color().equals(PLAYER_BLACK)
                ? opponent
                : null;
    }

    public String result() {
        String result = null;
        if (game.winner() != null) {
            if (PLAYER_WHITE.equals(game.winner())) {
                result = PLAYER_WHITE;
            } else if (PLAYER_BLACK.equals(game.winner())) {
                result = PLAYER_BLACK;
            }
        } else if (GAME_DRAW.equals(game.status().name())) {
            result = GAME_DRAW;
        }
        return RESULT.get(result);
    }

    @Override
    public Clock clock() {
        return clock;
    }

    @Override
    public Move[] moves() {
        if (moves == null) {
            moves = steps != null ? steps : treeParts;
            List<Move> movesList = new ArrayList<>(Arrays.asList(moves));
            movesList.removeIf(move -> move.move() == null);
            moves = movesList.toArray(
                    new Move[movesList.size()]
            );
        }
        return moves;
    }

}