79023ee190
Reviewed By: bhamodi Differential Revision: D33331422 fbshipit-source-id: 016e8dcc0c0c7f1fc353a348b54fda0d5e2ddc01
128 lines
3.7 KiB
Plaintext
128 lines
3.7 KiB
Plaintext
/*
|
|
* 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.
|
|
*/
|
|
|
|
#if FB_SONARKIT_ENABLED
|
|
#import "FlipperKitNetworkPlugin.h"
|
|
#import <iostream>
|
|
#import <memory>
|
|
#import <vector>
|
|
#import "SKBufferingPlugin+CPPInitialization.h"
|
|
#import "SKDispatchQueue.h"
|
|
#import "SKNetworkReporter.h"
|
|
#import "SonarKitNetworkPlugin+CPPInitialization.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 {
|
|
// Only track HTTP(S) calls, data URLs cannot be casted to NSHTTPURLResponse
|
|
if (![response.response isKindOfClass:[NSHTTPURLResponse class]]) {
|
|
return;
|
|
}
|
|
|
|
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
|