]> git.mxchange.org Git - quix0rs-gnu-social.git/blob - scripts/imdaemon.php
Merge commit 'refs/merge-requests/158' of git://gitorious.org/statusnet/mainline...
[quix0rs-gnu-social.git] / scripts / imdaemon.php
1 #!/usr/bin/env php
2 <?php
3 /*
4  * StatusNet - the distributed open-source microblogging tool
5  * Copyright (C) 2008, 2009, StatusNet, Inc.
6  *
7  * This program is free software: you can redistribute it and/or modify
8  * it under the terms of the GNU Affero General Public License as published by
9  * the Free Software Foundation, either version 3 of the License, or
10  * (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU Affero General Public License for more details.
16  *
17  * You should have received a copy of the GNU Affero General Public License
18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
19  */
20
21 define('INSTALLDIR', realpath(dirname(__FILE__) . '/..'));
22
23 $shortoptions = 'fi::a';
24 $longoptions = array('id::', 'foreground', 'all');
25
26 $helptext = <<<END_OF_IM_HELP
27 Daemon script for receiving new notices from IM users.
28
29     -i --id           Identity (default none)
30     -a --all          Handle XMPP for all local sites
31                       (requires Stomp queue handler, status_network setup)
32     -f --foreground   Stay in the foreground (default background)
33
34 END_OF_IM_HELP;
35
36 require_once INSTALLDIR.'/scripts/commandline.inc';
37
38 class ImDaemon extends SpawningDaemon
39 {
40     protected $allsites = false;
41
42     function __construct($id=null, $daemonize=true, $threads=1, $allsites=false)
43     {
44         if ($threads != 1) {
45             // This should never happen. :)
46             throw new Exception("IMDaemon can must run single-threaded");
47         }
48         parent::__construct($id, $daemonize, $threads);
49         $this->allsites = $allsites;
50     }
51
52     function runThread()
53     {
54         common_log(LOG_INFO, 'Waiting to listen to IM connections and queues');
55
56         $master = new ImMaster($this->get_id(), $this->processManager());
57         $master->init($this->allsites);
58         $master->service();
59
60         common_log(LOG_INFO, 'terminating normally');
61
62         return $master->respawn ? self::EXIT_RESTART : self::EXIT_SHUTDOWN;
63     }
64
65 }
66
67 class ImMaster extends IoMaster
68 {
69     protected $processManager;
70
71     function __construct($id, $processManager)
72     {
73         parent::__construct($id);
74         $this->processManager = $processManager;
75     }
76
77     /**
78      * Initialize IoManagers for the currently configured site
79      * which are appropriate to this instance.
80      */
81     function initManagers()
82     {
83         $classes = array();
84         if (Event::handle('StartImDaemonIoManagers', array(&$classes))) {
85             $qm = QueueManager::get();
86             $qm->setActiveGroup('im');
87             $classes[] = $qm;
88             $classes[] = $this->processManager;
89         }
90         Event::handle('EndImDaemonIoManagers', array(&$classes));
91         foreach ($classes as $class) {
92             $this->instantiate($class);
93         }
94     }
95 }
96
97 if (version_compare(PHP_VERSION, '5.2.6', '<')) {
98     $arch = php_uname('m');
99     if ($arch == 'x86_64' || $arch == 'amd64') {
100         print "Aborting daemon - 64-bit PHP prior to 5.2.6 has known bugs in stream_select; you are running " . PHP_VERSION . " on $arch.\n";
101         exit(1);
102     }
103 }
104
105 if (have_option('i', 'id')) {
106     $id = get_option_value('i', 'id');
107 } else if (count($args) > 0) {
108     $id = $args[0];
109 } else {
110     $id = null;
111 }
112
113 $foreground = have_option('f', 'foreground');
114 $all = have_option('a') || have_option('--all');
115
116 $daemon = new ImDaemon($id, !$foreground, 1, $all);
117
118 $daemon->runOnce();