Status.php

Go to the documentation of this file.

<?php
class Status {
        var $ok = true;
        var $value;

        public $successCount = 0, $failCount = 0;
        public $success = array();

        /*semi-private*/ var $errors = array();
        /*semi-private*/ var $cleanCallback = false;

        static function newFatal( $message /*, parameters...*/ ) {
                $params = func_get_args();
                $result = new self;
                call_user_func_array( array( &$result, 'error' ), $params );
                $result->ok = false;
                return $result;
        }

        static function newGood( $value = null ) {
                $result = new self;
                $result->value = $value;
                return $result;
        }

        function setResult( $ok, $value = null ) {
                $this->ok = $ok;
                $this->value = $value;
        }

        function isGood() {
                return $this->ok && !$this->errors;
        }

        function isOK() {
                return $this->ok;
        }

        function warning( $message /*, parameters... */ ) {
                $params = array_slice( func_get_args(), 1 );
                $this->errors[] = array(
                        'type' => 'warning',
                        'message' => $message,
                        'params' => $params );
        }

        function error( $message /*, parameters... */ ) {
                $params = array_slice( func_get_args(), 1 );
                $this->errors[] = array(
                        'type' => 'error',
                        'message' => $message,
                        'params' => $params );
        }

        function fatal( $message /*, parameters... */ ) {
                $params = array_slice( func_get_args(), 1 );
                $this->errors[] = array(
                        'type' => 'error',
                        'message' => $message,
                        'params' => $params );
                $this->ok = false;
        }

        function __wakeup() {
                $this->cleanCallback = false;
        }

        protected function cleanParams( $params ) {
                if ( !$this->cleanCallback ) {
                        return $params;
                }
                $cleanParams = array();
                foreach ( $params as $i => $param ) {
                        $cleanParams[$i] = call_user_func( $this->cleanCallback, $param );
                }
                return $cleanParams;
        }

        function getWikiText( $shortContext = false, $longContext = false ) {
                if ( count( $this->errors ) == 0 ) {
                        if ( $this->ok ) {
                                $this->fatal( 'internalerror_info',
                                        __METHOD__ . " called for a good result, this is incorrect\n" );
                        } else {
                                $this->fatal( 'internalerror_info',
                                        __METHOD__ . ": Invalid result object: no error text but not OK\n" );
                        }
                }
                if ( count( $this->errors ) == 1 ) {
                        $s = $this->getErrorMessage( $this->errors[0] );
                        if ( $shortContext ) {
                                $s = wfMessage( $shortContext, $s )->plain();
                        } elseif ( $longContext ) {
                                $s = wfMessage( $longContext, "* $s\n" )->plain();
                        }
                } else {
                        $s = '* '. implode( "\n* ",
                                $this->getErrorMessageArray( $this->errors ) ) . "\n";
                        if ( $longContext ) {
                                $s = wfMessage( $longContext, $s )->plain();
                        } elseif ( $shortContext ) {
                                $s = wfMessage( $shortContext, "\n$s\n" )->plain();
                        }
                }
                return $s;
        }

        protected function getErrorMessage( $error ) {
                if ( is_array( $error ) ) {
                        if( isset( $error['message'] ) && $error['message'] instanceof Message ) {
                                $msg = $error['message'];
                        } elseif ( isset( $error['message'] ) && isset( $error['params'] ) ) {
                                $msg = wfMessage( $error['message'],
                                        array_map( 'wfEscapeWikiText', $this->cleanParams( $error['params'] ) ) );
                        } else {
                                $msgName = array_shift( $error );
                                $msg = wfMessage( $msgName,
                                        array_map( 'wfEscapeWikiText', $this->cleanParams( $error ) ) );
                        }
                } else {
                        $msg = wfMessage( $error );
                }
                return $msg->plain();
        }

        public function getHTML( $shortContext = false, $longContext = false ) {
                $text = $this->getWikiText( $shortContext, $longContext );
                return MessageCache::singleton()->transform( $text, true );
        }

        protected function getErrorMessageArray( $errors ) {
                return array_map( array( $this, 'getErrorMessage' ), $errors );
        }

        function merge( $other, $overwriteValue = false ) {
                $this->errors = array_merge( $this->errors, $other->errors );
                $this->ok = $this->ok && $other->ok;
                if ( $overwriteValue ) {
                        $this->value = $other->value;
                }
                $this->successCount += $other->successCount;
                $this->failCount += $other->failCount;
        }

        function getErrorsArray() {
                return $this->getStatusArray( "error" );
        }

        function getWarningsArray() {
                return $this->getStatusArray( "warning" );
        }

        protected function getStatusArray( $type ) {
                $result = array();
                foreach ( $this->errors as $error ) {
                        if ( $error['type'] === $type ) {
                                if( $error['message'] instanceof Message ) {
                                        $result[] = $error['message'];
                                } elseif( $error['params'] ) {
                                        $result[] = array_merge( array( $error['message'] ), $error['params'] );
                                } else {
                                        $result[] = array( $error['message'] );
                                }
                        }
                }
                return $result;
        }

        public function getErrorsByType( $type ) {
                $result = array();
                foreach ( $this->errors as $error ) {
                        if ( $error['type'] === $type ) {
                                $result[] = $error;
                        }
                }
                return $result;
        }

        function hasMessage( $msg ) {
                foreach ( $this->errors as $error ) {
                        if ( $error['message'] === $msg ) {
                                return true;
                        }
                }
                return false;
        }

        function replaceMessage( $source, $dest ) {
                $replaced = false;
                foreach ( $this->errors as $index => $error ) {
                        if ( $error['message'] === $source ) {
                                $this->errors[$index]['message'] = $dest;
                                $replaced = true;
                        }
                }
                return $replaced;
        }

        public function getMessage() {
                return $this->getWikiText();
        }

        public function getValue() {
                return $this->value;
        }
}