DanMuRead/BarrageGrab/Server/WsBarrageServer.cs

using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Drawing;
using System.Dynamic;
using System.IO;
using System.IO.Compression;
using System.Linq;
using System.Net;
using System.Net.Http;
using System.Net.WebSockets;
using System.Text;
using System.Threading.Tasks;
using System.Timers;
using BarrageGrab.Modles;
using BarrageGrab.Modles.JsonEntity;
using BarrageGrab.Modles.ProtoEntity;
using BarrageGrab.Proxy.ProxyEventArgs;
using Fleck;
using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
using static BarrageGrab.Modles.ProtoEntity.Image;
using static BarrageGrab.Modles.WebCastGift;

namespace BarrageGrab
{
    /// <summary>
    /// WsBarrageServer 包装消息事件委托
    /// </summary>
    public delegate void PackMessageEventHandler(WsBarrageServer sender, WsBarrageServer.PackMsgEventArgs e);

    /// <summary>
    /// 弹幕服务
    /// </summary>
    public class WsBarrageServer : IDisposable
    {
        WebSocketServer socketServer; //Ws服务器对象
        Dictionary<string, UserState> socketList = new Dictionary<string, UserState>(); //客户端列表
        ConcurrentDictionary<string, Tuple<int, DateTime>> giftCountCache = new ConcurrentDictionary<string, Tuple<int, DateTime>>();//礼物计数缓存
        Timer dieout = new Timer(10000);//离线客户端清理计时器
        Timer giftCountTimer = new Timer(10000);//礼物缓存清理计时器
        WssBarrageGrab grab = new WssBarrageGrab();//抖音弹幕解析器核心
        KsBarrageGrab ksGrab = new KsBarrageGrab();//快手弹幕解析器核心
        BiliBarrageGrab biliGrab = new BiliBarrageGrab();//B站弹幕解析器核心
        AppSetting Appsetting = AppSetting.Current;//全局配置文件实例
        static int printCount = 0; //控制台输出计数，用于判断清理控制台

        /// <summary>
        /// WS服务器启动地址
        /// </summary>
        public string ServerLocation => socketServer.Location;

        /// <summary>
        /// 数据内核
        /// </summary>
        public WssBarrageGrab Grab => grab;

        /// <summary>
        /// 控制台打印事件
        /// </summary>
        public event EventHandler<PrintEventArgs> OnPrint;

        /// <summary>
        /// 是否已经释放资源
        /// </summary>
        public bool IsDisposed { get; private set; } = false;

        /// <summary>
        /// 服务关闭后触发
        /// </summary>
        public event EventHandler OnClose;

        /// <summary>
        /// 消息包装完成后触发
        /// </summary>
        public event PackMessageEventHandler OnPackMessage;

        public WsBarrageServer()
        {
            var socket = new WebSocketServer($"ws://0.0.0.0:{Appsetting.WsProt}");
            socket.RestartAfterListenError = true;//异常重启

            dieout.Elapsed += Dieout_Elapsed;
            giftCountTimer.Elapsed += GiftCountTimer_Elapsed;

            this.grab.OnChatMessage += Grab_OnChatMessage;
            this.grab.OnLikeMessage += Grab_OnLikeMessage;
            this.grab.OnMemberMessage += Grab_OnMemberMessage;
            this.grab.OnSocialMessage += Grab_OnSocialMessage;
            this.grab.OnSocialMessage += Grab_OnShardMessage;
            this.grab.OnGiftMessage += Grab_OnGiftMessage;
            this.grab.OnRoomUserSeqMessage += Grab_OnRoomUserSeqMessage;
            this.grab.OnFansclubMessage += Grab_OnFansclubMessage; ;
            this.grab.OnControlMessage += Grab_OnControlMessage;
            this.grab.Proxy.OnLiveCompanAction += Proxy_OnLiveCompanAction;

            //快手事件订阅
            this.ksGrab.OnChatMessage += KsGrab_OnChatMessage;
            this.ksGrab.OnLikeMessage += KsGrab_OnLikeMessage;
            this.ksGrab.OnGiftMessage += KsGrab_OnGiftMessage;
            this.ksGrab.OnShareMessage += KsGrab_OnShareMessage;
            this.ksGrab.OnRoomStatsMessage += KsGrab_OnRoomStatsMessage;

            //B站事件订阅
            this.biliGrab.OnChatMessage += BiliGrab_OnChatMessage;
            this.biliGrab.OnLikeMessage += BiliGrab_OnLikeMessage;
            this.biliGrab.OnGiftMessage += BiliGrab_OnGiftMessage;
            this.biliGrab.OnInteractMessage += BiliGrab_OnInteractMessage;
            this.biliGrab.OnShareMessage += BiliGrab_OnShareMessage;
            this.biliGrab.OnRoomStatsMessage += BiliGrab_OnRoomStatsMessage;

            //多平台WS数据路由 - 将代理收到的WS数据按平台分发到对应解析器
            this.grab.Proxy.OnWebSocketData += (s, e) =>
            {
                if (e.Platform == LivePlatform.Kuaishou)
                {
                    this.ksGrab.HandleWebSocketData(e);
                }
                else if (e.Platform == LivePlatform.Bilibili)
                {
                    this.biliGrab.HandleWebSocketData(e);
                }
            };

            //B站发送数据路由 - 解析AUTH认证包提取roomid
            this.grab.Proxy.OnWebSocketSentData += (s, e) =>
            {
                if (e.Platform == LivePlatform.Bilibili)
                {
                    this.biliGrab.HandleSentData(e);
                }
            };

            this.socketServer = socket;
            //dieout.Start();
            giftCountTimer.Start();
        }

      

        //礼物缓存清理计时器回调
        private void GiftCountTimer_Elapsed(object sender, ElapsedEventArgs e)
        {
            var now = DateTime.Now;
            var timeOutKeys = giftCountCache.Where(w => w.Value.Item2 < now.AddSeconds(-10) || w.Value == null).Select(s => s.Key).ToList();

            //淘汰过期的礼物计数缓存
            lock (giftCountCache)
            {
                timeOutKeys.ForEach(key =>
                {
                    Tuple<int, DateTime> _;
                    giftCountCache.TryRemove(key, out _);
                });
            }
        }

        //心跳淘汰计时器回调
        private void Dieout_Elapsed(object sender, ElapsedEventArgs e)
        {
            var now = DateTime.Now;
            var dieoutKvs = socketList.Where(w => w.Value.LastPing.AddSeconds(dieout.Interval * 3) < now).ToList();
            dieoutKvs.ForEach(f => f.Value.Socket.Close());
        }

        //判断Rommid是否符合拦截规则
        private bool CheckRoomId(long roomid)
        {
            if (!AppSetting.Current.WebRoomIds.Any()) return true;

            var webrid = AppRuntime.RoomCaches.GetCachedWebRoomid(roomid.ToString());
            if (webrid.IsNullOrWhiteSpace()) return true;
            if (webrid == "未知") return true;

            return AppSetting.Current.WebRoomIds.Contains(webrid);
        }

        //解析用户
        private MsgUser GetUser(User data)
        {
            if (data == null) return null;
            MsgUser user = new MsgUser()
            {
                DisplayId = data.displayId,
                ShortId = data.shortId,
                Gender = data.Gender,
                Id = data.Id,
                Level = data.Level,
                PayLevel = (int)(data.payGrade?.Level ?? -1),
                Nickname = data.Nickname ?? "用户" + data.displayId,
                HeadImgUrl = data.avatarThumb?.urlLists?.FirstOrDefault() ?? "",
                SecUid = data.sec_uid,
                FollowerCount = data.followInfo?.followerCount ?? -1,
                FollowingCount = data.followInfo?.followingCount ?? -1,
                FollowStatus = data.followInfo?.followStatus ?? -1,
            };
            user.FansClub = new FansClubInfo()
            {
                ClubName = "",
                Level = 0
            };

            if (data.fansClub != null && data.fansClub.Data != null)
            {
                user.FansClub.ClubName = data.fansClub.Data.clubName;
                user.FansClub.Level = data.fansClub.Data.Level;
            }

            return user;
        }

        //检查属性定义
        private bool HasProperty(dynamic obj, string propertyName)
        {
            if (obj is ExpandoObject)
            {
                return ((IDictionary<string, object>)obj).ContainsKey(propertyName);
            }
            return obj.GetType().GetProperty(propertyName) != null;
        }

        //创建消息对象
        private T CreateMsg<T>(dynamic msg) where T : Msg, new()
        {
            var roomid = msg.Common.roomId.ToString();
            RoomInfo roomInfo = AppRuntime.RoomCaches.GetCachedWebRoomInfo(roomid);

            //判断 Common 属性是否存在
            var hasUser = HasProperty(msg, nameof(msg.User));

            var enty = new T()
            {
                MsgId = msg.Common?.msgId,
                RoomId = roomid,
                WebRoomId = roomInfo?.WebRoomId ?? "",
                RoomTitle = roomInfo?.Title ?? "",
                IsAnonymous = roomInfo?.IsAnonymous ?? false,
                Appid = msg.Common.appId.ToString(),
                User = hasUser ? GetUser(msg.User) : null,
            };

            //判断是否是直播间管理员
            if (enty.User != null && roomInfo != null && roomInfo.AdminUserIds.Any())
            {
                enty.User.IsAdmin = roomInfo.AdminUserIds.Contains(enty.User.Id.ToString());
            }

            //判断是否是主播
            if (enty.User != null && roomInfo != null && roomInfo.Owner != null)
            {
                enty.User.IsAnchor = enty.User.Id.ToString() == roomInfo.Owner.UserId;
            }

            return enty;
        }

        //打印消息        
        private void PrintMsg(Msg msg, PackMsgType barType)
        {
            var rinfo = AppRuntime.RoomCaches.GetCachedWebRoomInfo(msg.RoomId.ToString());
            var roomName = (rinfo?.Owner?.Nickname ?? ((msg.WebRoomId.IsNullOrWhiteSpace() ? msg.RoomId.ToString() : msg.WebRoomId)));
            var isAdmin = msg.User != null && msg.User.IsAdmin;
            var isAnchor = msg.User != null && msg.User.IsAnchor;

            var text = $"{DateTime.Now.ToString("HH:mm:ss")} [{roomName}] [{barType}]";

            if (isAdmin)
            {
                text += " [管理员]";
            }

            if (isAnchor)
            {
                text += " [主播]";
            }

            if (msg.User != null)
            {
                text += $" [{msg.User?.GenderToString()}] ";
            }

            ConsoleColor color = AppSetting.Current.ColorMap[barType].Item1;
            var append = msg.Content;
            switch (barType)
            {
                case PackMsgType.弹幕消息: append = $"{msg?.User?.Nickname}: {msg.Content}"; break;
                case PackMsgType.下播: append = $"直播已结束"; break;
                default: break;
            }

            text += append;

            if (AppSetting.Current.BarrageLog)
            {
                Logger.LogBarrage(barType, msg);
            }

            if (!Appsetting.PrintBarrage) return;
            if (AppSetting.Current.PrintFilter.Any() && !AppSetting.Current.PrintFilter.Contains(barType.GetHashCode())) return;

            OnPrint?.Invoke(this, new PrintEventArgs()
            {
                Color = color,
                Message = text,
                MsgType = barType
            });

            if (++printCount > 10000)
            {
                Console.Clear();
                Logger.PrintColor("控制台已清理");
                printCount = 0;
            }
            Logger.PrintColor(text + "\n", color);
        }

        //发送消息包装事件
        private async void FirePack(Msg msg, PackMsgType barType)
        {
            if (OnPackMessage == null) return;
            var arg = new PackMsgEventArgs()
            {
                MsgType = barType,
                Message = msg
            };
            //异步执行
            OnPackMessage(this, arg);
        }

        //附加房间信息
        private void AttachRoomInfo(Msg msg)
        {
            if (msg == null) return;
            var roomInfo = AppRuntime.RoomCaches.GetCachedWebRoomInfo(msg.RoomId.ToString());
            if (roomInfo == null) return;

            if (roomInfo.Owner != null)
            {
                msg.Owner = new RoomAnchorInfo()
                {
                    Nickname = roomInfo.Owner.Nickname,
                    HeadUrl = roomInfo.Owner.HeadUrl,
                    FollowStatus = roomInfo.Owner.FollowStatus,
                    SecUid = roomInfo.Owner.SecUid,
                    UserId = roomInfo.Owner.UserId
                };
            }
           

            if (msg.WebRoomId.IsNullOrWhiteSpace())
            {
                msg.WebRoomId = roomInfo.WebRoomId;
                msg.RoomTitle = roomInfo.Title;
                msg.IsAnonymous = roomInfo.IsAnonymous;
            }
        }

        //粉丝团
        private void Grab_OnFansclubMessage(object sender, WssBarrageGrab.RoomMessageEventArgs<FansclubMessage> e)
        {
            var msg = e.Message;
            if (!CheckRoomId(msg.Common.roomId)) return;
            var enty = CreateMsg<FansclubMsg>(msg);

            enty.Content = msg.Content;
            enty.Type = msg.Type;
            enty.Level = enty.User.FansClub.Level;

            var msgType = PackMsgType.粉丝团消息;
            AttachRoomInfo(enty);
            FirePack(enty, msgType);
            PrintMsg(enty, msgType);
            Broadcast(new BarrageMsgPack(enty.ToJson(), msgType, e.Process));
        }

        //统计消息
        private void Grab_OnRoomUserSeqMessage(object sender, WssBarrageGrab.RoomMessageEventArgs<RoomUserSeqMessage> e)
        {
            var msg = e.Message;
            if (!CheckRoomId(msg.Common.roomId)) return;

            var enty = CreateMsg<UserSeqMsg>(msg);
            enty.OnlineUserCount = msg.Total;
            enty.TotalUserCount = msg.totalUser;
            enty.TotalUserCountStr = msg.totalPvForAnchor;
            enty.OnlineUserCountStr = msg.onlineUserForAnchor;
            enty.Content = $"当前直播间人数 {msg.onlineUserForAnchor}，累计直播间人数 {msg.totalPvForAnchor}";

            var msgType = PackMsgType.直播间统计;
            AttachRoomInfo(enty);
            FirePack(enty, msgType);
            PrintMsg(enty, msgType);
            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process);
            Broadcast(pack);
        }

        //礼物
        private void Grab_OnGiftMessage(object sender, WssBarrageGrab.RoomMessageEventArgs<GiftMessage> e)
        {
            var msg = e.Message;
            if (!CheckRoomId(msg.Common.roomId)) return;

            var key = msg.Common.roomId + "-" + msg.giftId + "-" + msg.groupId.ToString();

            int currCount = (int)msg.repeatCount;
            int lastCount = 0;

            //纠正赋值礼物数据，比如是否连击，弹幕回送会不准确
            //var findForData = giftData?.gifts?.FirstOrDefault(f => f.id == msg.giftId);
            //if (findForData != null)
            //{
            //    var ogift = msg.Gift;
            //    ogift.Name = findForData.name;
            //    ogift.Combo = findForData.combo;
            //    ogift.diamondCount = findForData.diamond_count;
            //    ogift.Name = findForData.name;
            //}

            //判断礼物重复
            if (msg.repeatEnd == 1 && giftCountCache.ContainsKey(key))
            {
                //清除缓存中的key
                if (msg.groupId > 0)
                {
                    Tuple<int, DateTime> _;
                    giftCountCache.TryRemove(key, out _);
                }
                return;
            }
            var backward = currCount <= lastCount;
            if (currCount <= 0) currCount = 1;

            if (giftCountCache.ContainsKey(key))
            {
                lastCount = giftCountCache[key].Item1;
                backward = currCount <= lastCount;
                if (!backward)
                {
                    lock (giftCountCache)
                    {
                        giftCountCache[key] = Tuple.Create(currCount, DateTime.Now);
                    }
                }
            }
            else
            {
                if (msg.groupId > 0 && !backward)
                {
                    giftCountCache.TryAdd(key, Tuple.Create(currCount, DateTime.Now));
                }
            }
            //比上次小，则说明先后顺序出了问题，直接丢掉，应为比它大的消息已经处理过了
            if (backward) return;

            var count = currCount - lastCount;

            var enty = CreateMsg<GiftMsg>(msg);
            enty.Content = $"{msg.User.Nickname} 送出 {msg.Gift.Name}{(msg.Gift.Combo ? "(可连击)" : "")} x {msg.repeatCount}个，增量{count}个";
            enty.DiamondCount = msg.Gift.diamondCount;
            enty.RepeatCount = msg.repeatCount;
            enty.GiftCount = count;
            enty.GroupId = msg.groupId;
            enty.GiftId = msg.giftId;
            enty.GiftName = msg.Gift.Name;
            enty.Combo = msg.Gift.Combo;
            enty.ImgUrl = msg.Gift.Image?.urlLists?.FirstOrDefault() ?? "";
            enty.ToUser = GetUser(msg.toUser);

            if (enty.ToUser != null)
            {
                enty.Content += "，给" + enty.ToUser.Nickname;
            }

            var msgType = PackMsgType.礼物消息;
            AttachRoomInfo(enty);
            FirePack(enty, msgType);
            PrintMsg(enty, msgType);
            var pack = new BarrageMsgPack(enty.ToJson(), PackMsgType.礼物消息, e.Process);
            Broadcast(pack);
        }

        //关注
        private void Grab_OnSocialMessage(object sender, WssBarrageGrab.RoomMessageEventArgs<SocialMessage> e)
        {
            var msg = e.Message;
            if (!CheckRoomId(msg.Common.roomId)) return;
            if (msg.Action != 1) return;
            var enty = CreateMsg<Msg>(msg);
            enty.Content = $"{msg.User.Nickname} 关注了主播";

            var msgType = PackMsgType.关注消息;
            AttachRoomInfo(enty);
            FirePack(enty, msgType);
            PrintMsg(enty, msgType);
            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process);
            var json = JsonConvert.SerializeObject(pack);
            Broadcast(pack);
        }

        //直播间分享
        private void Grab_OnShardMessage(object sender, WssBarrageGrab.RoomMessageEventArgs<SocialMessage> e)
        {
            var msg = e.Message;
            if (!CheckRoomId(msg.Common.roomId)) return;
            if (msg.Action != 3) return;
            ShareType type = ShareType.未知;
            if (Enum.IsDefined(type.GetType(), int.Parse(msg.shareTarget)))
            {
                type = (ShareType)int.Parse(msg.shareTarget);
            }

            var enty = CreateMsg<ShareMessage>(msg);
            enty.Content = $"{msg.User.Nickname} 分享了直播间到{type}";
            enty.ShareType = type;

            var msgType = PackMsgType.直播间分享;
            AttachRoomInfo(enty);
            FirePack(enty, msgType);
            PrintMsg(enty, msgType);

            //shareTarget: (112:好友),(1微信)(2朋友圈)(3微博)(5:qq)(4:qq空间),shareType: 1            
            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process);
            var json = JsonConvert.SerializeObject(pack);
            Broadcast(pack);
        }

        //来了
        private void Grab_OnMemberMessage(object sender, WssBarrageGrab.RoomMessageEventArgs<Modles.ProtoEntity.MemberMessage> e)
        {
            var msg = e.Message;
            if (!CheckRoomId(msg.Common.roomId)) return;

            var enterType = e.Message.userEnterTipType;
            var enty = CreateMsg<Modles.JsonEntity.MemberMessage>(msg);
            enty.Content = $"{msg.User.Nickname} ${(enterType == 6 ? " 通过分享" : "")}来了 直播间人数:{msg.memberCount}";
            enty.CurrentCount = msg.memberCount;
            enty.EnterTipType = enterType;

            var msgType = PackMsgType.进直播间;
            AttachRoomInfo(enty);
            FirePack(enty, msgType);
            PrintMsg(enty, msgType);
            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process);
            var json = JsonConvert.SerializeObject(pack);
            Broadcast(pack);
        }

        //点赞
        private void Grab_OnLikeMessage(object sender, WssBarrageGrab.RoomMessageEventArgs<LikeMessage> e)
        {
            var msg = e.Message;
            if (!CheckRoomId(msg.Common.roomId)) return;

            var enty = CreateMsg<LikeMsg>(msg);
            enty.Total = msg.Total;
            enty.Count = msg.Count;
            enty.Content = $"{msg.User.Nickname} 为主播点了{msg.Count}个赞，总点赞{msg.Total}";

            var msgType = PackMsgType.点赞消息;
            AttachRoomInfo(enty);
            FirePack(enty, msgType);
            PrintMsg(enty, msgType);
            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process);
            Broadcast(pack);
        }

        //弹幕
        private void Grab_OnChatMessage(object sender, WssBarrageGrab.RoomMessageEventArgs<ChatMessage> e)
        {
            var msg = e.Message;
            if (!CheckRoomId(msg.Common.roomId)) return;

            var enty = CreateMsg<Msg>(msg);
            enty.Content = msg.Content;

            var msgType = PackMsgType.弹幕消息;
            AttachRoomInfo(enty);
            FirePack(enty, msgType);
            PrintMsg(enty, msgType);

            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process);
            Broadcast(pack);
        }

        //直播间状态变更
        private void Grab_OnControlMessage(object sender, WssBarrageGrab.RoomMessageEventArgs<ControlMessage> e)
        {
            var msg = e.Message;
            if (!CheckRoomId(msg.Common.roomId)) return;
            BarrageMsgPack pack = null;
            //下播
            if (msg.Status == 3)
            {
                var enty = new Msg()
                {
                    MsgId = msg.Common.msgId,
                    Content = "直播已结束",
                    RoomId = msg.Common.roomId.ToString(),
                    WebRoomId = AppRuntime.RoomCaches.GetCachedWebRoomid(msg.Common.roomId.ToString()),
                    User = null,
                    Appid = e.Message.Common.appId.ToString()
                };

                var msgType = PackMsgType.下播;
                AttachRoomInfo(enty);
                FirePack(enty, msgType);
                PrintMsg(enty, msgType);
                pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process);
            }

            if (pack != null)
            {
                Broadcast(pack);
            }
        }

        //直播伴侣上下播
        private void Proxy_OnLiveCompanAction(object sender, LiveCompanEventArgs e)
        {            
            var msgid = DateTime.Now.Ticks;
            BarrageMsgPack pack = null;

            if (e.Action == 1)
            {
                RoomInfo roomInfo = e.Data as RoomInfo;
                if(roomInfo == null) return;

                var enty = new LiveCompanActionMsg()
                {
                    MsgId = msgid,
                    Content = $" [{roomInfo.Owner.Nickname}] 开播了",
                    RoomId = roomInfo.RoomId,
                    WebRoomId = AppRuntime.RoomCaches.GetCachedWebRoomid(roomInfo.RoomId),
                    User = null,
                    Appid = "0",
                    Data = roomInfo,
                };

                var msgType = PackMsgType.直播伴侣开播;
                AttachRoomInfo(enty);
                FirePack(enty, msgType);
                PrintMsg(enty, msgType);
                pack = new BarrageMsgPack(enty.ToJson(), msgType, e.ProcessName);
            }

            if (e.Action == 2)
            {                
                var info = e.Data as AnchorFinishInfo;
                if (info.RoomId.IsNullOrWhiteSpace()) return;

                var enty = new LiveCompanActionMsg()
                {
                    MsgId = msgid,
                    Content = $" [{info.Nickname}]下播，房间号:{info.RoomId}",
                    RoomId = info.RoomId,
                    WebRoomId = AppRuntime.RoomCaches.GetCachedWebRoomid(info.RoomId),                    
                    Appid = "0",
                    Data = info
                };

                var msgType = PackMsgType.直播伴侣下播;
                AttachRoomInfo(enty);
                FirePack(enty, msgType);
                PrintMsg(enty, msgType);
                pack = new BarrageMsgPack(enty.ToJson(), msgType, e.ProcessName);
            }


            if (pack != null)
            {
                Broadcast(pack);
            }
        }

        #region 快手消息处理

        private MsgUser KsUserToMsgUser(Modles.KsProtoEntity.KsSimpleUserInfo user)
        {
            if (user == null) return null;
            return new MsgUser()
            {
                Id = long.TryParse(user.PrincipalId, out var pid) ? pid : 0,
                DisplayId = user.PrincipalId ?? "",
                Nickname = user.UserName ?? "用户" + user.PrincipalId,
                HeadImgUrl = user.HeadUrl ?? "",
                Gender = 0,
                Level = 0,
                PayLevel = 0,
                ShortId = 0,
            };
        }

        private string KsRoomId;

        private void KsGrab_OnChatMessage(object sender, KsRoomMessageEventArgs<Modles.KsProtoEntity.KsWebCommentFeed> e)
        {
            var feed = e.Message;
            var enty = new Msg()
            {
                MsgId = long.TryParse(feed.Id, out var id) ? id : DateTime.Now.Ticks,
                User = KsUserToMsgUser(feed.User),
                Content = feed.Content ?? "",
                RoomId = KsRoomId ?? "",
                WebRoomId = KsRoomId ?? "",
                Appid = "0"
            };

            var msgType = PackMsgType.弹幕消息;
            PrintMsg(enty, msgType);
            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process, LivePlatform.Kuaishou);
            Broadcast(pack);
        }

        private void KsGrab_OnLikeMessage(object sender, KsRoomMessageEventArgs<Modles.KsProtoEntity.KsWebLikeFeed> e)
        {
            var feed = e.Message;
            var enty = new LikeMsg()
            {
                MsgId = long.TryParse(feed.Id, out var id) ? id : DateTime.Now.Ticks,
                User = KsUserToMsgUser(feed.User),
                Content = $"{feed.User?.UserName ?? "用户"} 为主播点了赞",
                Count = 1,
                Total = 0,
                RoomId = KsRoomId ?? "",
                WebRoomId = KsRoomId ?? "",
                Appid = "0"
            };

            var msgType = PackMsgType.点赞消息;
            PrintMsg(enty, msgType);
            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process, LivePlatform.Kuaishou);
            Broadcast(pack);
        }

        private void KsGrab_OnGiftMessage(object sender, KsRoomMessageEventArgs<Modles.KsProtoEntity.KsWebGiftFeed> e)
        {
            var feed = e.Message;
            var enty = new GiftMsg()
            {
                MsgId = long.TryParse(feed.Id, out var id) ? id : DateTime.Now.Ticks,
                User = KsUserToMsgUser(feed.User),
                GiftId = feed.GiftId,
                GiftName = KsGiftHelper.GetGiftName(feed.GiftId) ?? $"礼物#{feed.GiftId}",
                GroupId = 0,
                GiftCount = feed.ComboCount > 0 ? feed.ComboCount : 1,
                RepeatCount = feed.ComboCount > 0 ? feed.ComboCount : 1,
                DiamondCount = 0,
                Combo = feed.ComboCount > 1,
                Content = $"{feed.User?.UserName ?? "用户"} 送出 {KsGiftHelper.GetGiftName(feed.GiftId) ?? $"礼物#{feed.GiftId}"} x {feed.ComboCount}",
                RoomId = KsRoomId ?? "",
                WebRoomId = KsRoomId ?? "",
                Appid = "0"
            };

            var msgType = PackMsgType.礼物消息;
            PrintMsg(enty, msgType);
            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process, LivePlatform.Kuaishou);
            Broadcast(pack);
        }

        private void KsGrab_OnShareMessage(object sender, KsRoomMessageEventArgs<Modles.KsProtoEntity.KsWebShareFeed> e)
        {
            var feed = e.Message;
            var enty = new ShareMessage()
            {
                MsgId = long.TryParse(feed.Id, out var id) ? id : DateTime.Now.Ticks,
                User = KsUserToMsgUser(feed.User),
                Content = $"{feed.User?.UserName ?? "用户"} 分享了直播间",
                ShareType = ShareType.未知,
                RoomId = KsRoomId ?? "",
                WebRoomId = KsRoomId ?? "",
                Appid = "0"
            };

            var msgType = PackMsgType.直播间分享;
            PrintMsg(enty, msgType);
            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process, LivePlatform.Kuaishou);
            Broadcast(pack);
        }

        private void KsGrab_OnRoomStatsMessage(object sender, KsRoomMessageEventArgs<Modles.KsProtoEntity.KsSCWebFeedPush> e)
        {
            var feed = e.Message;
            var enty = new UserSeqMsg()
            {
                MsgId = DateTime.Now.Ticks,
                Content = $"当前直播间人数 {feed.DisplayWatchingCount}，点赞 {feed.DisplayLikeCount}",
                OnlineUserCount = long.TryParse(feed.DisplayWatchingCount, out var watching) ? watching : 0,
                TotalUserCount = 0,
                OnlineUserCountStr = feed.DisplayWatchingCount ?? "0",
                TotalUserCountStr = "0",
                RoomId = KsRoomId ?? "",
                WebRoomId = KsRoomId ?? "",
                Appid = "0"
            };

            var msgType = PackMsgType.直播间统计;
            PrintMsg(enty, msgType);
            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process, LivePlatform.Kuaishou);
            Broadcast(pack);
        }

        #endregion

        #region B站消息处理

        private MsgUser BiliUserToMsgUser(Modles.BiliProtoEntity.BiliDanmuMsg user)
        {
            if (user == null) return null;
            return new MsgUser()
            {
                Id = user.Uid,
                DisplayId = user.Uid.ToString(),
                Nickname = user.Uname ?? "用户" + user.Uid,
                IsAdmin = user.IsAdmin,
                IsAnchor = user.IsAnchor,
            };
        }

        private MsgUser BiliUserToMsgUser(long uid, string uname)
        {
            return new MsgUser()
            {
                Id = uid,
                DisplayId = uid.ToString(),
                Nickname = string.IsNullOrWhiteSpace(uname) ? "用户" + uid : uname,
            };
        }

        //B站弹幕
        private void BiliGrab_OnChatMessage(object sender, BiliRoomMessageEventArgs<Modles.BiliProtoEntity.BiliDanmuMsg> e)
        {
            var msg = e.Message;
            var enty = new Msg()
            {
                MsgId = DateTime.Now.Ticks,
                User = BiliUserToMsgUser(msg),
                Content = msg.Content ?? "",
                RoomId = biliGrab.CurrentRoomId,
                WebRoomId = biliGrab.CurrentRoomId,
                Appid = "0"
            };

            var msgType = PackMsgType.弹幕消息;
            PrintMsg(enty, msgType);
            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process, LivePlatform.Bilibili);
            Broadcast(pack);
        }

        //B站点赞
        private void BiliGrab_OnLikeMessage(object sender, BiliRoomMessageEventArgs<Modles.BiliProtoEntity.BiliLikeData> e)
        {
            var msg = e.Message;
            var enty = new LikeMsg()
            {
                MsgId = DateTime.Now.Ticks,
                User = BiliUserToMsgUser(msg.Uid, msg.Uname),
                Content = $"{msg.Uname ?? "用户"} 为主播点了赞",
                Count = 1,
                Total = 0,
                RoomId = biliGrab.CurrentRoomId,
                WebRoomId = biliGrab.CurrentRoomId,
                Appid = "0"
            };

            var msgType = PackMsgType.点赞消息;
            PrintMsg(enty, msgType);
            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process, LivePlatform.Bilibili);
            Broadcast(pack);
        }

        //B站礼物
        private void BiliGrab_OnGiftMessage(object sender, BiliRoomMessageEventArgs<Modles.BiliProtoEntity.BiliGiftData> e)
        {
            var msg = e.Message;
            var count = msg.Num > 0 ? msg.Num : 1;
            var enty = new GiftMsg()
            {
                MsgId = DateTime.Now.Ticks,
                User = BiliUserToMsgUser(msg.Uid, msg.Uname),
                GiftId = msg.GiftId,
                GiftName = msg.GiftName ?? $"礼物#{msg.GiftId}",
                GroupId = 0,
                GiftCount = count,
                RepeatCount = msg.SuperBatchNum > 0 ? msg.SuperBatchNum : count,
                DiamondCount = (int)(msg.Price * 1000),
                Combo = msg.ComboNum > 1,
                ImgUrl = msg.GifUrl ?? "",
                Content = $"{msg.Uname ?? "用户"} 送出 {msg.GiftName ?? $"礼物#{msg.GiftId}"} x {count}",
                RoomId = biliGrab.CurrentRoomId,
                WebRoomId = biliGrab.CurrentRoomId,
                Appid = "0"
            };

            var msgType = PackMsgType.礼物消息;
            PrintMsg(enty, msgType);
            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process, LivePlatform.Bilibili);
            Broadcast(pack);
        }

        //B站进房/关注
        private void BiliGrab_OnInteractMessage(object sender, BiliRoomMessageEventArgs<Modles.BiliProtoEntity.BiliInteractData> e)
        {
            var msg = e.Message;
            // msg_type: 1=进入 2=关注
            if (msg.MsgType == 2)
            {
                var enty = new Msg()
                {
                    MsgId = DateTime.Now.Ticks,
                    User = BiliUserToMsgUser(msg.Uid, msg.Uname),
                    Content = $"{msg.Uname ?? "用户"} 关注了主播",
                    RoomId = biliGrab.CurrentRoomId,
                    WebRoomId = biliGrab.CurrentRoomId,
                    Appid = "0"
                };
                var msgType = PackMsgType.关注消息;
                PrintMsg(enty, msgType);
                var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process, LivePlatform.Bilibili);
                Broadcast(pack);
            }
            else
            {
                var enty = new Modles.JsonEntity.MemberMessage()
                {
                    MsgId = DateTime.Now.Ticks,
                    User = BiliUserToMsgUser(msg.Uid, msg.Uname),
                    Content = $"{msg.Uname ?? "用户"} 来了直播间",
                    CurrentCount = 0,
                    EnterTipType = 0,
                    RoomId = biliGrab.CurrentRoomId,
                    WebRoomId = biliGrab.CurrentRoomId,
                    Appid = "0"
                };
                var msgType = PackMsgType.进直播间;
                PrintMsg(enty, msgType);
                var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process, LivePlatform.Bilibili);
                Broadcast(pack);
            }
        }

        //B站分享
        private void BiliGrab_OnShareMessage(object sender, BiliRoomMessageEventArgs<Modles.BiliProtoEntity.BiliInteractData> e)
        {
            var msg = e.Message;
            var enty = new ShareMessage()
            {
                MsgId = DateTime.Now.Ticks,
                User = BiliUserToMsgUser(msg.Uid, msg.Uname),
                Content = $"{msg.Uname ?? "用户"} 分享了直播间",
                ShareType = ShareType.未知,
                RoomId = biliGrab.CurrentRoomId,
                WebRoomId = biliGrab.CurrentRoomId,
                Appid = "0"
            };

            var msgType = PackMsgType.直播间分享;
            PrintMsg(enty, msgType);
            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process, LivePlatform.Bilibili);
            Broadcast(pack);
        }

        //B站房间统计
        private void BiliGrab_OnRoomStatsMessage(object sender, BiliRoomMessageEventArgs<Modles.BiliProtoEntity.BiliRoomStats> e)
        {
            var msg = e.Message;
            var enty = new UserSeqMsg()
            {
                MsgId = DateTime.Now.Ticks,
                Content = $"当前直播间人数 {msg.Online}，看过 {msg.Watched}",
                OnlineUserCount = msg.Online,
                TotalUserCount = msg.Watched,
                OnlineUserCountStr = msg.Online.ToString(),
                TotalUserCountStr = msg.Watched.ToString(),
                RoomId = biliGrab.CurrentRoomId,
                WebRoomId = biliGrab.CurrentRoomId,
                Appid = "0"
            };

            var msgType = PackMsgType.直播间统计;
            PrintMsg(enty, msgType);
            var pack = new BarrageMsgPack(enty.ToJson(), msgType, e.Process, LivePlatform.Bilibili);
            Broadcast(pack);
        }

        #endregion

        //监听用户连接
        private void Listen(IWebSocketConnection socket)
        {
            //客户端url
            string clientUrl = socket.ConnectionInfo.ClientIpAddress + ":" + socket.ConnectionInfo.ClientPort;
            if (!socketList.ContainsKey(clientUrl))
            {
                socketList.Add(clientUrl, new UserState(socket));
                Logger.PrintColor($"{DateTime.Now.ToLongTimeString()} 已经建立与[{clientUrl}]的连接", ConsoleColor.Green);
            }
            else
            {
                socketList[clientUrl].Socket = socket;
            }

            //接收指令
            socket.OnMessage = (message) =>
            {
                try
                {
                    var cmdPack = JsonConvert.DeserializeObject<Command>(message);
                    if (cmdPack == null) return;
                }
                catch (Exception) { }
            };

            socket.OnClose = () =>
            {
                socketList.Remove(clientUrl);
                Logger.PrintColor($"{DateTime.Now.ToLongTimeString()} 已经关闭与[{clientUrl}]的连接", ConsoleColor.Red);
            };

            socket.OnPing = (data) =>
            {
                socketList[clientUrl].LastPing = DateTime.Now;
                socket.SendPong(Encoding.UTF8.GetBytes("pong"));
            };
        }

        /// <summary>
        /// 广播消息
        /// </summary>
        /// <param name="msg"></param>
        public void Broadcast(BarrageMsgPack pack)
        {
            if (pack == null) return;
            if (AppSetting.Current.PushFilter.Any() && !AppSetting.Current.PushFilter.Contains(pack.Type.GetHashCode())) return;

            var offLines = new List<string>();
            foreach (var item in socketList)
            {
                var state = item.Value;
                if (item.Value.Socket.IsAvailable)
                {
                    state.Socket.Send(pack.ToJson());
                }
                else
                {
                    offLines.Add(item.Key);
                }
            }
            //删除掉线的套接字        
            offLines.ForEach(key => socketList.Remove(key));
        }

        /// <summary>
        /// 开始监听
        /// </summary>
        public void StartListen()
        {
            try
            {
                this.grab.Start(); //启动代理
                this.socketServer.Start(Listen);//启动监听         
            }
            catch (Exception)
            {
                this.Dispose();
                throw;
            }
        }

        /// <summary>
        /// 关闭服务器连接，并关闭系统代理
        /// </summary>
        public void Dispose()
        {
            socketList.Values.ToList().ForEach(f => f.Socket.Close());
            socketList.Clear();
            socketServer.Dispose();
            grab.Dispose();
            this.IsDisposed = true;
            this.OnClose?.Invoke(this, EventArgs.Empty);
        }

        /// <summary>
        /// 客户端套接字上下文
        /// </summary>
        class UserState
        {
            /// <summary>
            /// 套接字
            /// </summary>
            public IWebSocketConnection Socket { get; set; }

            /// <summary>
            /// 上次发起心跳包时间
            /// </summary>
            public DateTime LastPing { get; set; } = DateTime.Now;

            public UserState()
            {

            }
            public UserState(IWebSocketConnection socket)
            {
                Socket = socket;
            }
        }

        /// <summary>
        /// Print事件参数
        /// </summary>
        public class PrintEventArgs : EventArgs
        {
            public string Message { get; set; }

            public ConsoleColor Color { get; set; }

            public PackMsgType MsgType { get; set; }
        }

        /// <summary>
        /// 消息包事件参数
        /// </summary>
        public class PackMsgEventArgs
        {
            /// <summary>
            /// 消息类型
            /// </summary>
            public PackMsgType MsgType { get; set; }

            /// <summary>
            /// 消息内容
            /// </summary>
            public Msg Message { get; set; }
        }
    }
}