package chat.server.netty;

import chat.server.Configer;
import chat.server.im.IMDispatcher;
import chat.server.moquette.BytesMetricsCollector;
import chat.server.moquette.MessageMetricsCollector;
import chat.server.moquette.MqttBytesMetricsHandler;
import chat.server.moquette.MqttDecodHandler;
import chat.server.moquette.MqttEncodHandler;
import chat.server.moquette.MqttIdleTimeoutHandler;
import chat.server.moquette.MqttMessageLoggerHandler;
import chat.server.moquette.MqttMessageMetricsHandler;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.socket.SocketChannel;
import io.netty.handler.timeout.IdleStateHandler;

public class NettySSLChannelInitializer extends ChannelInitializer<SocketChannel> {

    private MessageMetricsCollector messageMetricsCollector;
    private BytesMetricsCollector bytesMetricsCollector;
	
	
	@Override
	protected void initChannel(SocketChannel channel) throws Exception {
		int timeoutSeconds = Configer.getInteger("timeout", 10);
		
		ChannelPipeline pipeline = channel.pipeline();
		
		ChannelHandler bytesMetricsHandler = new MqttBytesMetricsHandler(bytesMetricsCollector);
		ChannelHandler idleStateHandler = new IdleStateHandler(timeoutSeconds, 0, 0);
		ChannelHandler timeoutHandler = new MqttIdleTimeoutHandler();
//		ChannelHandler sslHandler = new SSLHandler();
		ChannelHandler decoder = new MqttDecodHandler();
		ChannelHandler encoder = MqttEncodHandler.getInstance();
		ChannelHandler messageMetricsHandler = new MqttMessageMetricsHandler(messageMetricsCollector);
		ChannelHandler messageLogger = new MqttMessageLoggerHandler();
		ChannelHandler dispatcher = new IMDispatcher();
		
		pipeline.addFirst("bytemetrics", bytesMetricsHandler);
		pipeline.addLast("idleStateHandler", idleStateHandler);
		pipeline.addAfter("idleStateHandler", "idleEventHandler", timeoutHandler);
//		pipeline.addLast("ssl", sslHandler);
		pipeline.addLast("decoder", decoder);
		pipeline.addLast("encoder", encoder);
		pipeline.addLast("metrics", messageMetricsHandler);
		pipeline.addLast("messageLogger", messageLogger);
		pipeline.addLast("handler", dispatcher);
	}
	
	public void setMessageMetricsCollector(MessageMetricsCollector messageMetricsCollector) {
		this.messageMetricsCollector = messageMetricsCollector;
	}

	public void setBytesMetricsCollector(BytesMetricsCollector bytesMetricsCollector) {
		this.bytesMetricsCollector = bytesMetricsCollector;
	}
}