includes/Status.php

   1 <?php
   2 /**
   3  * Generic operation result.
   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 /**
  24  * Generic operation result class
  25  * Has warning/error list, boolean status and arbitrary value
  26  *
  27  * "Good" means the operation was completed with no warnings or errors.
  28  *
  29  * "OK" means the operation was partially or wholly completed.
  30  *
  31  * An operation which is not OK should have errors so that the user can be
  32  * informed as to what went wrong. Calling the fatal() function sets an error
  33  * message and simultaneously switches off the OK flag.
  34  */
  35 class Status {
  36         var $ok = true;
  37         var $value;
  38
  39         /** Counters for batch operations */
  40         public $successCount = 0, $failCount = 0;
  41         /** Array to indicate which items of the batch operations were successful */
  42         public $success = array();
  43
  44         /*semi-private*/ var $errors = array();
  45         /*semi-private*/ var $cleanCallback = false;
  46
  47         /**
  48          * Factory function for fatal errors
  49          *
  50          * @param string|Message $message message name or object
  51          * @return Status
  52          */
  53         static function newFatal( $message /*, parameters...*/ ) {
  54                 $params = func_get_args();
  55                 $result = new self;
  56                 call_user_func_array( array( &$result, 'error' ), $params );
  57                 $result->ok = false;
  58                 return $result;
  59         }
  60
  61         /**
  62          * Factory function for good results
  63          *
  64          * @param $value Mixed
  65          * @return Status
  66          */
  67         static function newGood( $value = null ) {
  68                 $result = new self;
  69                 $result->value = $value;
  70                 return $result;
  71         }
  72
  73         /**
  74          * Change operation result
  75          *
  76          * @param $ok Boolean: whether the operation completed
  77          * @param $value Mixed
  78          */
  79         function setResult( $ok, $value = null ) {
  80                 $this->ok = $ok;
  81                 $this->value = $value;
  82         }
  83
  84         /**
  85          * Returns whether the operation completed and didn't have any error or
  86          * warnings
  87          *
  88          * @return Boolean
  89          */
  90         function isGood() {
  91                 return $this->ok && !$this->errors;
  92         }
  93
  94         /**
  95          * Returns whether the operation completed
  96          *
  97          * @return Boolean
  98          */
  99         function isOK() {
 100                 return $this->ok;
 101         }
 102
 103         /**
 104          * Add a new warning
 105          *
 106          * @param string|Message $message message name or object
 107          */
 108         function warning( $message /*, parameters... */ ) {
 109                 $params = array_slice( func_get_args(), 1 );
 110                 $this->errors[] = array(
 111                         'type' => 'warning',
 112                         'message' => $message,
 113                         'params' => $params );
 114         }
 115
 116         /**
 117          * Add an error, do not set fatal flag
 118          * This can be used for non-fatal errors
 119          *
 120          * @param string|Message $message message name or object
 121          */
 122         function error( $message /*, parameters... */ ) {
 123                 $params = array_slice( func_get_args(), 1 );
 124                 $this->errors[] = array(
 125                         'type' => 'error',
 126                         'message' => $message,
 127                         'params' => $params );
 128         }
 129
 130         /**
 131          * Add an error and set OK to false, indicating that the operation
 132          * as a whole was fatal
 133          *
 134          * @param string|Message $message message name or object
 135          */
 136         function fatal( $message /*, parameters... */ ) {
 137                 $params = array_slice( func_get_args(), 1 );
 138                 $this->errors[] = array(
 139                         'type' => 'error',
 140                         'message' => $message,
 141                         'params' => $params );
 142                 $this->ok = false;
 143         }
 144
 145         /**
 146          * Sanitize the callback parameter on wakeup, to avoid arbitrary execution.
 147          */
 148         function __wakeup() {
 149                 $this->cleanCallback = false;
 150         }
 151
 152         /**
 153          * @param $params array
 154          * @return array
 155          */
 156         protected function cleanParams( $params ) {
 157                 if ( !$this->cleanCallback ) {
 158                         return $params;
 159                 }
 160                 $cleanParams = array();
 161                 foreach ( $params as $i => $param ) {
 162                         $cleanParams[$i] = call_user_func( $this->cleanCallback, $param );
 163                 }
 164                 return $cleanParams;
 165         }
 166
 167         /**
 168          * Get the error list as a wikitext formatted list
 169          *
 170          * @param string $shortContext a short enclosing context message name, to
 171          *        be used when there is a single error
 172          * @param string $longContext a long enclosing context message name, for a list
 173          * @return String
 174          */
 175         function getWikiText( $shortContext = false, $longContext = false ) {
 176                 if ( count( $this->errors ) == 0 ) {
 177                         if ( $this->ok ) {
 178                                 $this->fatal( 'internalerror_info',
 179                                         __METHOD__ . " called for a good result, this is incorrect\n" );
 180                         } else {
 181                                 $this->fatal( 'internalerror_info',
 182                                         __METHOD__ . ": Invalid result object: no error text but not OK\n" );
 183                         }
 184                 }
 185                 if ( count( $this->errors ) == 1 ) {
 186                         $s = $this->getErrorMessage( $this->errors[0] );
 187                         if ( $shortContext ) {
 188                                 $s = wfMessage( $shortContext, $s )->plain();
 189                         } elseif ( $longContext ) {
 190                                 $s = wfMessage( $longContext, "* $s\n" )->plain();
 191                         }
 192                 } else {
 193                         $s = '* ' . implode( "\n* ",
 194                                 $this->getErrorMessageArray( $this->errors ) ) . "\n";
 195                         if ( $longContext ) {
 196                                 $s = wfMessage( $longContext, $s )->plain();
 197                         } elseif ( $shortContext ) {
 198                                 $s = wfMessage( $shortContext, "\n$s\n" )->plain();
 199                         }
 200                 }
 201                 return $s;
 202         }
 203
 204         /**
 205          * Return the message for a single error.
 206          * @param $error Mixed With an array & two values keyed by
 207          * 'message' and 'params', use those keys-value pairs.
 208          * Otherwise, if its an array, just use the first value as the
 209          * message and the remaining items as the params.
 210          *
 211          * @return String
 212          */
 213         protected function getErrorMessage( $error ) {
 214                 if ( is_array( $error ) ) {
 215                         if( isset( $error['message'] ) && $error['message'] instanceof Message ) {
 216                                 $msg = $error['message'];
 217                         } elseif ( isset( $error['message'] ) && isset( $error['params'] ) ) {
 218                                 $msg = wfMessage( $error['message'],
 219                                         array_map( 'wfEscapeWikiText', $this->cleanParams( $error['params'] ) ) );
 220                         } else {
 221                                 $msgName = array_shift( $error );
 222                                 $msg = wfMessage( $msgName,
 223                                         array_map( 'wfEscapeWikiText', $this->cleanParams( $error ) ) );
 224                         }
 225                 } else {
 226                         $msg = wfMessage( $error );
 227                 }
 228                 return $msg->plain();
 229         }
 230
 231         /**
 232          * Get the error message as HTML. This is done by parsing the wikitext error
 233          * message.
 234          *
 235          * @note: this does not perform a full wikitext to HTML conversion, it merely applies
 236          *        a message transformation.
 237          * @todo: figure out whether that is actually The Right Thing.
 238          */
 239         public function getHTML( $shortContext = false, $longContext = false ) {
 240                 $text = $this->getWikiText( $shortContext, $longContext );
 241                 return MessageCache::singleton()->transform( $text, true );
 242         }
 243
 244         /**
 245          * Return an array with the wikitext for each item in the array.
 246          * @param $errors Array
 247          * @return Array
 248          */
 249         protected function getErrorMessageArray( $errors ) {
 250                 return array_map( array( $this, 'getErrorMessage' ), $errors );
 251         }
 252
 253         /**
 254          * Merge another status object into this one
 255          *
 256          * @param $other Status Other Status object
 257          * @param $overwriteValue Boolean: whether to override the "value" member
 258          */
 259         function merge( $other, $overwriteValue = false ) {
 260                 $this->errors = array_merge( $this->errors, $other->errors );
 261                 $this->ok = $this->ok && $other->ok;
 262                 if ( $overwriteValue ) {
 263                         $this->value = $other->value;
 264                 }
 265                 $this->successCount += $other->successCount;
 266                 $this->failCount += $other->failCount;
 267         }
 268
 269         /**
 270          * Get the list of errors (but not warnings)
 271          *
 272          * @return Array
 273          */
 274         function getErrorsArray() {
 275                 return $this->getStatusArray( "error" );
 276         }
 277
 278         /**
 279          * Get the list of warnings (but not errors)
 280          *
 281          * @return Array
 282          */
 283         function getWarningsArray() {
 284                 return $this->getStatusArray( "warning" );
 285         }
 286
 287         /**
 288          * Returns a list of status messages of the given type
 289          * @param $type String
 290          *
 291          * @return Array
 292          */
 293         protected function getStatusArray( $type ) {
 294                 $result = array();
 295                 foreach ( $this->errors as $error ) {
 296                         if ( $error['type'] === $type ) {
 297                                 if( $error['message'] instanceof Message ) {
 298                                         $result[] = $error['message'];
 299                                 } elseif( $error['params'] ) {
 300                                         $result[] = array_merge( array( $error['message'] ), $error['params'] );
 301                                 } else {
 302                                         $result[] = array( $error['message'] );
 303                                 }
 304                         }
 305                 }
 306                 return $result;
 307         }
 308
 309         /**
 310          * Returns a list of status messages of the given type, with message and
 311          * params left untouched, like a sane version of getStatusArray
 312          *
 313          * @param $type String
 314          *
 315          * @return Array
 316          */
 317         public function getErrorsByType( $type ) {
 318                 $result = array();
 319                 foreach ( $this->errors as $error ) {
 320                         if ( $error['type'] === $type ) {
 321                                 $result[] = $error;
 322                         }
 323                 }
 324                 return $result;
 325         }
 326
 327         /**
 328          * Returns true if the specified message is present as a warning or error
 329          *
 330          * Note, due to the lack of tools for comparing Message objects, this
 331          * function will not work when using a Message object as a parameter.
 332          *
 333          * @param string $msg message name
 334          * @return Boolean
 335          */
 336         function hasMessage( $msg ) {
 337                 foreach ( $this->errors as $error ) {
 338                         if ( $error['message'] === $msg ) {
 339                                 return true;
 340                         }
 341                 }
 342                 return false;
 343         }
 344
 345         /**
 346          * If the specified source message exists, replace it with the specified
 347          * destination message, but keep the same parameters as in the original error.
 348          *
 349          * Note, due to the lack of tools for comparing Message objects, this
 350          * function will not work when using a Message object as the search parameter.
 351          *
 352          * @param $source Message|String: Message key or object to search for
 353          * @param $dest Message|String: Replacement message key or object
 354          * @return bool Return true if the replacement was done, false otherwise.
 355          */
 356         function replaceMessage( $source, $dest ) {
 357                 $replaced = false;
 358                 foreach ( $this->errors as $index => $error ) {
 359                         if ( $error['message'] === $source ) {
 360                                 $this->errors[$index]['message'] = $dest;
 361                                 $replaced = true;
 362                         }
 363                 }
 364                 return $replaced;
 365         }
 366
 367         /**
 368          * Backward compatibility function for WikiError -> Status migration
 369          *
 370          * @return String
 371          */
 372         public function getMessage() {
 373                 return $this->getWikiText();
 374         }
 375
 376         /**
 377          * @return mixed
 378          */
 379         public function getValue() {
 380                 return $this->value;
 381         }
 382 }