sublimejava.py

"""
Copyright (c) 2012 Fredrik Ehnbom

This software is provided 'as-is', without any express or implied
warranty. In no event will the authors be held liable for any damages
arising from the use of this software.

Permission is granted to anyone to use this software for any purpose,
including commercial applications, and to alter it and redistribute it
freely, subject to the following restrictions:

 1. The origin of this software must not be misrepresented; you must not
 claim that you wrote the original software. If you use this software
 in a product, an acknowledgment in the product documentation would be
 appreciated but is not required.

 2. Altered source versions must be plainly marked as such, and must not be
 misrepresented as being the original software.

 3. This notice may not be removed or altered from any source
 distribution.
"""
import sublime
import sublime_plugin
import os
import re

from sublimecompletioncommon import completioncommon
reload(completioncommon)

import classopener
reload(classopener)


class SublimeJavaDotComplete(completioncommon.CompletionCommonDotComplete):
 pass


class SublimeJavaCompletion(completioncommon.CompletionCommon):
 def __init__(self):
 super(SublimeJavaCompletion, self).__init__("SublimeJava.sublime-settings", os.path.dirname(os.path.abspath(__file__)))
 self.regex = [
 (re.compile(r"\[I([,)}]|$)"), r"int[]\1"),
 (re.compile(r"\[F([,)}]|$)"), r"float[]\1"),
 (re.compile(r"\[Z([,)}]|$)"), r"boolean[]\1"),
 (re.compile(r"\[B([,)}]|$)"), r"byte[]\1"),
 (re.compile(r"\[C([,)}]|$)"), r"char[]\1"),
 (re.compile(r"\[S([,)}]|$)"), r"short[]\1"),
 (re.compile(r"\[J([,)}]|$)"), r"long[]\1"),
 (re.compile(r"\[D([,)}]|$)"), r"double[]\1"),
 (re.compile(r"\[\L?([\w\./]+)(;)?"), r"\1[]")]

 def show_error(self, msg):
 if self.get_setting("sublimejava_no_visual_errors", False):
 print msg
 else:
 sublime.error_message(msg + "\n\nDisable visual error message dialogues with setting:\nsublimejava_no_visual_errors: true")

 def get_packages(self, data, thispackage, type):
 packages = re.findall(r"(?:^|\n)[ \t]*import[ \t]+(.*);", data)
 packages.append("java.lang.*")
 packages.append("") # for int, boolean, etc
 for package in packages:
 idx = type.find(".")
 if idx == -1:
 idx = len(type)
 subtype = type[:idx]
 if re.search("[\.\$]{1}%s$" % subtype, package):
 # Explicit imports, we want these to have the highest
 # priority when searching for the absolute type, so
 # insert them at the top of the package list.
 # Both the .* version and not is added so that
 # blah.<searchedForClass> and blah$<searchedForClass>
 # is tested
 add = package[:-(len(subtype)+1)]
 packages.insert(0, add + ".*")
 packages.insert(1, add)
 break
 packages.append(thispackage + ".*")
 return packages

 def get_cmd(self):
 classpath = self.get_setting("sublimejava_classpath", ["."])
 newclasspath = []
 window = sublime.active_window()
 for path in classpath:
 newclasspath.append(self.expand_path(path, window))
 classpath = newclasspath
 classpath.insert(0, ".")
 classpath = os.pathsep.join(classpath)
 return "java -classpath \"%s\" SublimeJava" % classpath

 def is_supported_language(self, view):
 if view.is_scratch() or not self.get_setting("sublimejava_enabled", True):
 return False
 language = self.get_language(view)
 return language == "java" or language == "jsp"

 def sub(self, regex, sub, data):
 olddata = data
 data = regex.sub(sub, data)
 while data != olddata:
 olddata = data
 data = regex.sub(sub, data)
 return data

 def fixnames(self, data):
 for regex, replace in self.regex:
 data = self.sub(regex, replace, data)
 return data

 def return_completions(self, comp):
 ret = []
 for display, insert in comp:
 ret.append((self.fixnames(display), self.fixnames(insert)))
 return super(SublimeJavaCompletion, self).return_completions(ret)

 def get_class_under_cursor(self):
 view = sublime.active_window().active_view()
 data = view.substr(sublime.Region(0, view.size()))
 word = view.substr(view.word(view.sel()[0].begin()))
 return self.find_absolute_of_type(data, data, word)

 def get_possible_imports(self, classname):
 imports = []

 if classname is not None:
 stdout = self.run_completion("-possibleimports;;--;;%s" % classname)
 imports = sorted(stdout.split("\n")[:-1])

 return imports


comp = SublimeJavaCompletion()


class SublimeJava(sublime_plugin.EventListener):

 def on_query_completions(self, view, prefix, locations):
 return comp.on_query_completions(view, prefix, locations)

 def on_query_context(self, view, key, operator, operand, match_all):
 if key == "sublimejava.dotcomplete":
 return comp.get_setting(key.replace(".", "_"), True)
 elif key == "sublimejava.supported_language":
 return comp.is_supported_language(view)
 else:
 return comp.on_query_context(view, key, operator, operand, match_all)


MSG_NO_CLASSES_FOUND = "No classes found to import for name %s."
MSG_ALREADY_IMPORTED = "Class %s has either already been imported, is \
in the current package, or is in the default package."

RE_IMPORT = "import( static)? ([\w\.]+)\.([\w]+|\*);"
RE_PACKAGE = "package ([\w]+.)*\w+;"


class ImportJavaClassCommand(sublime_plugin.TextCommand):

 def run(self, edit):
 view = self.view
 classname = view.substr(view.word(view.sel()[0].begin()))

 """
 if comp.get_class_under_cursor():
 comp.show_error(MSG_ALREADY_IMPORTED % classname)
 return
 """
 
 imports = comp.get_possible_imports(classname)

 def do_import(index):
 if index != -1:
 self._insert_import(imports[index], edit)

 if len(imports) > 0:
 view.window().show_quick_panel(imports, do_import)
 else:
 comp.show_error(MSG_NO_CLASSES_FOUND % classname)

 def _insert_import(self, full_classname, edit):
 insert_point = 0
 newlines_prepend = 0
 newlines_append = 1

 all_imports_region = self.view.find_all(RE_IMPORT)

 if len(all_imports_region) > 0:
 insert_point = all_imports_region[-1].b
 newlines_prepend = 1
 newlines_append = 0
 else:
 package_declaration_region = self.view.find(RE_PACKAGE, 0)

 if package_declaration_region is not None:
 insert_point = package_declaration_region.b
 newlines_prepend = 2
 newlines_append = 0

 import_classname = full_classname.replace("$", ".")
 import_statement = "%simport %s;%s" % ("\n" * newlines_prepend,
 import_classname,
 "\n" * newlines_append)

 self.view.insert(edit, insert_point, import_statement)


class OpenJavaSourceCommand(sublime_plugin.WindowCommand):

 def run(self, under_cursor=False):
 classopener.JavaSourceOpener(comp,
 self.window.active_view(),
 under_cursor).show()


class OpenJavaDocCommand(sublime_plugin.WindowCommand):

 def run(self, under_cursor=False):
 classopener.JavaDocOpener(comp,
 self.window.active_view(),
 under_cursor).show()