GoImport.py

import sublime
import sublime_plugin
import os
from .src import paths
from .src import utils
from .src import cache


def cachePaths():
    GO_STD_LIBRARY = paths.get_std_library_path()
    GOMODCACHE = paths.get_GOMODCACHE()

    if GO_STD_LIBRARY != "" and os.path.exists(GO_STD_LIBRARY):
        cache.cache_directory_paths_of_path(GO_STD_LIBRARY, optimizeCaching=True)

    if GOMODCACHE != "" and os.path.exists(GOMODCACHE):
        cache.cache_directory_paths_of_path(GOMODCACHE)


class GoImportCommand(sublime_plugin.TextCommand):
    isCached = False

    def run(self, edit, args=None):
        if "Go" not in self.view.syntax().name:
            return

        if not self.isCached:
            cachePaths()
            self.isCached = True

        if args and "words" in args:
            utils.import_words(self.view, edit, args["words"])
            return

        words = utils.filter_imported_words(self.view, utils.get_words(self.view))

        if len(words) == 0:
            sublime.status_message("GoImport: already imported.")
            return

        words = utils.get_full_word_names(self.view, self.get_searchable_paths(), words)

        if len(words) == 0:
            sublime.status_message("GoImport: keyword not found")
            return

        absoluteWords = [w for w in words if isinstance(w, str)]
        utils.import_words(self.view, edit, absoluteWords)
        sublime.status_message("GoImport: imported!")

        self.importList = [w for w in words if not isinstance(w, str)]
        self.show_quick_panel()

    # User selects intended import from panel
    def show_quick_panel(self):
        if len(self.importList) == 0:
            return
        sublime.active_window().show_quick_panel(
            self.importList[0], self.user_selected_import
        )

    def user_selected_import(self, index):
        if index == -1:
            return
        if len(self.importList) == 0:
            return

        self.view.run_command(
            "go_import", {"args": {"words": [self.importList[0][index]]}}
        )
        self.importList.pop(0)

        if len(self.importList) != 0:
            self.show_quick_panel()

    # Returns paths for searching import keywords
    def get_searchable_paths(self):
        searchablePaths = [
            utils.get_currect_project_path(self.view),
            paths.get_std_library_path(),
            paths.get_GOMODCACHE(),
        ]
        return [p for p in searchablePaths if p]


class GoImportEraseUnusedCommand(sublime_plugin.TextCommand):
    def run(self, edit):
        if "Go" not in self.view.syntax().name:
            return

        unusedWords = utils.get_unused_words(
            self.view, utils.get_imported_words(self.view)
        )

        if len(unusedWords) == 0:
            sublime.status_message("GoImport: no unused import found.")
            return

        utils.erase_imports(self.view, edit, unusedWords)
        sublime.status_message("GoImport: erased!")