Applied-Energistics-2-tilera-Edition/src/main/java/appeng/me/cache/SecurityCache.java

/*
 * This file is part of Applied Energistics 2.
 * Copyright (c) 2013 - 2015, 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.me.cache;


import java.util.ArrayList;
import java.util.EnumSet;
import java.util.HashMap;
import java.util.List;

import com.google.common.base.Preconditions;
import com.mojang.authlib.GameProfile;

import net.minecraft.entity.player.EntityPlayer;

import appeng.api.config.SecurityPermissions;
import appeng.api.networking.IGrid;
import appeng.api.networking.IGridHost;
import appeng.api.networking.IGridNode;
import appeng.api.networking.IGridStorage;
import appeng.api.networking.events.MENetworkEventSubscribe;
import appeng.api.networking.events.MENetworkSecurityChange;
import appeng.api.networking.security.ISecurityGrid;
import appeng.api.networking.security.ISecurityProvider;
import appeng.core.worlddata.WorldData;
import appeng.me.GridNode;


public class SecurityCache implements ISecurityGrid
{

	private final IGrid myGrid;
	private final List<ISecurityProvider> securityProvider = new ArrayList<ISecurityProvider>();
	private final HashMap<Integer, EnumSet<SecurityPermissions>> playerPerms = new HashMap<Integer, EnumSet<SecurityPermissions>>();
	private long securityKey = -1;

	public SecurityCache( final IGrid g )
	{
		this.myGrid = g;
	}

	@MENetworkEventSubscribe
	public void updatePermissions( final MENetworkSecurityChange ev )
	{
		this.playerPerms.clear();
		if( this.securityProvider.isEmpty() )
		{
			return;
		}

		this.securityProvider.get( 0 ).readPermissions( this.playerPerms );
	}

	public long getSecurityKey()
	{
		return this.securityKey;
	}

	@Override
	public void onUpdateTick()
	{

	}

	@Override
	public void removeNode( final IGridNode gridNode, final IGridHost machine )
	{
		if( machine instanceof ISecurityProvider )
		{
			this.securityProvider.remove( machine );
			this.updateSecurityKey();
		}
	}

	private void updateSecurityKey()
	{
		final long lastCode = this.securityKey;

		if( this.securityProvider.size() == 1 )
		{
			this.securityKey = this.securityProvider.get( 0 ).getSecurityKey();
		}
		else
		{
			this.securityKey = -1;
		}

		if( lastCode != this.securityKey )
		{
			this.getGrid().postEvent( new MENetworkSecurityChange() );
			for( final IGridNode n : this.getGrid().getNodes() )
			{
				( (GridNode) n ).setLastSecurityKey( this.securityKey );
			}
		}
	}

	@Override
	public void addNode( final IGridNode gridNode, final IGridHost machine )
	{
		if( machine instanceof ISecurityProvider )
		{
			this.securityProvider.add( (ISecurityProvider) machine );
			this.updateSecurityKey();
		}
		else
		{
			( (GridNode) gridNode ).setLastSecurityKey( this.securityKey );
		}
	}

	@Override
	public void onSplit( final IGridStorage destinationStorage )
	{

	}

	@Override
	public void onJoin( final IGridStorage sourceStorage )
	{

	}

	@Override
	public void populateGridStorage( final IGridStorage destinationStorage )
	{

	}

	@Override
	public boolean isAvailable()
	{
		return this.securityProvider.size() == 1 && this.securityProvider.get( 0 ).isSecurityEnabled();
	}

	@Override
	public boolean hasPermission( final EntityPlayer player, final SecurityPermissions perm )
	{
		Preconditions.checkNotNull( player );
		Preconditions.checkNotNull( perm );

		final GameProfile profile = player.getGameProfile();
		final int playerID = WorldData.instance().playerData().getPlayerID( profile );

		return this.hasPermission( playerID, perm );
	}

	@Override
	public boolean hasPermission( final int playerID, final SecurityPermissions perm )
	{
		if( this.isAvailable() )
		{
			final EnumSet<SecurityPermissions> perms = this.playerPerms.get( playerID );

			if( perms == null )
			{
				if( playerID == -1 ) // no default?
				{
					return false;
				}
				else
				{
					return this.hasPermission( -1, perm );
				}
			}

			return perms.contains( perm );
		}
		return true;
	}

	@Override
	public int getOwner()
	{
		if( this.isAvailable() )
		{
			return this.securityProvider.get( 0 ).getOwner();
		}
		return -1;
	}

	public IGrid getGrid()
	{
		return this.myGrid;
	}
}