Use gtk on linux

1 files changed, 35 insertions, 14 deletions

diff --git a/parrot_zik/indicator/linux.py b/parrot_zik/indicator/linux.py
index 278f781..f6f9cfc 100644
--- a/parrot_zik/indicator/linux.py
+++ b/parrot_zik/indicator/linux.py
@@ -6,17 +6,7 @@ from parrot_zik.indicator.base import BaseIndicator
 
 
 class LinuxIndicator(BaseIndicator):
-    def __init__(self, icon, menu):
-        import appindicator
-        self.icon_directory = os.path.join('/', 'usr', 'share', 'icons', 'zik')
-        if not os.path.isdir(self.icon_directory):
-            self.icon_directory = os.path.join('share', 'icons', 'zik')
-        statusicon = appindicator.Indicator(
-            "new-parrotzik-indicator", "indicator-messages",
-            appindicator.CATEGORY_APPLICATION_STATUS)
-        statusicon.set_status(appindicator.STATUS_ACTIVE)
-        statusicon.set_icon_theme_path(self.icon_directory)
-        statusicon.set_menu(menu.gtk_menu)
+    def __init__(self, icon, menu, statusicon):
         super(LinuxIndicator, self).__init__(icon, menu, statusicon)
 
     def gtk_right_click_event(self, icon, button, time):
@@ -28,9 +18,6 @@ class LinuxIndicator(BaseIndicator):
             self.menu_shown = False
             self.menu.poVpdown()
 
-    def setIcon(self, name):
-        self.statusicon.set_icon(name)
-
     @classmethod
     def main(cls):
         gtk.main()
@@ -46,3 +33,37 @@ class LinuxIndicator(BaseIndicator):
         about_dialog.set_authors(["Dmitry Moiseev m0sia@m0sia.ru"])
         about_dialog.run()
         about_dialog.destroy()
+
+
+class LinuxAppIndicator(LinuxIndicator):
+    def __init__(self, icon, menu):
+        import appindicator
+        self.icon_directory = os.path.join('/', 'usr', 'share', 'icons', 'zik')
+        if not os.path.isdir(self.icon_directory):
+            self.icon_directory = os.path.join('share', 'icons', 'zik')
+        statusicon = appindicator.Indicator(
+            "new-parrotzik-indicator", "indicator-messages",
+            appindicator.CATEGORY_APPLICATION_STATUS)
+        statusicon.set_status(appindicator.STATUS_ACTIVE)
+        statusicon.set_icon_theme_path(self.icon_directory)
+        statusicon.set_menu(menu.gtk_menu)
+        super(LinuxIndicator, self).__init__(icon, menu, statusicon)
+
+    def setIcon(self, name):
+        self.statusicon.set_icon(name)
+
+
+
+class LinuxGtkIndicator(LinuxIndicator):
+    def __init__(self, icon, menu):
+        self.icon_directory = os.path.join(
+            '/usr', 'share', 'icons/')
+        self.menu_shown = False
+        statusicon = gtk.StatusIcon()
+        statusicon.connect("popup-menu", self.gtk_right_click_event)
+        statusicon.set_tooltip("Parrot Zik")
+        super(LinuxIndicator, self).__init__(icon, menu, statusicon)
+
+    def setIcon(self, name):
+        self.statusicon.set_from_file(self.icon_directory + name + '.png')
+