Arakne / Araknemu

 * This file is part of Araknemu.

 *

 * Araknemu is free software: you can redistribute it and/or modify

 * it under the terms of the GNU Lesser General Public License as published by

 * the Free Software Foundation, either version 3 of the License, or

 * (at your option) any later version.

 *

 * Araknemu is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public License

 * along with Araknemu.  If not, see <https://www.gnu.org/licenses/>.

 *

 * Copyright (c) 2017-2019 Vincent Quatrevieux

 */

package fr.quatrevieux.araknemu.game.fight.state;

import fr.quatrevieux.araknemu.core.event.EventsSubscriber;

import fr.quatrevieux.araknemu.core.event.Listener;

import fr.quatrevieux.araknemu.game.fight.Fight;

import fr.quatrevieux.araknemu.game.fight.JoinFightError;

import fr.quatrevieux.araknemu.game.fight.ending.EndFightResults;

import fr.quatrevieux.araknemu.game.fight.ending.reward.FightRewardsSheet;

import fr.quatrevieux.araknemu.game.fight.event.FightJoined;

import fr.quatrevieux.araknemu.game.fight.event.FightLeaved;

import fr.quatrevieux.araknemu.game.fight.event.FighterAdded;

import fr.quatrevieux.araknemu.game.fight.event.FighterPlaceChanged;

import fr.quatrevieux.araknemu.game.fight.event.FighterRemoved;

import fr.quatrevieux.araknemu.game.fight.exception.FightException;

import fr.quatrevieux.araknemu.game.fight.exception.FightMapException;

import fr.quatrevieux.araknemu.game.fight.exception.InvalidFightStateException;

import fr.quatrevieux.araknemu.game.fight.exception.JoinFightException;

import fr.quatrevieux.araknemu.game.fight.fighter.Fighter;

import fr.quatrevieux.araknemu.game.fight.map.FightCell;

import fr.quatrevieux.araknemu.game.fight.map.util.PlacementCellsGenerator;

import fr.quatrevieux.araknemu.game.fight.team.FightTeam;

import fr.quatrevieux.araknemu.game.listener.fight.SendFighterPositions;

import fr.quatrevieux.araknemu.game.listener.fight.SendFighterReadyState;

import fr.quatrevieux.araknemu.game.listener.fight.SendJoinTeamOptionChangedMessage;

import fr.quatrevieux.araknemu.game.listener.fight.SendNeedHelpOptionChangedMessage;

import fr.quatrevieux.araknemu.game.listener.fight.SendNewFighter;

import fr.quatrevieux.araknemu.game.listener.fight.StartFightWhenAllReady;

import fr.quatrevieux.araknemu.game.listener.fight.fighter.ClearFighter;

import fr.quatrevieux.araknemu.game.listener.fight.fighter.SendFighterRemoved;

import org.checkerframework.checker.nullness.qual.EnsuresNonNullIf;

import org.checkerframework.checker.nullness.qual.MonotonicNonNull;

import org.checkerframework.checker.nullness.qual.RequiresNonNull;

import org.checkerframework.checker.nullness.util.NullnessUtil;

import java.util.Collection;

import java.util.Collections;

import java.util.HashMap;

import java.util.Map;

import java.util.concurrent.ScheduledFuture;

import java.util.stream.Collectors;

/**

 * Placement before start fight

 */

public final class PlacementState implements LeavableState, EventsSubscriber {

    private long startTime;

    private @MonotonicNonNull Fight fight;

    private Listener @MonotonicNonNull[] listeners;

    private @MonotonicNonNull Map<FightTeam, PlacementCellsGenerator> cellsGenerators;

    private @MonotonicNonNull ScheduledFuture<?> timer;

    private final boolean randomize;

    public PlacementState() {

        this(true);

    }

    public PlacementState(boolean randomize) {

        this.randomize = randomize;

    }

    @Override

    public void start(Fight fight) {

        this.fight = fight;

        this.cellsGenerators = new HashMap<>();

        for (FightTeam team : fight.teams()) {

            cellsGenerators.put(

                team,

                randomize

                    ? PlacementCellsGenerator.randomized(fight.map(), team.startPlaces())

                    : new PlacementCellsGenerator(fight.map(), team.startPlaces())

            );

        }

        fight.dispatcher().register(this);

        startTime = System.currentTimeMillis();

        // Add all fighters to fight

        // Note: fight.fighters() cannot be used because at this state fighters are not yet on fight

        addFighters(fight.teams().stream().flatMap(team -> team.fighters().stream()).collect(Collectors.toList()));

        if (fight.type().hasPlacementTimeLimit()) {

            timer = fight.schedule(this::innerStartFight, fight.type().placementDuration());

        }

    }

    @Override

    public int id() {

        return 2;

    }

    @Override

    public Listener[] listeners() {

        if (listeners != null) {

            return listeners;

        }

        if (fight == null) {

            throw new IllegalStateException("State must be started");

        }

        return listeners = new Listener[] {

            new SendFighterPositions(fight),

            new SendFighterReadyState(fight),

            new StartFightWhenAllReady(fight, this),

            new SendNewFighter(fight),

            new ClearFighter(),

            new SendFighterRemoved(fight),

            new SendJoinTeamOptionChangedMessage(),

            new SendNeedHelpOptionChangedMessage(),

        };

    }

    /**

     * Get the remaining placement time, in milliseconds

     */

    public long remainingTime() {

        if (fight == null) {

            throw new IllegalStateException("State must be started");

        }

        if (!fight.type().hasPlacementTimeLimit()) {

            throw new UnsupportedOperationException("The fight has no placement time limit");

        }

        return fight.type().placementDuration().toMillis() + startTime - System.currentTimeMillis();

    }

    /**

     * Try to change fighter start place

     *

     * @param fighter Fighter to move

     * @param cell The target cell

     */

    public synchronized void changePlace(Fighter fighter, FightCell cell) {

        if (invalidState()) {

            return;

        }

        if (fighter.ready()) {

            throw new FightException("The fighter is ready");

        }

        if (!cell.walkable()) {

            throw new FightMapException("Not walkable");

        }

        if (!fighter.team().startPlaces().contains(cell)) {

            throw new FightException("Bad start cell");

        }

        fighter.move(cell);

        fight.dispatch(new FighterPlaceChanged(fighter));

    }

    /**

     * Try to join a team

     *

     * @param fighter The fighter to add

     * @param team The team

     *

     * @throws JoinFightException When cannot join the team

     */

    public synchronized void joinTeam(Fighter fighter, FightTeam team) throws JoinFightException {

        if (invalidState()) {

            throw new JoinFightException(JoinFightError.CANT_DO_TOO_LATE);

        }

        team.join(fighter);

        addFighters(Collections.singleton(fighter));

    }

    @Override

    public synchronized void leave(Fighter fighter) {

        if (invalidState()) {

            throw new InvalidFightStateException(getClass());

        }

        // Not allowed to leave the fight : punish the fighter

        if (!fight.type().canCancel()) {

            punishDeserter(fighter);

        }

        // Remove fighter

        leaveFromFight(fighter);

    }

    /**

     * Kick a fighter during placement

     * Unlike leave, this method will not punish the fighter

     *

     * @param fighter Fighter to kick

     *

     * @throws InvalidFightStateException When the session state has changed

     */

    public synchronized void kick(Fighter fighter) {

        if (invalidState()) {

            throw new InvalidFightStateException(getClass());

        }

        // Remove fighter

        leaveFromFight(fighter);

    }

    /**

     * Manually start the fight

     */

    /**

     * Start the fight

     */

    private void innerStartFight() {

        if (invalidState()) {

            return;

        }

        fight.dispatcher().unregister(this);

        fight.nextState();

    }

    /**

     * Leave from a fight :

     * - If the fighter is the team leader, the team is dissolved

     * - The fighter is removed

     * - Check if the fight is valid (has at least two teams)

     */

    @RequiresNonNull("fight")

    private void leaveFromFight(Fighter fighter) {

        // The team leader quit the fight => Dissolve team

        if (fighter.isTeamLeader()) {

            fight.teams().remove(fighter.team());

            fighter.team().fighters().forEach(this::removeFighter);

        } else {

            fighter.team().kick(fighter);

            removeFighter(fighter);

        }

        checkFightValid();

    }

    /**

     * Punish the deserter fighter

     */

    @RequiresNonNull("fight")

    private void punishDeserter(Fighter fighter) {

        final FightRewardsSheet rewardsSheet = fight.type().rewards().generate(

            new EndFightResults(

                fight,

                Collections.emptyList(),

                Collections.singletonList(fighter)

            )

        );

        fighter.dispatch(new FightLeaved(rewardsSheet.rewards().get(0)));

    }

    @RequiresNonNull("fight")

    @SuppressWarnings("dereference.of.nullable") // cellsGenerators.get(fighter.team()) cannot be null

    private void addFighters(Collection<Fighter> fighters) {

        for (Fighter fighter : fighters) {

            fighter.joinFight(fight, NullnessUtil.castNonNull(cellsGenerators).get(fighter.team()).next());

        }

        for (Fighter fighter : fighters) {

            fighter.dispatch(new FightJoined(fight, fighter));

        }

        for (Fighter fighter : fighters) {

            fight.dispatch(new FighterAdded(fighter));

        }

    }

    /**

     * Notify fight that a fighter has leave

     */

    @RequiresNonNull("fight")

    private void removeFighter(Fighter fighter) {

        fighter.dispatch(new FightLeaved());

        fight.dispatch(new FighterRemoved(fighter, fight));

    }

    /**

     * Check if the fight is valid after fighter leaved

     */

    @RequiresNonNull("fight")

    private void checkFightValid() {

        if (fight.teams().stream().filter(FightTeam::alive).count() > 1) {

            return;

        }

        fight.cancel();

        if (timer != null) {

            timer.cancel(true);

        }

    }

    /**

     * Check if the fight state is not placement

     * This method will return true is the state is active (or following), or if it's cancelled or finished

     */

    @EnsuresNonNullIf(expression = "fight", result = false)

    private boolean invalidState() {

        return fight == null || fight.state() != this || !fight.alive();

    }

}