[nanny: 1/11] Add initial support of Consolekit

[Roberto Majadas <telemaco src gnome org>]

commit 6f09240dda1efaa7adacf7d95ac1ace6c8028255
Author: Roberto Majadas <roberto majadas openshine com>
Date:   Wed Mar 31 16:25:04 2010 +0200

    Add initial support of Consolekit

 daemon/src/LinuxSessionCKFiltering.py |  116 +++++++++++++++++++++++++++++++++
 daemon/src/Makefile.am                |    3 +-
 daemon/src/QuarterBack.py             |    2 +-
 3 files changed, 119 insertions(+), 2 deletions(-)
---
diff --git a/daemon/src/LinuxSessionCKFiltering.py b/daemon/src/LinuxSessionCKFiltering.py
new file mode 100644
index 0000000..80aceed
--- /dev/null
+++ b/daemon/src/LinuxSessionCKFiltering.py
@@ -0,0 +1,116 @@
+#!/usr/bin/env python
+#
+# Copyright (C) 2009,2010 Junta de Andalucia
+# 
+# Authors:
+#   Roberto Majadas <roberto.majadas at openshine.com>
+#
+# This program 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 2, or (at your option)
+# any later version.
+#
+# This program 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 this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301
+# USA
+
+import gobject
+import os
+
+from twisted.internet import reactor
+from time import localtime, strftime
+
+class LinuxSessionCKFiltering(gobject.GObject) :
+    def __init__(self, quarterback) :
+        gobject.GObject.__init__(self)
+        self.quarterback = quarterback
+
+        self.uids_blocked = []
+        
+        reactor.addSystemEventTrigger("before", "startup", self.start)
+        reactor.addSystemEventTrigger("before", "shutdown", self.stop)
+        
+        self.quarterback.connect('block-status', self.__update_cb)
+        self.quarterback.connect("update-blocks", self.__update_blocks_cb)
+
+    def start(self):
+        self.__update_blocks_cb(self.quarterback, self.quarterback.blocks)
+        print "Start Linux Session ConsoleKit Filtering"
+
+    def stop(self):
+        print "Stop Linux Session ConsoleKit Filtering"
+
+    def __update_blocks_cb(self, quarterback, blocks):
+        for user_id in blocks.keys() :
+            if not blocks[user_id].has_key(0) :
+                continue
+
+            block_status, next_change = quarterback.is_blocked(user_id, 0)
+            if quarterback.get_available_time(user_id, 0) == 0 or block_status == True:
+                if user_id in self.uids_blocked :
+                    return
+
+                users = quarterback.usersmanager.get_users()
+                for uid, uname, ufname in users :
+                    if str(uid) == user_id :
+                        self.uids_blocked.append(user_id)
+                        print "blocked session to user '%s'" % uname
+                        return
+            else:
+                if user_id in self.uids_blocked:
+                    users = quarterback.usersmanager.get_users()
+                    for uid, uname, ufname in users :
+                        if str(uid) == user_id :
+                            self.uids_blocked.pop(self.uids_blocked.index(user_id))
+                            print "Unblocked session to user '%s'" % uname
+
+
+    def __update_cb(self, quarterback, block_status, user_id, app_id, next_change, available_time):
+        if app_id != 0 :
+            return
+
+        if block_status == True or available_time == 0 :
+            if user_id in self.uids_blocked :
+                self.__logout_session_if_is_running(user_id)
+        
+        if available_time == 0 :
+            if user_id in self.uids_blocked :
+                return
+
+            users = quarterback.usersmanager.get_users()
+            for uid, uname, ufname in users :
+                if str(uid) == user_id :
+                    self.uids_blocked.append(user_id)
+                    self.__logout_session_if_is_running(user_id)
+                    print "blocked session to user '%s'" % uname
+                    return
+        
+        if block_status == False:
+            if user_id in self.uids_blocked:
+                users = quarterback.usersmanager.get_users()
+                for uid, uname, ufname in users :
+                    if str(uid) == user_id :
+                        self.uids_blocked.pop(self.uids_blocked.index(user_id))
+                        print "Unblocked session to user '%s'" % uname
+            return
+        else:
+            if user_id in self.uids_blocked :
+                return
+            
+            users = quarterback.usersmanager.get_users()
+            for uid, uname, ufname in users :
+                if str(uid) == user_id :
+                    self.uids_blocked.append(user_id)
+                    self.__logout_session_if_is_running(user_id)
+                    print "blocked session to user '%s'" % uname
+
+            return
+        
+    def __logout_session_if_is_running(self, user_id):
+        pass
diff --git a/daemon/src/Makefile.am b/daemon/src/Makefile.am
index 3ccf2de..d576e14 100644
--- a/daemon/src/Makefile.am
+++ b/daemon/src/Makefile.am
@@ -11,7 +11,8 @@ corelib_PYTHON =	 __init__.py 			\
 			LinuxFiltering.py		\
 			LinuxWebContentFiltering.py 	\
 			LinuxUsersManager.py		\
-			LinuxSessionFiltering.py
+			LinuxSessionFiltering.py	\
+			LinuxSessionCKFiltering.py
 
 if HAS_HACHOIR_REGEX
 dgimporterdir = $(pythondir)/nanny/daemon
diff --git a/daemon/src/QuarterBack.py b/daemon/src/QuarterBack.py
index 2a0502e..4f7eb75 100644
--- a/daemon/src/QuarterBack.py
+++ b/daemon/src/QuarterBack.py
@@ -33,7 +33,7 @@ from datetime import datetime, timedelta
 from LinuxFiltering import LinuxFiltering as FirewallFilter
 from LinuxWebContentFiltering import LinuxWebContentFiltering as WebContentFilter
 from LinuxUsersManager import LinuxUsersManager as UsersManager
-from LinuxSessionFiltering import LinuxSessionFiltering as SessionFilter
+from LinuxSessionCKFiltering import LinuxSessionCKFiltering as SessionFilter
 from FilterManager import FilterManager as FilterManager
 
 from Chrono import Chrono