bbee7a3357
Summary: Add the affiliates bit that the linter is checking for. Reviewed By: jknoxville Differential Revision: D15164826 fbshipit-source-id: 500ffe89ec0c2fd3acfbc374408d16a337cecfa4
120 lines
3.6 KiB
Plaintext
120 lines
3.6 KiB
Plaintext
/*
|
|
* Copyright (c) 2018-present, Facebook, Inc. and its affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the LICENSE
|
|
* file in the root directory of this source tree.
|
|
*
|
|
*/
|
|
#if FB_SONARKIT_ENABLED
|
|
#import <vector>
|
|
#import <iostream>
|
|
#import <memory>
|
|
#import "FlipperKitNetworkPlugin.h"
|
|
#import "SKNetworkReporter.h"
|
|
#import "SonarKitNetworkPlugin+CPPInitialization.h"
|
|
#import "SKBufferingPlugin+CPPInitialization.h"
|
|
#import "SKDispatchQueue.h"
|
|
|
|
@interface FlipperKitNetworkPlugin ()
|
|
|
|
@end
|
|
|
|
@implementation FlipperKitNetworkPlugin
|
|
|
|
- (void)setAdapter:(id<SKNetworkAdapterDelegate>)adapter {
|
|
_adapter = adapter;
|
|
_adapter.delegate = self;
|
|
}
|
|
|
|
- (instancetype)init {
|
|
if (self = [super initWithQueue:dispatch_queue_create("com.sonarkit.network.buffer", DISPATCH_QUEUE_SERIAL)]) {
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (instancetype)initWithNetworkAdapter:(id<SKNetworkAdapterDelegate>)adapter {
|
|
if (self = [super initWithQueue:dispatch_queue_create("com.sonarkit.network.buffer", DISPATCH_QUEUE_SERIAL)]) {
|
|
adapter.delegate = self;
|
|
_adapter = adapter;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (instancetype)initWithNetworkAdapter:(id<SKNetworkAdapterDelegate>)adapter queue:(dispatch_queue_t)queue; {
|
|
if (self = [super initWithQueue:queue]) {
|
|
adapter.delegate = self;
|
|
_adapter = adapter;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
#pragma mark - SKNetworkReporterDelegate
|
|
|
|
|
|
- (void)didObserveRequest:(SKRequestInfo *)request;
|
|
{
|
|
NSMutableArray<NSDictionary<NSString *, id> *> *headers = [NSMutableArray new];
|
|
for (NSString *key in [request.request.allHTTPHeaderFields allKeys]) {
|
|
NSDictionary<NSString *, id> *header = @{
|
|
@"key": key,
|
|
@"value": request.request.allHTTPHeaderFields[key]
|
|
};
|
|
[headers addObject: header];
|
|
}
|
|
|
|
NSString *body = request.body;
|
|
|
|
[self send:@"newRequest"
|
|
sonarObject:@{
|
|
@"id": @(request.identifier),
|
|
@"timestamp": @(request.timestamp),
|
|
@"method": request.request.HTTPMethod ?: [NSNull null],
|
|
@"url": [request.request.URL absoluteString] ?: [NSNull null],
|
|
@"headers": headers,
|
|
@"data": body ? body : [NSNull null]
|
|
}];
|
|
}
|
|
|
|
- (void)didObserveResponse:(SKResponseInfo *)response
|
|
{
|
|
NSHTTPURLResponse *httpResponse = (NSHTTPURLResponse*)response.response;
|
|
|
|
NSMutableArray<NSDictionary<NSString *, id> *> *headers = [NSMutableArray new];
|
|
for (NSString *key in [httpResponse.allHeaderFields allKeys]) {
|
|
NSDictionary<NSString *, id> *header = @{
|
|
@"key": key,
|
|
@"value": httpResponse.allHeaderFields[key]
|
|
};
|
|
[headers addObject: header];
|
|
}
|
|
|
|
NSString *body = response.body;
|
|
|
|
[self send:@"newResponse"
|
|
sonarObject:@{
|
|
@"id": @(response.identifier),
|
|
@"timestamp": @(response.timestamp),
|
|
@"status": @(httpResponse.statusCode),
|
|
@"reason": [NSHTTPURLResponse localizedStringForStatusCode: httpResponse.statusCode] ?: [NSNull null],
|
|
@"headers": headers,
|
|
@"data": body ? body : [NSNull null]
|
|
}];
|
|
|
|
}
|
|
|
|
@end
|
|
|
|
@implementation FlipperKitNetworkPlugin (CPPInitialization)
|
|
|
|
- (instancetype)initWithNetworkAdapter:(id<SKNetworkAdapterDelegate>)adapter dispatchQueue:(std::shared_ptr<facebook::flipper::DispatchQueue>)queue {
|
|
if (self = [super initWithDispatchQueue:queue]) {
|
|
adapter.delegate = self;
|
|
_adapter = adapter;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
@end
|
|
|
|
#endif
|