//
//  YH_SocketService.m
//  Yoho
//
//  Created by Arthur on 16/6/16.
//  Copyright © 2016年 YOHO. All rights reserved.
//

#import "YH_SocketService.h"
#import "GCDAsyncSocket+SocketHandler.h"
#import "UtilsMacros.h"
#import "YHL_ApiStringMacros.h"
#import "YH_NetworkAdapter+Live.h"
#import "OpenUDID.h"

#define TimeIntervalForHeartBeat    5
#define SOCKET_COMMAND          @"cmd"
#define SOCKET_CONNECT_MAX_COUNT        9

static const int ddLogLevel = DDLogLevelInfo;

@interface YH_SocketService ()<GCDAsyncSocketDelegate>
{
    BOOL _hasLoginWithNameSuccessed;
    BOOL _hasLoginWithDevieToken;
    int _currentLikeCount;
    int _heartBeatCount;
}
@property (strong, nonatomic) NSTimer *connectTimer;
@property (strong, nonatomic) NSTimer *heartBeatTimer;
@property (strong, nonatomic) GCDAsyncSocket *socket;
@property (strong, nonatomic) NSString *deviceToken;

@end


@implementation YH_SocketService

- (instancetype)init
{
    self = [super init];
    if (self) {
        self.socket = [[GCDAsyncSocket alloc]initWithDelegate:self delegateQueue:dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0)];
        
        [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(networkReachabilityDidChange:) name:AFNetworkingReachabilityDidChangeNotification object:nil];
        self.deviceToken = [NSString stringWithFormat:@"%f",[[NSDate date] timeIntervalSince1970]];
    }
    
    return self;
}

- (NSString *)getProperIPWithAddress:(NSString *)ipAddr port:(UInt32)port
{
    NSError *addresseError = nil;
    NSArray *addresseArray = [GCDAsyncSocket lookupHost:ipAddr
                                                   port:port
                                                  error:&addresseError];
    if (addresseError) {
        NSLog(@"");
    }
    
    NSString *ipv6Addr = @"";
    for (NSData *addrData in addresseArray) {
        if ([GCDAsyncSocket isIPv6Address:addrData]) {
            ipv6Addr = [GCDAsyncSocket hostFromAddress:addrData];
        }
    }
    
    if (ipv6Addr.length == 0) {
        ipv6Addr = ipAddr;
    }
    
    return ipv6Addr;
}

//网络发生变化时重连
- (void)networkReachabilityDidChange:(NSNotification *)notification
{
    [self reConnect];
}

- (void)reConnect
{
    [_socket disconnect];
    _hasLoginWithDevieToken = NO;
    self.socket = [[GCDAsyncSocket alloc]initWithDelegate:self delegateQueue:dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0)];
    [self connectToHostRepeat];
}

- (void)dealloc
{
    [[NSNotificationCenter defaultCenter] removeObserver:self];
}


//- (void)reset
//{
//    DLog(@"socket service reset!!");
//    if (_heartBeatTimer) {
//        [_heartBeatTimer invalidate];
//        _heartBeatTimer = nil;
//    }
//    
//    [_socket disconnect];
//}

#pragma mark - Private Methods
- (void)heartbeat{

    if (_currentLikeCount) {
        [self commitUserPraise];
    }
    
    if (++_heartBeatCount>=7) {
        NSDictionary *params = @{@"cmd":@(SOCKET_TAG_HEARTBEAT),
                                 @"msg":@"heart_ios",
                                 @"room":MakeStringNotNil(self.room),
                                 };
        [_socket writeDataWithParams:params tag:SOCKET_TAG_HEARTBEAT];
        [[YH_NetworkAdapter adapter] sendInterfaceHeartBeatWithRoomId:self.room];
        _heartBeatCount = 0;
    }

    //心跳并判断是否已经断掉
    if (![_socket isConnected]) {
        [self reConnect];
    }
    //判断上次是否登录成功
    if (!_hasLoginWithDevieToken) {
        [self loginToServer];
    }
    [self performSelector:@selector(heartbeat) withObject:nil afterDelay:5];
}

#pragma mark - Service
- (BOOL)connect
{
    [[YH_NetworkAdapter adapter] getLiveBarrageSocketAddressWithParam:@{@"type":@"tcp"} completion:^(NSDictionary *infoDict, NSError *error) {
        if (!error) {
            NSArray *data = infoDict[@"data"];
            NSString *result = data.firstObject;
            NSArray *resultAry = [result componentsSeparatedByString:@":"];
            self.socketLiveHost = resultAry.firstObject;
            self.socketLivePort = [resultAry.lastObject integerValue];
//            self.socketLiveHost = @"im.yoho.cn"; //test
            if (![self connectToHost:self.socketLiveHost onPort:self.socketLivePort]) {
                if(!_connectTimer){
                    self.connectTimer = [NSTimer timerWithTimeInterval:4.0 target:self selector:@selector(connectToHostRepeat) userInfo:nil repeats:YES];
                    [[NSRunLoop mainRunLoop] addTimer:self.connectTimer forMode:NSRunLoopCommonModes];
                }
            } else {
                NSLog(@"connent success!");
            }
        }
    }];
    
    return YES;
}

- (void)connectToHostRepeat
{
    [_socket disconnect];
    _hasLoginWithDevieToken = NO;
    self.socket = [[GCDAsyncSocket alloc]initWithDelegate:self delegateQueue:dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0)];

    if ([self connectToHost:self.socketLiveHost onPort:self.socketLivePort]) {
        [self.connectTimer invalidate];
        self.connectTimer = nil;
    }
}

- (BOOL)connectToHost:(NSString *)host onPort:(uint16_t)port
{
    NSString *hostProper = [self getProperIPWithAddress:host port:port];
    
    NSError *socketConnectError = nil;
    
    NSAssert(_socket, @"You must init YH_SocketService first!");
    
    [_socket connectToHost:hostProper onPort:port error:&socketConnectError];
    
    if (socketConnectError) {
        DLog(@"connect host error: %@", [socketConnectError localizedDescription]);
        
        return NO;
    } else {
        return YES;
    }
}

- (BOOL)writeDataWithParams:(NSDictionary *)params tag:(long)tag
{
    NSAssert(_socket, @"You must init YH_SocketService first!");
    return [_socket writeDataWithParams:params tag:tag];
}

#pragma mark - GCDAsyncSocketDelegate

- (void)socket:(GCDAsyncSocket *)sock didConnectToHost:(NSString *)host port:(uint16_t)port
{
    DLog(@"socket connect success!");
    if (self.heartBeatTimer) {
        [self.heartBeatTimer invalidate];
        self.heartBeatTimer = nil;
    }
    
    [self loginToServer];
    [self listenData];
    
    dispatch_async(dispatch_get_main_queue(), ^{
        [self heartbeat];
    });
}

- (void)socketDidDisconnect:(GCDAsyncSocket *)sock withError:(NSError *)err{
    if (err) {
        NSLog(@"%@",err);
        
        // 断线重连
        [self reConnect];
    }
}

// 数据成功发送到服务器
- (void)socket:(GCDAsyncSocket *)sock didWriteDataWithTag:(long)tag
{
    switch (tag) {
        case SOCKET_TAG_LOGIN:
        {

        }
            break;
            
        case SOCKET_TAG_RESIVE_MSG:
        {
            
            
        }
            break;
            
        default:
            break;
    }
    
    [self listenData];
}


- (void)socket:(GCDAsyncSocket *)sock didReadData:(NSData *)data withTag:(long)tag
{

    [sock didWriteDataWithParams:data tag:tag dataProcessingHandler:^(NSDictionary *resDic) {
        
        if (IsDictionaryClass(resDic) && [resDic.allKeys containsObject:SOCKET_COMMAND]) {
            int cmd = [[resDic objectForKey:SOCKET_COMMAND] intValue];
            
            NSString *likeCount = [resDic objectForKey:@"likeNums"]?:@"";
            NSString *onlineCount = [resDic objectForKey:@"onlineNums"]?:@"";
            NSString *audienceCount = [resDic objectForKey:@"audienceNums"]?:@"";
            NSString *videoLen = [resDic objectForKey:@"videoLen"]?:@"";
            switch (cmd) {
                case SOCKET_TAG_USER_JOIN:
                {
                    //保证用户名只显示一次
                    NSString *uid = [resDic objectForKey:@"uid"];
                    if (!IsStrEmpty(uid)&& [uid isEqualToString:self.uid]&& !IsStrEmpty(self.uid)) {
                        if (_hasLoginWithNameSuccessed) {
                            break;
                        } else {
                            _hasLoginWithNameSuccessed = YES;
                        }
                    }
                }
                case SOCKET_TAG_RESIVE_MSG:
                {
                    NSString *uid = [resDic objectForKey:@"uid"];
                    if (!IsStrEmpty(uid)&& [uid isEqualToString:self.uid]&& !IsStrEmpty(self.uid)) {
                        return ;
                    }
                    [self.barrageViewController insertItem:resDic];
                }
                    break;
                case SOCKET_TAG_NotifyPRAISE:
                {
                    if ([self.delegate respondsToSelector:@selector(userPraised:isSelfPraise:)] && [resDic.allKeys containsObject:@"msg"]) {
                        yh_dispatch_execute_in_main_queue(^{
                            BOOL isSelfPraise =(resDic[@"deviceToken"] && [resDic[@"deviceToken"] isEqualToString:self.deviceToken]);
                            [self.delegate userPraised:resDic[@"msg"] isSelfPraise:isSelfPraise];
                        });
                    }
                }
                    break;
                case SOCKET_TAG_NotifyPeopleNumber:
                {
                    if ([self.delegate respondsToSelector:@selector(currentPeopleNumber:)] && [resDic.allKeys containsObject:@"msg"]) {
                        yh_dispatch_execute_in_main_queue(^{
                            [self.delegate currentPeopleNumber:onlineCount];
                        });
                    }
                }
                    break;
                case SOCKET_TAG_NotifyEndPlay:
                {
                    if ([self.delegate respondsToSelector:@selector(livePalyEnd:likeNums:videoLen:)]) {
                        yh_dispatch_execute_in_main_queue(^{
                            [self.delegate livePalyEnd:audienceCount likeNums:likeCount videoLen:videoLen];
                        });
                    }
                }
                    break;
                case SOCKET_TAG_NotifyOnlineAndLikes://登录成功返回在线人数和点赞数
                {
                    _hasLoginWithDevieToken = YES;
                    if ([self.delegate respondsToSelector:@selector(liveOnlineNums:likes:)]) {
                        yh_dispatch_execute_in_main_queue(^{
                            [self.delegate liveOnlineNums:onlineCount likes:likeCount];
                        });
                    }
                }
                    break;
                case SOCKET_TAG_NotifyEndSuccess:
                {
                    if ([self.delegate respondsToSelector:@selector(liveNotifyEndSuccess)]) {
                        yh_dispatch_execute_in_main_queue(^{
                            [self.delegate liveNotifyEndSuccess];
                        });
                    }
                }
                    break;
                    
                case SOCKET_TAG_NotifyShowSknProduct:
                {
                    if ([_delegate respondsToSelector:@selector(liveSocketServiceDidReceiveProductSkn:)]) {
                        yh_dispatch_execute_in_main_queue(^{
                            NSInteger skn = [resDic[@"skn"] integerValue];
                            [_delegate liveSocketServiceDidReceiveProductSkn:@(skn).stringValue];
                        });
                    }
                }
                    break;
                case SOCKET_TAG_NotifyRefreshSknList:
                {
                    if ([_delegate respondsToSelector:@selector(liveSocketServiceDidReceiveShouldRefreshSknList)]) {
                        yh_dispatch_execute_in_main_queue(^{
                            [_delegate liveSocketServiceDidReceiveShouldRefreshSknList];
                        });
                    }
                }
                    break;
                    
                case SOCKET_TAG_NotifyRefreshCoupon: {
                    NSString *refresh = resDic[@"refresh"];
                    if ([refresh isEqualToString:@"Y"]&&[_delegate respondsToSelector:@selector(liveSocketServiceDidReceiveShouldRefreshCoupon)]) {
                        yh_dispatch_execute_in_main_queue(^{
                            [_delegate liveSocketServiceDidReceiveShouldRefreshCoupon];
                        });
                    }
                }
                    break;
                case SOCKET_TAG_RecieveBuyingRoad: {
                    NSString *name = [resDic objectForKey:@"name"]?:@"";
                    if (!name.length) {
                        return ;
                    }
                    [self.barrageViewController insertItem:resDic];
                }
                    break;
                case SOCKET_TAG_RedPacketRecieved: {
                    NSInteger couponId = [[resDic objectForKey:@"couponId"] integerValue];
                    if (couponId && [_delegate respondsToSelector:@selector(liveSocketServiceDidReceiveRedPacketWithCouponId:)]) {
                        yh_dispatch_execute_in_main_queue(^{
                            [_delegate liveSocketServiceDidReceiveRedPacketWithCouponId:couponId];
                        });
                    }
                }
                default:
                    break;
            }
        }
    }];
    
    [self listenData];

}

-(void)listenData {
    
    [_socket readDataToData:[GCDAsyncSocket LFData] withTimeout:-1 tag:1];
}

- (void)logout
{
    if (_heartBeatTimer) {
        [_heartBeatTimer invalidate];
        _heartBeatTimer = nil;
    }
    if (_connectTimer) {
        [self.connectTimer invalidate];
        self.connectTimer = nil;
    }
    [_socket disconnect];
    DDLogInfo(@"主播登出socket服务器 %@",self.room);
    [NSObject cancelPreviousPerformRequestsWithTarget:self];
}

//testOK
- (void)loginToServer
{
    NSDictionary *params = @{@"cmd":@(SOCKET_TAG_LOGIN),
                             @"uid":MakeStringNotNil(self.uid),
                             @"room":MakeStringNotNil(self.room),
                             @"name":MakeStringNotNil(self.userName),
                             @"avatar":MakeStringNotNil(self.avartar),
                             @"deviceToken":self.deviceToken,
                             };
    
    if (self.uid.length&&self.userName.length) {
        //如果向服务端发送的 uid和name 是有效的 则证明已经向服务器发送了有效的token信息
        _hasLoginWithDevieToken = YES;
    }
    DDLogInfo(@"主播登录socket服务器 %@",params);
    [_socket writeDataWithParams:params tag:SOCKET_TAG_LOGIN];
}

//testOK
- (void)sendMessage:(NSString *)msg
{
    
    NSString *trimStr = [msg stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceCharacterSet]];
    
    if (IsStrEmpty(trimStr)) {
        return;
    }
    
    NSDictionary *params = @{@"cmd":@(SOCKET_TAG_SEND_MSG),
                             @"uid":MakeStringNotNil(self.uid),
                             @"msg":MakeStringNotNil(msg),
                             @"name":MakeStringNotNil(self.userName),
                             @"avatar":MakeStringNotNil(self.avartar),
                             @"room":MakeStringNotNil(self.room)
                             };
    [_socket writeDataWithParams:params tag:SOCKET_TAG_SEND_MSG];
}

- (void)sendBuyingRoad {
    NSString * udidString = [OpenUDID value];
    if (udidString.length) {
        udidString = [udidString substringToIndex:7];
        udidString = [udidString stringByAppendingString:@"..."];
    }
    NSString *userName = self.userName;
    if (!userName.length) {
        userName = udidString?:@"xxx";
    }
    
    NSDictionary *params = @{@"cmd":@(SOCKET_TAG_SendBuyingRoad),
                             @"msg":@"",
                             @"name":MakeStringNotNil(userName),
                             @"room":MakeStringNotNil(self.room)
                             };
    [_socket writeDataWithParams:params tag:SOCKET_TAG_SendBuyingRoad];
    
    NSDictionary *fakeParams = @{@"cmd":@(SOCKET_TAG_RecieveBuyingRoad),
                             @"msg":@"",
                             @"name":MakeStringNotNil(userName),
                             @"room":MakeStringNotNil(self.room)
                             };
    [self.barrageViewController insertItem:fakeParams];
}

//testOK
- (void)userPraise
{
    if (++_currentLikeCount>=10) {
        [self commitUserPraise];
    }
    if ([self.delegate respondsToSelector:@selector(userPraised:isSelfPraise:)]) {
        [self.delegate userPraised:@"" isSelfPraise:YES];
    }
}

/**
 *  @author Kennaki Kai, 16-08-20 13:08:09
 *
 *  @brief _currentLikeCount 满10 或跟随heart beat一起调用.
 *
 *  @since 1.0
 */
- (void)commitUserPraise {
    NSDictionary *params = @{@"cmd":@(SOCKET_TAG_PRAISE),
                             @"uid":MakeStringNotNil(self.uid),
                             @"msg":@"",
                             @"room":MakeStringNotNil(self.room),
                             @"deviceToken":self.deviceToken,
                             @"clicks":@(_currentLikeCount),
                             };
    [_socket writeDataWithParams:params tag:SOCKET_TAG_SEND_MSG];
    _currentLikeCount = 0;
}

//结束播放//testOK
- (void)endPaly:(NSString *)videoLen
{
    if(![_socket isConnected]){
        [self reConnect];
    }

    NSDictionary *params = @{@"cmd":@(SOCKET_TAG_EndPlay),
                             @"uid":MakeStringNotNil(self.uid),
                             @"msg":@"",
                             @"videoLen":MakeStringNotNil(videoLen),
                             @"room":MakeStringNotNil(self.room)
                             };
    [_socket writeDataWithParams:params tag:SOCKET_TAG_SEND_MSG];
}

@end