table.py

import sublime_plugin
import threading


def toggle_table_mode(view, on=True):
    if on:
        view.settings().set("AlignTabTableMode", True)
        view.set_status("aligntab", "[Table Mode]")
    else:
        view.settings().erase("AlignTabTableMode")
        erase_table_rows(view)
        view.set_status("aligntab", "")


def get_table_rows(view):
    R = view.get_regions("AlignTabTableRegion")
    rows = []
    for s in R:
        for l in view.lines(s):
            rows.append(view.rowcol(l.begin())[0])
    return rows


def set_table_rows(view, rows):
    R = []
    for row in rows:
            R.append(view.line(view.text_point(row, 0)))

    view.add_regions("AlignTabTableRegion", R)


def erase_table_rows(view):
    view.erase_regions("AlignTabTableRegion")


class AlignTabModeController(sublime_plugin.EventListener):
    # aligntab thread
    thread = None
    # register for table mode
    registered_actions = ["insert", "left_delete", "right_delete",
                          "delete_word", "paste", "cut"]

    def on_modified(self, view):
        if view.is_scratch() or view.settings().get('is_widget'):
            return
        if self.table_mode(view):
            cmdhist = view.command_history(0)
            # print(cmdhist)
            if cmdhist[0] not in self.registered_actions:
                return
            delay = 0.2
            if cmdhist[0] == "insert" and cmdhist[1]['characters'].strip() == "":
                delay = 1
            if self.thread:
                self.thread.cancel()
            self.thread = threading.Timer(
                delay,
                lambda: view.run_command("align_tab",
                                         {"user_input": "last_regex", "mode": True})
            )
            self.thread.start()

    def on_text_command(self, view, cmd, args):
        if view.is_scratch() or view.settings().get('is_widget'):
            return
        if self.table_mode(view):
            if cmd == "undo":
                view.run_command("soft_undo")
                return ("soft_undo", None)
            return None

    def table_mode(self, view):
        return view.settings().has("AlignTabTableMode")


class AlignTabClearMode(sublime_plugin.TextCommand):
    def run(self, edit):
        view = self.view
        if view.is_scratch() or view.settings().get('is_widget'):
            return
        toggle_table_mode(view, False)

    def is_enabled(self):
        return self.view.settings().get("AlignTabTableMode", False)

    def is_visible(self):
        return self.view.settings().get("AlignTabTableMode", False)