# -----------------------------------------------------------------------
# Debugger command prompt with CustomViewers
# (c) Hex-Rays
#
import idaapi
import idc
from idaapi import simplecustviewer_t

def SendDbgCommand(cmd):
    """Sends a command to the debugger and returns the output string.
    An exception will be raised if the debugger is not running or the current debugger does not export
    the 'send_dbg_command' IDC command.
    """
    s = idc.eval('send_dbg_command("%s");' % cmd)
    if s.startswith("IDC_FAILURE"):
        raise Exception, "Debugger command is available only when the debugger is active!"
    return s

# -----------------------------------------------------------------------
class dbgcmd_t(simplecustviewer_t):
    def Create(self):
        # Form the title
        title = "Debugger command window"
        # Create the customview
        if not simplecustviewer_t.Create(self, title):
            return False
        self.last_cmd   = ""
        self.menu_clear = self.AddPopupMenu("Clear")
        self.menu_cmd   = self.AddPopupMenu("New command")

        self.ResetOutput()
        return True

    def IssueCommand(self):
        s = idaapi.ask_str(self.last_cmd, 0, "Please enter a debugger command")
        if not s:
            return

        # Save last command
        self.last_cmd = s

        # Add it using a different color
        self.AddLine("debugger>" + idaapi.COLSTR(s, idaapi.SCOLOR_VOIDOP))

        try:
            r = SendDbgCommand(s).split("\n")
            for s in r:
                self.AddLine(idaapi.COLSTR(s, idaapi.SCOLOR_LIBNAME))
        except:
            self.AddLine(idaapi.COLSTR("Debugger is not active or does not export send_dbg_command()", idaapi.SCOLOR_ERROR))
        self.Refresh()

    def ResetOutput(self):
        self.ClearLines()
        self.AddLine(idaapi.COLSTR("Please press INS to enter command; X to clear output", idaapi.SCOLOR_AUTOCMT))
        self.Refresh()

    def OnKeydown(self, vkey, shift):
        # ESCAPE?
        if vkey == 27:
            self.Close()
        # VK_INSERT
        elif vkey == 45:
            self.IssueCommand()
        elif vkey == ord('X'):
            self.ResetOutput()
        else:
            return False
        return True

    def OnPopupMenu(self, menu_id):
        if menu_id == self.menu_clear:
            self.ResetOutput()
        elif menu_id == self.menu_cmd:
            self.IssueCommand()
        else:
            # Unhandled
            return False
        return True

# -----------------------------------------------------------------------
def show_win():
    x = dbgcmd_t()
    if not x.Create():
        print "Failed to create debugger command line!"
        return None
    x.Show()
    return x

try:
    # created already?
    dbgcmd
    dbgcmd.Close()
    del dbgcmd
except:
    pass

dbgcmd = show_win()
if not dbgcmd:
    del dbgcmd