react-native/React/Base/RCTDevMenu.m

/**
 * Copyright (c) 2015-present, Facebook, Inc.
 * All rights reserved.
 *
 * This source code is licensed under the BSD-style license found in the
 * LICENSE file in the root directory of this source tree. An additional grant
 * of patent rights can be found in the PATENTS file in the same directory.
 */

#import "RCTDevMenu.h"

#import "RCTBridge.h"
#import "RCTDefines.h"
#import "RCTKeyCommands.h"
#import "RCTLog.h"
#import "RCTProfile.h"
#import "RCTRootView.h"
#import "RCTSourceCode.h"
#import "RCTUtils.h"

#if RCT_DEV

@interface RCTBridge (Profiling)

- (void)startProfiling;
- (void)stopProfiling;

@end

static NSString *const RCTShowDevMenuNotification = @"RCTShowDevMenuNotification";

@implementation UIWindow (RCTDevMenu)

- (void)RCT_motionEnded:(UIEventSubtype)motion withEvent:(UIEvent *)event
{
  if (event.subtype == UIEventSubtypeMotionShake) {
    [[NSNotificationCenter defaultCenter] postNotificationName:RCTShowDevMenuNotification object:nil];
  }
}

@end

@interface RCTDevMenu () <RCTBridgeModule, RCTInvalidating, UIActionSheetDelegate>

@end

@implementation RCTDevMenu
{
  NSTimer *_updateTimer;
  UIActionSheet *_actionSheet;
}

@synthesize bridge = _bridge;

RCT_EXPORT_MODULE()

+ (void)initialize
{
  // We're swizzling here because it's poor form to override methods in a category,
  // however UIWindow doesn't actually implement motionEnded:withEvent:, so there's
  // no need to call the original implementation.
  RCTSwapInstanceMethods([UIWindow class], @selector(motionEnded:withEvent:), @selector(RCT_motionEnded:withEvent:));
}

- (instancetype)init
{
  if ((self = [super init])) {

    _shakeToShow = YES;
    _liveReloadPeriod = 1.0; // 1 second
    [[NSNotificationCenter defaultCenter] addObserver:self
                                             selector:@selector(showOnShake)
                                                 name:RCTShowDevMenuNotification
                                               object:nil];

#if TARGET_IPHONE_SIMULATOR

    __weak RCTDevMenu *weakSelf = self;
    RCTKeyCommands *commands = [RCTKeyCommands sharedInstance];

    // Workaround around the first cmd+D not working: http://openradar.appspot.com/19613391
    // You can register just the cmd key and do nothing. This will trigger the bug and cmd+R
    // will work like a charm!
    [commands registerKeyCommandWithInput:@""
                            modifierFlags:UIKeyModifierCommand
                                   action:NULL];

    // reload in debug mode
    [commands registerKeyCommandWithInput:@"d"
                            modifierFlags:UIKeyModifierCommand
                                   action:^(UIKeyCommand *command) {
                                     __strong RCTDevMenu *strongSelf = weakSelf;
                                     [strongSelf show];
                                   }];
#endif

  }
  return self;
}

- (void)dealloc
{
  [[NSNotificationCenter defaultCenter] removeObserver:self];
}

- (void)showOnShake
{
  if (_shakeToShow) {
    [self show];
  }
}

- (void)show
{
  if (_actionSheet) {
    return;
  }

  NSString *debugTitleChrome = _bridge.executorClass && _bridge.executorClass == NSClassFromString(@"RCTWebSocketExecutor") ? @"Disable Chrome Debugging" : @"Enable Chrome Debugging";
  NSString *debugTitleSafari = _bridge.executorClass && _bridge.executorClass == NSClassFromString(@"RCTWebViewExecutor") ? @"Disable Safari Debugging" : @"Enable Safari Debugging";
  NSString *liveReloadTitle = _liveReloadEnabled ? @"Disable Live Reload" : @"Enable Live Reload";
  NSString *profilingTitle  = RCTProfileIsProfiling() ? @"Stop Profiling" : @"Start Profiling";

  UIActionSheet *actionSheet =
  [[UIActionSheet alloc] initWithTitle:@"React Native: Development"
                              delegate:self
                     cancelButtonTitle:@"Cancel"
                destructiveButtonTitle:nil
                     otherButtonTitles:@"Reload", debugTitleChrome, debugTitleSafari, liveReloadTitle, profilingTitle, nil];

  actionSheet.actionSheetStyle = UIBarStyleBlack;
  [actionSheet showInView:[UIApplication sharedApplication].keyWindow.rootViewController.view];
  _actionSheet = actionSheet;
}

- (void)actionSheet:(UIActionSheet *)actionSheet clickedButtonAtIndex:(NSInteger)buttonIndex
{
  _actionSheet = nil;

  switch (buttonIndex) {
    case 0: {
      [_bridge reload];
      break;
    }
    case 1: {
      Class cls = NSClassFromString(@"RCTWebSocketExecutor");
      _bridge.executorClass = (_bridge.executorClass != cls) ? cls : nil;
      [_bridge reload];
      break;
    }
    case 2: {
      Class cls = NSClassFromString(@"RCTWebViewExecutor");
      _bridge.executorClass = (_bridge.executorClass != cls) ? cls : Nil;
      [_bridge reload];
      break;
    }
    case 3: {
      self.liveReloadEnabled = !_liveReloadEnabled;
      break;
    }
    case 4: {
      self.profilingEnabled = !_profilingEnabled;
      break;
    }
    default:
      break;
  }
}

- (void)setProfilingEnabled:(BOOL)enabled
{
  if (_profilingEnabled == enabled) {
    return;
  }

  _profilingEnabled = enabled;
  if (RCTProfileIsProfiling()) {
    [_bridge stopProfiling];
  } else {
    [_bridge startProfiling];
  }
}

- (void)setLiveReloadEnabled:(BOOL)enabled
{
  if (_liveReloadEnabled == enabled) {
    return;
  }

  _liveReloadEnabled = enabled;
  if (_liveReloadEnabled) {

    _updateTimer = [NSTimer scheduledTimerWithTimeInterval:_liveReloadPeriod
                                                    target:self
                                                  selector:@selector(pollForUpdates)
                                                  userInfo:nil
                                                   repeats:YES];
  } else {

    [_updateTimer invalidate];
    _updateTimer = nil;
  }
}

- (void)setLiveReloadPeriod:(NSTimeInterval)liveReloadPeriod
{
  _liveReloadPeriod = liveReloadPeriod;
  if (_liveReloadEnabled) {
    self.liveReloadEnabled = NO;
    self.liveReloadEnabled = YES;
  }
}

- (void)pollForUpdates
{
  RCTSourceCode *sourceCodeModule = _bridge.modules[RCTBridgeModuleNameForClass([RCTSourceCode class])];
  if (!sourceCodeModule) {
    RCTLogError(@"RCTSourceCode module not found");
    self.liveReloadEnabled = NO;
  }

  NSURL *longPollURL = [[NSURL alloc] initWithString:@"/onchange" relativeToURL:sourceCodeModule.scriptURL];
  [NSURLConnection sendAsynchronousRequest:[NSURLRequest requestWithURL:longPollURL]
                                     queue:[[NSOperationQueue alloc] init]
                         completionHandler:^(NSURLResponse *response, NSData *data, NSError *connectionError) {

                           NSHTTPURLResponse *HTTPResponse = (NSHTTPURLResponse *)response;
                           if (_liveReloadEnabled && HTTPResponse.statusCode == 205) {
                             [_bridge reload];
                           }
                         }];
}

- (BOOL)isValid
{
  return !_liveReloadEnabled || _updateTimer != nil;
}

- (void)invalidate
{
  [_actionSheet dismissWithClickedButtonIndex:_actionSheet.cancelButtonIndex animated:YES];
  [_updateTimer invalidate];
  _updateTimer = nil;
}

@end

#else // Unvailable

@implementation RCTDevMenu

- (void)show {}

@end

#endif

@implementation  RCTBridge (RCTDevMenu)

- (RCTDevMenu *)devMenu
{
  return self.modules[RCTBridgeModuleNameForClass([RCTDevMenu class])];
}

@end