Dependents.py

import sublime
import sublime_plugin
import threading
import os
import sys

if sys.version_info < (3,):
    from BaseCommand import BaseCommand
    from BaseThread import BaseThread
    from lib.show_error import *
    from lib.trim_paths_of_root import trim_paths_of_root
    from lib.printer import p
    from node_dependents_editor_backend import backend
else:
    from .BaseCommand import BaseCommand
    from .BaseThread import BaseThread
    from .lib.show_error import *
    from .lib.trim_paths_of_root import trim_paths_of_root
    from .lib.printer import p
    from .node_dependents_editor_backend import backend


class DependentsCommand(BaseCommand, sublime_plugin.WindowCommand):
    def run(self, modifier=''):
        if super(DependentsCommand, self).run(modifier):
            self.init_thread(DependentsThread, 'Finding dependents')


class DependentsThread(BaseThread):
    """
    A thread to prevent the determination of the dependents from freezing the UI
    """
    def __init__(self, command):
        self.window = command.window
        self.view = command.view
        threading.Thread.__init__(self)

    def run(self):
        """
        Finds the dependents of the current file and jumps to that file or shows a panel of dependent files
        """
        dependents = self.get_dependents()
        p('fetched dependents: ', dependents)

        self.dependents = trim_paths_of_root(dependents, self.window.config['directory'])

        if self.view.modifier == 'OPEN_ALL':
            for dep in self.dependents:
                self.open_file(dep)

        elif self.view.modifier == 'COPY_ALL':
            extLess = map(lambda d: os.path.splitext(d)[0], self.dependents)
            sublime.set_timeout(lambda: sublime.set_clipboard('\n'.join(extLess)), 100)

        elif len(self.dependents) == 1:
            p('Opening the only dependent: ', self.dependents[0])
            self.open_file(self.dependents[0])

        else:
            sublime.set_timeout(self.show_quick_panel, 10)

    def get_dependents(self):
        """
        Asks the node tool for the dependents of the current module
        """
        args = {
            'filename': self.view.filename,
            'command': 'find-dependents'
        }

        dependents = [d for d in backend(args).split('\n') if d]

        p(len(dependents), 'dependents found:\n' + '\n'.join(dependents))

        return dependents

    def show_quick_panel(self):
        if not self.dependents:
            show_error('Can\'t find any file that depends on this file')
            return

        self.window.show_quick_panel(self.dependents, self.on_done)

    def on_done(self, picked):
        if picked == -1:
            return

        dependent = self.dependents[picked]
        self.open_file(dependent)

    def open_file(self, dependent):
        # We removed the root originally when populating the dependents list
        filename = os.path.normpath(os.path.join(self.window.config['directory'], dependent))
        super(DependentsThread, self).open_file(filename)