/*
 *  Copyright (c) 2018-present, Facebook, Inc.
 *
 *  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