setTitle is now a Screen method
[enigma2.git] / lib / python / Screens / PluginBrowser.py
index 1c9097465a000def4b0705ae3dd00d239a26ae4a..16f3aadd32774abbc8a0d49c0dd786453f83a606 100644 (file)
@@ -1,24 +1,35 @@
 from Screen import Screen
 
+from enigma import eConsoleAppContainer, loadPNG
+
 from Components.MenuList import MenuList
 from Components.ActionMap import ActionMap
 from Components.PluginComponent import plugins
 from Components.PluginList import *
 from Components.config import config
+from Components.Label import Label
+from Screens.MessageBox import MessageBox
+from Screens.Console import Console
 from Plugins.Plugin import PluginDescriptor
+from Tools.Directories import resolveFilename, SCOPE_PLUGINS, SCOPE_SKIN_IMAGE
 
 class PluginBrowser(Screen):
        def __init__(self, session):
                Screen.__init__(self, session)
                
+               self["red"] = Label(_("Remove Plugins"))
+               self["green"] = Label(_("Download Plugins"))
+               
                self.list = []
                self["list"] = PluginList(self.list)
                self.updateList()
                
-               self["actions"] = ActionMap(["WizardActions"], 
+               self["actions"] = ActionMap(["WizardActions", "ColorActions"],
                {
                        "ok": self.save,
                        "back": self.close,
+                       "red": self.delete,
+                       "green": self.download
                })
                
        def save(self):
@@ -37,3 +48,126 @@ class PluginBrowser(Screen):
                        self.list.append(PluginEntryComponent(plugin))
                
                self["list"].l.setList(self.list)
+
+       def delete(self):
+               self.session.openWithCallback(self.updateList, PluginDownloadBrowser, PluginDownloadBrowser.REMOVE)
+       
+       def download(self):
+               self.session.openWithCallback(self.updateList, PluginDownloadBrowser, PluginDownloadBrowser.DOWNLOAD)
+
+class PluginDownloadBrowser(Screen):
+       DOWNLOAD = 0
+       REMOVE = 1
+       
+       def __init__(self, session, type):
+               Screen.__init__(self, session)
+               
+               self.type = type
+               
+               self.container = eConsoleAppContainer()
+               self.container.appClosed.get().append(self.runFinished)
+               self.container.dataAvail.get().append(self.dataAvail)
+               self.onLayoutFinish.append(self.startRun)
+               self.onShown.append(self.setWindowTitle)
+               
+               self.list = []
+               self["list"] = PluginList(self.list)
+               self.pluginlist = []
+               self.expanded = []
+               
+               if self.type == self.DOWNLOAD:
+                       self["text"] = Label(_("Downloading plugin information. Please wait..."))
+               elif self.type == self.REMOVE:
+                       self["text"] = Label(_("Getting plugin information. Please wait..."))
+               
+               self.run = 0
+                               
+               self["actions"] = ActionMap(["WizardActions"], 
+               {
+                       "ok": self.go,
+                       "back": self.close,
+               })
+               
+       def go(self):
+               if type(self["list"].l.getCurrentSelection()[0]) is str: # category
+                       if self["list"].l.getCurrentSelection()[0] in self.expanded:
+                               self.expanded.remove(self["list"].l.getCurrentSelection()[0])
+                       else:
+                               self.expanded.append(self["list"].l.getCurrentSelection()[0])
+                       self.updateList()
+               else:
+                       if self.type == self.DOWNLOAD:
+                               self.session.openWithCallback(self.runInstall, MessageBox, _("Do you really want to download\nthe plugin \"" + self["list"].l.getCurrentSelection()[0].name + "\"?"))
+                       elif self.type == self.REMOVE:
+                               self.session.openWithCallback(self.runInstall, MessageBox, _("Do you really want to REMOVE\nthe plugin \"" + self["list"].l.getCurrentSelection()[0].name + "\"?"))
+
+       def runInstall(self, val):
+               if val:
+                       if self.type == self.DOWNLOAD:
+                               self.session.openWithCallback(self.installFinished, Console, ["ipkg install " + "enigma2-plugin-" + self["list"].l.getCurrentSelection()[0].name])
+                       elif self.type == self.REMOVE:
+                               self.session.openWithCallback(self.installFinished, Console, ["ipkg remove " + "enigma2-plugin-" + self["list"].l.getCurrentSelection()[0].name])
+
+       def setWindowTitle(self):
+               if self.type == self.DOWNLOAD:
+                       self.setTitle(_("Downloadable new plugins"))
+               elif self.type == self.REMOVE:
+                       self.setTitle(_("Remove plugins"))
+
+       def startRun(self):
+               self["list"].instance.hide()
+               if self.type == self.DOWNLOAD:
+                       self.container.execute("ipkg update")
+               elif self.type == self.REMOVE:
+                       self.run = 1
+                       self.container.execute("ipkg list_installed enigma2-plugin-*")
+               
+       def installFinished(self):
+               plugins.readPluginList(resolveFilename(SCOPE_PLUGINS))
+               self.close()
+               
+       def runFinished(self, retval):
+               if self.run == 0:
+                       self.run = 1
+                       if self.type == self.DOWNLOAD:
+                               self.container.execute("ipkg list enigma2-plugin-*")
+               else:
+                       if len(self.pluginlist) > 0:
+                               self.updateList()
+                               self["list"].instance.show()
+                       else:
+                               self["text"].setText("No plugins found")
+
+       def dataAvail(self, str):
+               for x in str.split('\n'):
+                       plugin = x.split(" - ")
+                       if len(plugin) == 3:
+                               plugin.append(plugin[0][15:])
+
+                               self.pluginlist.append(plugin)
+       
+       def updateList(self):
+               self.list = []
+               expandableIcon = loadPNG(resolveFilename(SCOPE_SKIN_IMAGE, "expandable-plugins.png"))
+               expandedIcon = loadPNG(resolveFilename(SCOPE_SKIN_IMAGE, "expanded-plugins.png"))
+               verticallineIcon = loadPNG(resolveFilename(SCOPE_SKIN_IMAGE, "verticalline-plugins.png"))
+               
+               self.plugins = {}
+               for x in self.pluginlist:
+                       split = x[3].split('-')
+                       if len(split) < 2:
+                               continue
+                       if not self.plugins.has_key(split[0]):
+                               self.plugins[split[0]] = []
+                               
+                       self.plugins[split[0]].append((PluginDescriptor(name = x[3], description = x[2], icon = verticallineIcon), split[1]))
+                       
+               for x in self.plugins.keys():
+                       if x in self.expanded:
+                               self.list.append(PluginCategoryComponent(x, expandedIcon))
+                               for plugin in self.plugins[x]:
+                                       self.list.append(PluginDownloadComponent(plugin[0], plugin[1]))
+                       else:
+                               self.list.append(PluginCategoryComponent(x, expandableIcon))
+               self["list"].l.setList(self.list)
+