Signal-iOS/Signal/src/ViewControllers/DebugUI/DebugUITableViewController.m

//
//  Copyright (c) 2021 Open Whisper Systems. All rights reserved.
//

#import "DebugUITableViewController.h"
#import "DebugUIContacts.h"
#import "DebugUIDiskUsage.h"
#import "DebugUIMessages.h"
#import "DebugUIMisc.h"
#import "DebugUIScreenshots.h"
#import "DebugUISessionState.h"
#import "DebugUIStress.h"
#import "DebugUISyncMessages.h"
#import "Signal-Swift.h"
#import <SignalServiceKit/TSContactThread.h>
#import <SignalServiceKit/TSThread.h>

#ifdef DEBUG

NS_ASSUME_NONNULL_BEGIN

@implementation DebugUITableViewController

- (void)viewWillAppear:(BOOL)animated
{
    [super viewWillAppear:animated];

    // Block device from sleeping while in the Debug UI.
    //
    // This is useful if you're using long-running actions in the
    // Debug UI, like "send 1k messages", etc.
    [DeviceSleepManager.shared addBlockWithBlockObject:self];
}

- (void)viewWillDisappear:(BOOL)animated
{
    [super viewWillDisappear:animated];

    [DeviceSleepManager.shared removeBlockWithBlockObject:self];
}

#pragma mark - Factory Methods

- (void)pushPageWithSection:(OWSTableSection *)section
{
    DebugUITableViewController *viewController = [DebugUITableViewController new];
    OWSTableContents *contents = [OWSTableContents new];
    contents.title = section.headerTitle;
    [contents addSection:section];
    viewController.contents = contents;
    [self.navigationController pushViewController:viewController animated:YES];
}

+ (OWSTableItem *)itemForSubsection:(DebugUIPage *)page
                     viewController:(DebugUITableViewController *)viewController
                             thread:(nullable TSThread *)thread
{
    OWSAssertDebug(page);
    OWSAssertDebug(viewController);

    __weak DebugUITableViewController *weakSelf = viewController;
    return [OWSTableItem disclosureItemWithText:page.name
                                    actionBlock:^{
                                        [weakSelf pushPageWithSection:[page sectionForThread:thread]];
                                    }];
}

+ (void)presentDebugUIForThread:(TSThread *)thread fromViewController:(UIViewController *)fromViewController
{
    OWSAssertDebug(thread);
    OWSAssertDebug(fromViewController);

    DebugUITableViewController *viewController = [DebugUITableViewController new];

    OWSTableContents *contents = [OWSTableContents new];
    contents.title = @"Debug: Conversation";

    NSMutableArray<OWSTableItem *> *subsectionItems = [NSMutableArray new];

    [subsectionItems
        addObject:[self itemForSubsection:[DebugUIMessages new] viewController:viewController thread:thread]];
    [subsectionItems
        addObject:[self itemForSubsection:[DebugUIContacts new] viewController:viewController thread:thread]];
    [subsectionItems
        addObject:[self itemForSubsection:[DebugUIDiskUsage new] viewController:viewController thread:thread]];
    [subsectionItems
        addObject:[self itemForSubsection:[DebugUISessionState new] viewController:viewController thread:thread]];
    if ([thread isKindOfClass:[TSContactThread class]]) {
        [subsectionItems
            addObject:[self itemForSubsection:[DebugUICalling new] viewController:viewController thread:thread]];
    }
    [subsectionItems
        addObject:[self itemForSubsection:[DebugUINotifications new] viewController:viewController thread:thread]];
    [subsectionItems addObject:[self itemForSubsection:[DebugUIProfile new] viewController:viewController thread:thread]];
    [subsectionItems
        addObject:[self itemForSubsection:[DebugUIStress new] viewController:viewController thread:thread]];
    [subsectionItems addObject:[self itemForSubsection:[DebugUISyncMessages new]
                                        viewController:viewController
                                                thread:thread]];
    [subsectionItems addObject:[self itemForSubsection:[DebugUIScreenshots new]
                                        viewController:viewController
                                                thread:thread]];
    OWSTableItem *sharedDataFileBrowserItem = [OWSTableItem
        disclosureItemWithText:@"📁 Shared Container"
                   actionBlock:^{
                       NSURL *baseURL = [NSURL URLWithString:[OWSFileSystem appSharedDataDirectoryPath]];
                       DebugUIFileBrowser *fileBrowser = [[DebugUIFileBrowser alloc] initWithFileURL:baseURL];
                       [viewController.navigationController pushViewController:fileBrowser animated:YES];
                   }];
    [subsectionItems addObject:sharedDataFileBrowserItem];
    OWSTableItem *documentsFileBrowserItem = [OWSTableItem
        disclosureItemWithText:@"📁 App Container"
                   actionBlock:^{
                       NSString *libraryPath = [OWSFileSystem appLibraryDirectoryPath];
                       NSString *containerPath = [libraryPath stringByDeletingLastPathComponent];
                       NSURL *baseURL = [NSURL fileURLWithPath:containerPath];
                       DebugUIFileBrowser *fileBrowser = [[DebugUIFileBrowser alloc] initWithFileURL:baseURL];
                       [viewController.navigationController pushViewController:fileBrowser animated:YES];
                   }];
    [subsectionItems addObject:documentsFileBrowserItem];
    OWSTableItem *dataStoreItem = [OWSTableItem disclosureItemWithText:@"Data Store Reports"
                                                           actionBlock:^{
                                                               [viewController.navigationController
                                                                   pushViewController:[DebugUIReportsViewController new]
                                                                             animated:YES];
                                                           }];
    [subsectionItems addObject:dataStoreItem];
    [subsectionItems addObject:[self itemForSubsection:[DebugUIGroupsV2 new]
                                        viewController:viewController
                                                thread:thread]];
    [subsectionItems addObject:[self itemForSubsection:[DebugUIPayments new]
                                        viewController:viewController
                                                thread:thread]];
    [subsectionItems addObject:[self itemForSubsection:[DebugUIMisc new] viewController:viewController thread:thread]];

    [contents addSection:[OWSTableSection sectionWithTitle:@"Sections" items:subsectionItems]];

    viewController.contents = contents;
    [viewController presentFromViewController:fromViewController];
}

+ (void)presentDebugUIFromViewController:(UIViewController *)fromViewController
{
    OWSAssertDebug(fromViewController);

    DebugUITableViewController *viewController = [DebugUITableViewController new];

    OWSTableContents *contents = [OWSTableContents new];
    contents.title = @"Debug UI";

    NSMutableArray<OWSTableItem *> *subsectionItems = [NSMutableArray new];
    [subsectionItems addObject:[self itemForSubsection:[DebugUIContacts new] viewController:viewController thread:nil]];
    [subsectionItems
        addObject:[self itemForSubsection:[DebugUIDiskUsage new] viewController:viewController thread:nil]];
    [subsectionItems
        addObject:[self itemForSubsection:[DebugUISessionState new] viewController:viewController thread:nil]];
    [subsectionItems
        addObject:[self itemForSubsection:[DebugUISyncMessages new] viewController:viewController thread:nil]];
    [subsectionItems addObject:[self itemForSubsection:[DebugUIGroupsV2 new] viewController:viewController thread:nil]];
    [subsectionItems addObject:[self itemForSubsection:[DebugUIPayments new] viewController:viewController thread:nil]];
    [subsectionItems addObject:[self itemForSubsection:[DebugUIScreenshots new]
                                        viewController:viewController
                                                thread:nil]];
    [subsectionItems addObject:[self itemForSubsection:[DebugUIMisc new] viewController:viewController thread:nil]];
    [contents addSection:[OWSTableSection sectionWithTitle:@"Sections" items:subsectionItems]];

    viewController.contents = contents;
    [viewController presentFromViewController:fromViewController];
}

+ (BOOL)useDebugUI
{
#ifdef USE_DEBUG_UI
    return YES;
#else
    return NO;
#endif
}

@end

NS_ASSUME_NONNULL_END

#endif