Server/src/engine/net/client/handlers/UpgradeAssetMsgHandler.java

package engine.net.client.handlers;

import engine.Enum;
import engine.Enum.GameObjectType;
import engine.exception.MsgSendException;
import engine.gameManager.BuildingManager;
import engine.gameManager.DbManager;
import engine.gameManager.SessionManager;
import engine.net.Dispatch;
import engine.net.DispatchMessage;
import engine.net.client.ClientConnection;
import engine.net.client.msg.*;
import engine.objects.Building;
import engine.objects.PlayerCharacter;
import org.pmw.tinylog.Logger;

import java.time.LocalDateTime;

import static engine.net.client.msg.ErrorPopupMsg.sendErrorPopup;

/*
 *
 * @Summary: Processes application protocol message where a
 * client requests that a building be upgraded.
 */
public class UpgradeAssetMsgHandler extends AbstractClientMsgHandler {

	// Constructor
	public UpgradeAssetMsgHandler() {

		super(UpgradeAssetMessage.class);
	}

	@Override
	protected boolean _handleNetMsg(ClientNetMsg baseMsg, ClientConnection origin) throws MsgSendException {

		// Member variable declaration

		UpgradeAssetMessage msg;
		ManageCityAssetsMsg outMsg;
		PlayerCharacter player;
		int buildingUUID;
		Building buildingToRank;
		LocalDateTime dateToUpgrade;
		int nextRank;
		int rankCost;
		Dispatch dispatch;

		// Assign member variables

		msg = (UpgradeAssetMessage) baseMsg;

		// Grab pointer to the requesting player

		player = SessionManager.getPlayerCharacter(origin);

		// Grab pointer to the building from the cache

		buildingUUID = msg.getBuildingUUID();

		buildingToRank = (Building) DbManager.getObject(GameObjectType.Building, buildingUUID);

		// Early exit if building not in cache.

		if (buildingToRank == null) {
			Logger.error("Attempt to upgrade null building by " + player.getName());
			return true;
		}

		// Early exit for building that is already ranking

		if (buildingToRank.isRanking()) {
			Logger.error("Attempt to upgrade a building already ranking by " + player.getName());
			return true;
		}

		// Calculate and set time/cost to upgrade

		nextRank = (buildingToRank.getRank() + 1);

		if (buildingToRank.getBlueprint() == null)
			return true;
		if (buildingToRank.getBlueprint().getMaxRank() < nextRank || nextRank == 8){
			ErrorPopupMsg.sendErrorMsg(player, "Building is already at it's Max rank.");
			return true;
		}

		rankCost = buildingToRank.getBlueprint().getRankCost(nextRank);

		// SEND NOT ENOUGH GOLD ERROR

		if (!buildingToRank.hasFunds(rankCost)){
			ErrorPopupMsg.sendErrorPopup(player, 127); // Not enough gold in strongbox
			return true;
		}

		if (rankCost > buildingToRank.getStrongboxValue()) {
			sendErrorPopup(player, 127);
			return true;
		}

		// Validation appears good.  Let's now process the upgrade
		
		try {
			if (buildingToRank.getCity() != null){
				buildingToRank.getCity().transactionLock.writeLock().lock();
				try{
					if (!buildingToRank.transferGold(-rankCost,false)) {
						sendErrorPopup(player, 127);
						return true;
					}
				}catch(Exception e){
					Logger.error(e);
				}finally{
					buildingToRank.getCity().transactionLock.writeLock().unlock();
				}
			}else
			if (!buildingToRank.transferGold(-rankCost,false)) {
				sendErrorPopup(player, 127);
				return true;
			}

			dateToUpgrade = LocalDateTime.now().plusHours(buildingToRank.getBlueprint().getRankTime(nextRank));

			BuildingManager.setUpgradeDateTime(buildingToRank, dateToUpgrade, 0);

			// Schedule upgrade job

			BuildingManager.submitUpgradeJob(buildingToRank);

			// Refresh the client's manage asset window
			// *** Refactor : We have some of these unknowns

			outMsg = new ManageCityAssetsMsg(player, buildingToRank);

			// Action TYPE
			outMsg.actionType = 3;
			outMsg.setTargetType(buildingToRank.getObjectType().ordinal());
			outMsg.setTargetID(buildingToRank.getObjectUUID());
			outMsg.setTargetType3(buildingToRank.getObjectType().ordinal());
			outMsg.setTargetID3(buildingToRank.getObjectUUID());
			outMsg.setAssetName1(buildingToRank.getName());
			outMsg.setUnknown54(1);

			dispatch = Dispatch.borrow(player, outMsg);
			DispatchMessage.dispatchMsgDispatch(dispatch, Enum.DispatchChannel.SECONDARY);

		} catch (Exception e) {
			PlaceAssetMsg.sendPlaceAssetError(player.getClientConnection(), 1, "A Serious error has occurred. Please post details for to ensure transaction integrity");
		}

		return true;
	}
}