Merge pull request #150 from alanzeino/master

Add present and dismiss commands for UIViewController

commands/FBDisplayCommands.py

@@ -10,6 +10,7 @@
 import lldb
 
 import fblldbviewhelpers as viewHelpers
+import fblldbviewcontrollerhelpers as viewControllerHelpers
 import fblldbbase as fb
 import fblldbobjcruntimehelpers as runtimeHelpers
 
@@ -22,6 +23,8 @@ def lldbcommands():
     FBUnmaskViewCommand(),
     FBShowViewCommand(),
     FBHideViewCommand(),
+    FBPresentViewControllerCommand(),
+    FBDismissViewControllerCommand(),
     FBSlowAnimationCommand(),
     FBUnslowAnimationCommand()
   ]
@@ -204,6 +207,34 @@ def run(self, args, options):
     viewHelpers.setViewHidden(args[0], True)
 
 
+class FBPresentViewControllerCommand(fb.FBCommand):
+  def name(self):
+    return 'present'
+
+  def description(self):
+    return 'Present a view controller.'
+
+  def args(self):
+    return [ fb.FBCommandArgument(arg='viewController', type='UIViewController *', help='The view controller to present.') ]
+
+  def run(self, args, option):
+    viewControllerHelpers.presentViewController(args[0])
+
+
+class FBDismissViewControllerCommand(fb.FBCommand):
+  def name(self):
+    return 'dismiss'
+
+  def description(self):
+    return 'Dismiss a presented view controller.'
+
+  def args(self):
+    return [ fb.FBCommandArgument(arg='viewController', type='UIViewController *', help='The view controller to dismiss.') ]
+
+  def run(self, args, option):
+    viewControllerHelpers.dismissViewController(args[0])
+
+
 class FBSlowAnimationCommand(fb.FBCommand):
   def name(self):
     return 'slowanim'

fblldbviewcontrollerhelpers.py

@@ -12,6 +12,32 @@
 import fblldbbase as fb
 import fblldbobjcruntimehelpers as runtimeHelpers
 
+def presentViewController(viewController):
+  vc = '(%s)' % (viewController)
+
+  if fb.evaluateBooleanExpression('%s != nil && ((BOOL)[(id)%s isKindOfClass:(Class)[UIViewController class]])' % (vc, vc)):
+    notPresented = fb.evaluateBooleanExpression('[%s presentingViewController] == nil' % vc)
+
+    if notPresented:
+      lldb.debugger.HandleCommand('expr (void)[[[[UIApplication sharedApplication] keyWindow] rootViewController] presentViewController:%s animated:YES completion:nil]' % vc)
+    else:
+      raise Exception('Argument is already presented')
+  else:
+    raise Exception('Argument must be a UIViewController')
+
+def dismissViewController(viewController):
+  vc = '(%s)' % (viewController)
+
+  if fb.evaluateBooleanExpression('%s != nil && ((BOOL)[(id)%s isKindOfClass:(Class)[UIViewController class]])' % (vc, vc)):
+    isPresented = fb.evaluateBooleanExpression('[%s presentingViewController] != nil' % vc)
+
+    if isPresented:
+      lldb.debugger.HandleCommand('expr (void)[(UIViewController *)%s dismissViewControllerAnimated:YES completion:nil]' % vc)
+    else:
+      raise Exception('Argument must be presented')
+  else:
+    raise Exception('Argument must be a UIViewController')
+
 def viewControllerRecursiveDescription(vc):
   return _recursiveViewControllerDescriptionWithPrefixAndChildPrefix(fb.evaluateObjectExpression(vc), '', '', '')