move_text.py

from functools import cmp_to_key

import sublime
import sublime_plugin
from sublime import Region


class MoveTextHorizCommand(sublime_plugin.TextCommand):
    def move_text_horiz(self, edit, direction, selections=None):
        selections = list(self.view.sel())
        if direction > 1:
            selections.reverse()
        if all(region.empty() for region in selections):
            window = self.view.window()
            view = self.view
            group, index = window.get_view_index(view)
            view_len = len(window.views_in_group(group))
            if direction > 0 and index < view_len - 1:
                window.set_view_index(view, group, index + 1)
            elif direction < 0 and index > 0:
                window.set_view_index(view, group, index - 1)
            return
        for region in selections:
            if region.empty():
                continue

            orig_region = region
            sel_region = Region(region.begin() + direction, region.end() + direction)

            if sel_region.a < 0 or sel_region.b > self.view.size():
                continue

            if direction < 0:
                dest_region = Region(region.begin() + direction, region.end())
                move_text = self.view.substr(region) + self.view.substr(Region(region.begin() + direction, region.begin()))
            else:
                dest_region = Region(region.begin(), region.end() + direction)
                move_text = self.view.substr(Region(region.end(), region.end() + direction)) + self.view.substr(region)

            # Remove selection from RegionSet
            self.view.sel().subtract(orig_region)
            # Replace the selection with transformed text
            self.view.replace(edit, dest_region, move_text)
            # Add the new selection
            self.view.sel().add(sel_region)


class MoveTextLeftCommand(MoveTextHorizCommand):
    def run(self, edit):
        self.move_text_horiz(edit, -1)


class MoveTextRightCommand(MoveTextHorizCommand):
    def run(self, edit):
        self.move_text_horiz(edit, 1)


class MoveTextVertCommand(sublime_plugin.TextCommand):
    def __init__(self, view):
        super(MoveTextVertCommand, self).__init__(view)
        view.move_text_vert_column = None

    def move_text_vert(self, region, edit, direction):
        orig_region = region
        select_begin = None
        if region.empty():
            row, col = self.view.rowcol(region.begin())
            select_begin = col
            region = self.view.full_line(region.begin())

        move_text = self.view.substr(region)

        # calculate number of characters to the left
        row, col = self.view.rowcol(region.begin())

        # if the last command was a vertical move, use that column
        # the column is stored on the view - each command has its own instance,
        # and we don't want two buffers to modify the same object (e.g. MoveTextVertCommand)
        cmd, _, _ = self.view.command_history(0, True)
        if cmd != 'move_text_up' and cmd != 'move_text_down':
            self.view.move_text_vert_column = col
        elif self.view.move_text_vert_column:
            col = self.view.move_text_vert_column

        dest_row = row + direction

        max_row = self.view.rowcol(self.view.size())[0]
        if dest_row < 0:
            dest_row = 0
        elif dest_row > max_row:
            dest_row = max_row

        self.view.sel().subtract(orig_region)
        self.view.replace(edit, region, '')

        # starting at the destination row at col 0, count off "col" characters
        # it's possible that there aren't enough characters in the destination row,
        # so stop if we end up on the wrong row, or past the buffer
        dest_point = self.view.text_point(dest_row, 0)
        if dest_point is None:
            dest_point = self.view.size()
        else:
            dest_line = self.view.line(dest_point)
            if dest_point + col > dest_line.b:
                dest_point = dest_line.b
            else:
                dest_point = dest_point + col

        self.view.insert(edit, dest_point, move_text)

        if select_begin is None:
            sel_region = Region(dest_point, dest_point + len(move_text))
        else:
            sel_region = Region(dest_point + select_begin)
        return sel_region


class MoveTextUpCommand(MoveTextVertCommand):
    def run(self, edit):
        regions = list(self.view.sel())
        new_regions = []
        for region in regions:
            new_regions.append(self.move_text_vert(region, edit, -1))

        self.view.sel().add_all(new_regions)
        try:
            region = new_regions[0]
            self.view.show(region.begin())
        except IndexError:
            pass


class MoveTextDownCommand(MoveTextVertCommand):
    def run(self, edit):
        new_regions = []
        for region in reversed(self.view.sel()):
            new_regions.append(self.move_text_vert(region, edit, 1))

        self.view.sel().add_all(new_regions)
        try:
            region = new_regions[0]
            self.view.show(region.begin())
        except IndexError:
            pass