shakebug-react-native 2.3.2 → 2.3.4

package/ios/RecordScreen/ShakebugSDKRecordScreen.m

@@ -15,295 +15,374 @@ UIBackgroundTaskIdentifier _ShakebugSDKbackgroundRenderingID;
 
 - (NSDictionary *)errorResponse:(NSDictionary *)result;
 {
-    NSDictionary *json = [NSDictionary dictionaryWithObjectsAndKeys:
-        @"error", @"status",
-        result, @"result",nil];
-    return json;
-
+  NSDictionary *json = [NSDictionary
+      dictionaryWithObjectsAndKeys:@"error", @"status", result, @"result", nil];
+  return json;
 }
 
-- (NSDictionary *) successResponse:(NSDictionary *)result;
+- (NSDictionary *)successResponse:(NSDictionary *)result;
 {
-    NSDictionary *json = [NSDictionary dictionaryWithObjectsAndKeys:
-        @"success", @"status",
-        result, @"result",nil];
-    return json;
-
+  NSDictionary *json =
+      [NSDictionary dictionaryWithObjectsAndKeys:@"success", @"status", result,
+                                                 @"result", nil];
+  return json;
 }
 
-- (void) muteAudioInBuffer:(CMSampleBufferRef)sampleBuffer
-{
+- (void)muteAudioInBuffer:(CMSampleBufferRef)sampleBuffer {
 
-    CMItemCount numSamples = CMSampleBufferGetNumSamples(sampleBuffer);
-    NSUInteger channelIndex = 0;
+  CMItemCount numSamples = CMSampleBufferGetNumSamples(sampleBuffer);
+  NSUInteger channelIndex = 0;
 
-    CMBlockBufferRef audioBlockBuffer = CMSampleBufferGetDataBuffer(sampleBuffer);
-    size_t audioBlockBufferOffset = (channelIndex * numSamples * sizeof(SInt16));
-    size_t lengthAtOffset = 0;
-    size_t totalLength = 0;
-    SInt16 *samples = NULL;
-    CMBlockBufferGetDataPointer(audioBlockBuffer, audioBlockBufferOffset, &lengthAtOffset, &totalLength, (char **)(&samples));
+  CMBlockBufferRef audioBlockBuffer = CMSampleBufferGetDataBuffer(sampleBuffer);
+  size_t audioBlockBufferOffset = (channelIndex * numSamples * sizeof(SInt16));
+  size_t lengthAtOffset = 0;
+  size_t totalLength = 0;
+  SInt16 *samples = NULL;
+  CMBlockBufferGetDataPointer(audioBlockBuffer, audioBlockBufferOffset,
+                              &lengthAtOffset, &totalLength,
+                              (char **)(&samples));
 
-    for (NSInteger i=0; i<numSamples; i++) {
-        samples[i] = (SInt16)0;
-    }
+  for (NSInteger i = 0; i < numSamples; i++) {
+    samples[i] = (SInt16)0;
+  }
 }
 
-// For H264, unless the value is a multiple of 2 or 4, a green border will appear, so a function to adjust it
-- (int) adjustMultipleOf2:(int)value;
+// For H264, unless the value is a multiple of 2 or 4, a green border will
+// appear, so a function to adjust it
+- (int)adjustMultipleOf2:(int)value;
 {
-    if (value % 2 == 1) {
-        return value + 1;
-    }
-    return value;
+  if (value % 2 == 1) {
+    return value + 1;
+  }
+  return value;
 }
 
-
 RCT_EXPORT_MODULE()
 
-RCT_EXPORT_METHOD(setup: (NSDictionary *)config)
-{
-    self.screenWidth = [RCTConvert int: config[@"width"]];
-    self.screenHeight = [RCTConvert int: config[@"height"]];
-    self.enableMic = [RCTConvert BOOL: config[@"mic"]];
-    self.bitrate = [RCTConvert int: config[@"bitrate"]];
-    self.fps = [RCTConvert int: config[@"fps"]];
+RCT_EXPORT_METHOD(setup : (NSDictionary *)config) {
+  self.screenWidth = [RCTConvert int:config[@"width"]];
+  self.screenHeight = [RCTConvert int:config[@"height"]];
+  self.enableMic = [RCTConvert BOOL:config[@"mic"]];
+  self.bitrate = [RCTConvert int:config[@"bitrate"]];
+  self.fps = [RCTConvert int:config[@"fps"]];
 }
 
-RCT_REMAP_METHOD(startRecording, resolve:(RCTPromiseResolveBlock)resolve rejecte:(RCTPromiseRejectBlock)reject)
-{
-    UIApplication *app = [UIApplication sharedApplication];
-    _ShakebugSDKbackgroundRenderingID = [app beginBackgroundTaskWithExpirationHandler:^{
+RCT_REMAP_METHOD(startRecording,
+                 resolve : (RCTPromiseResolveBlock)
+                     resolve rejecte : (RCTPromiseRejectBlock)reject) {
+  UIApplication *app = [UIApplication sharedApplication];
+  _ShakebugSDKbackgroundRenderingID =
+      [app beginBackgroundTaskWithExpirationHandler:^{
         [app endBackgroundTask:_ShakebugSDKbackgroundRenderingID];
         _ShakebugSDKbackgroundRenderingID = UIBackgroundTaskInvalid;
-    }];
-
-    self.screenRecorder = [RPScreenRecorder sharedRecorder];
-    if (self.screenRecorder.isRecording) {
-        return;
-    }
-
-    self.encounteredFirstBuffer = NO;
-
-    NSArray *pathDocuments = NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES);
-    NSString *outputURL = pathDocuments[0];
-
-    NSString *videoOutPath = [[outputURL stringByAppendingPathComponent:[NSString stringWithFormat:@"%u", arc4random() % 1000]] stringByAppendingPathExtension:@"mp4"];
-
-    NSError *error;
-    self.writer = [AVAssetWriter assetWriterWithURL:[NSURL fileURLWithPath:videoOutPath] fileType:AVFileTypeMPEG4 error:&error];
-    if (!self.writer) {
-        NSLog(@"writer: %@", error);
-        abort();
-    }
-
-    AudioChannelLayout acl = { 0 };
-    acl.mChannelLayoutTag = kAudioChannelLayoutTag_Mono;
-    self.audioInput = [[AVAssetWriterInput alloc] initWithMediaType:AVMediaTypeAudio outputSettings:@{ AVFormatIDKey: @(kAudioFormatMPEG4AAC), AVSampleRateKey: @(44100),  AVChannelLayoutKey: [NSData dataWithBytes: &acl length: sizeof( acl ) ], AVEncoderBitRateKey: @(64000)}];
-    self.micInput = [[AVAssetWriterInput alloc] initWithMediaType:AVMediaTypeAudio outputSettings:@{ AVFormatIDKey: @(kAudioFormatMPEG4AAC), AVSampleRateKey: @(44100),  AVChannelLayoutKey: [NSData dataWithBytes: &acl length: sizeof( acl ) ], AVEncoderBitRateKey: @(64000)}];
-
-    self.audioInput.preferredVolume = 0.0;
-    self.micInput.preferredVolume = 0.0;
-
-    NSDictionary *compressionProperties = @{AVVideoProfileLevelKey         : AVVideoProfileLevelH264HighAutoLevel,
-                                            AVVideoH264EntropyModeKey      : AVVideoH264EntropyModeCABAC,
-                                            AVVideoAverageBitRateKey       : @(self.bitrate),
-                                            AVVideoMaxKeyFrameIntervalKey  : @(self.fps),
-                                            AVVideoAllowFrameReorderingKey : @NO};
-
-    NSLog(@"width: %d", [self adjustMultipleOf2:self.screenWidth]);
-    NSLog(@"height: %d", [self adjustMultipleOf2:self.screenHeight]);
-    if (@available(iOS 11.0, *)) {
-        NSDictionary *videoSettings = @{AVVideoCompressionPropertiesKey : compressionProperties,
-                                        AVVideoCodecKey                 : AVVideoCodecTypeH264,
-                                        AVVideoWidthKey                 : @([self adjustMultipleOf2:self.screenWidth]),
-                                        AVVideoHeightKey                : @([self adjustMultipleOf2:self.screenHeight])};
-
-        self.videoInput = [AVAssetWriterInput assetWriterInputWithMediaType:AVMediaTypeVideo outputSettings:videoSettings];
-    } else {
-        // Fallback on earlier versions
-    }
-
-    [self.writer addInput:self.audioInput];
-    [self.writer addInput:self.micInput];
-    [self.writer addInput:self.videoInput];
-    [self.videoInput setMediaTimeScale:60];
-    [self.writer setMovieTimeScale:60];
-    [self.videoInput setExpectsMediaDataInRealTime:YES];
-
-    if (self.enableMic) {
-        self.screenRecorder.microphoneEnabled = YES;
-    }
-
-    [AVCaptureDevice requestAccessForMediaType:AVMediaTypeVideo completionHandler:^(BOOL granted) {
-        if (granted) {
-            if (@available(iOS 11.0, *)) {
-                [self.screenRecorder startCaptureWithHandler:^(CMSampleBufferRef sampleBuffer, RPSampleBufferType bufferType, NSError* error) {
-                    // Failure to do so will result in a memory error when accessing the sampleBuffer in the main thread.
-                    CFRetain(sampleBuffer);
-                    dispatch_async(dispatch_get_main_queue(), ^{
-                        if (CMSampleBufferDataIsReady(sampleBuffer)) {
-                            if (self.writer.status == AVAssetWriterStatusUnknown && !self.encounteredFirstBuffer && bufferType == RPSampleBufferTypeVideo) {
+      }];
+
+  self.screenRecorder = [RPScreenRecorder sharedRecorder];
+  if (self.screenRecorder.isRecording) {
+    return;
+  }
+
+  self.encounteredFirstBuffer = NO;
+
+  NSArray *pathDocuments = NSSearchPathForDirectoriesInDomains(
+      NSDocumentDirectory, NSUserDomainMask, YES);
+  NSString *outputURL = pathDocuments[0];
+
+  NSString *videoOutPath = [[outputURL
+      stringByAppendingPathComponent:[NSString
+                                         stringWithFormat:@"%u",
+                                                          arc4random() % 1000]]
+      stringByAppendingPathExtension:@"mp4"];
+
+  NSError *error;
+  self.writer =
+      [AVAssetWriter assetWriterWithURL:[NSURL fileURLWithPath:videoOutPath]
+                               fileType:AVFileTypeMPEG4
+                                  error:&error];
+  if (!self.writer) {
+    NSLog(@"writer: %@", error);
+    abort();
+  }
+
+  AudioChannelLayout acl = {0};
+  acl.mChannelLayoutTag = kAudioChannelLayoutTag_Mono;
+  self.audioInput = [[AVAssetWriterInput alloc]
+      initWithMediaType:AVMediaTypeAudio
+         outputSettings:@{
+           AVFormatIDKey : @(kAudioFormatMPEG4AAC),
+           AVSampleRateKey : @(44100),
+           AVChannelLayoutKey : [NSData dataWithBytes:&acl length:sizeof(acl)],
+           AVEncoderBitRateKey : @(64000)
+         }];
+  self.micInput = [[AVAssetWriterInput alloc]
+      initWithMediaType:AVMediaTypeAudio
+         outputSettings:@{
+           AVFormatIDKey : @(kAudioFormatMPEG4AAC),
+           AVSampleRateKey : @(44100),
+           AVChannelLayoutKey : [NSData dataWithBytes:&acl length:sizeof(acl)],
+           AVEncoderBitRateKey : @(64000)
+         }];
+
+  self.audioInput.preferredVolume = 0.0;
+  self.micInput.preferredVolume = 0.0;
+
+  NSDictionary *compressionProperties = @{
+    AVVideoProfileLevelKey : AVVideoProfileLevelH264HighAutoLevel,
+    AVVideoH264EntropyModeKey : AVVideoH264EntropyModeCABAC,
+    AVVideoAverageBitRateKey : @(self.bitrate),
+    AVVideoMaxKeyFrameIntervalKey : @(self.fps),
+    AVVideoAllowFrameReorderingKey : @NO
+  };
+
+  NSLog(@"width: %d", [self adjustMultipleOf2:self.screenWidth]);
+  NSLog(@"height: %d", [self adjustMultipleOf2:self.screenHeight]);
+  if (@available(iOS 11.0, *)) {
+    NSDictionary *videoSettings = @{
+      AVVideoCompressionPropertiesKey : compressionProperties,
+      AVVideoCodecKey : AVVideoCodecTypeH264,
+      AVVideoWidthKey : @([self adjustMultipleOf2:self.screenWidth]),
+      AVVideoHeightKey : @([self adjustMultipleOf2:self.screenHeight])
+    };
+
+    self.videoInput =
+        [AVAssetWriterInput assetWriterInputWithMediaType:AVMediaTypeVideo
+                                           outputSettings:videoSettings];
+  } else {
+    // Fallback on earlier versions
+  }
+
+  [self.writer addInput:self.audioInput];
+  [self.writer addInput:self.micInput];
+  [self.writer addInput:self.videoInput];
+  [self.videoInput setMediaTimeScale:60];
+  [self.writer setMovieTimeScale:60];
+  [self.videoInput setExpectsMediaDataInRealTime:YES];
+
+  if (self.enableMic) {
+    self.screenRecorder.microphoneEnabled = YES;
+  }
+
+  [AVCaptureDevice
+      requestAccessForMediaType:AVMediaTypeVideo
+              completionHandler:^(BOOL granted) {
+                if (granted) {
+                  if (@available(iOS 11.0, *)) {
+                    [self.screenRecorder
+                        startCaptureWithHandler:^(
+                            CMSampleBufferRef sampleBuffer,
+                            RPSampleBufferType bufferType, NSError *error) {
+                          // Failure to do so will result in a memory error when
+                          // accessing the sampleBuffer in the main thread.
+                          CFRetain(sampleBuffer);
+                          dispatch_async(dispatch_get_main_queue(), ^{
+                            if (CMSampleBufferDataIsReady(sampleBuffer)) {
+                              if (self.writer.status ==
+                                      AVAssetWriterStatusUnknown &&
+                                  !self.encounteredFirstBuffer &&
+                                  bufferType == RPSampleBufferTypeVideo) {
                                 self.encounteredFirstBuffer = YES;
                                 NSLog(@"First buffer video");
                                 [self.writer startWriting];
-                                [self.writer startSessionAtSourceTime:CMSampleBufferGetPresentationTimeStamp(sampleBuffer)];
-                            } else if (self.writer.status == AVAssetWriterStatusFailed) {
-
-                            }
-
-                            if ([[UIApplication sharedApplication] applicationState] == UIApplicationStateBackground) {
+                                [self.writer
+                                    startSessionAtSourceTime:
+                                        CMSampleBufferGetPresentationTimeStamp(
+                                            sampleBuffer)];
+                              } else if (self.writer.status ==
+                                         AVAssetWriterStatusFailed) {
+                              }
+
+                              if ([[UIApplication sharedApplication]
+                                      applicationState] ==
+                                  UIApplicationStateBackground) {
                                 CMSampleBufferRef copiedBuffer;
-                                CMSampleBufferCreateCopy(kCFAllocatorDefault, sampleBuffer, &copiedBuffer);
+                                CMSampleBufferCreateCopy(kCFAllocatorDefault,
+                                                         sampleBuffer,
+                                                         &copiedBuffer);
                                 switch (bufferType) {
-                                    case RPSampleBufferTypeVideo:
-                                        self.afterAppBackgroundVideoSampleBuffer = copiedBuffer;
-                                        break;
-                                    case RPSampleBufferTypeAudioApp:
-                                        self.afterAppBackgroundAudioSampleBuffer = copiedBuffer;
-                                        break;
-                                    case RPSampleBufferTypeAudioMic:
-                                        self.afterAppBackgroundMicSampleBuffer = copiedBuffer;
-                                        break;
-                                    default:
-                                        break;
+                                case RPSampleBufferTypeVideo:
+                                  self.afterAppBackgroundVideoSampleBuffer =
+                                      copiedBuffer;
+                                  break;
+                                case RPSampleBufferTypeAudioApp:
+                                  self.afterAppBackgroundAudioSampleBuffer =
+                                      copiedBuffer;
+                                  break;
+                                case RPSampleBufferTypeAudioMic:
+                                  self.afterAppBackgroundMicSampleBuffer =
+                                      copiedBuffer;
+                                  break;
+                                default:
+                                  break;
                                 }
-                            } else if ([[UIApplication sharedApplication] applicationState] == UIApplicationStateActive) {
-                                if (bufferType == RPSampleBufferTypeVideo && self.afterAppBackgroundVideoSampleBuffer != nil && self.afterAppBackgroundAudioSampleBuffer != nil && self.afterAppBackgroundMicSampleBuffer != nil) {
-                                    CMTime timeWhenAppBackground = CMTimeSubtract(CMSampleBufferGetPresentationTimeStamp(sampleBuffer), CMSampleBufferGetPresentationTimeStamp(self.afterAppBackgroundVideoSampleBuffer));
-                                    // Calc loop count for appendSampleBuffer.
-                                    long bufferCount = floor(CMTimeGetSeconds(timeWhenAppBackground) * (1 / CMTimeGetSeconds(CMSampleBufferGetDuration(self.afterAppBackgroundAudioSampleBuffer))));
-
-                                    // Make mute audio.
-                                    [self muteAudioInBuffer:self.afterAppBackgroundAudioSampleBuffer];
-                                    [self muteAudioInBuffer:self.afterAppBackgroundMicSampleBuffer];
-
-                                    for (int i = 0; i < bufferCount; i ++) {
-                                        if (self.audioInput.isReadyForMoreMediaData) {
-                                            [self.audioInput appendSampleBuffer:self.afterAppBackgroundAudioSampleBuffer];
-                                        }
+                              } else if ([[UIApplication sharedApplication]
+                                             applicationState] ==
+                                         UIApplicationStateActive) {
+                                if (bufferType == RPSampleBufferTypeVideo &&
+                                    self.afterAppBackgroundVideoSampleBuffer !=
+                                        nil &&
+                                    self.afterAppBackgroundAudioSampleBuffer !=
+                                        nil &&
+                                    self.afterAppBackgroundMicSampleBuffer !=
+                                        nil) {
+                                  CMTime timeWhenAppBackground = CMTimeSubtract(
+                                      CMSampleBufferGetPresentationTimeStamp(
+                                          sampleBuffer),
+                                      CMSampleBufferGetPresentationTimeStamp(
+                                          self.afterAppBackgroundVideoSampleBuffer));
+                                  // Calc loop count for appendSampleBuffer.
+                                  long bufferCount = floor(
+                                      CMTimeGetSeconds(timeWhenAppBackground) *
+                                      (1 /
+                                       CMTimeGetSeconds(CMSampleBufferGetDuration(
+                                           self.afterAppBackgroundAudioSampleBuffer))));
+
+                                  // Make mute audio.
+                                  [self
+                                      muteAudioInBuffer:
+                                          self.afterAppBackgroundAudioSampleBuffer];
+                                  [self
+                                      muteAudioInBuffer:
+                                          self.afterAppBackgroundMicSampleBuffer];
+
+                                  for (int i = 0; i < bufferCount; i++) {
+                                    if (self.audioInput
+                                            .isReadyForMoreMediaData) {
+                                      [self.audioInput
+                                          appendSampleBuffer:
+                                              self.afterAppBackgroundAudioSampleBuffer];
                                     }
-                                    for (int i = 0; i < bufferCount; i ++) {
-                                        if (self.enableMic && self.micInput.isReadyForMoreMediaData) {
-                                            [self.micInput appendSampleBuffer:self.afterAppBackgroundMicSampleBuffer];
-                                        }
+                                  }
+                                  for (int i = 0; i < bufferCount; i++) {
+                                    if (self.enableMic &&
+                                        self.micInput.isReadyForMoreMediaData) {
+                                      [self.micInput
+                                          appendSampleBuffer:
+                                              self.afterAppBackgroundMicSampleBuffer];
                                     }
-
-                                    // Clean
-                                    CFRelease(self.afterAppBackgroundAudioSampleBuffer);
-                                    CFRelease(self.afterAppBackgroundMicSampleBuffer);
-                                    CFRelease(self.afterAppBackgroundVideoSampleBuffer);
-                                    self.afterAppBackgroundAudioSampleBuffer = nil;
-                                    self.afterAppBackgroundMicSampleBuffer = nil;
-                                    self.afterAppBackgroundVideoSampleBuffer = nil;
+                                  }
+
+                                  // Clean
+                                  CFRelease(
+                                      self.afterAppBackgroundAudioSampleBuffer);
+                                  CFRelease(
+                                      self.afterAppBackgroundMicSampleBuffer);
+                                  CFRelease(
+                                      self.afterAppBackgroundVideoSampleBuffer);
+                                  self.afterAppBackgroundAudioSampleBuffer =
+                                      nil;
+                                  self.afterAppBackgroundMicSampleBuffer = nil;
+                                  self.afterAppBackgroundVideoSampleBuffer =
+                                      nil;
                                 }
-                            }
+                              }
 
-                            if (self.writer.status == AVAssetWriterStatusWriting) {
+                              if (self.writer.status ==
+                                  AVAssetWriterStatusWriting) {
                                 switch (bufferType) {
-                                    case RPSampleBufferTypeVideo:
-                                        if (self.videoInput.isReadyForMoreMediaData) {
-                                            [self.videoInput appendSampleBuffer:sampleBuffer];
-                                        }
-                                        break;
-                                    case RPSampleBufferTypeAudioApp:
-                                        if (self.audioInput.isReadyForMoreMediaData) {
-                                            if(self.enableMic){
-                                                [self.audioInput appendSampleBuffer:sampleBuffer];
-                                            } else {
-                                                [self muteAudioInBuffer:sampleBuffer];
-                                            }
-                                        }
-                                        break;
-                                    case RPSampleBufferTypeAudioMic:
-                                        if (self.micInput.isReadyForMoreMediaData) {
-                                            if(self.enableMic){
-                                                [self.micInput appendSampleBuffer:sampleBuffer];
-                                            } else {
-                                                [self muteAudioInBuffer:sampleBuffer];
-                                            }
-                                        }
-                                        break;
-                                    default:
-                                        break;
+                                case RPSampleBufferTypeVideo:
+                                  if (self.videoInput.isReadyForMoreMediaData) {
+                                    [self.videoInput
+                                        appendSampleBuffer:sampleBuffer];
+                                  }
+                                  break;
+                                case RPSampleBufferTypeAudioApp:
+                                  if (self.audioInput.isReadyForMoreMediaData) {
+                                    if (self.enableMic) {
+                                      [self.audioInput
+                                          appendSampleBuffer:sampleBuffer];
+                                    } else {
+                                      [self muteAudioInBuffer:sampleBuffer];
+                                    }
+                                  }
+                                  break;
+                                case RPSampleBufferTypeAudioMic:
+                                  if (self.micInput.isReadyForMoreMediaData) {
+                                    if (self.enableMic) {
+                                      [self.micInput
+                                          appendSampleBuffer:sampleBuffer];
+                                    } else {
+                                      [self muteAudioInBuffer:sampleBuffer];
+                                    }
+                                  }
+                                  break;
+                                default:
+                                  break;
                                 }
+                              }
                             }
-                        }
 
-                        CFRelease(sampleBuffer);
-                    });
-                } completionHandler:^(NSError* error) {
-                    NSLog(@"startCapture: %@", error);
-                    if (error) {
-                        resolve(@"permission_error");
-                    } else {
-                        resolve(@"started");
-                    }
-                }];
-            } else {
-                // Fallback on earlier versions
-            }
-        } else {
-            NSError* err = nil;
-            reject(0, @"Permission denied", err);
-        }
-    }];
-
-    if (self.enableMic) {
-        self.screenRecorder.microphoneEnabled = YES;
-    }
-}
-
-RCT_REMAP_METHOD(stopRecording, resolver:(RCTPromiseResolveBlock)resolve rejecter:(RCTPromiseRejectBlock)reject)
-{
-    [[UIApplication sharedApplication] endBackgroundTask:_ShakebugSDKbackgroundRenderingID];
-
-    dispatch_async(dispatch_get_main_queue(), ^{
-        if (@available(iOS 11.0, *)) {
-            [[RPScreenRecorder sharedRecorder] stopCaptureWithHandler:^(NSError * _Nullable error) {
-                if (!error) {
-                    [self.audioInput markAsFinished];
-                    [self.micInput markAsFinished];
-                    [self.videoInput markAsFinished];
-                    [self.writer finishWritingWithCompletionHandler:^{
-
-                        NSDictionary *result = [NSDictionary dictionaryWithObject:self.writer.outputURL.absoluteString forKey:@"outputURL"];
-                        resolve([self successResponse:result]);
-
-                        NSLog(@"finishWritingWithCompletionHandler: Recording stopped successfully. Cleaning up... %@", result);
-                        self.audioInput = nil;
-                        self.micInput = nil;
-                        self.videoInput = nil;
-                        self.writer = nil;
-                        self.screenRecorder = nil;
-                    }];
+                            CFRelease(sampleBuffer);
+                          });
+                        }
+                        completionHandler:^(NSError *error) {
+                          NSLog(@"startCapture: %@", error);
+                          if (error) {
+                            resolve(@"permission_error");
+                          } else {
+                            resolve(@"started");
+                          }
+                        }];
+                  } else {
+                    // Fallback on earlier versions
+                  }
+                } else {
+                  NSError *err = nil;
+                  reject(0, @"Permission denied", err);
                 }
-            }];
-        } else {
-            // Fallback on earlier versions
-        }
-    });
+              }];
+
+  if (self.enableMic) {
+    self.screenRecorder.microphoneEnabled = YES;
+  }
 }
 
-RCT_REMAP_METHOD(clean,
-                 cleanResolve:(RCTPromiseResolveBlock)resolve
-                 cleanRejecte:(RCTPromiseRejectBlock)reject)
-{
+RCT_REMAP_METHOD(stopRecording,
+                 resolver : (RCTPromiseResolveBlock)
+                     resolve rejecter : (RCTPromiseRejectBlock)reject) {
+  [[UIApplication sharedApplication]
+      endBackgroundTask:_ShakebugSDKbackgroundRenderingID];
 
-    NSArray *pathDocuments = NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES);
-    NSString *path = pathDocuments[0];
-    NSLog(@"startCapture: %@", path);
-    [[NSFileManager defaultManager] removeItemAtPath:path error:nil];
-    resolve(@"cleaned");
+  dispatch_async(dispatch_get_main_queue(), ^{
+    if (@available(iOS 11.0, *)) {
+      [[RPScreenRecorder sharedRecorder]
+          stopCaptureWithHandler:^(NSError *_Nullable error) {
+            if (!error) {
+              [self.audioInput markAsFinished];
+              [self.micInput markAsFinished];
+              [self.videoInput markAsFinished];
+              [self.writer finishWritingWithCompletionHandler:^{
+                NSDictionary *result = [NSDictionary
+                    dictionaryWithObject:self.writer.outputURL.absoluteString
+                                  forKey:@"outputURL"];
+                resolve([self successResponse:result]);
+
+                NSLog(@"finishWritingWithCompletionHandler: Recording stopped "
+                      @"successfully. Cleaning up... %@",
+                      result);
+                self.audioInput = nil;
+                self.micInput = nil;
+                self.videoInput = nil;
+                self.writer = nil;
+                self.screenRecorder = nil;
+              }];
+            }
+          }];
+    } else {
+      // Fallback on earlier versions
+    }
+  });
 }
 
-// Don't compile this code when we build for the old architecture.
-#ifdef RCT_NEW_ARCH_ENABLED
-- (std::shared_ptr<facebook::react::TurboModule>)getTurboModule:
-    (const facebook::react::ObjCTurboModule::InitParams &)params
-{
-    return std::make_shared<facebook::react::NativeRecordScreenSpecJSI>(params);
+RCT_REMAP_METHOD(clean,
+                 cleanResolve : (RCTPromiseResolveBlock)
+                     resolve cleanRejecte : (RCTPromiseRejectBlock)reject) {
+
+  NSArray *pathDocuments = NSSearchPathForDirectoriesInDomains(
+      NSDocumentDirectory, NSUserDomainMask, YES);
+  NSString *path = pathDocuments[0];
+  NSLog(@"startCapture: %@", path);
+  [[NSFileManager defaultManager] removeItemAtPath:path error:nil];
+  resolve(@"cleaned");
 }
-#endif
 
 @end

package/ios/ScreenShot/ShakebugCapture.h

@@ -0,0 +1,5 @@
+#import <React/RCTBridgeModule.h>
+
+@interface ShakebugCapture : NSObject <RCTBridgeModule>
+
+@end