print_json.py

#!/usr/bin/python

# ---------------------------------------------------------------------
# Be sure to add the python path that points to the LLDB shared library.
#
# # To use this in the embedded python interpreter using "lldb" just
# import it with the full path using the "command script import"
# command
#   (lldb) command script import /path/to/cmdtemplate.py
# ---------------------------------------------------------------------

from __future__ import print_function

import inspect
import lldb
import argparse
import shlex
import sys
import os


class PrintJSONDataCommand:
    program = 'pjsond'

    @classmethod
    def register_lldb_command(cls, debugger, module_name):
        parser = cls.create_options()
        cls.__doc__ = parser.format_help()
        # Add any commands contained in this module to LLDB
        command = 'command script add -c %s.%s %s' % (module_name,
                                                      cls.__name__,
                                                      cls.program)
        debugger.HandleCommand(command)
        print('The "{0}" command has been installed, type "help {0}" or "{0} '
              '--help" for detailed help.'.format(cls.program))

    @classmethod
    def create_options(cls):

        description = ('This command prints Data as JSON String')

        
        parser = argparse.ArgumentParser(
            description=description,
            prog=cls.program)
        parser.add_argument('data')

        return parser

    def get_short_help(self):
        return "Prints Data as JSON String"

    def get_long_help(self):
        return self.help_string

    def __init__(self, debugger, unused):
        self.parser = self.create_options()
        self.help_string = self.parser.format_help()

    def __call__(self, debugger, command, exe_ctx, result):
        # Use the Shell Lexer to properly parse up command options just like a
        # shell would
        command_args = shlex.split(command)
        try:
            args = self.parser.parse_args(command_args)
        except:
            # if you don't handle exceptions, passing an incorrect argument to
            # the OptionParser will cause LLDB to exit (courtesy of OptParse
            # dealing with argument errors by throwing SystemExit)
            result.SetError("option parsing failed")
            return
        
        # Run the command and store the result
        print_json_command = "po String(data: try! JSONSerialization.data(withJSONObject: try! JSONSerialization.jsonObject(with: " + args.data + ", options: [.allowFragments]), options: .prettyPrinted), encoding: .utf8)! as NSString"
        debugger.GetCommandInterpreter().HandleCommand(print_json_command, result)

class PrintJSONObjectCommand:
    program = 'pjsono'

    @classmethod
    def register_lldb_command(cls, debugger, module_name):
        parser = cls.create_options()
        cls.__doc__ = parser.format_help()
        # Add any commands contained in this module to LLDB
        command = 'command script add -c %s.%s %s' % (module_name,
                                                      cls.__name__,
                                                      cls.program)
        debugger.HandleCommand(command)
        print('The "{0}" command has been installed, type "help {0}" or "{0} '
              '--help" for detailed help.'.format(cls.program))

    @classmethod
    def create_options(cls):

        description = ('This command prints Encodable or NSObject as JSON String')

        
        parser = argparse.ArgumentParser(
            description=description,
            prog=cls.program)
        parser.add_argument('object')

        return parser

    def get_short_help(self):
        return "Prints Encodable or NSObject as JSON String"

    def get_long_help(self):
        return self.help_string

    def __init__(self, debugger, unused):
        self.parser = self.create_options()
        self.help_string = self.parser.format_help()

    def __call__(self, debugger, command, exe_ctx, result):
        # Use the Shell Lexer to properly parse up command options just like a
        # shell would
        command_args = shlex.split(command)
        try:
            args = self.parser.parse_args(command_args)
        except:
            # if you don't handle exceptions, passing an incorrect argument to
            # the OptionParser will cause LLDB to exit (courtesy of OptParse
            # dealing with argument errors by throwing SystemExit)
            result.SetError("option parsing failed")
            return
        
        # Run the command and store the result
        print_json_command = "po String(data: try! JSONSerialization.data(withJSONObject: " + args.object + ", options: .prettyPrinted), encoding: .utf8)! as NSString"
        debugger.GetCommandInterpreter().HandleCommand(print_json_command, result)

def __lldb_init_module(debugger, dict):
    # Register all classes that have a register_lldb_command method
    for _name, cls in inspect.getmembers(sys.modules[__name__]):
        if inspect.isclass(cls) and callable(getattr(cls,
                                                     "register_lldb_command",
                                                     None)):
            cls.register_lldb_command(debugger, __name__)