scripts/ombqueuehandler.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 = 'i::';
  24 $longoptions = array('id::');
  25
  26 $helptext = <<<END_OF_OMB_HELP
  27 Daemon script for pushing new notices to OpenMicroBlogging subscribers.
  28
  29     -i --id           Identity (default none)
  30
  31 END_OF_OMB_HELP;
  32
  33 require_once INSTALLDIR.'/scripts/commandline.inc';
  34
  35 require_once INSTALLDIR . '/lib/omb.php';
  36 require_once INSTALLDIR . '/lib/queuehandler.php';
  37
  38 set_error_handler('common_error_handler');
  39
  40 class OmbQueueHandler extends QueueHandler
  41 {
  42
  43     function transport()
  44     {
  45         return 'omb';
  46     }
  47
  48     function start()
  49     {
  50         $this->log(LOG_INFO, "INITIALIZE");
  51         return true;
  52     }
  53
  54     function handle_notice($notice)
  55     {
  56         if ($this->is_remote($notice)) {
  57             $this->log(LOG_DEBUG, 'Ignoring remote notice ' . $notice->id);
  58             return true;
  59         } else {
  60             return omb_broadcast_notice($notice);
  61         }
  62     }
  63
  64     function finish()
  65     {
  66     }
  67
  68     function is_remote($notice)
  69     {
  70         $user = User::staticGet($notice->profile_id);
  71         return is_null($user);
  72     }
  73 }
  74
  75 if (have_option('i')) {
  76     $id = get_option_value('i');
  77 } else if (have_option('--id')) {
  78     $id = get_option_value('--id');
  79 } else if (count($args) > 0) {
  80     $id = $args[0];
  81 } else {
  82     $id = null;
  83 }
  84
  85 $handler = new OmbQueueHandler($id);
  86
  87 $handler->runOnce();