gsfmt.py

import sublime, sublime_plugin
import gscommon as gs
import thatcher

class SublimeEditor(thatcher.Editor):
    def __init__(self, view, edit):
        self.view = view
        self.edit = edit
        self.dirty = False

    def update_regions(self):
        self.regions = self.view.split_by_newlines(sublime.Region(0, self.view.size()))

    def match_line(self, line_index, content):
        self.update_regions()
        if line_index < len(self.regions):
            return self.view.substr(self.regions[line_index]) == content
        return False

    def insert_line(self, line_index, content):
        self.update_regions()
        if line_index <= len(self.regions):
            if line_index < len(self.regions):
                pos = self.regions[line_index].begin()
            else:
                pos = self.view.size()
            self.view.insert(self.edit, pos, content+'\n')
            self.dirty = True
            return True
        return False

    def delete_line(self, line_index):
        self.update_regions()
        if line_index < len(self.regions):
            self.view.erase(self.edit, self.view.full_line(self.regions[line_index]))
            self.dirty = True
            return True
        return False

class GsFmtCommand(sublime_plugin.TextCommand):
    def run(self, edit):
        if not gs.is_go_source_view(self.view):
            return

        region = sublime.Region(0, self.view.size())
        src = self.view.substr(region)

        args = [gs.setting("gofmt_cmd", "gofmt"), "-d"]
        diff, err = gs.runcmd(args, src)
        if err:
            fn = self.view.file_name()
            err = err.replace('<standard input>', fn)
            gs.notice('GsFmt', 'File %s contains errors' % fn)
        elif diff:
            err = ''
            try:
                edit = self.view.begin_edit()
                ed = SublimeEditor(self.view, edit)
                err = thatcher.patch(ed, diff)
            except Exception as e:
                err = "%s\n\n%s" % (err, e)
            finally:
                self.view.end_edit(edit)
            
            if err:
                def cb():
                    if ed.dirty:
                        self.view.run_command('undo')
                    gs.notice("GsFmt", "Could not patch the buffer: %s" % err)
                sublime.set_timeout(cb, 0)