klx_teacher/KulexiuForTeacher/KulexiuForTeacher/Common/Tools/KSUploadManager.m

//
//  KSUploadManager.m
//  KulexiuForTeacher
//
//  Created by 王智 on 2022/5/8.
//

#import "KSUploadManager.h"
#import "NSDate+Extension.h"
#import <QCloudCOSXML/QCloudCOSXMLTransfer.h>

@interface KSUploadManager ()<QCloudSignatureProvider>


@property (assign, nonatomic) long long fileSize;

// 腾讯cos bucket name
@property (nonatomic, strong) NSString *cosBucketName;


@property (nonatomic, strong) NSString *cosFileName;

@end

@implementation KSUploadManager
+ (instancetype)shareInstance {
    static KSUploadManager *manager = nil;
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        manager = [[KSUploadManager alloc] init];
        [manager configCilentBucket];
        [manager configTXCloud];
    });
    return manager;
}

- (void)configTXCloud {
    QCloudServiceConfiguration *configuration = [QCloudServiceConfiguration new];
    QCloudCOSXMLEndPoint *endpoint = [[QCloudCOSXMLEndPoint alloc] initWithLiteralURL:[NSURL URLWithString:@"oss.dayaedu.com"]];
    endpoint.regionName = @"ap-nanjing";
    // 适使用HTTPS
    endpoint.useHTTPS = YES;
    configuration.endpoint = endpoint;
    configuration.signatureProvider = self;
    
    // 初始化 cos 服务实例
    [QCloudCOSXMLService registerDefaultCOSXMLWithConfiguration:configuration];
    [QCloudCOSTransferMangerService registerDefaultCOSTransferMangerWithConfiguration:configuration];
}

- (void)configBucketName:(NSString *)bucketName {

}

- (void)configCilentBucket {
    self.cosBucketName = @"daya-online-1303457149";
}

// 上传改造 添加目录 /yyyyMM/dd/文件名
- (NSString *)getDayFolder {
    NSDateFormatter *formatter = [NSObject getDateformatter];
    [formatter setDateFormat:@"yyyyMM"];
    NSDate *date = [NSDate date];
    NSString *month = [formatter stringFromDate:date];
    [formatter setDateFormat:@"dd"];
    NSString *dayString = [formatter stringFromDate:date];
    return [NSString stringWithFormat:@"%@/%@/",month, dayString];
}

- (void)uploadImage:(NSData *)imageData fileName:(NSString *)fileName successCallback:(KSUploadSuccess)success faliure:(KSUploadFailer)faliure {

    NSString *folderString = [self getDayFolder];
    NSString *uploadFileName = [NSString stringWithFormat:@"%@%@%@_%@%@%@",FILE_DOMAIN,folderString,[NSDate getCurrentTimestampFor64], UserDefault(UIDKey), fileName,[UIImage typeForImageData:imageData]];

    [self uploadFileToOssWithSource:uploadFileName fileData:imageData progress:^(int64_t bytesWritten, int64_t totalBytes) {
        
    } successCallback:^(NSString *fileUrl) {
        NSMutableArray *fileUrlArray = [NSMutableArray array];
        [fileUrlArray addObject:fileUrl];
        if (success) {
            success(fileUrlArray);
        }
    } faliure:^{
        if (faliure) {
            faliure(nil, @"上传文件失败");
        }
    }];
}

- (void)mutilUploadImage:(NSMutableArray *)fileDataArray fileName:(NSString *)fileName successCallback:(KSUploadSuccess)success faliure:(KSUploadFailer)faliure {

    NSString *folderString = [self getDayFolder];
    NSMutableArray *fileNameArray = [NSMutableArray array];
    if (fileDataArray.count > 1) {
        for (NSInteger i = 0 ; i < fileDataArray.count; i++) {
            NSData *fileData = fileDataArray[i];
            NSString *uploadFileName = [NSString stringWithFormat:@"%@%@%@_%@%@%zd%@",FILE_DOMAIN,folderString,[NSDate getCurrentTimestampFor64], UserDefault(UIDKey), fileName,i,[UIImage typeForImageData:fileData]];
            [fileNameArray addObject:uploadFileName];
        }
    }
    else {
        NSData *fileData = fileDataArray[0];
        NSString *uploadFileName = [NSString stringWithFormat:@"%@%@%@_%@%@%@",FILE_DOMAIN,folderString,[NSDate getCurrentTimestampFor64], UserDefault(UIDKey), fileName,[UIImage typeForImageData:fileData]];
        [fileNameArray addObject:uploadFileName];
    }
    [self MutilUploadFile:fileNameArray fileDataArray:fileDataArray successArray:[NSMutableArray array] failArray:[NSMutableArray array] fileIndex:0 successCallback:success faliure:faliure];
}

- (void)MutilUploadFile:(NSMutableArray *)fileNameArray fileDataArray:(NSMutableArray *)fileDataArray successArray:(NSMutableArray *)successArray failArray:(NSMutableArray *)failArray fileIndex:(NSInteger)fileIndex successCallback:(KSUploadSuccess)successCallback faliure:(KSUploadFailer)faliureCallback {
    if (fileIndex >= fileNameArray.count) {
        return;
    }
    NSString *uploadFileName = fileNameArray[fileIndex];
    NSData *fileData = fileDataArray[fileIndex];
    [self uploadFileToOssWithSource:uploadFileName fileData:fileData progress:^(int64_t bytesWritten, int64_t totalBytes) {
        
    } successCallback:^(NSString *fileUrl) {
        
        NSMutableArray *success = [NSMutableArray arrayWithArray:successArray];
        [success addObject:fileUrl];
        if (fileIndex == fileNameArray.count - 1) {
            [self finishMutilUpload:success failArray:failArray successCallback:successCallback faliure:faliureCallback];
        }
        else {
            [self MutilUploadFile:fileNameArray fileDataArray:fileDataArray successArray:success failArray:failArray fileIndex:fileIndex+1 successCallback:successCallback faliure:faliureCallback];
        }
        
    } faliure:^{
        
        NSString *desc = [NSString stringWithFormat:@"第%zd次上传文件失败",fileIndex];
        NSMutableArray *failerArray = [NSMutableArray arrayWithArray:failArray];
        [failerArray addObject:desc];
        
        if (fileIndex == fileNameArray.count - 1) {
            [self finishMutilUpload:successArray failArray:failerArray successCallback:successCallback faliure:faliureCallback];
        }
        else {
            [self MutilUploadFile:fileNameArray fileDataArray:fileDataArray successArray:successArray failArray:failerArray fileIndex:fileIndex+1 successCallback:successCallback faliure:faliureCallback];
        }
    }];
}

- (void)finishMutilUpload:(NSMutableArray *)successArray failArray:(NSMutableArray *)array successCallback:(KSUploadSuccess)successCallback faliure:(KSUploadFailer)faliureCallback {
    if (successArray.count > 0) {
        if (successCallback) {
            successCallback([successArray copy]);
        }
    }
    else {
        if (faliureCallback) {
            faliureCallback(nil,@"上传文件失败");
        }
    }
}


- (void)videoUpload:(NSData *)fileData fileName:(NSString *)fileName fileSuffix:(NSString *)fileSuffix progress:(KSUploadProgress)uploadProgress successCallback:(KSUploadSuccess)success faliure:(KSUploadFailer)faliure {

    NSString *folderString = [self getDayFolder];
    NSString *uploadFileName = [NSString stringWithFormat:@"%@%@%@_%@%@%@",FILE_DOMAIN,folderString,[NSDate getCurrentTimestampFor64], UserDefault(UIDKey),fileName,fileSuffix];

    [self uploadFileToOssWithSource:uploadFileName fileData:fileData progress:^(int64_t bytesWritten, int64_t totalBytes) {
        if (uploadProgress) {
            uploadProgress(bytesWritten, totalBytes);
        }
    } successCallback:^(NSString *fileUrl) {
        NSMutableArray *fileUrlArray = [NSMutableArray array];
        [fileUrlArray addObject:fileUrl];
        if (success) {
            success(fileUrlArray);
        }
    } faliure:^{
        if (faliure) {
            faliure(nil, @"上传文件失败");
        }
    }];
}

- (void)uploadFile:(NSData *)fileData fileName:(NSString *)fileName fileSuffix:(NSString *)fileSuffix progress:(KSUploadProgress)uploadProgress successCallback:(KSUploadSuccess)success faliure:(KSUploadFailer)faliure {

    NSString *folderString = [self getDayFolder];
    NSString *uploadFileName = [NSString stringWithFormat:@"%@%@%@_%@%@%@",FILE_DOMAIN,folderString,[NSDate getCurrentTimestampFor64],UserDefault(UIDKey), fileName,fileSuffix];
    
    [self uploadFileToOssWithSource:uploadFileName fileData:fileData progress:^(int64_t bytesWritten, int64_t totalBytes) {
        if (uploadProgress) {
            uploadProgress(bytesWritten, totalBytes);
        }
    } successCallback:^(NSString *fileUrl) {
        NSMutableArray *fileUrlArray = [NSMutableArray array];
        [fileUrlArray addObject:fileUrl];
        if (success) {
            success(fileUrlArray);
        }
    } faliure:^{
        if (faliure) {
            faliure(nil, @"上传文件失败");
        }
    }];
}


- (void)uploadFileToOssWithSource:(NSString *)fileName fileData:(NSData *)fileData progress:(KSUploadProgress)uploadProgress successCallback:(void(^)(NSString *fileUrl))success faliure:(void(^)(void))faliure {
    if (fileData == nil) {
        if (faliure) {
            dispatch_main_async_safe(^{
                faliure();
            });
        }
        return;
    }
    self.cosFileName = fileName;
    QCloudCOSXMLUploadObjectRequest *put = [QCloudCOSXMLUploadObjectRequest new];
    put.bucket = self.cosBucketName;
    put.object = fileName;
    put.body = fileData;
    
    [put setSendProcessBlock:^(int64_t bytesSent, int64_t totalBytesSent, int64_t totalBytesExpectedToSend) {
            // 进度
        dispatch_main_async_safe(^{
            if (uploadProgress) {
                uploadProgress(totalBytesSent, totalBytesExpectedToSend);
            }
        });
    }];
    // 监听上传结果
    [put setFinishBlock:^(QCloudUploadObjectResult * _Nullable result, NSError * _Nullable error) {
        if (result) {
            NSString * fileUrl = result.location;
            if (success) {
                dispatch_main_async_safe(^{
                    success(fileUrl);
                });
                
            }
        }
        else {
            if (faliure) {
                dispatch_main_async_safe(^{
                    faliure();
                });
            }
        }
    }];
    
    [[QCloudCOSTransferMangerService defaultCOSTransferManager] UploadObject:put];
}

- (void)signatureWithFields:(QCloudSignatureFields *)fileds request:(QCloudBizHTTPRequest *)request urlRequest:(NSMutableURLRequest *)urlRequst compelete:(QCloudHTTPAuthentationContinueBlock)continueBlock {
    [KSNetworkingManager getTXCosUploadSignRequest:KS_POST fileName:self.cosFileName keyName:self.cosFileName bucketName:self.cosBucketName success:^(NSDictionary * _Nonnull dic) {
        if ([dic ks_integerValueForKey:@"code"] == 200) {
            NSString *signatureString = [[dic ks_dictionaryValueForKey:@"data"] ks_stringValueForKey:@"signature"];
//            NSString *
//            QCloudCredential* credential = [QCloudCredential new];
//            credential.secretID = @"AKIDSo12YszsjQwKdkbFjBThokdO6HI7lFbq";
//            credential.secretKey = @"5ykLhplZDhLo7E7rQnDSWQ9eyAAyhP0T";
//            QCloudAuthentationV5Creator* creator = [[QCloudAuthentationV5Creator alloc]
//                                                    initWithCredential:credential];
            // 注意 这里不要对urlRequst 进行copy以及mutableCopy操作
            
//            NSTimeInterval endTime = [[[dic ks_dictionaryValueForKey:@"data"] ks_stringValueForKey:@"expiration"] longLongValue];
//            NSDate *expiration = [NSDate dateWithTimeIntervalSince1970:endTime/1000.0];
            QCloudSignature *signature = [[QCloudSignature alloc] initWithSignature:signatureString expiration:nil];
//            QCloudSignature *signature = [creator signatureForData:urlRequst];
            continueBlock(signature, nil);
        }
        else {
            continueBlock(nil, nil);
        }
    } faliure:^(NSError * _Nonnull error) {
        continueBlock(nil, nil);
    }];
}

@end