import os

import bdb

import types

from Tkinter import *

from WindowList import ListedToplevel

from ScrolledList import ScrolledList

import macosxSupport

class Idb(bdb.Bdb):

    def __init__(self, gui):

        self.gui = gui

        bdb.Bdb.__init__(self)

    def user_line(self, frame):

        if self.in_rpc_code(frame):

            self.set_step()

            return

        message = self.__frame2message(frame)

        self.gui.interaction(message, frame)

    def user_exception(self, frame, info):

        if self.in_rpc_code(frame):

            self.set_step()

            return

        message = self.__frame2message(frame)

        self.gui.interaction(message, frame, info)

    def in_rpc_code(self, frame):

        if frame.f_code.co_filename.count('rpc.py'):

            return True

        else:

            prev_frame = frame.f_back

            if prev_frame.f_code.co_filename.count('Debugger.py'):

                # (that test will catch both Debugger.py and RemoteDebugger.py)

                return False

            return self.in_rpc_code(prev_frame)

    def __frame2message(self, frame):

        code = frame.f_code

        filename = code.co_filename

        lineno = frame.f_lineno

        basename = os.path.basename(filename)

        message = "%s:%s" % (basename, lineno)

        if code.co_name != "?":

            message = "%s: %s()" % (message, code.co_name)

        return message

class Debugger:

    vstack = vsource = vlocals = vglobals = None

    def __init__(self, pyshell, idb=None):

        if idb is None:

            idb = Idb(self)

        self.pyshell = pyshell

        self.idb = idb

        self.frame = None

        self.make_gui()

        self.interacting = 0

    def run(self, *args):

        try:

            self.interacting = 1

            return self.idb.run(*args)

        finally:

            self.interacting = 0

    def close(self, event=None):

        if self.interacting:

            self.top.bell()

            return

        if self.stackviewer:

            self.stackviewer.close(); self.stackviewer = None

        # Clean up pyshell if user clicked debugger control close widget.

        # (Causes a harmless extra cycle through close_debugger() if user

        # toggled debugger from pyshell Debug menu)

        self.pyshell.close_debugger()

        # Now close the debugger control window....

        self.top.destroy()

    def make_gui(self):

        pyshell = self.pyshell

        self.flist = pyshell.flist

        self.root = root = pyshell.root

        self.top = top = ListedToplevel(root)

        self.top.wm_title("Debug Control")

        self.top.wm_iconname("Debug")

        top.wm_protocol("WM_DELETE_WINDOW", self.close)

        self.top.bind("<Escape>", self.close)

        #

        self.bframe = bframe = Frame(top)

        self.bframe.pack(anchor="w")

        self.buttons = bl = []

        #

        self.bcont = b = Button(bframe, text="Go", command=self.cont)

        bl.append(b)

        self.bstep = b = Button(bframe, text="Step", command=self.step)

        bl.append(b)

        self.bnext = b = Button(bframe, text="Over", command=self.next)

        bl.append(b)

        self.bret = b = Button(bframe, text="Out", command=self.ret)

        bl.append(b)

        self.bret = b = Button(bframe, text="Quit", command=self.quit)

        bl.append(b)

        #

        for b in bl:

            b.configure(state="disabled")

            b.pack(side="left")

        #

        self.cframe = cframe = Frame(bframe)

        self.cframe.pack(side="left")

        #

        if not self.vstack:

            self.__class__.vstack = BooleanVar(top)

            self.vstack.set(1)

        self.bstack = Checkbutton(cframe,

            text="Stack", command=self.show_stack, variable=self.vstack)

        self.bstack.grid(row=0, column=0)

        if not self.vsource:

            self.__class__.vsource = BooleanVar(top)

        self.bsource = Checkbutton(cframe,

            text="Source", command=self.show_source, variable=self.vsource)

        self.bsource.grid(row=0, column=1)

        if not self.vlocals:

            self.__class__.vlocals = BooleanVar(top)

            self.vlocals.set(1)

        self.blocals = Checkbutton(cframe,

            text="Locals", command=self.show_locals, variable=self.vlocals)

        self.blocals.grid(row=1, column=0)

        if not self.vglobals:

            self.__class__.vglobals = BooleanVar(top)

        self.bglobals = Checkbutton(cframe,

            text="Globals", command=self.show_globals, variable=self.vglobals)

        self.bglobals.grid(row=1, column=1)

        #

        self.status = Label(top, anchor="w")

        self.status.pack(anchor="w")

        self.error = Label(top, anchor="w")

        self.error.pack(anchor="w", fill="x")

        self.errorbg = self.error.cget("background")

        #

        self.fstack = Frame(top, height=1)

        self.fstack.pack(expand=1, fill="both")

        self.flocals = Frame(top)

        self.flocals.pack(expand=1, fill="both")

        self.fglobals = Frame(top, height=1)

        self.fglobals.pack(expand=1, fill="both")

        #

        if self.vstack.get():

            self.show_stack()

        if self.vlocals.get():

            self.show_locals()

        if self.vglobals.get():

            self.show_globals()

    def interaction(self, message, frame, info=None):

        self.frame = frame

        self.status.configure(text=message)

        #

        if info:

            type, value, tb = info

            try:

                m1 = type.__name__

            except AttributeError:

                m1 = "%s" % str(type)

            if value is not None:

                try:

                    m1 = "%s: %s" % (m1, str(value))

                except:

                    pass

            bg = "yellow"

        else:

            m1 = ""

            tb = None

            bg = self.errorbg

        self.error.configure(text=m1, background=bg)

        #

        sv = self.stackviewer

        if sv:

            stack, i = self.idb.get_stack(self.frame, tb)

            sv.load_stack(stack, i)

        #

        self.show_variables(1)

        #

        if self.vsource.get():

            self.sync_source_line()

        #

        for b in self.buttons:

            b.configure(state="normal")

        #

        self.top.wakeup()

        self.root.mainloop()

        #

        for b in self.buttons:

            b.configure(state="disabled")

        self.status.configure(text="")

        self.error.configure(text="", background=self.errorbg)

        self.frame = None

    def sync_source_line(self):

        frame = self.frame

        if not frame:

            return

        filename, lineno = self.__frame2fileline(frame)

        if filename[:1] + filename[-1:] != "<>" and os.path.exists(filename):

            self.flist.gotofileline(filename, lineno)

    def __frame2fileline(self, frame):

        code = frame.f_code

        filename = code.co_filename

        lineno = frame.f_lineno

        return filename, lineno

    def cont(self):

        self.idb.set_continue()

        self.root.quit()

    def step(self):

        self.idb.set_step()

        self.root.quit()

    def next(self):

        self.idb.set_next(self.frame)

        self.root.quit()

    def ret(self):

        self.idb.set_return(self.frame)

        self.root.quit()

    def quit(self):

        self.idb.set_quit()

        self.root.quit()

    stackviewer = None

    def show_stack(self):

        if not self.stackviewer and self.vstack.get():

            self.stackviewer = sv = StackViewer(self.fstack, self.flist, self)

            if self.frame:

                stack, i = self.idb.get_stack(self.frame, None)

                sv.load_stack(stack, i)

        else:

            sv = self.stackviewer

            if sv and not self.vstack.get():

                self.stackviewer = None

                sv.close()

            self.fstack['height'] = 1

    def show_source(self):

        if self.vsource.get():

            self.sync_source_line()

    def show_frame(self, (frame, lineno)):

        self.frame = frame

        self.show_variables()

    localsviewer = None

    globalsviewer = None

    def show_locals(self):

        lv = self.localsviewer

        if self.vlocals.get():

            if not lv:

                self.localsviewer = NamespaceViewer(self.flocals, "Locals")

        else:

            if lv:

                self.localsviewer = None

                lv.close()

                self.flocals['height'] = 1

        self.show_variables()

    def show_globals(self):

        gv = self.globalsviewer

        if self.vglobals.get():

            if not gv:

                self.globalsviewer = NamespaceViewer(self.fglobals, "Globals")

        else:

            if gv:

                self.globalsviewer = None

                gv.close()

                self.fglobals['height'] = 1

        self.show_variables()

    def show_variables(self, force=0):

        lv = self.localsviewer

        gv = self.globalsviewer

        frame = self.frame

        if not frame:

            ldict = gdict = None

        else:

            ldict = frame.f_locals

            gdict = frame.f_globals

            if lv and gv and ldict is gdict:

                ldict = None

        if lv:

            lv.load_dict(ldict, force, self.pyshell.interp.rpcclt)

        if gv:

            gv.load_dict(gdict, force, self.pyshell.interp.rpcclt)

    def set_breakpoint_here(self, filename, lineno):

        self.idb.set_break(filename, lineno)

    def clear_breakpoint_here(self, filename, lineno):

        self.idb.clear_break(filename, lineno)

    def clear_file_breaks(self, filename):

        self.idb.clear_all_file_breaks(filename)

    def load_breakpoints(self):

        "Load PyShellEditorWindow breakpoints into subprocess debugger"

        pyshell_edit_windows = self.pyshell.flist.inversedict.keys()

        for editwin in pyshell_edit_windows:

            filename = editwin.io.filename

            try:

                for lineno in editwin.breakpoints:

                    self.set_breakpoint_here(filename, lineno)

            except AttributeError:

                continue

class StackViewer(ScrolledList):

    def __init__(self, master, flist, gui):

        if macosxSupport.runningAsOSXApp():

            # At least on with the stock AquaTk version on OSX 10.4 you'll

            # get an shaking GUI that eventually kills IDLE if the width

            # argument is specified.

            ScrolledList.__init__(self, master)

        else:

            ScrolledList.__init__(self, master, width=80)

        self.flist = flist

        self.gui = gui

        self.stack = []

    def load_stack(self, stack, index=None):

        self.stack = stack

        self.clear()

        for i in range(len(stack)):

            frame, lineno = stack[i]

            try:

                modname = frame.f_globals["__name__"]

            except:

                modname = "?"

            code = frame.f_code

            filename = code.co_filename

            funcname = code.co_name

            import linecache

            sourceline = linecache.getline(filename, lineno)

            import string

            sourceline = string.strip(sourceline)

            if funcname in ("?", "", None):

                item = "%s, line %d: %s" % (modname, lineno, sourceline)

            else:

                item = "%s.%s(), line %d: %s" % (modname, funcname,

                                                 lineno, sourceline)

            if i == index:

                item = "> " + item

            self.append(item)

        if index is not None:

            self.select(index)

    def popup_event(self, event):

        "override base method"

        if self.stack:

            return ScrolledList.popup_event(self, event)

    def fill_menu(self):

        "override base method"

        menu = self.menu

        menu.add_command(label="Go to source line",

                         command=self.goto_source_line)

        menu.add_command(label="Show stack frame",

                         command=self.show_stack_frame)

    def on_select(self, index):

        "override base method"

        if 0 <= index < len(self.stack):

            self.gui.show_frame(self.stack[index])

    def on_double(self, index):

        "override base method"

        self.show_source(index)

    def goto_source_line(self):

        index = self.listbox.index("active")

        self.show_source(index)

    def show_stack_frame(self):

        index = self.listbox.index("active")

        if 0 <= index < len(self.stack):

            self.gui.show_frame(self.stack[index])

    def show_source(self, index):

        if not (0 <= index < len(self.stack)):

            return

        frame, lineno = self.stack[index]

        code = frame.f_code

        filename = code.co_filename

        if os.path.isfile(filename):

            edit = self.flist.open(filename)

            if edit:

                edit.gotoline(lineno)

class NamespaceViewer:

    def __init__(self, master, title, dict=None):

        width = 0

        height = 40

        if dict:

            height = 20*len(dict) # XXX 20 == observed height of Entry widget

        self.master = master

        self.title = title

        import repr

        self.repr = repr.Repr()

        self.repr.maxstring = 60

        self.repr.maxother = 60

        self.frame = frame = Frame(master)

        self.frame.pack(expand=1, fill="both")

        self.label = Label(frame, text=title, borderwidth=2, relief="groove")

        self.label.pack(fill="x")

        self.vbar = vbar = Scrollbar(frame, name="vbar")

        vbar.pack(side="right", fill="y")

        self.canvas = canvas = Canvas(frame,

                                      height=min(300, max(40, height)),

                                      scrollregion=(0, 0, width, height))

        canvas.pack(side="left", fill="both", expand=1)

        vbar["command"] = canvas.yview

        canvas["yscrollcommand"] = vbar.set

        self.subframe = subframe = Frame(canvas)

        self.sfid = canvas.create_window(0, 0, window=subframe, anchor="nw")

        self.load_dict(dict)

    dict = -1

    def load_dict(self, dict, force=0, rpc_client=None):

        if dict is self.dict and not force:

            return

        subframe = self.subframe

        frame = self.frame

        for c in subframe.children.values():

            c.destroy()

        self.dict = None

        if not dict:

            l = Label(subframe, text="None")

            l.grid(row=0, column=0)

        else:

            names = dict.keys()

            names.sort()

            row = 0

            for name in names:

                value = dict[name]

                svalue = self.repr.repr(value) # repr(value)

                # Strip extra quotes caused by calling repr on the (already)

                # repr'd value sent across the RPC interface:

                if rpc_client:

                    svalue = svalue[1:-1]

                l = Label(subframe, text=name)

                l.grid(row=row, column=0, sticky="nw")

                l = Entry(subframe, width=0, borderwidth=0)

                l.insert(0, svalue)

                l.grid(row=row, column=1, sticky="nw")

                row = row+1

        self.dict = dict

        # XXX Could we use a <Configure> callback for the following?

        subframe.update_idletasks() # Alas!

        width = subframe.winfo_reqwidth()

        height = subframe.winfo_reqheight()

        canvas = self.canvas

        self.canvas["scrollregion"] = (0, 0, width, height)

        if height > 300:

            canvas["height"] = 300

            frame.pack(expand=1)

        else:

            canvas["height"] = height

            frame.pack(expand=0)

    def close(self):

        self.frame.destroy()