content: /*
   * Copyright (c) 2019-2020 5zig Reborn
   * Copyright (c) 2015-2019 5zig
   *
   * This file is part of The 5zig Mod
   * The 5zig Mod is free software: you can redistribute it and/or modify
   * it under the terms of the GNU General Public License as published by
   * the Free Software Foundation, either version 3 of the License, or
   * (at your option) any later version.
   *
   * The 5zig Mod 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 General Public License for more details.
   *
   * You should have received a copy of the GNU General Public License
   * along with The 5zig Mod.  If not, see <http://www.gnu.org/licenses/>.
   */
  
  package eu.the5zig.mod.chat.network;
  
  import eu.the5zig.mod.The5zigMod;
  import eu.the5zig.mod.chat.network.packets.Packet;
  import eu.the5zig.mod.chat.network.packets.PacketBuffer;
  import io.netty.buffer.ByteBuf;
  import io.netty.buffer.Unpooled;
  import io.netty.channel.ChannelHandlerContext;
  import io.netty.handler.codec.MessageToByteEncoder;
  
  /**
   * Created by 5zig.
   * All rights reserved © 2015
   */
  public class NetworkEncoder extends MessageToByteEncoder<Packet> {
  
  	private NetworkManager networkManager;
  	private NettyEncryptionTranslator enc;
  
  	public NetworkEncoder(NetworkManager networkManager) {
  		this.networkManager = networkManager;
  	}
  
  	@Override
  	protected void encode(ChannelHandlerContext channelHandlerContext, Packet packet, ByteBuf byteBuf) throws Exception {
  		if(enc != null) {
  			ByteBuf beforeEncryption = null;
  			try {
  				beforeEncryption = Unpooled.buffer();
  				PacketBuffer.writeVarIntToBuffer(beforeEncryption, networkManager.getProtocol().getPacketId(packet));
  				packet.write(beforeEncryption);
  
  				enc.cipher(beforeEncryption, byteBuf);
  			}
  			finally {
  				if(beforeEncryption != null)
  					beforeEncryption.release();
  			}
  		}
  		else {
  			PacketBuffer.writeVarIntToBuffer(byteBuf, networkManager.getProtocol().getPacketId(packet));
  			packet.write(byteBuf);
  		}
  		The5zigMod.getDataManager().getNetworkStats().onPacketSend(byteBuf);
  		The5zigMod.logger.debug(The5zigMod.networkMarker, "OUT| {} ({} bytes)", packet.toString(), byteBuf.readableBytes());
  	}
  
  	public void setEnc(NettyEncryptionTranslator enc) {
  		this.enc = enc;
  	}
  }
  