Add VALIDATE_DOCUMENT macro

Author: RadAzzouz

ios/RCTPSPDFKit/RCTPSPDFKitView.m

@@ -11,6 +11,8 @@
 #import <React/RCTUtils.h>
 #import "RCTConvert+PSPDFAnnotation.h"
 
+#define VALIDATE_DOCUMENT(document, ...) { if (!document.isValid) { NSLog(@"Document is invalid."); return __VA_ARGS__; }}
+
 @interface RCTPSPDFKitView ()<PSPDFDocumentDelegate, PSPDFViewControllerDelegate, PSPDFFlexibleToolbarContainerDelegate>
 
 @property (nonatomic, nullable) UIViewController *topController;
@@ -174,10 +176,7 @@ - (void)flexibleToolbarContainerDidHide:(PSPDFFlexibleToolbarContainer *)contain
 
 - (NSDictionary<NSString *, NSArray<NSDictionary *> *> *)getAnnotations:(PSPDFPageIndex)pageIndex type:(PSPDFAnnotationType)type {
   PSPDFDocument *document = self.pdfController.document;
-  if (!document.isValid) {
-    NSLog(@"Document is invalid.");
-    return nil;
-  }
+  VALIDATE_DOCUMENT(document, nil);
 
   NSArray <PSPDFAnnotation *> *annotations = [document annotationsForPageAtIndex:pageIndex type:type];
   NSArray <NSDictionary *> *annotationsJSON = [RCTConvert instantJSONFromAnnotations:annotations];
@@ -196,10 +195,7 @@ - (BOOL)addAnnotation:(id)jsonAnnotation {
   }
 
   PSPDFDocument *document = self.pdfController.document;
-  if (!document.isValid) {
-    NSLog(@"Document is invalid.");
-    return NO;
-  }
+  VALIDATE_DOCUMENT(document, NO)
   PSPDFDocumentProvider *documentProvider = document.documentProviders.firstObject;
 
   BOOL success = NO;
@@ -217,11 +213,7 @@ - (BOOL)addAnnotation:(id)jsonAnnotation {
 
 - (BOOL)removeAnnotationWithUUID:(NSString *)annotationUUID {
   PSPDFDocument *document = self.pdfController.document;
-  if (!document.isValid) {
-    NSLog(@"Document is invalid.");
-    return NO;
-  }
-  
+  VALIDATE_DOCUMENT(document, NO)
   BOOL success = NO;
 
   NSArray<PSPDFAnnotation *> *allAnnotations = [[document allAnnotationsOfType:PSPDFAnnotationTypeAll].allValues valueForKeyPath:@"@unionOfArrays.self"];
@@ -241,10 +233,7 @@ - (BOOL)removeAnnotationWithUUID:(NSString *)annotationUUID {
 
 - (NSDictionary<NSString *, NSArray<NSDictionary *> *> *)getAllUnsavedAnnotations {
   PSPDFDocument *document = self.pdfController.document;
-  if (!document.isValid) {
-    NSLog(@"Document is invalid.");
-    return nil;
-  }
+  VALIDATE_DOCUMENT(document, nil)
 
   PSPDFDocumentProvider *documentProvider = document.documentProviders.firstObject;
   NSData *data = [document generateInstantJSONFromDocumentProvider:documentProvider error:NULL];
@@ -265,11 +254,7 @@ - (BOOL)addAnnotations:(id)jsonAnnotations {
 
   PSPDFDataContainerProvider *dataContainerProvider = [[PSPDFDataContainerProvider alloc] initWithData:data];
   PSPDFDocument *document = self.pdfController.document;
-  if (!document.isValid) {
-    NSLog(@"Document is invalid.");
-    return NO;
-  }
-  
+  VALIDATE_DOCUMENT(document, NO)
   PSPDFDocumentProvider *documentProvider = document.documentProviders.firstObject;
   BOOL success = [document applyInstantJSONFromDataProvider:dataContainerProvider toDocumentProvider:documentProvider lenient:NO error:NULL];
   if (!success) {
@@ -289,10 +274,7 @@ - (BOOL)addAnnotations:(id)jsonAnnotations {
   }
 
   PSPDFDocument *document = self.pdfController.document;
-  if (!document.isValid) {
-    NSLog(@"Document is invalid.");
-    return nil;
-  }
+  VALIDATE_DOCUMENT(document, nil)
 
   for (PSPDFFormElement *formElement in document.formParser.forms) {
     if ([formElement.fullyQualifiedFieldName isEqualToString:fullyQualifiedName]) {
@@ -311,10 +293,7 @@ - (void)setFormFieldValue:(NSString *)value fullyQualifiedName:(NSString *)fully
   }
 
   PSPDFDocument *document = self.pdfController.document;
-  if (!document.isValid) {
-    NSLog(@"Document is invalid.");
-    return;
-  }
+  VALIDATE_DOCUMENT(document)
 
   for (PSPDFFormElement *formElement in document.formParser.forms) {
     if ([formElement.fullyQualifiedFieldName isEqualToString:fullyQualifiedName]) {