//
//  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 "YHL_Utils.h"
//#import "YH_LiveGlobalData.h"
#import "YH_NetworkAdapter+Live.h"

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

@interface YH_SocketService ()<GCDAsyncSocketDelegate>
{
    BOOL _hasLoginWithNameSuccessed;
    BOOL _hasLoginWithDevieToken;
    BOOL _needSendStartPackage;
}
@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) {
        _needSendStartPackage = YES;
        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{
    
    NSDictionary *params = @{@"cmd":@(SOCKET_TAG_HEARTBEAT),
                             @"msg":@"heart_ios",
                             @"room":MakeStringNotNil(self.room),
                             };
    [_socket writeDataWithParams:params tag:SOCKET_TAG_HEARTBEAT];

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

#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 (_needSendStartPackage) {
        NSDictionary *params = @{@"cmd":@(SOCKET_TAG_Start),
                                 @"room":self.room?:@"666",
                                 @"msg":@"",
                                 @"uid":@""
                                 };
        [_socket writeDataWithParams:params tag:SOCKET_TAG_Start];
    }
    dispatch_async(dispatch_get_main_queue(), ^{
        [self heartbeat];
    });
    [self listenData];
}



- (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;
         
        case SOCKET_TAG_Start:
        {
            [self loginToServer];
            _needSendStartPackage = NO;
        }
            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];
            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:
                {
                    [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:resDic[@"onlineNums"]];
                        });
                    }
                }
                    break;
                case SOCKET_TAG_NotifyEndPlay:
                {
                    if ([self.delegate respondsToSelector:@selector(livePalyEnd:likeNums:videoLen:)]) {
                        yh_dispatch_execute_in_main_queue(^{
                           [self.delegate livePalyEnd:resDic[@"audienceNums"] likeNums:resDic[@"likeNums"] videoLen:resDic[@"videoLen"]];
                        });
                    }
                }
                    break;
                case SOCKET_TAG_NotifyOnlineAndLikes://登录成功返回在线人数和点赞数
                {
                    _hasLoginWithDevieToken = YES;
                    if ([self.delegate respondsToSelector:@selector(liveOnlineNums:likes:)]) {
                        yh_dispatch_execute_in_main_queue(^{
                            [self.delegate liveOnlineNums:resDic[@"onlineNums"] likes:resDic[@"likeNums"]];
                        });
                    }
                }
                    break;
                case SOCKET_TAG_NotifyEndSuccess:
                {
                    if ([self.delegate respondsToSelector:@selector(liveNotifyEndSuccess)]) {
                        yh_dispatch_execute_in_main_queue(^{
                            [self.delegate liveNotifyEndSuccess];
                        });
                    }
                }
                    break;
                    
                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;
    }
    [NSObject cancelPreviousPerformRequestsWithTarget:self];
    _needSendStartPackage = YES;
    [_socket disconnect];
}

//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,
                             };
    
    [_socket writeDataWithParams:params tag:SOCKET_TAG_LOGIN];

}

//testOK
- (void)sendMessage:(NSString *)msg
{
    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];
}

//testOK
- (void)userPraise
{
    NSDictionary *params = @{@"cmd":@(SOCKET_TAG_PRAISE),
                             @"uid":MakeStringNotNil(self.uid),
                             @"msg":@"",
                             @"room":MakeStringNotNil(self.room),
                             @"deviceToken":self.deviceToken,
                             };
    [_socket writeDataWithParams:params tag:SOCKET_TAG_SEND_MSG];
}

//结束播放//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