Feature: Accessibility Tree printing

README.md

@@ -3,6 +3,8 @@
 
 [[Installation](#installation) • [Commands](#commands) • [Custom Commands](#custom-commands) • [Development Workflow](#development-workflow) [Contributing](#contributing) • [License](#license)]
 
+For a comprehensive overview of LLDB, and how Chisel compliments it, read Ari Grant's [Dancing in the Debugger — A Waltz with LLDB](http://www.objc.io/issue-19/lldb-debugging.html) in issue 19 of [objc.io](http://www.objc.io/).
+
 ## Installation
 
 ```

commands/FBAutoLayoutCommands.py

@@ -36,13 +36,13 @@ def run(self, arguments, options):
 def setBorderOnAmbiguousViewRecursive(view, width, color):
   if not fb.evaluateBooleanExpression('[(id)%s isKindOfClass:(Class)[UIView class]]' % view):
     return
-  
+
   isAmbiguous = fb.evaluateBooleanExpression('(BOOL)[%s hasAmbiguousLayout]' % view)
   if isAmbiguous:
     layer = viewHelpers.convertToLayer(view)
     lldb.debugger.HandleCommand('expr (void)[%s setBorderWidth:(CGFloat)%s]' % (layer, width))
     lldb.debugger.HandleCommand('expr (void)[%s setBorderColor:(CGColorRef)[(id)[UIColor %sColor] CGColor]]' % (layer, color))
-  
+
   subviews = fb.evaluateExpression('(id)[%s subviews]' % view)
   subviewsCount = int(fb.evaluateExpression('(int)[(id)%s count]' % subviews))
   if subviewsCount > 0:

commands/FBDebugCommands.py

@@ -31,12 +31,12 @@ def run(self, arguments, options):
 
     objectAddress = int(fb.evaluateObjectExpression(commandForObject), 0)
 
-    ivarOffsetCommand = '(ptrdiff_t)ivar_getOffset((Ivar)object_getInstanceVariable((id){}, "{}", 0))'.format(objectAddress, ivarName)
+    ivarOffsetCommand = '(ptrdiff_t)ivar_getOffset((void*)object_getInstanceVariable((id){}, "{}", 0))'.format(objectAddress, ivarName)
     ivarOffset = fb.evaluateIntegerExpression(ivarOffsetCommand)
 
     # A multi-statement command allows for variables scoped to the command, not permanent in the session like $variables.
     ivarSizeCommand = ('unsigned int size = 0;'
-                       'char *typeEncoding = (char *)ivar_getTypeEncoding((Ivar)class_getInstanceVariable((Class)object_getClass((id){}), "{}"));'
+                       'char *typeEncoding = (char *)ivar_getTypeEncoding((void*)class_getInstanceVariable((Class)object_getClass((id){}), "{}"));'
                        '(char *)NSGetSizeAndAlignment(typeEncoding, &size, 0);'
                        'size').format(objectAddress, ivarName)
     ivarSize = int(fb.evaluateExpression(ivarSizeCommand), 0)
@@ -82,7 +82,17 @@ def args(self):
   def run(self, arguments, options):
     expression = arguments[0]
 
-    match = re.match(r'([-+])*\[(.*) (.*)\]', expression)
+    methodPattern = re.compile(r"""
+      (?P<scope>[-+])?
+      \[
+        (?P<target>.*?)
+        (?P<category>\(.+\))?
+        \s+
+        (?P<selector>.*)
+      \]
+""", re.VERBOSE)
+
+    match = methodPattern.match(expression)
 
     if not match:
       print 'Failed to parse expression. Do you even Objective-C?!'
@@ -93,9 +103,10 @@ def run(self, arguments, options):
       print 'Your architecture, {}, is truly fantastic. However, I don\'t currently support it.'.format(arch)
       return
 
-    methodTypeCharacter = match.group(1)
-    classNameOrExpression = match.group(2)
-    selector = match.group(3)
+    methodTypeCharacter = match.group('scope')
+    classNameOrExpression = match.group('target')
+    category = match.group('category')
+    selector = match.group('selector')
 
     methodIsClassMethod = (methodTypeCharacter == '+')
 
@@ -135,7 +146,8 @@ def run(self, arguments, options):
       return
 
     breakpointClassName = objc.class_getName(nextClass)
-    breakpointFullName = '{}[{} {}]'.format(methodTypeCharacter, breakpointClassName, selector)
+    formattedCategory = category if category else ''
+    breakpointFullName = '{}[{}{} {}]'.format(methodTypeCharacter, breakpointClassName, formattedCategory, selector)
 
     breakpointCondition = None
     if targetIsClass:
@@ -145,7 +157,11 @@ def run(self, arguments, options):
 
     print 'Setting a breakpoint at {} with condition {}'.format(breakpointFullName, breakpointCondition)
 
-    lldb.debugger.HandleCommand('breakpoint set --fullname "{}" --condition "{}"'.format(breakpointFullName, breakpointCondition))
+    if category:
+      lldb.debugger.HandleCommand('breakpoint set --fullname "{}" --condition "{}"'.format(breakpointFullName, breakpointCondition))
+    else:
+      breakpointPattern = '{}\[{}(\(.+\))? {}\]'.format(methodTypeCharacter, breakpointClassName, selector)
+      lldb.debugger.HandleCommand('breakpoint set --func-regex "{}" --condition "{}"'.format(breakpointPattern, breakpointCondition))
 
 def classItselfImplementsSelector(klass, selector):
   thisMethod = objc.class_getInstanceMethod(klass, selector)

commands/FBDisplayCommands.py

@@ -46,10 +46,10 @@ def options(self):
   def run(self, args, options):
     colorClassName = 'UIColor'
     isMac = runtimeHelpers.isMacintoshArch()
-    
+
     if isMac:
       colorClassName = 'NSColor'
-    
+
     layer = viewHelpers.convertToLayer(args[0])
     lldb.debugger.HandleCommand('expr (void)[%s setBorderWidth:(CGFloat)%s]' % (layer, options.width))
     lldb.debugger.HandleCommand('expr (void)[%s setBorderColor:(CGColorRef)[(id)[%s %sColor] CGColor]]' % (layer, colorClassName, options.color))

commands/FBFindCommands.py

@@ -98,7 +98,7 @@ def printMatchesInViewOutputStringAndCopyFirstToClipboard(needle, haystack):
     view = match.groups()[-1]
     className = fb.evaluateExpressionValue('(id)[(' + view + ') class]').GetObjectDescription()
     print('{} {}'.format(view, className))
-    if first == None:
+    if first is None:
       first = view
       cmd = 'echo %s | tr -d "\n" | pbcopy' % view
       os.system(cmd)
@@ -125,7 +125,7 @@ def run(self, arguments, options):
       if re.match(r'.*' + needle + '.*', a11yLabel, re.IGNORECASE):
         print('{} {}'.format(view, a11yLabel))
 
-        if first == None:
+        if first is None:
           first = view
           cmd = 'echo %s | tr -d "\n" | pbcopy' % first
           os.system(cmd)

commands/FBFlickerCommands.py

@@ -110,10 +110,10 @@ def inputCallback(self, input):
     elif input == 'p':
       recusionName = 'recursiveDescription'
       isMac = runtimeHelpers.isMacintoshArch()
-      
+
       if isMac:
         recursionName = '_subtreeDescription'
-      
+
       lldb.debugger.HandleCommand('po [(id)' + oldView + ' ' + recusionName + ']')
     else:
       print '\nI really have no idea what you meant by \'' + input + '\'... =\\\n'

commands/FBInvocationCommands.py

@@ -89,7 +89,7 @@ def stackStartAddressInSelectedFrame(frame):
     return int(frame.EvaluateExpression('($esp + 8)').GetValue())
   else:
     return int(frame.EvaluateExpression('($ebp + 8)').GetValue())
-    
+
 
 def findArgAtIndexFromStackFrame(frame, index):
   return fb.evaluateExpression('*(int *)' + str(findArgAdressAtIndexFromStackFrame(frame, index)))
@@ -134,36 +134,36 @@ def prettyPrintInvocation(frame, invocation):
         print readableString
       else:
         if encoding[0] == '{':
-          encoding = encoding[1:len(encoding)-1]
+          encoding = encoding[1:]
         print (hex(address) + ', address of ' + encoding + ' ' + description).strip()
 
       index += 1
 
 def argumentAsString(frame, address, encoding):
   if encoding[0] == '{':
-    encoding = encoding[1:len(encoding)-1]
+    encoding = encoding[1:]
 
   encodingMap = {
-    'c' : 'char',
-    'i' : 'int',
-    's' : 'short',
-    'l' : 'long',
-    'q' : 'long long',
-
-    'C' : 'unsigned char',
-    'I' : 'unsigned int',
-    'S' : 'unsigned short',
-    'L' : 'unsigned long',
-    'Q' : 'unsigned long long',
-
-    'f' : 'float',
-    'd' : 'double',
-    'B' : 'bool',
-    'v' : 'void',
-    '*' : 'char *',
-    '@' : 'id',
-    '#' : 'Class',
-    ':' : 'SEL',
+    'c': 'char',
+    'i': 'int',
+    's': 'short',
+    'l': 'long',
+    'q': 'long long',
+
+    'C': 'unsigned char',
+    'I': 'unsigned int',
+    'S': 'unsigned short',
+    'L': 'unsigned long',
+    'Q': 'unsigned long long',
+
+    'f': 'float',
+    'd': 'double',
+    'B': 'bool',
+    'v': 'void',
+    '*': 'char *',
+    '@': 'id',
+    '#': 'Class',
+    ':': 'SEL',
   }
 
   pointers = ''
@@ -202,5 +202,5 @@ def argumentAsString(frame, address, encoding):
         description = value.GetSummary()
       if description:
         return type + ': ' + description
-  
+
   return None

commands/FBPrintCommands.py

@@ -15,6 +15,7 @@
 import fblldbviewcontrollerhelpers as vcHelpers
 import fblldbviewhelpers as viewHelpers
 import fblldbobjcruntimehelpers as runtimeHelpers
+import fblldbobjecthelpers as objectHelpers
 
 def lldbcommands():
   return [
@@ -28,6 +29,8 @@ def lldbcommands():
     FBPrintOnscreenTableViewCells(),
     FBPrintInternals(),
     FBPrintInstanceVariable(),
+    FBPrintKeyPath(),
+    FBPrintAccessibilityTree(),
   ]
 
 class FBPrintViewHierarchyCommand(fb.FBCommand):
@@ -49,10 +52,10 @@ def args(self):
   def run(self, arguments, options):
     maxDepth = int(options.depth)
     isMac = runtimeHelpers.isMacintoshArch()
-    
-    if (arguments[0] == '__keyWindow_dynamic__'):
+
+    if arguments[0] == '__keyWindow_dynamic__':
       arguments[0] = '(id)[[UIApplication sharedApplication] keyWindow]'
-      
+
       if isMac:
         arguments[0] = '(id)[[[[NSApplication sharedApplication] windows] objectAtIndex:0] contentView]'
 
@@ -65,9 +68,9 @@ def run(self, arguments, options):
         print 'Failed to walk view hierarchy. Make sure you pass a view, not any other kind of object or expression.'
     else:
       printingMethod = 'recursiveDescription'
-      if (isMac):
+      if isMac:
         printingMethod = '_subtreeDescription'
-      
+
       description = fb.evaluateExpressionValue('(id)[' + arguments[0] + ' ' + printingMethod + ']').GetObjectDescription()
       if maxDepth > 0:
         separator = re.escape("   | ")
@@ -100,10 +103,14 @@ def args(self):
 
   def run(self, arguments, options):
     isMac = runtimeHelpers.isMacintoshArch()
-
-    if (arguments[0] == '__keyWindow_rootVC_dynamic__'):
+
+    if arguments[0] == '__keyWindow_rootVC_dynamic__':
+      if fb.evaluateBooleanExpression('[UIViewController respondsToSelector:@selector(_printHierarchy)]'):
+        lldb.debugger.HandleCommand('po [UIViewController _printHierarchy]')
+        return
+
       arguments[0] = '(id)[(id)[[UIApplication sharedApplication] keyWindow] rootViewController]'
-      if (isMac):
+      if isMac:
         arguments[0] = '(id)[[[[NSApplication sharedApplication] windows] objectAtIndex:0] contentViewController]'
 
     print vcHelpers.viewControllerRecursiveDescription(arguments[0])
@@ -123,7 +130,7 @@ def run(self, arguments, options):
 def _printIterative(initialValue, generator):
   indent = 0
   for currentValue in generator(initialValue):
-    print '   | '*indent + currentValue
+    print '   | ' * indent + currentValue
     indent += 1
 
 
@@ -267,8 +274,46 @@ def run(self, arguments, options):
     object = fb.evaluateObjectExpression(commandForObject)
     objectClass = fb.evaluateExpressionValue('(id)[(' + object + ') class]').GetObjectDescription()
 
-    ivarTypeCommand = '((char *)ivar_getTypeEncoding((Ivar)object_getInstanceVariable((id){}, \"{}\", 0)))[0]'.format(object, ivarName)
+    ivarTypeCommand = '((char *)ivar_getTypeEncoding((void*)object_getInstanceVariable((id){}, \"{}\", 0)))[0]'.format(object, ivarName)
     ivarTypeEncodingFirstChar = fb.evaluateExpression(ivarTypeCommand)
 
     printCommand = 'po' if ('@' in ivarTypeEncodingFirstChar) else 'p'
     lldb.debugger.HandleCommand('{} (({} *)({}))->{}'.format(printCommand, objectClass, object, ivarName))
+
+class FBPrintKeyPath(fb.FBCommand):
+  def name(self):
+    return 'pkp'
+
+  def description(self):
+    return "Print out the value of the key path expression using -valueForKeyPath:"
+
+  def args(self):
+    return [
+      fb.FBCommandArgument(arg='keypath', type='NSString *', help='The keypath to print'),
+    ]
+
+  def run(self, arguments, options):
+    command = arguments[0]
+    if len(command.split('.')) == 1:
+      lldb.debugger.HandleCommand("po " + command)
+    else:
+      objectToMessage, keypath = command.split('.', 1)
+      object = fb.evaluateObjectExpression(objectToMessage)
+      printCommand = 'po [{} valueForKeyPath:@"{}"]'.format(object, keypath)
+      lldb.debugger.HandleCommand(printCommand)
+
+class FBPrintAccessibilityTree(fb.FBCommand):
+  def name(self):
+    return 'pae'
+
+  def description(self):
+    return "Print out the accessibility heirarchy.   Traverses the accessibilityElements if present and the accessibility inspector is  enabled, otherwise the subviews."
+
+  def args(self):
+    return [
+      fb.FBCommandArgument(arg='object', type='id', default="[[UIApplication sharedApplication] keyWindow]", help='The object to print accessibility information for'),
+    ]
+
+  def run(self, arguments, options):
+    object = fb.evaluateObjectExpression(arguments[0])
+    return viewHelpers.accessibilityRecursiveDescription(object, "")