git_plugin.py

"""
Terminator plugin to find git repo.

License: GPLv2
"""
import inspect, os, shlex, subprocess
import gtk
from terminatorlib.util import err, dbg
from terminatorlib import plugin
from terminatorlib import config

AVAILABLE = ['GitPlugin']

class GitPlugin(plugin.MenuItem):
    """ Process URLs returned by commands. """
    capabilities = ['terminal_menu']

    def __init__(self):
        plugin.MenuItem.__init__(self)
        self.plugin_name = self.__class__.__name__
        self.current_path = None
        
    def get_cwd(self):
        """ Return current working directory. """
        # HACK: Because the current working directory is not available to plugins,
        # we need to use the inspect module to climb up the stack to the Terminal
        # object and call get_cwd() from there.
        for frameinfo in inspect.stack():
            frameobj = frameinfo[0].f_locals.get('self')
            if frameobj and frameobj.__class__.__name__ == 'Terminal':
                return frameobj.get_cwd()
        return None        

    def callback(self, menuitems, menu, terminal):
        filepath = self.get_cwd()
        gitpath = filepath + '/.git/'
        #BUG si ls -l /etc/ suis encore dans home :( filepath pas bon !!
        print 'mon dossier: ' + gitpath
        
        if os.path.exists(gitpath):
            #item = gtk.MenuItem('Git Status')
            #menuitems.append(item)
            #item.connect("activate", self._execute, {'terminal' : terminal, 'command' : 'git status' })
            #dbg('Menu items git appended')
            
            item = gtk.MenuItem('Git')
            menuitems.append(item)
            submenu = gtk.Menu()
            item.set_submenu(submenu)
            
            menuitem = gtk.MenuItem('Status')
            submenu.append(menuitem)
            menuitem.connect("activate", self._execute, {'terminal' : terminal, 'command' : 'git status' })
            menuitem = gtk.SeparatorMenuItem()
            submenu.append(menuitem)
            
            myBranche = subprocess.check_output(['git', 'symbolic-ref','HEAD', '--short']) 
            menuitem = gtk.MenuItem('Branches')
            submenu.append(menuitem)
            #menuitem.connect("activate", self._execute, {'terminal' : terminal, 'command' : 'git branch -a' })
            myBranches = subprocess.check_output(['git', 'branch']).split('\n')
            
            ssubmenu = gtk.Menu()
            menuitem.set_submenu(ssubmenu)

            for element in myBranches:
                if len(element)>1:
                    smenuitem = gtk.MenuItem(element)
                    ssubmenu.append(smenuitem)
                    smenuitem.connect("activate", self._execute, {'terminal' : terminal, 'command' : 'git checkout '+element.replace('* ','') })
            menuitem = gtk.SeparatorMenuItem()
            ssubmenu.append(menuitem)
            menuitem = gtk.MenuItem('List Branches')
            ssubmenu.append(menuitem)
            menuitem.connect("activate", self._execute, {'terminal' : terminal, 'command' : 'git branch -a' })
                
            
            menuitem = gtk.MenuItem('Logs')
            menuitem.connect("activate", self._execute, {'terminal' : terminal, 'command' : 'git log --oneline -n 6' })
            submenu.append(menuitem)
            
            
        else:
            dbg('Menu items git remove')
            
    def _execute(self, _widget, data):
        command = data['command']+"\n"
        terminal = data['terminal']
        #print 'exec: ?  ' + command
        terminal.feed(command)
        return command