from code import InteractiveConsole
from tempfile import mkstemp
import atexit
import os
import readline
import rlcompleter
import sys


readline.parse_and_bind("tab: complete")


class TermColors(dict):
    color_templates = (
        ("Normal",      "0"),
        ("Black",       "0;30"),
        ("Red",         "0;31"),
        ("Green",       "0;32"),
        ("Brown",       "0;33"),
        ("Blue",        "0;34"),
        ("Purple",      "0;35"),
        ("Cyan",        "0;36"),
        ("LightGray",   "0;37"),
        ("DarkGray",    "1;30"),
        ("LightRed",    "1;31"),
        ("LightGreen",  "1;32"),
        ("Yellow",      "1;33"),
        ("LightBlue",   "1;34"),
        ("LightPurple", "1;35"),
        ("LightCyan",   "1;36"),
        ("White",       "1;37")
    )
    color_base = '\001\033[%sm\002'

    def __init__(self):
        self.update(dict(
            [(k, self.color_base % v) for k, v in self.color_templates]))


class Completer(object):
    def save_history(self):
        import readline

        readline.write_history_file(self.python_histfile)

    def __init__(self):
        self.python_dir = os.path.expanduser(
            "%s/python" % os.environ["XDG_DATA_HOME"])

        if not os.path.exists(self.python_dir):
            os.mkdir(self.python_dir)

        self.python_histfile = os.path.expanduser(
            "%s/history" % self.python_dir)

        if os.path.exists(self.python_histfile):
            readline.read_history_file(self.python_histfile)

        readline.set_history_length(1000)
        atexit.register(self.save_history)


def DisplayHook(value):
    if value is not None:
        try:
            import __builtin__
            __builtin__._ = value
        except ImportError:
            __builtins__._ = value

        import pprint

        pprint.pprint(value)
        del pprint


class EditableBufferInteractiveConsole(InteractiveConsole):
    def __init__(self, *args, **kwargs):
        self.last_buffer = []
        InteractiveConsole.__init__(self, *args, **kwargs)

    def runsource(self, source, *args):
        self.last_buffer = [source.encode('utf-8')]
        return InteractiveConsole.runsource(self, source, *args)

    def raw_input(self, *args):
        line = InteractiveConsole.raw_input(self, *args)

        if line == EDIT_CMD:
            tmp_fd, tmp_file = mkstemp('.py')

            os.write(tmp_fd, b'\n'.join(self.last_buffer))
            os.close(tmp_fd)

            os.system('%s %s' % (EDITOR, tmp_file))

            line = open(tmp_file).read()

            os.unlink(tmp_file)
            tmp_file = ''

            lines = line.split('\n')

            for i in range(len(lines) - 1):
                self.push(lines[i])

            line = lines[-1]
        return line


TC = TermColors()
ps1 = "%sλ%s %s>%s "
sys.ps1 = ps1 % (TC['Blue'],  TC['Normal'],
                 TC['White'], TC['Normal'])
ps2 = "    %s…%s %s>%s "
sys.ps2 = ps2 % (TC['Blue'],  TC['Normal'],
                 TC['White'], TC['Normal'])
sys.displayhook = DisplayHook

C = Completer()
EDITOR = os.environ.get('EDITOR', 'vim')
EDIT_CMD = ':e'
C = EditableBufferInteractiveConsole(locals=locals())
C.interact(banner="")

sys.exit()

# vim: filetype=python