crystal_format.py

import json
import subprocess
import sys
import re
import os

import sublime_plugin
import sublime
from .diff_match_patch import diff_match_patch

class CrystalPluginListener(sublime_plugin.EventListener):
  def on_pre_save(self, view):
    settings = sublime.load_settings('Crystal.sublime-settings')
    if settings.get('auto_format'):
      view.run_command('crystal_format')

class CrystalFormatCommand(sublime_plugin.TextCommand):
  def is_enabled(self):
    caret = self.view.sel()[0].a
    syntax_name = self.view.scope_name(caret)
    return "source.crystal" in syntax_name

  def has_redo(self):
    cmd, args, repeat = self.view.command_history(1)
    return cmd != ''

  def run(self, edit):
    vsize = self.view.size()
    region = sublime.Region(0, vsize)
    src = self.view.substr(region)
    window = self.view.window()

    settings = sublime.load_settings('Crystal.sublime-settings')
    #command = [settings.get("crystal_cmd"), "tool", "format", "-", "--format", "json"]
    command = [settings.get("crystal_cmd"), "tool", "format", "-", "--no-color"]

    # for Windows Subsystem for Linux
    if os.name == "nt": command.insert(0, "wsl")

    popen_args = dict(args=command, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    # Prevent flashing terminal windows
    if sys.platform.startswith('win'):
      popen_args['startupinfo'] = subprocess.STARTUPINFO()
      popen_args['startupinfo'].dwFlags |= subprocess.STARTF_USESHOWWINDOW

    proc = subprocess.Popen(**popen_args)
    stdout, stderr = proc.communicate(src.encode('utf-8'))
    stdout = stdout.decode('utf-8')
    stderr = stderr.decode('utf-8')
    exit = proc.returncode

    pos = 0
    if exit == 0:
      if not self.has_redo():
        for op, text in diff_match_patch().diff_main(src, stdout):
          if op == diff_match_patch.DIFF_DELETE:
            self.view.erase(edit, sublime.Region(pos, pos + len(text)))
          if op == diff_match_patch.DIFF_INSERT:
            self.view.insert(edit, pos, text)
            pos += len(text)
          if op == diff_match_patch.DIFF_EQUAL:
            pos += len(text)

      self.view.erase_regions('crystal_errors')
      window.run_command("hide_panel")

    else:
      error_pos = None
      pattern = r"Error: Syntax error in .+?:(\d+): (.+)"
      match = re.match(pattern, stderr)
      if match:
        error_pos = int(match.group(1))
        error = match.group(2)
      else:
        error_pos = None
        error = stderr
      # error = json.loads(stderr)
      # error_pos = self.view.text_point(error[0]["line"] - 1, error[0]["column"] - 1)

      if error_pos:
        line_region = self.view.full_line(error_pos)
        self.view.add_regions('crystal_errors', [line_region], 'comment', 'dot', sublime.HIDDEN)

      error_panel = window.create_output_panel('crystal_errors')
      # error_panel.run_command("append", {"characters":
      #   "Error at line %d, column %d: %s" % (error[0]["line"], error[0]["column"], error[0]['message'])
      # })

      if error_pos:
        error_panel.run_command("append", {"characters":
          "Error at line %d: %s" % (error_pos, error)
        })
      else:
        error_panel.run_command("append", {"characters": error})

      window.run_command("show_panel", {"panel": "output.crystal_errors"})