ForestryMC/src/main/java/forestry/mail/tiles/TileMailbox.java

/*******************************************************************************
 * Copyright (c) 2011-2014 SirSengir. All rights reserved. This program and the accompanying materials are made
 * available under the terms of the GNU Lesser Public License v3 which accompanies this distribution, and is available
 * at http://www.gnu.org/licenses/lgpl-3.0.txt
 *
 * Various Contributors including, but not limited to: SirSengir (original work), CovertJaguar, Player, Binnie,
 * MysteriousAges
 ******************************************************************************/
package forestry.mail.tiles;

import java.util.Collection;
import java.util.LinkedList;

import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.inventory.IInventory;
import net.minecraft.item.ItemStack;
import net.minecraft.tileentity.TileEntity;
import net.minecraft.util.ChatComponentTranslation;
import net.minecraft.world.World;
import net.minecraftforge.common.util.ForgeDirection;

import com.mojang.authlib.GameProfile;

import buildcraft.api.statements.ITriggerExternal;
import cpw.mods.fml.common.Optional;
import forestry.api.mail.ILetter;
import forestry.api.mail.IMailAddress;
import forestry.api.mail.IPostalState;
import forestry.api.mail.PostManager;
import forestry.core.inventory.InventoryAdapter;
import forestry.core.tiles.TileBase;
import forestry.mail.EnumDeliveryState;
import forestry.mail.POBox;
import forestry.mail.PostRegistry;
import forestry.mail.gui.ContainerMailbox;
import forestry.mail.gui.GuiMailbox;
import forestry.mail.triggers.MailTriggers;

public class TileMailbox extends TileBase implements IMailContainer {

    private boolean isLinked = false;

    public TileMailbox() {
        super("mailbox");
        setInternalInventory(new InventoryAdapter(POBox.SLOT_SIZE, "Letters").disableAutomation());
    }

    /* GUI */
    @Override
    public void openGui(EntityPlayer player) {
        if (worldObj.isRemote) {
            return;
        }

        ItemStack held = player.getCurrentEquippedItem();

        // Handle letter sending
        if (PostManager.postRegistry.isLetter(held)) {
            IPostalState result = this.tryDispatchLetter(held);
            if (!result.isOk()) {
                player.addChatMessage(new ChatComponentTranslation("for.chat.mail." + result.getIdentifier()));
            } else {
                held.stackSize--;
            }
        } else {
            super.openGui(player);
        }
    }

    /* UPDATING */
    @Override
    public void updateServerSide() {
        if (!isLinked) {
            getOrCreateMailInventory(worldObj, getAccessHandler().getOwner());
            isLinked = true;
        }
    }

    /* MAIL HANDLING */
    public IInventory getOrCreateMailInventory(World world, GameProfile playerProfile) {
        if (world.isRemote) {
            return getInternalInventory();
        }

        IMailAddress address = PostManager.postRegistry.getMailAddress(playerProfile);
        return PostRegistry.getOrCreatePOBox(worldObj, address);
    }

    private IPostalState tryDispatchLetter(ItemStack letterstack) {
        ILetter letter = PostManager.postRegistry.getLetter(letterstack);
        IPostalState result;

        if (letter != null) {
            result = PostManager.postRegistry.getPostOffice(worldObj).lodgeLetter(worldObj, letterstack, true);
        } else {
            result = EnumDeliveryState.NOT_MAILABLE;
        }

        return result;
    }

    /* IMAILCONTAINER */
    @Override
    public boolean hasMail() {

        IInventory mailInventory = getOrCreateMailInventory(worldObj, getAccessHandler().getOwner());
        for (int i = 0; i < mailInventory.getSizeInventory(); i++) {
            if (mailInventory.getStackInSlot(i) != null) {
                return true;
            }
        }

        return false;
    }

    /* ITRIGGERPROVIDER */
    @Optional.Method(modid = "BuildCraftAPI|statements")
    @Override
    public Collection<ITriggerExternal> getExternalTriggers(ForgeDirection side, TileEntity tile) {
        LinkedList<ITriggerExternal> res = new LinkedList<>();
        res.add(MailTriggers.triggerHasMail);
        return res;
    }

    @Override
    public Object getGui(EntityPlayer player, int data) {
        return new GuiMailbox(player.inventory, this);
    }

    @Override
    public Object getContainer(EntityPlayer player, int data) {
        return new ContainerMailbox(player.inventory, this);
    }
}