package com.fzzy.protocol.youxian0.client;

import com.fzzy.api.utils.BytesUtil;
import com.fzzy.protocol.ProtocolUtils;
import com.ld.io.api.InvokeResult;
import io.netty.bootstrap.Bootstrap;
import io.netty.buffer.Unpooled;
import io.netty.channel.*;
import io.netty.channel.oio.OioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.oio.OioSocketChannel;
import io.netty.handler.codec.bytes.ByteArrayDecoder;
import io.netty.handler.codec.bytes.ByteArrayEncoder;
import lombok.extern.slf4j.Slf4j;

import java.util.HashMap;
import java.util.Map;

/**
 * 游仙主库，通讯协议
 */
@Slf4j
public class ClientEngine implements Runnable {

    /**
     * 客户端模式创建的连接通道
     */
    public static Map<String, Channel> clientChannelMap = new HashMap<>();


    private String host;
    private int port;
    public Channel defaultChannel;


    public ClientEngine(String host, int port) {
        this.host = host;
        this.port = port;
    }

    public void start() {
        Thread thread = new Thread(this);
        thread.start();
    }

    @Override
    public void run() {
        try {
            startClient();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public void startClient() throws Exception {
        EventLoopGroup group = new OioEventLoopGroup();
        Bootstrap b = new Bootstrap();
        //默认长连接
        b.option(ChannelOption.SO_KEEPALIVE, true);
        b.group(group).channel(OioSocketChannel.class)
                .handler(new ChannelInitializer<SocketChannel>() {
                    @Override
                    protected void initChannel(SocketChannel ch)
                            throws Exception {
                        ChannelPipeline p = ch.pipeline();
                        // 字符串解码 和 编码
                        p.addLast("decoder", new ByteArrayDecoder());
                        p.addLast("encoder", new ByteArrayEncoder());
                        // 自己的逻辑Handler
                        p.addLast("handler", new ClientHandler());
                    }
                });

        // 发起异步连接请求，绑定连接端口和host信息
        ChannelFuture channelFuture = b.connect(host, port);
        this.defaultChannel = channelFuture.channel();
        // channelFuture.channel().closeFuture().sync();

        this.add2ChannelMap(host, defaultChannel);

        channelFuture.addListener(new ChannelFutureListener() {
            @Override
            public void operationComplete(ChannelFuture arg0) throws Exception {
                if (channelFuture.isSuccess()) {
                    log.info("-----IP={},连接成功");
                } else {
                    log.info("-----IP={},连接失败，自动关闭线程");
                    channelFuture.cause().printStackTrace();
                    group.shutdownGracefully(); // 关闭线程组
                }
            }
        });
    }

    public InvokeResult send(byte[] array) throws InterruptedException {

        if (null == defaultChannel) {
            return InvokeResult.SOCKET_NOT_CREATE;
        }
        if (!defaultChannel.isActive()) {
            return InvokeResult.CHANNEL_CLOSED;
        }

        defaultChannel.writeAndFlush(Unpooled.copiedBuffer(array)).sync();

        return InvokeResult.SUCCESS;
    }


    public Channel getChannel() {
        return defaultChannel;
    }

    public static void add2ChannelMap(String key, Channel channel) {
        clientChannelMap.put(key, channel);
    }

    /**
     * 如果
     *
     * @param key
     * @return
     */
    public static  Channel getChannel(String key) {
        Channel channel = clientChannelMap.get(key);
        if (null == channel) return null;
        if (channel.isActive()) {
            return channel;
        } else {
            channel.close();
        }
        return null;
    }

    public static InvokeResult send2(String hex,Channel channel) throws InterruptedException {
        if (null == channel) {
            return InvokeResult.SOCKET_NOT_CREATE;
        }
        if (!channel.isActive()) {
            return InvokeResult.CHANNEL_CLOSED;
        }
        channel.writeAndFlush(Unpooled.copiedBuffer(BytesUtil.hexStrToBytes(hex))).sync();
        return InvokeResult.SUCCESS;
    }


}