#!/usr/bin/env python3

#   Interactive Python-GTK Console
#   Copyright (C), 1998 James Henstridge <james@daa.com.au>
#
#    This program is free software; you can redistribute it and/or modify
#   it under the terms of the GNU General Public License as published by
#   the Free Software Foundation; either version 2 of the License, or
#   (at your option) any later version.
#
#   This program is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#   GNU General Public License for more details.
#
#   You should have received a copy of the GNU General Public License
#   along with this program; if not, write to the Free Software
#   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

# This module implements an interactive python session in a GTK window.  To
# start the session, use the gtk_console command.  Its specification is:
#   gtk_console(namespace, title, copyright)
# where namespace is a dictionary representing the namespace of the session,
#       title     is the title on the window and
#       copyright is any additional copyright info to print.
#
# As well as the starting attributes in namespace, the session will also
# have access to the list __history__, which is the command history.

import warnings
import sys, traceback

import gi

gi.require_version('Pango', '1.0')
gi.require_version('Gtk', '2.0')

with warnings.catch_warnings():
    warnings.filterwarnings("ignore", category=RuntimeWarning)
    from gi.repository import GObject, Pango, Gtk, Gdk

import gettext

_ = gettext.gettext

stdout = sys.stdout

if not hasattr(sys, "ps1"):
    sys.ps1 = ">>> "
if not hasattr(sys, "ps2"):
    sys.ps2 = "... "

# some functions to help recognise breaks between commands
def remQuotStr(s):
    """Returns s with any quoted strings removed (leaving quote marks)"""
    r = ""
    inq = 0
    qt = ""
    prev = "_"
    while len(s):
        s0, s = s[0], s[1:]
        if inq and (s0 != qt or prev == "\\"):
            prev = s0
            continue
        prev = s0
        if s0 in "'\"":
            if inq:
                inq = 0
            else:
                inq = 1
                qt = s0
        r = r + s0
    return r


def bracketsBalanced(s):
    """Returns true iff the brackets in s are balanced"""
    s = [x for x in s if x in "()[]{}"]
    stack = []
    brackets = {"(": ")", "[": "]", "{": "}"}
    while len(s) != 0:
        if s[0] in ")]}":
            if len(stack) != 0 and brackets[stack[-1]] == s[0]:
                del stack[-1]
            else:
                return 0
        else:
            stack.append(s[0])
        s = s[1:]
    return len(stack) == 0


class gtkoutfile:
    """A fake output file object.  It sends output to a TK test widget,
    and if asked for a file number, returns one set on instance creation"""

    def __init__(self, w, fn, tag):
        self.__fn = fn
        self.__w = w
        self.__tag = tag

    def close(self):
        pass

    flush = close

    def fileno(self):
        return self.__fn

    def isatty(self):
        return 0

    def read(self, a):
        return ""

    def readline(self):
        return ""

    def readlines(self):
        return []

    def write(self, s):
        # stdout.write(str(self.__w.get_point()) + '\n')
        self.__w.text_insert(self.__tag, s)

    def writelines(self, l):
        self.__w.text_insert(self.__tag, l)

    def seek(self, a):
        raise IOError(29, "Illegal seek")

    def tell(self):
        raise IOError(29, "Illegal seek")

    truncate = tell


class Console(Gtk.VBox):
    def __init__(self, namespace={}, copyright="", quit_cb=None):
        GObject.GObject.__init__(self, spacing=2)
        self.copyright = copyright

        self.quit_cb = quit_cb

        self.inp = Gtk.HBox()
        self.pack_start(self.inp, True, True, 0)
        self.inp.show()

        self.scroll = Gtk.ScrolledWindow()
        self.inp.pack_start(self.scroll, True, True, 0)
        self.scroll.set_policy(Gtk.PolicyType.NEVER, Gtk.PolicyType.ALWAYS)
        self.scroll.show()

        self.text = Gtk.TextView()
        self.text.modify_font(Pango.FontDescription("monospace 8"))
        self.text.set_editable(False)
        self.text.set_wrap_mode(Gtk.WrapMode.WORD)
        self.text.set_size_request(500, 400)
        self.scroll.add(self.text)
        self.text.show()

        buffer = self.text.get_buffer()
        self.normal = buffer.create_tag("normal")
        self.title = buffer.create_tag("title")
        self.title.set_property("weight", 600)
        self.error = buffer.create_tag("error")
        self.error.set_property("foreground", "red")
        self.command = buffer.create_tag("command")
        self.command.set_property("weight", 600)
        self.command.set_property("foreground", "blue")

        self.inputbox = Gtk.HBox(spacing=2)
        self.pack_end(self.inputbox, False, True, 0)
        self.inputbox.show()

        self.prompt = Gtk.Label(label=sys.ps1)
        self.prompt.set_padding(2, 0)
        self.prompt.set_size_request(26, -1)
        self.inputbox.pack_start(self.prompt, False, False, 0)
        self.prompt.show()

        self.line = Gtk.Entry()
        self.line.set_size_request(400, -1)
        self.line.connect("key_press_event", self.key_function)
        self.inputbox.pack_start(self.line, True, True, 2)
        self.line.show()

        # now let the text box be resized
        self.text.set_size_request(0, 0)
        self.line.set_size_request(0, -1)

        self.namespace = namespace

        self.cmd = ""
        self.cmd2 = ""

        # set up hooks for standard output.
        self.stdout = gtkoutfile(self, sys.stdout.fileno(), self.normal)
        try:
            # this will mostly fail on win32 ...
            self.stderr = gtkoutfile(self, sys.stderr.fileno(), self.error)
        except:
            # ... but gtkoutfile is not using the fileno anyway
            self.stderr = gtkoutfile(self, -1, self.error)
        # set up command history
        self.history = [""]
        self.histpos = 0
        self.namespace["__history__"] = self.history

    def scroll_to_end(self):
        iter = self.text.get_buffer().get_end_iter()
        self.text.scroll_to_iter(iter, 0, False, 0, 0)
        return False  # don't requeue this handler

    def text_insert(self, tag, s):
        buffer = self.text.get_buffer()
        iter = buffer.get_end_iter()
        buffer.insert_with_tags(iter, s, tag)
        GObject.idle_add(self.scroll_to_end)

    def init(self):
        self.text.realize()
        # ??? self.text.style = self.text.get_style()
        # self.text.fg = self.text.style.fg[Gtk.StateType.NORMAL]
        # self.text.bg = self.text.style.white

        self.text_insert(
            self.title,
            "Python %s\n%s\n\n" % (sys.version, sys.copyright)
            + "Interactive Python-GTK Console - "
            + "Copyright (C) 1998 James Henstridge\n\n"
            + self.copyright
            + "\n",
        )
        self.line.grab_focus()

    def quit(self, *args):
        self.destroy()
        if self.quit_cb:
            self.quit_cb()

    def key_function(self, entry, event):
        if event.keyval == Gdk.KEY_Return:
            self.line.emit_stop_by_name("key_press_event")
            self.eval()
        if event.keyval == Gdk.KEY_Tab:
            self.line.emit_stop_by_name("key_press_event")
            self.line.append_text("\t")
            GObject.idle_add(self.focus_text)
        elif event.keyval in (Gdk.KEY_KP_Up, Gdk.KEY_Up):
            self.line.emit_stop_by_name("key_press_event")
            self.historyUp()
            GObject.idle_add(self.focus_text)
        elif event.keyval in (Gdk.KEY_KP_Down, Gdk.KEY_Down):
            self.line.emit_stop_by_name("key_press_event")
            self.historyDown()
            GObject.idle_add(self.focus_text)
        elif (
            event.keyval in (Gdk.KEY_D, Gdk.KEY_d)
            and event.get_state() & Gdk.ModifierType.CONTROL_MASK
        ):
            self.line.emit_stop_by_name("key_press_event")
            self.ctrld()

    def focus_text(self):
        self.line.grab_focus()
        return False  # don't requeue this handler

    def ctrld(self):
        self.quit()
        pass

    def historyUp(self):
        if self.histpos > 0:
            l = self.line.get_text()
            if len(l) > 0 and l[0] == "\n":
                l = l[1:]
            if len(l) > 0 and l[-1] == "\n":
                l = l[:-1]
            self.history[self.histpos] = l
            self.histpos = self.histpos - 1
            self.line.set_text(self.history[self.histpos])

    def historyDown(self):
        if self.histpos < len(self.history) - 1:
            l = self.line.get_text()
            if len(l) > 0 and l[0] == "\n":
                l = l[1:]
            if len(l) > 0 and l[-1] == "\n":
                l = l[:-1]
            self.history[self.histpos] = l
            self.histpos = self.histpos + 1
            self.line.set_text(self.history[self.histpos])

    def eval(self):
        l = self.line.get_text() + "\n"
        if len(l) > 1 and l[0] == "\n":
            l = l[1:]
        self.histpos = len(self.history) - 1
        if len(l) > 0 and l[-1] == "\n":
            self.history[self.histpos] = l[:-1]
        else:
            self.history[self.histpos] = l
        self.line.set_text("")
        self.text_insert(self.command, self.prompt.get_label() + l)
        if l == "\n":
            self.run(self.cmd)
            self.cmd = ""
            self.cmd2 = ""
            return
        self.histpos = self.histpos + 1
        self.history.append("")
        self.cmd = self.cmd + l
        self.cmd2 = self.cmd2 + remQuotStr(l)
        l = l.rstrip()
        if (
            not bracketsBalanced(self.cmd2)
            or l[-1] == ":"
            or l[-1] == "\\"
            or l[0] in " \11"
        ):
            self.prompt.set_text(sys.ps2)
            self.prompt.queue_draw()
            return
        self.run(self.cmd)
        self.cmd = ""
        self.cmd2 = ""

    def run(self, cmd):
        sys.stdout, self.stdout = self.stdout, sys.stdout
        sys.stderr, self.stderr = self.stderr, sys.stderr
        try:
            try:
                r = eval(cmd, self.namespace, self.namespace)
                if r is not None:
                    print(repr(r))
            except SyntaxError:
                exec(cmd, self.namespace)
        except:
            if hasattr(sys, "last_type") and sys.last_type == SystemExit:
                self.quit()
            else:
                traceback.print_exc()
        self.prompt.set_text(sys.ps1)
        self.prompt.queue_draw()
        # adj = self.text.get_vadjustment()
        # adj.set_value(adj.upper - adj.page_size)
        sys.stdout, self.stdout = self.stdout, sys.stdout
        sys.stderr, self.stderr = self.stderr, sys.stderr


def gtk_console(ns, title="Python", copyright="", menu=None):
    win = Gtk.Window()
    win.set_default_size(475, 300)
    # win.connect("destroy", mainquit)
    win.set_title(title)

    def quit(win=win):
        win.destroy()

    cons = Console(namespace=ns, copyright=copyright, quit_cb=quit)
    if menu:
        box = Gtk.VBox()
        win.add(box)
        box.show()
        box.pack_start(menu, False, True, 0)
        menu.show()
        box.pack_start(cons, True, True, 0)
    else:
        win.add(cons)
    cons.show()
    win.show()
    cons.init()


# set up as a dia plugin
try:
    import dia

    def open_console(data, flags):
        gtk_console(
            {
                "__builtins__": __builtins__,
                "__name__": "__main__",
                "__doc__": None,
                "dia": dia,
            },
            _("Python Dia Console"),
        )

    dia.register_action(
        "DialogsPythonconsole",
        _("_Python Console"),
        "/DisplayMenu/Dialogs/DialogsExtensionStart",
        open_console,
    )

except:
    print("Failed to import Dia ...")
    gtk_console({"__builtins__": __builtins__, "__name__": "__main__", "__doc__": None})
    Gtk.main()
