includes/user/UserArray.php

   1 <?php
   2 /**
   3  * Class to walk into a list of User objects.
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with this program; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  18  * http://www.gnu.org/copyleft/gpl.html
  19  *
  20  * @file
  21  */
  22
  23 use Wikimedia\Rdbms\IResultWrapper;
  24
  25 abstract class UserArray implements Iterator {
  26         /**
  27          * @param IResultWrapper $res
  28          * @return UserArrayFromResult
  29          */
  30         static function newFromResult( $res ) {
  31                 $userArray = null;
  32                 if ( !Hooks::run( 'UserArrayFromResult', [ &$userArray, $res ] ) ) {
  33                         return null;
  34                 }
  35                 return $userArray ?? new UserArrayFromResult( $res );
  36         }
  37
  38         /**
  39          * @param array $ids
  40          * @return UserArrayFromResult|ArrayIterator
  41          */
  42         static function newFromIDs( $ids ) {
  43                 $ids = array_map( 'intval', (array)$ids ); // paranoia
  44                 if ( !$ids ) {
  45                         // Database::select() doesn't like empty arrays
  46                         return new ArrayIterator( [] );
  47                 }
  48                 $dbr = wfGetDB( DB_REPLICA );
  49                 $userQuery = User::getQueryInfo();
  50                 $res = $dbr->select(
  51                         $userQuery['tables'],
  52                         $userQuery['fields'],
  53                         [ 'user_id' => array_unique( $ids ) ],
  54                         __METHOD__,
  55                         [],
  56                         $userQuery['joins']
  57                 );
  58                 return self::newFromResult( $res );
  59         }
  60
  61         /**
  62          * @since 1.25
  63          * @param array $names
  64          * @return UserArrayFromResult|ArrayIterator
  65          */
  66         static function newFromNames( $names ) {
  67                 $names = array_map( 'strval', (array)$names ); // paranoia
  68                 if ( !$names ) {
  69                         // Database::select() doesn't like empty arrays
  70                         return new ArrayIterator( [] );
  71                 }
  72                 $dbr = wfGetDB( DB_REPLICA );
  73                 $userQuery = User::getQueryInfo();
  74                 $res = $dbr->select(
  75                         $userQuery['tables'],
  76                         $userQuery['fields'],
  77                         [ 'user_name' => array_unique( $names ) ],
  78                         __METHOD__,
  79                         [],
  80                         $userQuery['joins']
  81                 );
  82                 return self::newFromResult( $res );
  83         }
  84 }