PrivateBoxiOS/创维盒子/code/cloudPhone/websocket/webSocketManager+upLoadFile.m

//
//  webSocketManager+upLoadFile.m
//  双子星云手机
//
//  Created by xd h on 2024/6/20.
//

#import "webSocketManager+upLoadFile.h"

@implementation webSocketManager (upLoadFile)
#pragma mark 监听文件开始的的通知
- (void)beginGotoUploadFileFunBy:(uploadFileDataModel *)uploadFileDataMod
{
    self.isSuspendUploadType = NO;
    
    //HLog(@"%@",uploadFileDataMod);

    self.curUploadFileDataMod = uploadFileDataMod;
    
    if(![uploadFileDataMod isKindOfClass:[uploadFileDataMod class]]){
        HLog(@"上传的文件类型不对");
        return;
    }
    
    [self ApplyUploadFileFun];
}

- (void)applyUploadFileServiceResponseFun:(NSDictionary *)dataDict
{
    if([dataDict isKindOfClass:[NSDictionary class]]){
        if ([[dataDict allKeys] containsObject:@"data"]) {
            NSDictionary *data = dataDict[@"data"];

            if ([[data allKeys] containsObject:@"status"]) {
                NSString  *status = [data objectForKey:@"status"];
                if(![status isKindOfClass:[NSString class]]){
                    status = [[NSString  alloc] initWithFormat:@"%@",status];
                }
                
                //if(1)
                if([status isEqualToString:@"1"])
                {
                    //[self tryAgainApplyUploadFun];
                    [self clearUploadFileDataFun];
                    [[uploadFileManager shareInstance] uploadFileFailFun];
                    
                }else if ([status isEqualToString:@"0"]){
                    if ([[data allKeys] containsObject:@"taskUid"]) {
                        NSString *taskUid = data[@"taskUid"];
                        if(![status isKindOfClass:[NSString class]]){
                            taskUid = [[NSString  alloc] initWithFormat:@"%@",taskUid];
                        }
                        
                        if(taskUid &&  [taskUid isEqualToString:self.taskUid]){
                            
                            NSNumber * fileLengthNum = @0;
                            if ([[data allKeys] containsObject:@"fileLength"])
                            {
                                fileLengthNum = data[@"fileLength"];
                            }
                            
                        //    [[uploadFileManager shareInstance] changeUploadFileState:uploadStateUploading withDidUploadBytes:[fileLengthNum integerValue] withModel:self.curUploadFileDataMod complete:^(BOOL isSuccess) {
                        //
                        //    }];
                            
                            if([fileLengthNum integerValue] == self.curUploadFileDataMod.totalBytes && self.curUploadFileDataMod.totalBytes != 0){
                                self.taskUid = nil;
                                
                                //self.curUploadFileDataMod.curUploadStateType = uploadStateDone;
                                
                                NSMutableArray *delArr = [NSMutableArray new];
                                [delArr addObject:self.curUploadFileDataMod];
                                [[uploadFileManager shareInstance] deleteUploadFileRecordBy:delArr withDelCache:NO complete:^(BOOL isSuccess) {
                                    if (isSuccess) {
                                        
                                    }
                                }];
                                
                                mainBlock(^{
                                    [[iToast makeText:NSLocalizedString(@"File_upload_file_already_exists",nil)] show];
                                });
                                
                                HLog(@"hxd already 上传文件已存在")
                                return;
                            }
                            
                            self.curUploadPostion = [fileLengthNum longValue];
                            
                            if(self.curUploadFileDataMod.curUploadFileType == uploadFileTypeImage){
                                [self beginGotoUploadImageFun:fileLengthNum];
                            }
                            else{
                                [self beginGotoUploadVidelFun:fileLengthNum];
                            }
                        }
                    }
                   
                }
            }
        }
        else{
            //异常
            [self tryAgainApplyUploadFun];
        }
    }
    else{
        //异常
        [self tryAgainApplyUploadFun];
    }
}


#pragma mark 图片上传
-(void)beginGotoUploadImageFun:(NSNumber*)fileLengthNum{
    
    self.cutFileDataArr = [self fileCutPartsBy:self.AllFileData withDidUpLoadSize:[fileLengthNum integerValue]];
    
    if(!self.cutFileDataArr || self.cutFileDataArr.count == 0){
        [self clearUploadFileDataFun];
        [[uploadFileManager shareInstance] uploadFileFailFun];
        
        HLog(@"hxd fail 上传任务异常终止")
        return;
    }
    
    //答复的taskUid 跟申请的一致 开发发送内容
    [self upLoadFileFun];
}


#pragma mark 视频上传
//每次从文件重新切一片 传一片
-(void)beginGotoUploadVidelFun:(NSNumber*)fileLengthNum{
    
    NSData *data = [self cutVideoFileFunAtIndex:fileLengthNum.longLongValue];
    HLog(@"data leng:%ld",[data length]);

    if([data length] == 0){
        if(self.curUploadFileDataMod.curUploadStateType == uploadStateDone){
            [[uploadFileManager shareInstance] uploadFileDoneFun];
        }
        else{
            [[uploadFileManager shareInstance] uploadFileFailFun];
        }
        
        [self clearUploadFileDataFun];
        
        return;
    }
    
    self.AllFileData = data;
    self.cutFileDataArr = [self fileCutPartsBy:self.AllFileData withDidUpLoadSize:0];
    self.indexOfUploadFlie = 0;
    
    //return;

//    NSString*videoPathStr = [cachesFileManager getFilePathWithName:self.curUploadFileDataMod.filename type:uploadFileTypeVideo];
//    self.AllFileData = [NSData dataWithContentsOfFile:videoPathStr];

   // self.cutFileDataArr = [self fileCutPartsBy:self.AllFileData withDidUpLoadSize:[fileLengthNum integerValue]];
    
    if(!self.cutFileDataArr ||self.cutFileDataArr.count ==0){
        [self clearUploadFileDataFun];
        [[uploadFileManager shareInstance] uploadFileFailFun];
        
        HLog(@"hxd fail 上传任务异常终止")
        return;
    }
    
    //答复的taskUid 跟申请的一致 开发发送内容
    [self upLoadFileFun];
}

#pragma mark 分段读视频文件
-(NSData*)cutVideoFileFunAtIndex:(NSUInteger)dataIndex{
    NSString *filePath = [cachesFileManager getFilePathWithName:self.curUploadFileDataMod.filename type:uploadFileTypeVideo]; // 文件路径
    
    NSFileManager *manager0 = [NSFileManager defaultManager];
    if(![manager0 fileExistsAtPath:filePath]) {
     
        return [NSData new];
    }
    
    NSFileHandle *fileHandle = [NSFileHandle fileHandleForReadingAtPath:filePath]; // 创建文件句柄
      
    // 设置分段读取的大小，这里以每次读取1KB为例
    //const NSUInteger chunkSize = 50 * 1024 *1024;
    const NSUInteger chunkSize = cutVideoPieceSzie;//5 * 1024 *1024;
    NSMutableData *data = [NSMutableData data];
    
    if (fileHandle) {
       
        long long endOfFile = [fileHandle seekToEndOfFile];
        
        if(self.curUploadFileDataMod.totalBytes == 0
           || self.curUploadFileDataMod.totalBytes < endOfFile){//异常处理
            self.curUploadFileDataMod.totalBytes = endOfFile;
        }
        
        //异常处理
        if(endOfFile == dataIndex){
            self.curUploadFileDataMod.totalBytes = endOfFile;
            self.curUploadFileDataMod.didUploadBytes = endOfFile;
            self.curUploadFileDataMod.curUploadStateType = uploadStateDone;
            [fileHandle closeFile];
            return  data;
        }
        
        if (endOfFile >=  chunkSize) {
            
            // 读取文件的分段数据到某个位置
            [fileHandle seekToFileOffset:dataIndex];
            
            // 读取文件的分段数据
            NSData* chunk = [fileHandle readDataOfLength:chunkSize];
            if (chunk) {
                [data appendData:chunk];
            }
        }
        else{
            // 读取文件的分段数据到某个位置
            [fileHandle seekToFileOffset:dataIndex];
            
            [data appendData:[fileHandle readDataToEndOfFile]];
        }
        
        // 在这里可以对文件内容进行处理
        // ...
        
        // 关闭文件句柄
        [fileHandle closeFile];
    }
    
    return data;
}


- (void)upLoadFileFunServiceResponseFun:(NSDictionary *)dataDict
{
    //{"type":"uploadFileRandomRet","data":{"progress":"9\/10","filePath":"/sdcard/Download/IMG_0036.JPG","status":"0","msg":"success"}}
    
    if([dataDict isKindOfClass:[NSDictionary class]]){
        if ([[dataDict allKeys] containsObject:@"data"]) {
            NSDictionary *data = dataDict[@"data"];
            
            if ([[data allKeys] containsObject:@"taskId"]){
                NSString *taskId = data[@"taskId"];
                if(![taskId isEqualToString:self.taskUid]){
                    return;
                }
            }
            
            if ([[data allKeys] containsObject:@"status"]) {
                NSString  *status = [data objectForKey:@"status"];
                if(![status isKindOfClass:[NSString class]]){
                    status = [[NSString  alloc] initWithFormat:@"%@",status];
                }
                
                 if ([status isEqualToString:@"0"]){
                    
                    NSNumber * fileLengthNum = @0;
                    if ([[data allKeys] containsObject:@"fileLenth"])
                    {
                        fileLengthNum = data[@"fileLenth"];
                    }
                    
                     if (fileLengthNum && fileLengthNum.longLongValue >= self.curUploadFileDataMod.totalBytes)
                     {//
                         //告诉代理 完成了
                         NSString *commandStr = [RCCommandHelp uploadOrBackupsDoneBy:self.taskUid withFileName:self.fileName withType:YES];
                        
                         //[self.commandChannelManager rc_sendData:commandStr];
                         
                         [self clearUploadFileDataFun];
                         [[uploadFileManager shareInstance] uploadFileDoneFun];
                     }
                     else{
                         //异常处理
                         if([fileLengthNum longValue] - self.curUploadPostion > EachPieceSzie){
                             HLog(@"发现上传后返回的长度 出错");
                             
                             if(self.curUploadFileDataMod.curUploadFileType == uploadFileTypeVideo)
                             {//清除已经取出来的数据  后面会根据fileLengthNum 重新拿数据
                                 self.cutFileDataArr = [NSMutableArray new];
                             }
                             else{
                                 NSInteger curIndex = [fileLengthNum longValue] % EachPieceSzie;
                                 //把当前的上传位置指定已经上传的前一片，后续++
                                 self.indexOfUploadFlie = curIndex -1;
                             }
                         }
                         
                         self.curUploadPostion = [fileLengthNum longValue];
                         [self handleUploadingFun:fileLengthNum];
                     }

                }
                 else if ([status isEqualToString:@"2"]) {
                     [self clearUploadFileDataFun];
                     [[uploadFileManager shareInstance] suspendUploadFileFun:YES withModel:nil];
                     
                     NSString *tipStr = NSLocalizedString(@"disk_not_find_tip",nil);
                     mainBlock(^{
                         [[iToast makeText:tipStr] show];
                     });
                 }
                 else if ([status isEqualToString:@"5"]) {
                     [self clearUploadFileDataFun];
                     [[uploadFileManager shareInstance] suspendUploadFileFun:YES withModel:nil];
                     
                     NSString *tipStr = NSLocalizedString(@"disk_space_not_tip",nil);
                     mainBlock(^{
                         [[iToast makeText:tipStr] show];
                     });
                 }
                else //if ([status isEqualToString:@"1"])
                {
                    //[self tryAgainApplyUploadFun];
                    [self clearUploadFileDataFun];
                    [[uploadFileManager shareInstance] uploadFileFailFun];
                }
            }
        }
        else{
            //异常
            [self tryAgainApplyUploadFun];
        }
    }
    else{
        //异常
        [self tryAgainApplyUploadFun];
    }
    
}

#pragma mark 申请上传文件
- (void)ApplyUploadFileFun
{
    //if(!self.taskUid){
        self.taskUid = [iTools getTaskUidStr];
    
        if([self.taskUid isEqualToString:self.downloadFileTaskUid]
           ||[self.taskUid isEqualToString:self.backupsTaskUid]
           ||[self.taskUid isEqualToString:self.downloadThumbnailTaskUid])
        {//重新生成taskUid
            [self ApplyUploadFileFun];
            return;
        }
    
        self.fileName = self.curUploadFileDataMod.filename;
        if(self.curUploadFileDataMod.curUploadFileType == uploadFileTypeVideo){
            //self.AllFileData = self.curUploadFileDataMod.videoData;
        }
        else{
            self.AllFileData = self.curUploadFileDataMod.imageData;
        }
        
        self.indexOfUploadFlie = 0;
        
        //HLog(@"%@  %@",self.taskUid,self.fileName);
        //self.cutFileDataArr = [self fileCutPartsBy:self.AllFileData];
    //}
    
    NSString * commandStr = [RCCommandHelp applyForUploadFileBy:self.taskUid  filePath:self.fileName];
   // [self.commandChannelManager rc_sendData:commandStr];
}

#pragma mark 遇到异常时 重新上传
- (void)tryAgainApplyUploadFun
{
    self.taskUid = nil;
    [self ApplyUploadFileFun];
}

#pragma mark 处理上传过程
- (void)handleUploadingFun:(NSNumber*)fileLengthNum
{
    //异常处理
    if(fileLengthNum.longValue < 0)
    {
        return;
    }
    
    BOOL isCanUseCellular = [HWDataManager getBoolWithKey:stringKeyAddSn(Const_file_Transfe_canUse_Cellular_all)];
    
    if(!isCanUseCellular){//不允许流量备份
        //
        if([AFNetworkReachabilityManager sharedManager].networkReachabilityStatus == AFNetworkReachabilityStatusReachableViaWWAN){
            self.isSuspendUploadType = YES;
        }
        
    }
    
    if(self.isSuspendUploadType){
        //暂停
        self.taskUid = nil;
        //NSInteger didUploadSize = (self.indexOfUploadFlie+1) * (self.cutFileEachPieceSzie);
        NSInteger didUploadSize = fileLengthNum.longLongValue;
        [[uploadFileManager shareInstance] changeUploadFileState:uploadStateSuspend withDidUploadBytes:didUploadSize  withModel:self.curUploadFileDataMod complete:^(BOOL isSuccess) {

        }];
        
        return;
    }
    
    if(self.cutFileDataArr.count > 0
       && self.indexOfUploadFlie < (self.cutFileDataArr.count -1)) {
        
        //NSInteger didUploadSize = (self.indexOfUploadFlie+1) * (self.cutFileEachPieceSzie);
        NSInteger didUploadSize = fileLengthNum.longLongValue;
        
        [[uploadFileManager shareInstance] changeUploadFileState:uploadStateUploading withDidUploadBytes:didUploadSize withModel:self.curUploadFileDataMod complete:^(BOOL isSuccess) {

        }];
        
        self.indexOfUploadFlie ++;
        [self upLoadFileFun];
    }
    else if(self.curUploadFileDataMod.curUploadFileType == uploadFileTypeVideo && fileLengthNum && fileLengthNum.longLongValue < self.curUploadFileDataMod.totalBytes){
        //视频做了二次切割 不一定上传完了
        HLog(@"视频上传传完一片 再来一片");
        [self beginGotoUploadVidelFun:fileLengthNum];
    }
    else{
        HLog(@"上传完成");
        
        //告诉代理 完成了
        NSString *commandStr = [RCCommandHelp uploadOrBackupsDoneBy:self.taskUid withFileName:self.fileName withType:YES];
       
        
        //[self.commandChannelManager rc_sendData:commandStr];
        
        [self clearUploadFileDataFun];
        [[uploadFileManager shareInstance] uploadFileDoneFun];
    }
}

-(void)clearUploadFileDataFun{
    //上传数据清空
    self.taskUid = nil;
    self.fileName = nil;
    self.AllFileData = nil;
    self.indexOfUploadFlie = 0;
    self.cutFileDataArr = nil;
    self.curUploadFileDataMod = nil;
    self.isSuspendUploadType = NO;
    
}

- (Byte)getBCC:(NSData *)data {
    Byte bcc = 0;
    unsigned char *bytes = (unsigned char *)[data bytes];
    for (int i = 0; i < [data length]; i++) {
        bcc = (Byte)(bcc ^ bytes[i]);
    }
    return bcc;
}


#pragma mark 文件切片
-(NSMutableArray *)fileCutPartsBy:(NSData*)curData withDidUpLoadSize:(NSInteger)didLoadSize{
 
    NSData * data = curData;
    
    NSUInteger allLength = data.length;
    
    NSUInteger subs = EachPieceSzie;//1*1024*1024;//4096;//要切片的大小，我这里设置的是4096字节
    //NSUInteger subs = 31*1024;//4096;//要切片的大小，内存没有暴增
    
    self.cutFileEachPieceSzie = subs;
    
    NSInteger index = 0;//起始位置
    
    if(didLoadSize != 0){
        if (didLoadSize % subs != 0){
            //出错了
            return nil;
        }
        else{
//            index = didLoadSize/subs;
//            allLength = allLength - didLoadSize;
            
            self.indexOfUploadFlie = didLoadSize/subs;
        }
    }

    NSMutableArray *dataArray =[NSMutableArray new];
    
    do {
        //@autoreleasepool {
            if (allLength>subs) {
                
                NSRange range =NSMakeRange(index*subs, subs);
                
                index++;
                
                //NSLog(@"%@",NSStringFromRange(range));
                
                [dataArray addObject:[data subdataWithRange:range]];
                
                allLength = allLength - subs;
                
            }else{
                NSRange range = NSMakeRange(index*subs, allLength);
                
                //NSLog(@"%@",NSStringFromRange(range));
                NSData *subdata = [data subdataWithRange:range];
                
                if(subdata){
                    [dataArray addObject:subdata];
                }
                
                allLength = 0;
            }
        //}
        
    } while (allLength>0);
    
    HLog(@"dataArray leng:%ld",dataArray.count);//最后得到切片的结果，数组里面是NSData对象

    return dataArray;
}

// int转NSData
- (NSData *)int2Data:(int)i{
    Byte b0 = i & 0xff;
    Byte b1 = i >> 8 & 0xff;
    Byte b2 = i >> 16 & 0xff;
    Byte b3 = i >> 24 & 0xff;
    // 有大小端模式问题？
    //Byte result[] = {b0, b1, b2, b3};
    Byte result[] = {b3, b2, b1, b0};
    return [NSData dataWithBytes:result length:sizeof(result)];
}

// long转NSData
- (NSData *)long2Data:(long)i{
    Byte b0 = i & 0xff;
    Byte b1 = i >> 8 & 0xff;
    Byte b2 = i >> 16 & 0xff;
    Byte b3 = i >> 24 & 0xff;
    Byte b4 = i >> 32 & 0xff;
    Byte b5 = i >> 40 & 0xff;
    Byte b6 = i >> 48 & 0xff;
    Byte b7 = i >> 56 & 0xff;
    
    // 有大小端模式问题？
    //Byte result[] = {b0, b1, b2, b3};
    //Byte result[] = {b3, b2, b1, b0};
    Byte result[] = {b7, b6, b5,b4,b3, b2, b1, b0};
    return [NSData dataWithBytes:result length:sizeof(result)];
}

// short转NSData
- (NSData *)short2Data:(short)i{
    Byte b0 = i & 0xff;
    Byte b1 = i >> 8 & 0xff;
    // 有大小端模式问题？
    //Byte result[] = {b0, b1, b2, b3};
    Byte result[] = { b1, b0};
    return [NSData dataWithBytes:result length:sizeof(result)];
}

#pragma mark 可以发送文件了
/*
文件数据字节流byte[]
  格式1:下载日志文件压缩包
       |head|msgType|taskId|    datalen            | curNum | totalNum|   data   |bcc校验|end|
  长度：  1      1      6 （curNum+totalNum+data）        4         4      datalen- 8   1     1
       0x68  0x0e                                                        0x16
    bcc校验从第1位 也就是从msgType起 到 校验位前（不包含校验位）数据作bcc校验
 */
- (void)upLoadFileFun
{
    //HLog(@"上传文件%ld/%ld",self.indexOfUploadFlie +1,self.cutFileDataArr.count);
    
    if(self.cutFileDataArr.count <= self.indexOfUploadFlie)
    {
        return;
    }
    
    //data
    NSData *curData = self.cutFileDataArr[self.indexOfUploadFlie];
    //HLog(@"headData:%@",curData);
    
    //head
    unsigned char headNum = 104;
    NSData *headData = [NSMutableData dataWithBytes:&headNum length:sizeof(headNum)];
    //HLog(@"headData:%@",headData);
    
    //msgType
    //unsigned char msgType = 12;
    //unsigned char msgType = 15;
    unsigned char msgType = 20;
    NSData *msgTypeData = [NSMutableData dataWithBytes:&msgType length:sizeof(msgType)];
    //HLog(@"msgTypeData:%@",msgTypeData);
    
    //taskId
    NSData *taskIdData = [self.taskUid dataUsingEncoding:NSUTF8StringEncoding];
    //HLog(@"taskIdData:%@",taskIdData);
    
    //datalen
    //int datalenNum = int([curData length] +(4+4));
    int datalenNum = [curData length] + (4+4);
    //NSData *datalenData = [NSData dataWithBytes:&datalenNum length:sizeof(datalenNum)];
    NSData *datalenData = [self int2Data:datalenNum];
    //HLog(@"datalenData %@",datalenData);
    
    
    //curNum
    int curNum = (int)self.indexOfUploadFlie ;//+ 1;
    //NSData *curNumData = [NSData dataWithBytes:&curNum length:sizeof(curNum)];
    NSData * curNumData = [self int2Data:curNum];
    //HLog(@"curNumData:%@",curNumData);
    
    //totalNum
    int totalNum = (int)self.cutFileDataArr.count;
    //NSData *totalNumData = [NSData dataWithBytes:&totalNum length:sizeof(totalNum)];
    NSData * totalNumData = [self int2Data:totalNum];
    //HLog(@"totalNumData:%@",totalNumData);
    
    long Postion = self.curUploadPostion;
    //long Postion = 100000000000;
    NSData * PostionData = [self long2Data:Postion];
    HLog(@"PostionData:%@",PostionData);

    NSMutableData *checkBCCData = [NSMutableData new];
    [checkBCCData appendData:msgTypeData];
    [checkBCCData appendData:taskIdData];
    [checkBCCData appendData:datalenData];
    [checkBCCData appendData:curNumData];
    [checkBCCData appendData:totalNumData];
    [checkBCCData appendData:PostionData];
    [checkBCCData appendData:curData];
    
    long length = [curData length];
    //HLog(@"curData length %d",length);
    
    Byte bccCheck = [self getBCC:checkBCCData];
    //HLog(@"bcc 000 %d",bccCheck);
    
    //bcc校验
    char bccNum = (char)bccCheck;
    NSData *bccData = [NSMutableData dataWithBytes:&bccNum length:sizeof(bccNum)];
    //HLog(@"%@",bccData);
    
    //end
    unsigned char endNum = 22;
    NSData *endData = [NSMutableData dataWithBytes:&endNum length:sizeof(endNum)];
    //HLog(@"%@",endData);
    
    NSMutableData *sendData = [NSMutableData new];
    [sendData appendData:headData];
    [sendData appendData:checkBCCData];
    [sendData appendData:bccData];
    [sendData appendData:endData];
    
    //HLog(@"sendData length %ld",[sendData length]);
    //[self.commandChannelManager rc_sendData:sendData];
    //HLog(@"文件上传sendData %@",sendData);
}

- (void)suspendUploadFileFun
{
    self.isSuspendUploadType = YES;
}


@end