def run(self, edit):
if self.data is not None:
self.update_buffer(edit)
return
try:
messages = {
'start': 'Auto formatting file...',
'end': 'done!',
'fail': 'The auto formatting failed!',
'timeout': 'The auto formatiing timed out!'
}
self.pbar = ProgressBar(messages)
self.pbar.start()
self.view.set_read_only(True)
rustfmt = get_settings(self.view, 'rustfmt_binary_path', 'rustfmt')
if rustfmt == '':
rustfmt = 'rustfmt'
self.code = self.view.substr(sublime.Region(0, self.view.size()))
# the JonServer deletes the temp file so we don't worry
fd, path = tempfile.mkstemp(suffix=".rs", dir=file_directory())
try:
with os.fdopen(fd, "wb") as tmp:
tmp.write(self.code.encode('utf-8'))
except Exception as e:
self.clean_tmp_file(path)
raise e
config_path = get_settings(self.view, 'rust_rustfmt_config_path')
if config_path is None or config_path == '':
config_path = self._get_working_directory()
data = {
'vid': self.view.id(),
'filename': path,
'settings': {
'rustfmt_binary_path': rustfmt,
'config_path': config_path
},
'method': 'format',
'handler': 'rustfmt'
}
timeout = get_settings(self.view, 'rust_rustfmt_timeout', 1)
callback = Callback(timeout=timeout)
callback.on(success=self.prepare_data)
callback.on(error=self.on_failure)
callback.on(timeout=partial(self.clean_tmp_file, path))
Worker().execute(callback, **data)
except:
logging.error(traceback.format_exc())
def run(self, edit):
if self.data is not None:
self.update_buffer(edit)
return
try:
messages = {
'start': 'Auto formatting file...',
'end': 'done!',
'fail': 'The auto formatting failed!',
'timeout': 'The auto formatiing timed out!'
}
self.pbar = ProgressBar(messages)
self.pbar.start()
self.view.set_read_only(True)
rustfmt = get_settings(
self.view, 'rustfmt_binary_path', 'rustfmt'
)
if rustfmt == '':
rustfmt = 'rustfmt'
self.code = self.view.substr(
sublime.Region(0, self.view.size())
)
# the JonServer deletes the temp file so we don't worry
fd, path = tempfile.mkstemp(suffix=".rs", dir=file_directory())
with os.fdopen(fd, "w", encoding="utf-8") as tmp:
tmp.write(self.code)
config_path = get_settings(self.view, 'rust_rustfmt_config_path', '')
data = {
'vid': self.view.id(),
'filename': path,
'settings': {
'rustfmt_binary_path': rustfmt,
'config_path': config_path
},
'method': 'format',
'handler': 'rustfmt'
}
timeout = get_settings(self.view, 'rust_rustfmt_timeout', 1)
callback = Callback(timeout=timeout)
callback.on(success=self.prepare_data)
callback.on(error=self.on_failure)
callback.on(timeout=partial(self.clean_tmp_file, path))
Worker().execute(callback, **data)
except:
logging.error(traceback.format_exc())
class AnacondaRustFmt(sublime_plugin.TextCommand):
"""Execute rustfmt command in a file
"""
data = None
def run(self, edit):
if self.data is not None:
self.update_buffer(edit)
return
try:
messages = {
'start': 'Auto formatting file...',
'end': 'done!',
'fail': 'The auto formatting failed!',
'timeout': 'The auto formatiing timed out!'
}
self.pbar = ProgressBar(messages)
self.pbar.start()
self.view.set_read_only(True)
rustfmt = get_settings(self.view, 'rustfmt_binary_path', 'rustfmt')
if rustfmt == '':
rustfmt = 'rustfmt'
self.code = self.view.substr(sublime.Region(0, self.view.size()))
# the JonServer deletes the temp file so we don't worry
fd, path = tempfile.mkstemp(suffix=".rs", dir=file_directory())
with os.fdopen(fd, "w") as tmp:
tmp.write(self.code)
config_path = get_settings(self.view, 'rust_rustfmt_config_path')
if config_path is None or config_path == '':
config_path = self._get_working_directory()
data = {
'vid': self.view.id(),
'filename': path,
'settings': {
'rustfmt_binary_path': rustfmt,
'config_path': config_path
},
'method': 'format',
'handler': 'rustfmt'
}
timeout = get_settings(self.view, 'rust_rustfmt_timeout', 1)
callback = Callback(timeout=timeout)
callback.on(success=self.prepare_data)
callback.on(error=self.on_failure)
callback.on(timeout=partial(self.clean_tmp_file, path))
Worker().execute(callback, **data)
except:
logging.error(traceback.format_exc())
def is_enabled(self):
"""Determine if this command is enabled or not
"""
return is_code(self.view, lang='rust', ignore_comments=True)
def on_failure(self, *args, **kwargs):
"""Called when callback return a failure or times out
"""
self.pbar.terminate(status=self.pbar.Status.FAILURE)
self.view.set_read_only(False)
print(args[0]['error'])
def prepare_data(self, data):
"""Prepare the returned data to overwrite our buffer
"""
self.data = data
self.pbar.terminate()
self.view.set_read_only(False)
self.view.run_command('anaconda_rust_fmt')
def update_buffer(self, edit):
"""Update and reload the buffer
"""
view = get_window_view(self.data['vid'])
if self.sanitize(self.code) != self.sanitize(self.data.get('output')):
region = sublime.Region(0, view.size())
view.replace(edit, region, self.data.get('output'))
if get_settings(view, 'rust_format_on_save'):
sublime.set_timeout(lambda: view.run_command('save'), 0)
self.data = None
self.code = None
def sanitize(self, text):
"""Remove blank lines from text and trim it
"""
return os.linesep.join([s for s in text.splitlines() if s]).strip()
def clean_tmp_file(self, path):
"""Clean the tmp file at timeout
"""
try:
os.remove(path)
except:
pass
def _get_working_directory(self):
"""Return back the project file directory if any or current file one
"""
pfilename = sublime.active_window().project_file_name()
if pfilename is not None:
return os.path.dirname(pfilename)
return os.path.dirname(self.view.file_name())
class AnacondaRustFmt(sublime_plugin.TextCommand):
"""Execute rustfmt command in a file
"""
data = None
def run(self, edit):
if self.data is not None:
self.update_buffer(edit)
return
try:
messages = {
'start': 'Auto formatting file...',
'end': 'done!',
'fail': 'The auto formatting failed!',
'timeout': 'The auto formatiing timed out!'
}
self.pbar = ProgressBar(messages)
self.pbar.start()
self.view.set_read_only(True)
rustfmt = get_settings(
self.view, 'rustfmt_binary_path', 'rustfmt'
)
if rustfmt == '':
rustfmt = 'rustfmt'
self.code = self.view.substr(
sublime.Region(0, self.view.size())
)
# the JonServer deletes the temp file so we don't worry
fd, path = tempfile.mkstemp(suffix=".rs", dir=file_directory())
with os.fdopen(fd, "w", encoding="utf-8") as tmp:
tmp.write(self.code)
config_path = get_settings(self.view, 'rust_rustfmt_config_path', '')
data = {
'vid': self.view.id(),
'filename': path,
'settings': {
'rustfmt_binary_path': rustfmt,
'config_path': config_path
},
'method': 'format',
'handler': 'rustfmt'
}
timeout = get_settings(self.view, 'rust_rustfmt_timeout', 1)
callback = Callback(timeout=timeout)
callback.on(success=self.prepare_data)
callback.on(error=self.on_failure)
callback.on(timeout=partial(self.clean_tmp_file, path))
Worker().execute(callback, **data)
except:
logging.error(traceback.format_exc())
def is_enabled(self):
"""Determine if this command is enabled or not
"""
return is_code(self.view, lang='rust', ignore_comments=True)
def on_failure(self, *args, **kwargs):
"""Called when callback return a failure or times out
"""
self.pbar.terminate(status=self.pbar.Status.FAILURE)
self.view.set_read_only(False)
print(args[0]['error'])
def prepare_data(self, data):
"""Prepare the returned data to overwrite our buffer
"""
self.data = data
self.pbar.terminate()
self.view.set_read_only(False)
self.view.run_command('anaconda_rust_fmt')
def update_buffer(self, edit):
"""Update and reload the buffer
"""
view = get_window_view(self.data['vid'])
output = self.sanitize(self.data.get('output'))
if output and self.sanitize(self.code) != output:
region = sublime.Region(0, view.size())
view.replace(edit, region, self.data.get('output'))
if get_settings(view, 'rust_format_on_save'):
sublime.set_timeout(lambda: view.run_command('save'), 0)
self.data = None
self.code = None
def sanitize(self, text):
"""Remove blank lines from text and trim it
"""
return os.linesep.join([s for s in text.splitlines() if s]).strip()
def clean_tmp_file(self, path):
"""Clean the tmp file at timeout
"""
try:
os.remove(path)
except:
pass
def _get_working_directory(self):
"""Return back the project file directory if any or current file one
"""
pfilename = sublime.active_window().project_file_name()
if pfilename is not None:
return os.path.dirname(pfilename)
return os.path.dirname(self.view.file_name())