Applied-Energistics-2/src/main/java/appeng/items/tools/ToolBiometricCard.java

/*
 * This file is part of Applied Energistics 2.
 * Copyright (c) 2013 - 2014, AlgorithmX2, All rights reserved.
 *
 * Applied Energistics 2 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.
 *
 * Applied Energistics 2 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 Applied Energistics 2.  If not, see <http://www.gnu.org/licenses/lgpl>.
 */

package appeng.items.tools;


import java.util.EnumSet;
import java.util.List;

import com.mojang.authlib.GameProfile;

import net.minecraft.entity.EntityLivingBase;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTUtil;
import net.minecraft.util.ActionResult;
import net.minecraft.util.EnumActionResult;
import net.minecraft.util.EnumHand;
import net.minecraft.world.World;

import appeng.api.config.SecurityPermissions;
import appeng.api.features.IPlayerRegistry;
import appeng.api.implementations.items.IBiometricCard;
import appeng.api.networking.security.ISecurityRegistry;
import appeng.core.localization.GuiText;
import appeng.items.AEBaseItem;
import appeng.util.Platform;


public class ToolBiometricCard extends AEBaseItem implements IBiometricCard
{
	public ToolBiometricCard()
	{
		this.setMaxStackSize( 1 );
	}

	@Override
	public ActionResult<ItemStack> onItemRightClick( final ItemStack is, final World w, final EntityPlayer p, final EnumHand hand )
	{
		if( p.isSneaking() )
		{
			this.encode( is, p );
			p.swingArm( hand );
			return new ActionResult<ItemStack>( EnumActionResult.SUCCESS, is );
		}

		return new ActionResult<ItemStack>( EnumActionResult.PASS, is );
	}
	
	@Override
	public boolean itemInteractionForEntity( ItemStack is, final EntityPlayer player, final EntityLivingBase target, final EnumHand hand )
	{
		if( target instanceof EntityPlayer && !player.isSneaking() )
		{
			if( player.capabilities.isCreativeMode )
			{
				is = player.getHeldItem( hand );
			}
			this.encode( is, (EntityPlayer) target );
			player.swingArm( hand );
			return true;
		}
		return false;
	}

	@Override
	public String getItemStackDisplayName( final ItemStack is )
	{
		final GameProfile username = this.getProfile( is );
		return username != null ? super.getItemStackDisplayName( is ) + " - " + username.getName() : super.getItemStackDisplayName( is );
	}

	private void encode( final ItemStack is, final EntityPlayer p )
	{
		final GameProfile username = this.getProfile( is );

		if( username != null && username.equals( p.getGameProfile() ) )
		{
			this.setProfile( is, null );
		}
		else
		{
			this.setProfile( is, p.getGameProfile() );
		}
	}

	@Override
	public void setProfile( final ItemStack itemStack, final GameProfile profile )
	{
		final NBTTagCompound tag = Platform.openNbtData( itemStack );

		if( profile != null )
		{
			final NBTTagCompound pNBT = new NBTTagCompound();
			NBTUtil.writeGameProfile( pNBT, profile );
			tag.setTag( "profile", pNBT );
		}
		else
		{
			tag.removeTag( "profile" );
		}
	}

	@Override
	public GameProfile getProfile( final ItemStack is )
	{
		final NBTTagCompound tag = Platform.openNbtData( is );
		if( tag.hasKey( "profile" ) )
		{
			return NBTUtil.readGameProfileFromNBT( tag.getCompoundTag( "profile" ) );
		}
		return null;
	}

	@Override
	public EnumSet<SecurityPermissions> getPermissions( final ItemStack is )
	{
		final NBTTagCompound tag = Platform.openNbtData( is );
		final EnumSet<SecurityPermissions> result = EnumSet.noneOf( SecurityPermissions.class );

		for( final SecurityPermissions sp : SecurityPermissions.values() )
		{
			if( tag.getBoolean( sp.name() ) )
			{
				result.add( sp );
			}
		}

		return result;
	}

	@Override
	public boolean hasPermission( final ItemStack is, final SecurityPermissions permission )
	{
		final NBTTagCompound tag = Platform.openNbtData( is );
		return tag.getBoolean( permission.name() );
	}

	@Override
	public void removePermission( final ItemStack itemStack, final SecurityPermissions permission )
	{
		final NBTTagCompound tag = Platform.openNbtData( itemStack );
		if( tag.hasKey( permission.name() ) )
		{
			tag.removeTag( permission.name() );
		}
	}

	@Override
	public void addPermission( final ItemStack itemStack, final SecurityPermissions permission )
	{
		final NBTTagCompound tag = Platform.openNbtData( itemStack );
		tag.setBoolean( permission.name(), true );
	}

	@Override
	public void registerPermissions( final ISecurityRegistry register, final IPlayerRegistry pr, final ItemStack is )
	{
		register.addPlayer( pr.getID( this.getProfile( is ) ), this.getPermissions( is ) );
	}

	@Override
	public void addCheckedInformation( final ItemStack stack, final EntityPlayer player, final List<String> lines, final boolean displayMoreInfo )
	{
		final EnumSet<SecurityPermissions> perms = this.getPermissions( stack );
		if( perms.isEmpty() )
		{
			lines.add( GuiText.NoPermissions.getLocal() );
		}
		else
		{
			String msg = null;

			for( final SecurityPermissions sp : perms )
			{
				if( msg == null )
				{
					msg = Platform.gui_localize( sp.getUnlocalizedName() );
				}
				else
				{
					msg = msg + ", " + Platform.gui_localize( sp.getUnlocalizedName() );
				}
			}
			lines.add( msg );
		}
	}
}