FixMyJS.py

import sublime
import sublime_plugin
import json
from os.path import dirname, realpath, join, splitext, basename

try:
	# Python 2
	from node_bridge import node_bridge
except:
	from .node_bridge import node_bridge

# monkeypatch `Region` to be iterable
sublime.Region.totuple = lambda self: (self.a, self.b)
sublime.Region.__iter__ = lambda self: self.totuple().__iter__()

BIN_PATH = join(sublime.packages_path(), dirname(realpath(__file__)), 'fixmyjs.js')

def get_setting(view, key):
	settings = view.settings().get('FixMyJS')
	if settings is None:
		settings = sublime.load_settings('FixMyJS.sublime-settings')
	return settings.get(key)

def is_javascript(view):
	return splitext(basename(view.settings().get('syntax')))[0] in ('JavaScript', 'JavaScript (Babel)')

class FixCommand(sublime_plugin.TextCommand):
	def run(self, edit):
		if not self.has_selection():
			region = sublime.Region(0, self.view.size())
			originalBuffer = self.view.substr(region)
			fixed = self.fix(originalBuffer)
			if fixed:
				self.view.replace(edit, region, fixed)
			return
		for region in self.view.sel():
			if region.empty():
				continue
			originalBuffer = self.view.substr(region)
			fixed = self.fix(originalBuffer)
			if fixed:
				self.view.replace(edit, region, fixed)

	def fix(self, data):
		try:
			return node_bridge(data, BIN_PATH, [json.dumps({
				'legacy': get_setting(self.view, 'legacy'),
				'filepath': self.view.file_name()
			})])
		except Exception as e:
			sublime.error_message('FixMyJS\n%s' % e)

	def has_selection(self):
		for sel in self.view.sel():
			start, end = sel
			if start != end:
				return True
		return False

class FixPreSaveCommand(sublime_plugin.EventListener):
	def on_pre_save(self, view):
		if get_setting(view, 'fixOnSave') is True and is_javascript(view):
				view.run_command('fix')