pretty_command.py

#! /usr/bin/env python
# -*- coding: utf-8 -*-

import sys
import subprocess
import sublime
import sublime_plugin
from .proto_formatter import ProtoSettings, ProtoFormatter

class PrettyProtobufCommand(sublime_plugin.TextCommand):
    def run(self, edit):
        region = sublime.Region(0, self.view.size())
        lines = self.view.substr(region)

        # Avoid flashing an ugly cmd prompt on Windows when invoking clang-format
        startupinfo = None
        if sys.platform.startswith('win32'):
          startupinfo = subprocess.STARTUPINFO()
          startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
          startupinfo.wShowWindow = subprocess.SW_HIDE

        command = [ProtoSettings().clang_format_path]
        if self.view.file_name():
            command += ['-assume-filename', self.view.file_name()]
        proc = subprocess.Popen(command, stdout=subprocess.PIPE,
                stderr=subprocess.PIPE, stdin=subprocess.PIPE, startupinfo=startupinfo)
        stdout, stderr = proc.communicate(input=lines.encode())
        if stderr:
          print(f'{stderr = }')
        if not stdout:
          print('No output from clang-format, maybe crashed...\n'
                'Please report to https://github.com/hanfezh/pretty-protobuf/issues')
          return

        # Replace with formatted content
        self.view.replace(edit, region, stdout.decode())

class PrettyDebugStringCommand(sublime_plugin.TextCommand):
    def run(self, edit):
        if len(self.view.sel()) < 1:
            return
        first_reg = self.view.sel()[0]
        if first_reg.empty() and ProtoSettings().use_entire_file:
            first_reg = sublime.Region(0, self.view.size())
        lines = self.view.substr(first_reg)
        if not lines:
            return
        lines = ProtoFormatter(lines).format()
        if lines:
            self.view.replace(edit, first_reg, lines)