package_installer.py

#!/usr/bin/python

from Tkinter import *

import webbrowser
import os
import tkMessageBox
from threading import Thread
import tkFileDialog

import operator
import subprocess

#Pakage Generation Script
from brocre_package_tools import *
import tkSimpleDialog
import tkFont

global checkbuttonValues
global listbox
global allPackages
global textField
global consoleOutput
global buttonInstall
global buttonUninstall
global INSTALL_PATH

allPackages = []
MAKEFILE_NAME = "Makefile"
ROBOT_PACKAGE_PATH = "./"
INSTALL_PATH = ""


class DeletePackageDialog(tkSimpleDialog.Dialog):
    message = ""
    isOKpressed = False
    
    
    def __init__(self, parent, title = None, text = None):
        self.message = text
        tkSimpleDialog.Dialog.__init__(self, parent, title)

    def body(self, master):
        w = Label(master, text=self.message,font=(16))
        f = tkFont.Font() 
        f['weight'] = 'bold' 
        w['font'] = f.name 
        w.grid(row=0)
        self.deleteCheckbuttonState = IntVar()
        self.cb = Checkbutton(master, variable=self.deleteCheckbuttonState, text="Delete ALL files in the package folder?")
        self.cb.grid(row=1, sticky=N)
        return self.cb # initial focus

    def apply(self):
        self.isOKpressed = True
        
def checkIfBROCREisBootstraped(window):
  BROCRE_IS_INSTALLED = False
  try:
    brocreconf = open('./brocre.conf','r')
    for line in  brocreconf:
      if "ROBOTPKG_BASE=" in line:
        global INSTALL_PATH
        INSTALL_PATH = line.replace("ROBOTPKG_BASE=",'')
          
    os.environ["ROBOTPKG_BASE"]=INSTALL_PATH
    if (os.path.isfile(INSTALL_PATH+"/sbin/robotpkg_info")):   
      BROCRE_IS_INSTALLED = True;
  except IOError as e:
    print 'BROCRE is NOT installed'
    
    
  if(BROCRE_IS_INSTALLED):
    updateAllPackageDescriptions()
    
  else:
    if( tkMessageBox.askokcancel("BROCRE installation", "BROCRE is NOT installed yet!\nDo you want to install it now?") == 1):
        INSTALL_PATH = tkFileDialog.askdirectory(initialdir="~/",title='Please select a directory')
        if(INSTALL_PATH == ""):
          exit() 
    
        brocreconf = open("./brocre.conf", 'w')
        brocreconf.write("ROBOTPKG_BASE="+INSTALL_PATH)
        os.environ["ROBOTPKG_BASE"]=INSTALL_PATH
        command = ["./bootstrap/bootstrap","--prefix="+INSTALL_PATH]
        t = Thread(target=executeCommandWithFunction, args=(command,updateBashrc))
        t.start()
        updateAllPackageDescriptions()
    else: 
      exit()
      
def updateBashrc():
    if( tkMessageBox.askokcancel("BROCRE installation","Do you want to add the\n"+INSTALL_PATH+"\npath to the ROS_PACKAGE_PATH?\n") == 1):
      bashrc = open(os.environ["HOME"]+"/.bashrc", 'a')
      bashrc.write("export ROS_PACKAGE_PATH=$ROS_PACKAGE_PATH:"+INSTALL_PATH)
      printIntoConsoleBox("Add the "+INSTALL_PATH+" path to the ROS_PACKAGE_PATH.")
      
    printIntoConsoleBox("\n\nInstallation is finished")
      
      
def runProcess(exe):  
    p = subprocess.Popen(exe, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
    while(True):
        retcode = p.poll() #returns None while subprocess is running
        line = p.stdout.readline()
        yield line
        if(retcode is not None):
            # if process ends read all line with still remain in the pipe 
            newline = ""
            while(True):
                newline = p.stdout.readline()
                if newline == "":
                    break
                line = line + newline
                yield line
            break

def executeCommand(exe):
    if not len(listbox.curselection()) == 0:
        listbox.select_clear(listbox.curselection())
    listbox.config(state=DISABLED)
    buttonInstall.config(state=DISABLED)
    buttonUninstall.config(state=DISABLED)
    buttonDownload.config(state=DISABLED)
    buttonupdateBROCRE.config(state=DISABLED)

    data = runProcess(exe) 
    for i in data:
        printIntoConsoleBox(i)

    printIntoConsoleBox("Done!\n")
    buttonupdateBROCRE.config(state=NORMAL)
    listbox.config(state=NORMAL)
    updateAllPackageDescriptions()
    
def printIntoConsoleBox(line):
    consoleOutput.config(state=NORMAL)
    consoleOutput.insert(END,line)
    consoleOutput.yview_moveto(1)  
    consoleOutput.config(state=DISABLED)
    
def executeTwoCommandSequential(exe, exe2):
    executeCommand(exe)
    executeCommand(exe2)
    
def executeCommandWithRestart(exe):
    executeCommand(exe)
    python = sys.executable
    os.execl(python, python, * sys.argv)
    
def executeCommandWithFunction(exe,function):
    executeCommand(exe)
    function()

def updateAllPackageDescriptions():
    global allPackages
    allPackages = extractPackageDescriptions(INSTALL_PATH)
    selectCategorieEvent()
    updateCurrentPackageDescription()

def downloadbutton():
    #Show Error if values are not entered
    error = ""
    toInstalledPackage = []
    
    if len(listbox.curselection()) == 0:
        error = "No package is selected!"
    else:        
        currentPackage = packageDescription()
        for package in allPackages:
            if package.name ==listbox.get(listbox.curselection()):
                currentPackage = package
    
    
    if not error == "":
        tkMessageBox.showerror(message=error)
    else:
        command = ['make', '-C', ROBOT_PACKAGE_PATH + currentPackage.folder +"/" +  currentPackage.name, 'fetch']
        t = Thread(target=executeCommand, args=(command,))
        t.start()


def installbutton():
    #Show Error if values are not entered
    error = ""
    toInstalledPackage = []
    
    if len(listbox.curselection()) == 0:
        error = "No package is selected!"
    else:        
        currentPackage = packageDescription()
        for package in allPackages:
            if package.name ==listbox.get(listbox.curselection()):
                currentPackage = package
    
    
    if not error == "":
        tkMessageBox.showerror(message=error)
    else:
        if( tkMessageBox.askokcancel("Install package", "Package " + currentPackage.name + " will be installed!") == 1):
            command = ['make', '-C', ROBOT_PACKAGE_PATH + currentPackage.folder +"/" +  currentPackage.name, 'update']
            t = Thread(target=executeCommand, args=(command,))
            t.start()
        
        
def uninstallbutton():
    #Show Error if values are not entered
    error = ""
    toInstalledPackage = []
    
    if len(listbox.curselection()) == 0:
        error = "No package is selected!"
    else:        
        currentPackage = packageDescription()
        for package in allPackages:
            if package.name ==listbox.get(listbox.curselection()):
                currentPackage = package
    
    if not error == "":
        tkMessageBox.showerror(message=error)
    else:
        deleteDialogRoot = DeletePackageDialog(root,"Uninstall package",  "Package " + currentPackage.name + " will be uninstalled!")
        if(deleteDialogRoot.isOKpressed):
            printIntoConsoleBox("Uninstalling ...")
            if(deleteDialogRoot.deleteCheckbuttonState.get() == 1):
                command = [INSTALL_PATH+"/sbin/robotpkg_delete", currentPackage.name]
                packagePath = os.environ['ROBOTPKG_BASE'] +"/"+ currentPackage.folder+"/"+currentPackage.name
                command2 = ["rm","-r", packagePath]
                t = Thread(target=executeTwoCommandSequential, args=(command,command2,))
                t.start()
            else:
                command = [INSTALL_PATH+"/sbin/robotpkg_delete", currentPackage.name]
                t = Thread(target=executeCommand, args=(command,))
                t.start()

        
def compilebutton():
    #Show Error if values are not entered
    error = ""
    toInstalledPackage = []
    
    if len(listbox.curselection()) == 0:
        error = "No package is selected!"
    else:        
        currentPackage = packageDescription()
        for package in allPackages:
            if package.name ==listbox.get(listbox.curselection()):
                currentPackage = package
    
    if not error == "":
        tkMessageBox.showerror(message=error)
    else:
        if( tkMessageBox.askokcancel("Compile package", "Package " + currentPackage.name + " will be compiled!") == 1):
            command = ["rosmake" , currentPackage.name]
            t = Thread(target=executeCommand, args=(command,))
            t.start()

        
def updateBROCREbutton():
    #Show Error if values are not entered
 #   tkMessageBox.showinfo("Result", "BROCRE will be updated!" )
    if( tkMessageBox.askokcancel("BROCRE update", "BROCRE will be updated!") == 1):
        printIntoConsoleBox("Updating ...")
        command = ["git","pull", "origin", "master"]
        t = Thread(target=executeCommandWithRestart, args=(command,))
        t.start()

  
def selectCategorieEvent():
    listbox.delete(0,END)
    selectedpackages = []
    for package in allPackages:
        for selectedCategories in checkbuttonValues:
            if package.categories.count(selectedCategories.get()) > 0:
                selectedpackages.append(package)
    
    selectedpackages = list(set(selectedpackages))
    selectedpackages.sort(cmp=None, key=operator.attrgetter('name'), reverse=False)
    
    for package in selectedpackages:
        listbox.insert(END, package.name)
        if not package.installedVersion == "":
            if package.installedVersion == package.version:
                listbox.itemconfig(END, background="green")
            if package.installedVersion != package.version:
                listbox.itemconfig(END, background="orange")
                    
def URL_click(event):
    if not len(listbox.curselection()) == 0:
        currentPackage = packageDescription()
        for package in allPackages:
            if package.name ==listbox.get(listbox.curselection()):
                currentPackage = package
        webbrowser.open_new(currentPackage.homepage)
    
def updateCurrentPackageDescriptionEvent(event):
    if event.widget == textField or event.widget == consoleOutput:
        return
    updateCurrentPackageDescription()
        
def updateCurrentPackageDescription():

    if not len(listbox.curselection()) == 0:
        currentPackage = packageDescription()
        
        for package in allPackages:
            if package.name ==listbox.get(listbox.curselection()):
                currentPackage = package
                
        textField.config(state=NORMAL)
        textField.delete(1.0, END)
        
        text = "Package: " + currentPackage.name + "\n" + \
        "Version: " + currentPackage.version +"\n" \
        "Installed version: " + currentPackage.installedVersion +"\n" \
        "Categories: " + currentPackage.categories.__str__() +"\n" \
        "Maintainer: " + currentPackage.maintainer +"\n" \
        "Homepage: "
        textField.insert(END, text)
        textField.insert(INSERT, currentPackage.homepage, "hyper")
        text =  "\nLicense: " + currentPackage.license +"\n\n" \
        "Description: " + currentPackage.description +"\n"
        textField.insert(END, text)
        
        textField.config(state=DISABLED)
        
        if currentPackage.installedVersion == "":
            buttonInstall.config(state=NORMAL)
            buttonDownload.config(state=NORMAL)
            buttonUninstall.config(state=DISABLED)
        elif currentPackage.installedVersion != currentPackage.version:
            buttonInstall.config(state=NORMAL)
            buttonDownload.config(state=NORMAL)
            buttonUninstall.config(state=NORMAL)
        else:
            buttonUninstall.config(state=NORMAL)
            buttonInstall.config(state=DISABLED)
            buttonDownload.config(state=NORMAL)
            
        
    else:
        textField.config(state=NORMAL)
        textField.delete(1.0, END)
        textField.config(state=DISABLED)
        buttonInstall.config(state=DISABLED)
        buttonDownload.config(state=DISABLED)
        buttonUninstall.config(state=DISABLED)
        

if __name__ == "__main__":
                    
    ''' Window Creation '''
    root = Tk()
    root.title("BROCRE Package Installer")
          
    allPackages = extractPackageDescriptions(INSTALL_PATH)
        
    allCategories = list()
    for package in allPackages:
         for category in package.categories:
             allCategories.append(category)
             
    allCategories = list(set(allCategories))
    allCategories.sort(cmp=None, key=None, reverse=False)
    
    # Categories Frame
    checkbuttonRow = 0
    checkbuttonValues = list()
    
    groupCategories = LabelFrame(root, text="Categories", padx=5, pady=5)
    groupCategories.grid(row=0, column=0, sticky=N+S+W)
    
    var = StringVar()
    for categorie in allCategories:
        checkbuttonValues.append(StringVar())
        cb = Checkbutton(groupCategories, text=categorie, variable=checkbuttonValues[checkbuttonRow], onvalue=categorie, offvalue="", command=selectCategorieEvent)
        cb.grid(row=checkbuttonRow, column=0, sticky=W)
        cb.select()
        checkbuttonRow = checkbuttonRow + 1
        
        
    # Packages Listbox Frame    
    listboxframe = LabelFrame(root, text="BROCRE Packages", bd=2, relief=SUNKEN)
    yscrollbarlistbox = Scrollbar(listboxframe)
    yscrollbarlistbox.grid(row=0, column=1, sticky=N+S)
    listboxframe.grid(row=0, column=1, sticky=N+S+W)
    listboxheight = checkbuttonRow
    if listboxheight < 21:
        listboxheight = 21
    listbox = Listbox(listboxframe, selectmode=SINGLE, height=listboxheight , width = 30, yscrollcommand=yscrollbarlistbox.set)
    listbox.grid(row=0, column=0, sticky=N+S+W)
    yscrollbarlistbox.config(command=listbox.yview)
    
    root.bind("<Button-1>", updateCurrentPackageDescriptionEvent)
    root.bind("<Key>", updateCurrentPackageDescriptionEvent)
    
    # Legned Frame   
    
    Legendframe = LabelFrame(root, text="BROCRE Packages Legend", bd=2, relief=SUNKEN)
    Legendframe.grid(row=1, column=1, sticky=N+S+W+E)
    legend1 = Label(Legendframe, text="available", background='white', relief=RIDGE)
    legend1.grid(row=0, column=0, sticky=W)
    legend2 = Label(Legendframe, text=" ", )
    legend2.grid(row=0, column=1, sticky=W)
    legend3 = Label(Legendframe, text="installed", background='green', relief=RIDGE)
    legend3.grid(row=0, column=2, sticky=N)
    legend4 = Label(Legendframe, text=" ", )
    legend4.grid(row=0, column=3, sticky=W)
    legend5 = Label(Legendframe, text="new version available", background='orange', relief=RIDGE)
    legend5.grid(row=0, column=4, sticky=E)
    
    # Description Frame   

    descriptionFrame = LabelFrame(root, text="Description", padx=5, pady=5)
    descriptionFrame.grid(row=0, column=2, sticky=N+S+W+E)
    descriptionFrame.grid_columnconfigure(0,weight=1)
    textField = Text(descriptionFrame, wrap=WORD, height = 20)
    textField.grid(row=0, column=0, sticky=N+S+W+E)
    textField.tag_config("hyper", foreground="blue", underline=1)
       # textField.tag_bind("a", "<Enter>", URL_click)
       # textField.tag_bind("a", "<Leave>", URL_click)
    textField.tag_bind("hyper", "<Button-1>", URL_click)
    textField.config(cursor="arrow")
    
    
    buttonInstall = Button(descriptionFrame, text="Install", command=installbutton)
    buttonInstall.grid( row=1,column=0,rowspan=2, sticky=W)
    buttonUninstall = Button(descriptionFrame, text="Uninstall", command=uninstallbutton)
    buttonUninstall.grid( row=1,column=0, sticky=N)
    buttonDownload = Button(descriptionFrame, text="Download", command=downloadbutton)
    buttonDownload.grid( row=1,column=0, sticky=E)
   # buttonCompile = Button(root, text="Compile", command=compilebutton)
   # buttonCompile.grid()
    buttonupdateBROCRE = Button(root, text="Update BROCRE", command=updateBROCREbutton)
    buttonupdateBROCRE.grid( row=1,column=2, sticky=E)
    
    buttonInstall.config(state=DISABLED)
    buttonUninstall.config(state=DISABLED)
    buttonDownload.config(state=DISABLED)
    
    
    # Console Frame
    consoleframe = LabelFrame(root, text="Console", bd=2, relief=SUNKEN)
    
    xscrollbar = Scrollbar(consoleframe, orient=HORIZONTAL)
    xscrollbar.grid(row=1, column=0, sticky=E+W)
    
    yscrollbar = Scrollbar(consoleframe)
    yscrollbar.grid(row=0, column=1, sticky=N+S)
    
    consoleOutput = Text(consoleframe, wrap=WORD, bd=0, height = 15, width = 115,
                xscrollcommand=xscrollbar.set,
                yscrollcommand=yscrollbar.set)
    
    consoleOutput.grid(row=0, column=0, sticky=N+S+E+W)
    

    xscrollbar.config(command=consoleOutput.xview)
    yscrollbar.config(command=consoleOutput.yview)
    consoleframe.grid(row=2,columnspan=3, column=0, sticky=N+S+E+W)
    
    
 #   consoleOutput.grid_columnconfigure(0,weight=1)
    consoleframe.grid_columnconfigure(0,weight=1)
    consoleframe.grid_rowconfigure(0, weight=1)
    
    
    root.grid_columnconfigure(2,weight=1)
    root.grid_rowconfigure(0,weight=1)
    #root.grid_columnconfigure(1,weight=1)
    #root.grid_rowconfigure(2,weight=1)
    #root.grid_columnconfigure(1,weight=1)
    root.grid_rowconfigure(2,weight=10000)
    
    
    selectCategorieEvent()

    checkIfBROCREisBootstraped(root)

    root.mainloop()