// Copyright 2015-present 650 Industries. All rights reserved.

#import "EXAppViewController.h"
#import "EXErrorRecoveryManager.h"
#import "EXKernel.h"
#import "EXKernelAppRecord.h"
#import "EXReactAppManager.h"
#import "EXReactAppExceptionHandler.h"
#import "EXUtil.h"

#import <Crashlytics/Crashlytics.h>
#import <React/RCTBridge.h>
#import <React/RCTRedBox.h>

RCTFatalHandler handleFatalReactError = ^(NSError *error) {
  [EXUtil performSynchronouslyOnMainThread:^{
    EXKernelAppRecord *record = [[EXKernel sharedInstance].serviceRegistry.errorRecoveryManager appRecordForError:error];
    if (!record) {
      // show the error on Home or on the main standalone app if we can't figure out who this error belongs to
      if ([EXKernel sharedInstance].appRegistry.homeAppRecord) {
        record = [EXKernel sharedInstance].appRegistry.homeAppRecord;
      } else if ([EXKernel sharedInstance].appRegistry.standaloneAppRecord) {
        record = [EXKernel sharedInstance].appRegistry.standaloneAppRecord;
      }
    }
    if (record) {
      [record.viewController maybeShowError:error];
    }
  }];
};

NS_ASSUME_NONNULL_BEGIN

@interface EXReactAppExceptionHandler ()

@property (nonatomic, weak) EXKernelAppRecord *appRecord;

@end

@implementation EXReactAppExceptionHandler

- (instancetype)initWithAppRecord:(EXKernelAppRecord *)appRecord
{
  if (self = [super init]) {
    _appRecord = appRecord;
  }
  return self;
}

RCT_NOT_IMPLEMENTED(- (instancetype)init)

- (void)handleSoftJSExceptionWithMessage:(NSString *)message
                                   stack:(NSArray<NSDictionary<NSString *, id> *> *)stack
                             exceptionId:(NSNumber *)exceptionId
{
  [[self _bridgeForRecord].redBox showErrorMessage:message withStack:stack];
}

- (void)handleFatalJSExceptionWithMessage:(NSString *)message
                                    stack:(NSArray<NSDictionary<NSString *, id> *> *)stack
                              exceptionId:(NSNumber *)exceptionId
{
  [[self _bridgeForRecord].redBox showErrorMessage:message withStack:stack];

  NSString *description = [@"Unhandled JS Exception: " stringByAppendingString:message];
  NSDictionary *errorInfo = @{ NSLocalizedDescriptionKey: description, RCTJSStackTraceKey: stack };
  NSError *error = [NSError errorWithDomain:RCTErrorDomain code:0 userInfo:errorInfo];

  [[EXKernel sharedInstance].serviceRegistry.errorRecoveryManager setError:error forExperienceId:_appRecord.experienceId];

  if ([self _isProdHome]) {
    [self _recordCrashlyticsExceptionFromMessage:message stack:stack];
    RCTFatal(error);
  }
}

- (void)updateJSExceptionWithMessage:(NSString *)message
                               stack:(NSArray *)stack
                         exceptionId:(NSNumber *)exceptionId
{
  [[self _bridgeForRecord].redBox updateErrorMessage:message withStack:stack];
}

#pragma mark - internal

- (RCTBridge *)_bridgeForRecord
{
  return _appRecord.appManager.reactBridge;
}

- (BOOL)_isProdHome
{
  if (RCT_DEBUG) {
    return NO;
  }
  return (_appRecord && _appRecord == [EXKernel sharedInstance].appRegistry.homeAppRecord);
}

- (void)_recordCrashlyticsExceptionFromMessage:(NSString *)message stack:(NSArray *)stack
{
  NSError *error;
  NSRegularExpression *regex = [NSRegularExpression regularExpressionWithPattern:@"^([a-z$_][a-z\\d$_]*):\\s*"
                                                                         options:NSRegularExpressionCaseInsensitive
                                                                           error:&error];
  if (!regex) {
    DDLogError(@"Error creating regular expression: %@", error.localizedDescription);
    return;
  }
  
  NSString *errorName;
  NSString *errorReason;
  NSTextCheckingResult *match = [regex firstMatchInString:message options:0 range:NSMakeRange(0, message.length)];
  if (match) {
    errorName = [message substringWithRange:[match rangeAtIndex:1]];
    errorName = [message substringFromIndex:match.range.location + match.range.length];
  } else {
    errorName = @"UnknownError";
    errorReason = message;
  }
  
  NSMutableArray *frameArray = [NSMutableArray arrayWithCapacity:stack.count];
  for (NSDictionary *frameInfo in stack) {
    CLSStackFrame *frame = [CLSStackFrame stackFrame];
    if ([frameInfo[@"file"] isKindOfClass:[NSString class]]) {
      frame.fileName = frameInfo[@"file"];
    }
    if ([frameInfo[@"methodName"] isKindOfClass:[NSString class]]) {
      frame.symbol = frameInfo[@"methodName"];
    }
    if ([frameInfo[@"lineNumber"] isKindOfClass:[NSNumber class]]) {
      frame.lineNumber = (uint32_t)MAX([frameInfo[@"lineNumber"] integerValue], 0);
    }
    [frameArray addObject:frame];
  }
  
  [[Crashlytics sharedInstance] recordCustomExceptionName:errorName reason:errorReason frameArray:frameArray];
}

@end

NS_ASSUME_NONNULL_END