Add python script to control redshift through a status icon (tray icon).

2 files changed, 90 insertions, 0 deletions

diff --git a/Makefile.am b/Makefile.am
index 6bf9c7b..6f7be58 100644
--- a/Makefile.am
+++ b/Makefile.am
@@ -30,6 +30,9 @@ redshift_LDADD += \
 	$(XF86VM_LIBS) $(XF86VM_CFLAGS)
 endif
 
+# gtk-redshift Script
+dist_bin_SCRIPTS = src/gtk-redshift/gtk-redshift
+
 # Icons
 icondir = @datadir@/icons/hicolor/scalable/apps
 icon_DATA = data/icons/hicolor/scalable/apps/redshift.svg
diff --git a/src/gtk-redshift/gtk-redshift b/src/gtk-redshift/gtk-redshift
new file mode 100755
index 0000000..cd35113
--- /dev/null
+++ b/src/gtk-redshift/gtk-redshift
@@ -0,0 +1,87 @@
+#!/usr/bin/env python
+# gtk-redshift -- GTK+ status icon source
+# This file is part of Redshift.
+
+# Redshift is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+
+# Redshift is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+
+# You should have received a copy of the GNU General Public License
+# along with Redshift.  If not, see <http://www.gnu.org/licenses/>.
+
+# Copyright (c) 2010  Jon Lund Steffensen <jonlst@gmail.com>
+
+
+import sys
+import subprocess, signal
+
+import pygtk
+pygtk.require("2.0")
+
+import gtk, glib
+
+
+if __name__ == '__main__':
+    # Start redshift with arguments from the command line
+    args = sys.argv[1:]
+    args.insert(0, 'redshift')
+    process = subprocess.Popen(args)
+
+    try:
+        # Create status icon
+        status_icon = gtk.StatusIcon()
+        status_icon.set_from_icon_name('redshift')
+        status_icon.set_tooltip('Redshift')
+
+        def toggle_cb(widget, data=None):
+            process.send_signal(signal.SIGUSR1)
+
+        def destroy_cb(widget, data=None):
+            status_icon.set_visible(False)
+            gtk.main_quit()
+            return False
+
+        # Create popup menu
+        status_menu = gtk.Menu()
+
+        toggle_item = gtk.ImageMenuItem('Toggle')
+        toggle_item.connect('activate', toggle_cb)
+        status_menu.append(toggle_item)
+
+        quit_item = gtk.ImageMenuItem(gtk.STOCK_QUIT)
+        quit_item.connect('activate', destroy_cb)
+        status_menu.append(quit_item)
+
+        def popup_menu_cb(widget, button, time, data=None):
+            status_menu.show_all()
+            status_menu.popup(None, None, gtk.status_icon_position_menu,
+                              button, time, status_icon)
+
+        # Connect signals for status icon and show
+        status_icon.connect('activate', toggle_cb)
+        status_icon.connect('popup-menu', popup_menu_cb)
+        status_icon.set_visible(True)
+
+        def child_cb(pid, cond, data=None):
+            sys.exit(-1)
+
+        # Add awtch on child process
+        glib.child_watch_add(process.pid, child_cb)
+
+        # Run main loop
+        gtk.main()
+
+    except KeyboardInterrupt:
+        # Ignore user interruption
+        pass
+
+    finally:
+        # Always terminate redshift
+        process.terminate()
+        process.wait()