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\ResultWrapper;
  24
  25 abstract class UserArray implements Iterator {
  26         /**
  27          * @param ResultWrapper $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                 if ( $userArray === null ) {
  36                         $userArray = self::newFromResult_internal( $res );
  37                 }
  38                 return $userArray;
  39         }
  40
  41         /**
  42          * @param array $ids
  43          * @return UserArrayFromResult|ArrayIterator
  44          */
  45         static function newFromIDs( $ids ) {
  46                 $ids = array_map( 'intval', (array)$ids ); // paranoia
  47                 if ( !$ids ) {
  48                         // Database::select() doesn't like empty arrays
  49                         return new ArrayIterator( [] );
  50                 }
  51                 $dbr = wfGetDB( DB_REPLICA );
  52                 $res = $dbr->select(
  53                         'user',
  54                         User::selectFields(),
  55                         [ 'user_id' => array_unique( $ids ) ],
  56                         __METHOD__
  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                 $res = $dbr->select(
  74                         'user',
  75                         User::selectFields(),
  76                         [ 'user_name' => array_unique( $names ) ],
  77                         __METHOD__
  78                 );
  79                 return self::newFromResult( $res );
  80         }
  81
  82         /**
  83          * @param ResultWrapper $res
  84          * @return UserArrayFromResult
  85          */
  86         protected static function newFromResult_internal( $res ) {
  87                 return new UserArrayFromResult( $res );
  88         }
  89 }