plugins/Autocomplete/autocomplete.php

   1 <?php
   2 /**
   3  * StatusNet, the distributed open-source microblogging tool
   4  *
   5  * List users for autocompletion
   6  *
   7  * PHP version 5
   8  *
   9  * LICENCE: This program is free software: you can redistribute it and/or modify
  10  * it under the terms of the GNU Affero General Public License as published by
  11  * the Free Software Foundation, either version 3 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU Affero General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Affero General Public License
  20  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21  *
  22  * @category  Plugin
  23  * @package   StatusNet
  24  * @author    Craig Andrews <candrews@integralblue.com>
  25  * @copyright 2008-2009 StatusNet, Inc.
  26  * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
  27  * @link      http://status.net/
  28  */
  29
  30 if (!defined('STATUSNET') && !defined('LACONICA')) {
  31     exit(1);
  32 }
  33
  34 /**
  35  * List users for autocompletion
  36  *
  37  * This is the form for adding a new g
  38  *
  39  * @category Plugin
  40  * @package  StatusNet
  41  * @author   Craig Andrews <candrews@integralblue.com>
  42  * @license  http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
  43  * @link     http://status.net/
  44  */
  45
  46 class AutocompleteAction extends Action
  47 {
  48     private $result;
  49
  50     function prepare($args)
  51     {
  52         parent::prepare($args);
  53         $this->results = array();
  54         $q = $this->arg('q');
  55         $limit = $this->arg('limit');
  56         if($limit > 200) $limit=200; //prevent DOS attacks
  57         if(substr($q,0,1)=='@'){
  58             //user search
  59             $q=substr($q,1);
  60             $user = new User();
  61             $user->limit($limit);
  62             $user->whereAdd('nickname like \'' . trim($user->escape($q), '\'') . '%\'');
  63             $user->find();
  64             while($user->fetch()) {
  65                 $profile = Profile::staticGet($user->id);
  66                 $this->results[]=array('nickname' => $user->nickname, 'fullname'=> $profile->fullname, 'type'=>'user');
  67             }
  68         }
  69         if(substr($q,0,1)=='!'){
  70             //group search
  71             $q=substr($q,1);
  72             $group = new User_group();
  73             $group->limit($limit);
  74             $group->whereAdd('nickname like \'' . trim($group->escape($q), '\'') . '%\'');
  75             $group->find();
  76             while($group->fetch()) {
  77                 $this->results[]=array('nickname' => $group->nickname, 'fullname'=> $group->fullname, 'type'=>'group');
  78             }
  79         }
  80         return true;
  81     }
  82
  83     function handle($args)
  84     {
  85         parent::handle($args);
  86         foreach($this->results as $result) {
  87             print json_encode($result) . "\n";
  88         }
  89     }
  90 }