Allow registration of content type viewers (#241)

Classes/FLEXManager.h

@@ -9,6 +9,8 @@
 #import <Foundation/Foundation.h>
 #import <UIKit/UIKit.h>
 
+typedef UIViewController *(^FLEXCustomContentViewerFuture)(NSData *data);
+
 @interface FLEXManager : NSObject
 
 + (instancetype)sharedManager;
@@ -77,4 +79,13 @@
 - (void)registerGlobalEntryWithName:(NSString *)entryName
           viewControllerFutureBlock:(UIViewController * (^)(void))viewControllerFutureBlock;
 
+/// Sets custom viewer for specific content type.
+/// @param contentType Mime type like application/json
+/// @param viewControllerFutureBlock Viewer (view controller) creation block
+/// @note This method must be called from the main thread.
+/// The viewControllerFutureBlock will be invoked from the main thread and may not return nil.
+/// @note The passed block will be copied and retain for the duration of the application, you may want to use __weak references.
+- (void)setCustomViewerForContentType:(NSString *)contentType
+            viewControllerFutureBlock:(FLEXCustomContentViewerFuture)viewControllerFutureBlock;
+
 @end

Classes/Manager/FLEXManager+Private.h

@@ -15,4 +15,6 @@
 /// An array of FLEXGlobalsTableViewControllerEntry objects that have been registered by the user.
 @property (nonatomic, readonly, strong) NSArray<FLEXGlobalsTableViewControllerEntry *> *userGlobalEntries;
 
+@property (nonatomic, readonly, strong) NSDictionary<NSString *, FLEXCustomContentViewerFuture> *customContentTypeViewers;
+
 @end

Classes/Manager/FLEXManager.m

@@ -25,6 +25,7 @@
 @property (nonatomic, strong) FLEXExplorerViewController *explorerViewController;
 
 @property (nonatomic, readonly, strong) NSMutableArray<FLEXGlobalsTableViewControllerEntry *> *userGlobalEntries;
+@property (nonatomic, readonly, strong) NSMutableDictionary<NSString *, FLEXCustomContentViewerFuture> *customContentTypeViewers;
 
 @end
 
@@ -45,6 +46,7 @@
     self = [super init];
     if (self) {
         _userGlobalEntries = [NSMutableArray array];
+        _customContentTypeViewers = [NSMutableDictionary dictionary];
     }
     return self;
 }
@@ -288,6 +290,15 @@
     [self.userGlobalEntries addObject:entry];
 }
 
+- (void)setCustomViewerForContentType:(NSString *)contentType viewControllerFutureBlock:(FLEXCustomContentViewerFuture)viewControllerFutureBlock
+{
+    NSParameterAssert(contentType.length);
+    NSParameterAssert(viewControllerFutureBlock);
+    NSAssert([NSThread isMainThread], @"This method must be called from the main thread.");
+
+    self.customContentTypeViewers[contentType.lowercaseString] = viewControllerFutureBlock;
+}
+
 - (void)tryScrollDown
 {
     UIScrollView *firstScrollView = [self firstScrollView];

Classes/Network/FLEXNetworkTransactionDetailTableViewController.m

@@ -14,6 +14,7 @@
 #import "FLEXImagePreviewViewController.h"
 #import "FLEXMultilineTableViewCell.h"
 #import "FLEXUtility.h"
+#import "FLEXManager+Private.h"
 
 typedef UIViewController *(^FLEXNetworkDetailRowSelectionFuture)(void);
 
@@ -425,6 +426,16 @@ typedef UIViewController *(^FLEXNetworkDetailRowSelectionFuture)(void);
 
 + (UIViewController *)detailViewControllerForMIMEType:(NSString *)mimeType data:(NSData *)data
 {
+    FLEXCustomContentViewerFuture makeCustomViewer = [FLEXManager sharedManager].customContentTypeViewers[mimeType.lowercaseString];
+
+    if (makeCustomViewer) {
+        UIViewController *viewer = makeCustomViewer(data);
+
+        if (viewer) {
+            return viewer;
+        }
+    }
+
     // FIXME (RKO): Don't rely on UTF8 string encoding
     UIViewController *detailViewController = nil;
     if ([FLEXUtility isValidJSONData:data]) {