79023ee190
Reviewed By: bhamodi Differential Revision: D33331422 fbshipit-source-id: 016e8dcc0c0c7f1fc353a348b54fda0d5e2ddc01
54 lines
1.5 KiB
Objective-C
54 lines
1.5 KiB
Objective-C
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
#import "SKResponseInfo.h"
|
|
|
|
@implementation SKResponseInfo
|
|
@synthesize identifier = _identifier;
|
|
@synthesize timestamp = _timestamp;
|
|
@synthesize response = _response;
|
|
@synthesize body = _body;
|
|
|
|
- (instancetype)initWithIndentifier:(int64_t)identifier
|
|
timestamp:(uint64_t)timestamp
|
|
response:(NSURLResponse*)response
|
|
data:(NSData*)data {
|
|
if (self = [super init]) {
|
|
_identifier = identifier;
|
|
_timestamp = timestamp;
|
|
_response = response;
|
|
_body = [SKResponseInfo shouldStripReponseBodyWithResponse:response]
|
|
? nil
|
|
: [data base64EncodedStringWithOptions:0];
|
|
}
|
|
return self;
|
|
}
|
|
|
|
+ (BOOL)shouldStripReponseBodyWithResponse:(NSURLResponse*)response {
|
|
// Only HTTP(S) responses have Content-Type headers
|
|
if (![response isKindOfClass:[NSHTTPURLResponse class]]) {
|
|
return YES;
|
|
}
|
|
|
|
NSHTTPURLResponse* httpResponse = (NSHTTPURLResponse*)response;
|
|
NSString* contentType = httpResponse.allHeaderFields[@"content-type"];
|
|
if (!contentType) {
|
|
return NO;
|
|
}
|
|
|
|
return [contentType containsString:@"video/"] ||
|
|
[contentType containsString:@"application/zip"];
|
|
}
|
|
|
|
- (void)setBodyFromData:(NSData* _Nullable)data {
|
|
self.body = [SKResponseInfo shouldStripReponseBodyWithResponse:self.response]
|
|
? nil
|
|
: [data base64EncodedStringWithOptions:0];
|
|
}
|
|
|
|
@end
|