Check scope, else a privacy leaks happens this way:

[quix0rs-gnu-social.git] / scripts / imdaemon.php

diff --git a/scripts/imdaemon.php b/scripts/imdaemon.php
index ffb5ecf0d945fd98b3613bd920928b14836937b0..7d9ff1fe7dc7d1d65d0007c14e76219e713e196d 100755 (executable)
--- a/scripts/imdaemon.php
+++ b/scripts/imdaemon.php
@@ -20,36 +20,41 @@
 
 define('INSTALLDIR', realpath(dirname(__FILE__) . '/..'));
 
-$shortoptions = 'fi::';
-$longoptions = array('id::', 'foreground');
+$shortoptions = 'fi::a';
+$longoptions = array('id::', 'foreground', 'all');
 
 $helptext = <<<END_OF_IM_HELP
 Daemon script for receiving new notices from IM users.
 
     -i --id           Identity (default none)
+    -a --all          Handle XMPP for all local sites
+                      (requires Stomp queue handler, status_network setup)
     -f --foreground   Stay in the foreground (default background)
 
 END_OF_IM_HELP;
 
-require_once INSTALLDIR.'/scripts/commandline.inc';
+require_once INSTALLDIR.'/scripts/commandline.inc.php';
 
 class ImDaemon extends SpawningDaemon
 {
-    function __construct($id=null, $daemonize=true, $threads=1)
+    protected $allsites = false;
+
+    function __construct($id=null, $daemonize=true, $threads=1, $allsites=false)
     {
         if ($threads != 1) {
             // This should never happen. :)
             throw new Exception("IMDaemon can must run single-threaded");
         }
         parent::__construct($id, $daemonize, $threads);
+        $this->allsites = $allsites;
     }
 
     function runThread()
     {
         common_log(LOG_INFO, 'Waiting to listen to IM connections and queues');
 
-        $master = new ImMaster($this->get_id());
-        $master->init();
+        $master = new ImMaster($this->get_id(), $this->processManager());
+        $master->init($this->allsites);
         $master->service();
 
         common_log(LOG_INFO, 'terminating normally');
@@ -61,6 +66,14 @@ class ImDaemon extends SpawningDaemon
 
 class ImMaster extends IoMaster
 {
+    protected $processManager;
+
+    function __construct($id, $processManager)
+    {
+        parent::__construct($id);
+        $this->processManager = $processManager;
+    }
+
     /**
      * Initialize IoManagers for the currently configured site
      * which are appropriate to this instance.
@@ -69,7 +82,10 @@ class ImMaster extends IoMaster
     {
         $classes = array();
         if (Event::handle('StartImDaemonIoManagers', array(&$classes))) {
-            $classes[] = 'QueueManager';
+            $qm = QueueManager::get();
+            $qm->setActiveGroup('im');
+            $classes[] = $qm;
+            $classes[] = $this->processManager;
         }
         Event::handle('EndImDaemonIoManagers', array(&$classes));
         foreach ($classes as $class) {
@@ -87,7 +103,8 @@ if (have_option('i', 'id')) {
 }
 
 $foreground = have_option('f', 'foreground');
+$all = have_option('a') || have_option('--all');
 
-$daemon = new ImDaemon($id, !$foreground);
+$daemon = new ImDaemon($id, !$foreground, 1, $all);
 
 $daemon->runOnce();