menu.py

# -*- coding: utf-8 -*-

from functools import partial

import sublime

from .st3_CommandsBase.WindowCommand import stWindowCommand


class Action(object):
    def __init__(self, text, func, id=None):
        self.id = id if id else text
        self.text = text
        self.func = func

    def isCheckbox(self):
        return False


class CheckBox(Action):
    def __init__(self, text, id=None, checked=False):
        Action.__init__(self, text, self.change, id)
        self.checked = checked
        self._text = text
        self.text = self.cb() + text

    def change(self, parent, selectedId, options):
        self.checked = not self.checked
        self.text = self.cb() + self._text
        parent()

    def cb(self):
        return '[x] ' if self.checked else '[ ] '

    def isCheckbox(self):
        return True


def  menu(refresh=False, temp=False):
    def _menu(getActions):
        def impl(self, *args, **kwargs):
            return self.menu(
                getActions=partial(getActions, self, *args, **kwargs),
                refresh=refresh,
                temp=temp)

        return impl

    return _menu


def action(terminate=False):
    def _action(func):
        def impl(self, *args, **kwargs):
            return self.action(
                func=partial(func, self, *args, **kwargs),
                terminate=terminate)

        return impl

    return _action


class Menu:
    def menu(self, getActions, refresh=False, temp=False):
        def impl(parent=None, selectedId=None, options=None):
            actions = getActions()
            defaultSelectedId = None
            if isinstance(actions, tuple):
                assert len(actions) == 2
                actions, defaultSelectedId = actions[0], actions[1]

            assert isinstance(actions, list)
            actions = [Action(a[0], a[1]) if isinstance(a, tuple) else a for a in actions]

            selectedIndex = 0
            if parent:
                selectedIndex = 1
                caption = ".." if not actions or isinstance(actions[0].text, str) else ["..", ""]
                actions = [Action(caption, parent)] + actions

            if not selectedId:
                selectedId = defaultSelectedId

            if selectedId:
                for i in range(len(actions)):
                    if actions[i].id == selectedId:
                        selectedIndex = i
                        break

            def getOptions():
                options = [cb for cb in actions if cb.isCheckbox()]
                if not options:
                    return None

                return [cb.id for cb in options if cb.checked]

            def show(selectedIndex):
                def onSelect(index):
                    if parent and index == 0:
                        parent()
                        return

                    action = actions[index]
                    thisMenu = (
                        parent if temp and not action.isCheckbox() else
                        partial(show, index) if not refresh else
                        partial(impl, parent=parent, selectedId=action.id))

                    action.func(parent=thisMenu, selectedId=None, options=getOptions())

                self.SelectItem(
                    [a.text for a in actions],
                    onSelect,
                    selectedIndex=selectedIndex)

            show(selectedIndex)

        return impl

    def action(self, func, terminate=False):
        def impl(parent, selectedId, options):
            if options is None:
                func()
            else:
                func(options=options)

            if not terminate and parent:
                parent()

        return impl

    def none(self):
        def impl(parent, selectedId, options):
            pass

        return impl


class TestMenuCommand(stWindowCommand, Menu):
    def run(self):
        self.buildMenu(prefix="")(parent=None, selectedId=None)

    @menu()
    def buildMenu(self, prefix):
        return [
            (prefix + "menu ...", self.buildMenu(prefix=prefix+"menu/")),
            (prefix + "temporary menu", self.temporaryMenu()),
            (prefix + "action", self.action(func=partial(sublime.message_dialog, prefix+"action is called"))),
            (prefix + "test action", self.testAction()),
            (prefix + "terminated action", self.terminatedAction()),
            (prefix + "static menu ...", self.staticMenu()),
            (prefix + "Checkbox example ...", self.checkboxMenu())
        ]

    @menu(temp=True)
    def temporaryMenu(self):
        return [
            ("usual menu", self.buildMenu(prefix="temp/")),
            ("print", self.testAction()),
            ("print and terminate", self.terminatedAction),
        ]

    @action()
    def testAction(self):
        print("test action runned")

    @action(terminate=True)
    def terminatedAction(self):
        print("action runned and menu should not appears again")

    @menu()
    def staticMenu(self):
        return [
            ('submenu 1', self.menu(lambda: [
                ('submemu 1/item 1', self.action(partial(sublime.message_dialog, 'submemu 1/item 1'))),
                ('submenu 1/item 2', self.action(partial(sublime.message_dialog, 'submemu 1/item 2'))),
                ])
            ),
            ('submenu 2', self.menu(lambda: [
                ('submemu 2/one more submenu', self.menu(lambda:[
                    ('make some action', self.action(partial(sublime.message_dialog, 'some action'))),
                    ('make another action', self.action(partial(sublime.message_dialog, 'another action'))),
                    ])
                ),
                ('submenu 2/item 2', self.action(partial(sublime.message_dialog, 'submemu 2/item 2'))),
                ('submenu 2/Checkbox menu', self.checkboxMenu()),
                ], temp=True)
            ),
        ]

    @menu(temp=True)
    def checkboxMenu(self):
        return [
            ('Show options', self.action(lambda options: sublime.message_dialog(str(options)))),
            CheckBox("option A", id="A"),
            CheckBox("option B"),
            CheckBox("option C"),
        ]