q_menu.py

import sublime, sublime_plugin
from . import QCon as Q
from . import Settings as S

class PerViewCommand(sublime_plugin.WindowCommand):
    def view(self):
        return self.window.active_view()

    def getCurrentConnection(self):
        return S.Settings.get_view_conn(self.view())

    def isCurrentConnection(self, qcon):
        current = self.getCurrentConnection()
        return current and current.equals(qcon)

    def setConnection(self, qcon):
        S.Settings.set_view_conn(self.view(), qcon)

    def set_status(self, tag, message, temp=False):
        self.view().set_status(tag, message)
        if temp:
            sublime.set_timeout(lambda:self.set_status(tag, ''), 3000)

    def loadQCons(self):
        con_dicts = S.Settings.get_connections()
        self.qcons = []
        for con_dict in con_dicts:
            self.qcons.append(Q.QCon.fromDict(con_dict))
        return self.qcons

class ShowConnectionListCommand(PerViewCommand):
    ACTIONS = ['Use', 'Update', 'Delete', 'Rename']

    def run(self):
        self.prompt_connection_list()

    def prompt_connection_list(self):
        qcons = self.loadQCons()

        l = [['Add...', 'New Connection']]  #add new connection to first element in the list
        #array of dict to array of array format as reqruired by show_quick_panel()
        current = self.getCurrentConnection()
        if current:
            l.append([current.name, "[CONNECTED] " + current.h()])
        for qcon in qcons:
            if current and qcon.equals(current):
                continue 
            l.append([qcon.name, qcon.h()])
        self.l = l
        self.window.show_quick_panel(l, self.prompt_actions, 0, 1)  #start at index 1 (so user can press up for new)

    def prompt_actions(self, i):
        if i < 0:       #we get index=-1 if cancel
            return        
        elif i == 0:    #new connection
            self.window.run_command('new_connection')
            return
        
        self.conn = self.l[i]

        self.window.show_quick_panel(ShowConnectionListCommand.ACTIONS, self.on_done)

    def on_done(self, i):
        if i < 0:       #we get index=-1 if cancel
            return
        action = ShowConnectionListCommand.ACTIONS[i]
        action = action.replace(' ', '_').lower()
        print(action + ' ' + str(self.conn[1]))
        self.window.run_command(action + '_connection', {'name': self.conn[0], 'h': self.conn[1].replace("[CONNECTED] ","")})


class NewConnectionCommand(PerViewCommand):
    def run(self):
        self.prompt_new_connection()

    def prompt_new_connection(self):
        self.window.show_input_panel('New q Connection handle:', S.Settings.default_new_connection(), self.prompt_name, None, None)

    def prompt_name(self, h):
        self.h = h
        self.window.show_input_panel('Connection Name:', h, self.on_done, None, self.on_cancel)

    def on_cancel(self):
        self.on_done('')

    def on_done(self, name):
        self.window.run_command('use_connection', {'name': name, 'h': self.h})


class CommandWithQCon(PerViewCommand):
    def run(self, name, h):
        selected_qcon = Q.QCon.new(name, h)
        current = self.getCurrentConnection()
        if current and current.equals(selected_qcon):
            self.current = True
            self.qcon = current
        else:
            self.current = False
            self.qcon = selected_qcon
        self.do()

    def done(self):
        if self.current:
            self.setConnection(self.qcon)
        self.set_status('q', self.getCurrentConnection().status())

class UseConnectionCommand(CommandWithQCon):
    def do(self):
        self.setConnection(self.qcon)
        self.done()

class DeleteConnectionCommand(CommandWithQCon):
    def do(self):
        S.Settings.delete_connection(self.qcon)
        self.set_status('result', 'Deleted "' + self.qcon.name + '"', True)
        

class RenameConnectionCommand(CommandWithQCon):
    def do(self):
        self.window.show_input_panel('Rename Connection', self.qcon.name, self.on_done, None, None)

    def on_done(self, new_name):
        new_qcon = self.qcon.clone()
        new_qcon.name = new_name
        S.Settings.update_connection(self.qcon, new_qcon)
        self.set_status('result', 'Renamed "' + new_qcon.h() + '" to "' + new_qcon.name  + '"', True)
        self.qcon = new_qcon
        self.done()

class UpdateConnectionCommand(CommandWithQCon):
    def do(self):
        self.window.show_input_panel('Update Connection', self.qcon.h(), self.on_done, None, None)

    def on_done(self, new_h):
        new_qcon = self.qcon.clone()
        new_qcon.h(new_h)
        S.Settings.update_connection(self.qcon, new_qcon)
        self.set_status('result', 'Updated "' + self.qcon.name + '" to "' + new_qcon.h()  + '"', True)
        self.qcon = new_qcon
        self.done()