/// <summary>
/// 网络服务器抽象类
/// </summary>
public class NetServer : SingletonTemplate<NetServer>
{
    /// <summary>
    /// 服务器是否启动
    /// </summary>
    public bool IsServerStart
    {
        get;
        private set;
    }

    /// <summary>
    /// 网络连接器
    /// </summary>
    private static NetConnector mNetConnector;

    /// <summary>
    /// 限定长度的客户端消息接受二进制数组
    /// </summary>
    private static byte[] ClientMsg = new byte[1024 * 1024];

    public NetServer()
    {
        mNetConnector = new NetConnector();
        IsServerStart = false;
    }

    public NetServer(ProtocolType protocoltype = ProtocolType.Tcp)
    {
        mNetConnector = new NetConnector(protocoltype);
        IsServerStart = false;
    }

    /// <summary>
    /// 启动服务器端口监听
    /// </summary>
    public void StartServer()
    {
        if(IsServerStart == false)
        {
            // 启动服务器端口监听以及等待客户端连接线程
            IsServerStart = true;
            mNetConnector.ListenTo("192.168.1.3", 8888);
            Thread clientconnectthread = new Thread(ClientConnectListener);
            clientconnectthread.Start();                
        }
        else
        {
            Console.WriteLine("服务器端已启动!");
        }
    }

    /// <summary>
    /// 发送消息到客户端
    /// </summary>
    /// <param name="msg"></param>
    public void SendMessage(NetMessage msg)
    {
        //clientsocket.Send(WriteMessage(msg));
    }

    /// <summary>
    /// 客户端连接请求监听线程
    /// </summary>
    private void ClientConnectListener()
    {
        while(true)
        {
            // 检查是否有客户端连接(阻塞的)
            Socket clientsocket = mNetConnector.NetSocket.Accept();
            Console.WriteLine("客户端{0}成功连接", clientsocket.RemoteEndPoint.ToString());
            // 想客户端发送连接成功数据(这里的二进制数据可以换成Protobuf序列化的)
            NetMessage netmsg = new NetMessage(1, Encoding.UTF8.GetBytes("连接成功!"));
            clientsocket.Send(WriteMessage(netmsg));
            // 为连接的客户端创建接受消息的线程
            Thread resmsgthread = new Thread(ReceiveMessage);
            resmsgthread.Start(clientsocket);
        }
    }

    /// <summary>
    /// 写入网络传输所需数据
    /// 1. 数据长度(short - 2字节) -- 用于存储消息数据长度
    /// 2. 消息ID(short - 2字节) -- 用于区分消息类型
    /// 3. 二进制数据(Protobuf) -- 用于序列化和反序列化数据
    /// </summary>
    /// <param name="netmsg"></param>
    private byte[] WriteMessage(NetMessage netmsg)
    {
        ByteBufferWriter bbw = new ByteBufferWriter();
        bbw.WriteShort(netmsg.MessageLength);
        bbw.WriteShort(netmsg.MessageID);
        bbw.WriteBytes(netmsg.MessageData);
        bbw.Flush();
        var bytesdata = bbw.ToBytes();
        bbw.Close();
        return bytesdata;
    }

    /// <summary>
    /// 接受客户端消息
    /// </summary>
    /// <param name="clientsocket"></param>
    private void ReceiveMessage(object clientsocket)
    {
        Socket cltsocket = (Socket)clientsocket;
        while(true)
        {
            try
            {
                int receivenumber = cltsocket.Receive(ClientMsg);
                if(receivenumber == 0)
                {
                    Console.WriteLine("客户端 : {0}断开了连接!", cltsocket.RemoteEndPoint.ToString());
                    cltsocket.Shutdown(SocketShutdown.Both);
                    cltsocket.Close();
                    break;
                }
                else
                {
                    Console.WriteLine(string.Format("接收客户端{0}消息， 长度为{1}", cltsocket.RemoteEndPoint.ToString(), receivenumber));
                    ByteBufferReader bbr = new ByteBufferReader(ClientMsg);
                    // 消息数据长度
                    int msglen = bbr.ReadShort();
                    // 消息id
                    int msgid = bbr.ReadShort();
                    // 数据二进制内容
                    byte[] msgdata = bbr.ReadBytes(msglen);
                    var msgcontent = Encoding.UTF8.GetString(msgdata);
                    Console.WriteLine(string.Format("消息数据长度 : {0} 消息ID : {1} 消息数据内容：{2}", msglen, msgid, msgcontent));
                }
            }
            catch (Exception ex)
            {
                Console.WriteLine(ex.Message);
                cltsocket.Shutdown(SocketShutdown.Both);
                cltsocket.Close();
                break;
            }
        }
    }
}

/// <summary>
/// NetClient.cs
/// 网络客户端抽象
/// </summary>
public class NetClient : SingletonTemplate<NetClient>
{
    /// <summary>
    /// 服务器端的二进制数据缓存区
    /// </summary>
    private static byte[] ServerMsg = new byte[1024 * 1024];

    /// <summary>
    /// 客户端网络连接器
    /// </summary>
    private NetConnector mNetConnector;

    /// <summary>
    /// 是否已连接服务器
    /// </summary>
    public bool IsServerConnected
    {
        get
        {
            return mNetConnector.IsConnected;
        }
    }

    /// <summary>
    /// 消息接受线程
    /// </summary>
    private Thread mMsgReceiveThread;

    public NetClient()
    {
        mNetConnector = new NetConnector();
        mMsgReceiveThread = null;
    }

    public NetClient(ProtocolType protocoltype = ProtocolType.Tcp)
    {
        mNetConnector = new NetConnector(protocoltype);
        mMsgReceiveThread = null;
    }

    /// <summary>
    /// 连接服务器
    /// </summary>
    /// <param name="ip">ip地址</param>
    /// <param name="port">端口号</param>
    public void ConnectToServer(string ipaddress, int port)
    {
        if(IsServerConnected)
        {
            mNetConnector.Close();
        }
        try
        {
            // 连接服务器
            mNetConnector.ConnectTo(ipaddress, port);
            Debug.Log(string.Format("连接服务器:{0}成功!", mNetConnector.IPEndPoint.ToString()));
        }
        catch
        {
            Debug.Log(string.Format("连接服务器:{0}失败!", mNetConnector.IPEndPoint.ToString()));
        }

        if (IsServerConnected)
        {
            // 启动客户端线程去监听接受服务器消息
            mMsgReceiveThread = new Thread(ReceiveMessage);
            mMsgReceiveThread.Start(mNetConnector.NetSocket);
        }
    }

    /// <summary>
    /// 断开连接
    /// </summary>
    public void Close()
    {
        if (IsServerConnected)
        {
            Debug.Log(string.Format("断开服务器地址 : {0}连接!", mNetConnector.NetSocket.RemoteEndPoint.ToString()));
            mNetConnector.Close();
        }
    }

    /// <summary>
    /// 发送消息到服务器
    /// </summary>
    /// <param name="msg"></param>
    public void SendMessage(NetMessage msg)
    {
        if (IsServerConnected == false)
        {
            Debug.Log("服务器未连接成功，无法发送数据!");
            return;
        }
        else
        {
            try
            {
                mNetConnector.NetSocket.Send(WriteMessage(msg));
            }
            catch
            {
                mNetConnector.Close();
            }
        }
    }


    /// <summary>
    /// 写入网络传输所需数据
    /// 1. 数据长度(short - 2字节) -- 用于存储消息数据长度
    /// 2. 消息ID(short - 2字节) -- 用于区分消息类型
    /// 3. 二进制数据(Protobuf) -- 用于序列化和反序列化数据
    /// </summary>
    /// <param name="netmsg"></param>
    private byte[] WriteMessage(NetMessage netmsg)
    {
        ByteBufferWriter bbw = new ByteBufferWriter();
        bbw.WriteShort(netmsg.MessageLength);
        bbw.WriteShort(netmsg.MessageID);
        bbw.WriteBytes(netmsg.MessageData);
        bbw.Flush();
        var bytesdata = bbw.ToBytes();
        bbw.Close();
        return bytesdata;
    }

    /// <summary>
    /// 服务器连接消息接受监听线程
    /// </summary>
    private void ReceiveMessage(object clientsocket)
    {
        Socket cltsocket = (Socket)clientsocket;
        while (true)
        {
            try
            {
                if(mNetConnector.IsConnected)
                {
                    if(mNetConnector.NetSocket.Available > 0)
                    {
                        int receivenumber = cltsocket.Receive(ServerMsg);
                        if (receivenumber == 0)
                        {
                            Debug.Log(string.Format("服务器端 : {0}断开了连接!", cltsocket.RemoteEndPoint.ToString()));
                            break;
                        }
                        else
                        {
                            ByteBufferReader bbr = new ByteBufferReader(ServerMsg);
                            var msglength = bbr.ReadShort();
                            var msgid = bbr.ReadShort();
                            var msgdata = bbr.ReadBytes(msglength);
                            string msgcontent = Encoding.UTF8.GetString(msgdata);
                            Debug.Log(string.Format("服务器返回数据: 消息ID : {0} 消息长度 : {1} 消息内容 : {2}", msgid, msglength, msgcontent));
                        }
                    }
                }
                else
                {
                    break;
                }                    
            }
            catch (Exception ex)
            {
                Debug.Log(ex.Message);
                mNetConnector.NetSocket.Close();
                break;
            }
        }
    }
}