# vim:fileencoding=utf-8:noet
from weakref import ref
from atexit import register as atexit

from IPython.terminal.prompts import Prompts
from pygments.token import Token  # NOQA

from powerline.ipython import IPythonPowerline
from powerline.renderers.ipython.since_7 import PowerlinePromptStyle
from powerline.bindings.ipython.post_0_11 import PowerlineMagics


class ConfigurableIPythonPowerline(IPythonPowerline):
    def init(self, ip):
        config = ip.config.Powerline
        self.config_overrides = config.get('config_overrides')
        self.theme_overrides = config.get('theme_overrides', {})
        self.config_paths = config.get('config_paths')
        super(ConfigurableIPythonPowerline, self).init(
            renderer_module='.since_7')

    def do_setup(self, ip, prompts):
        prompts.powerline = self

        msfn_missing = ()
        saved_msfn = getattr(ip, '_make_style_from_name', msfn_missing)

        if hasattr(saved_msfn, 'powerline_original'):
            saved_msfn = saved_msfn.powerline_original

        def _make_style_from_name(ip, name):
            prev_style = saved_msfn(name)
            new_style = PowerlinePromptStyle(lambda: prev_style)
            return new_style

        _make_style_from_name.powerline_original = saved_msfn

        if not isinstance(ip._style, PowerlinePromptStyle):
            prev_style = ip._style
            ip._style = PowerlinePromptStyle(lambda: prev_style)

        if not isinstance(saved_msfn, type(self.init)):
            _saved_msfn = saved_msfn
            saved_msfn = lambda: _saved_msfn(ip)

        if saved_msfn is not msfn_missing:
            ip._make_style_from_name = _make_style_from_name

        magics = PowerlineMagics(ip, self)
        ip.register_magics(magics)

        atexit(self.shutdown)


class PowerlinePrompts(Prompts):
    '''Class that returns powerline prompts
    '''
    def __init__(self, shell):
        powerline = ConfigurableIPythonPowerline(shell)
        self.shell = shell
        powerline.do_setup(shell, self)
        self.last_output_count = None
        self.last_output = {}

    for prompt in ('in', 'continuation', 'rewrite', 'out'):
        exec((
            'def {0}_prompt_tokens(self, *args, **kwargs):\n'
            '    if self.last_output_count != self.shell.execution_count:\n'
            '        self.last_output.clear()\n'
            '        self.last_output_count = self.shell.execution_count\n'
            '    if "{0}" not in self.last_output:\n'
            '        self.last_output["{0}"] = self.powerline.render('
            '            side="left",'
            '            matcher_info="{1}",'
            '            segment_info=self.shell,'
            '        ) + [(Token.Generic.Prompt, " ")]\n'
            '    return self.last_output["{0}"]'
        ).format(prompt, 'in2' if prompt == 'continuation' else prompt))