SpecialRedirect.php

Go to the documentation of this file.

<?php
class SpecialRedirect extends FormSpecialPage {
 
    protected $mType;
 
    protected $mValue;
 
    function __construct() {
        parent::__construct( 'Redirect' );
        $this->mType = null;
        $this->mValue = null;
    }
 
    function setParameter( $subpage ) {
        // parse $subpage to pull out the parts
        $parts = explode( '/', $subpage, 2 );
        $this->mType = count( $parts ) > 0 ? $parts[0] : null;
        $this->mValue = count( $parts ) > 1 ? $parts[1] : null;
    }
 
    function dispatchUser() {
        if ( !ctype_digit( $this->mValue ) ) {
            return null;
        }
        $user = User::newFromId( (int)$this->mValue );
        $username = $user->getName(); // load User as side-effect
        if ( $user->isAnon() ) {
            return null;
        }
        $userpage = Title::makeTitle( NS_USER, $username );
 
        return $userpage->getFullURL( '', false, PROTO_CURRENT );
    }
 
    function dispatchFile() {
        $title = Title::makeTitleSafe( NS_FILE, $this->mValue );
 
        if ( !$title instanceof Title ) {
            return null;
        }
        $file = wfFindFile( $title );
 
        if ( !$file || !$file->exists() ) {
            return null;
        }
        // Default behavior: Use the direct link to the file.
        $url = $file->getURL();
        $request = $this->getRequest();
        $width = $request->getInt( 'width', -1 );
        $height = $request->getInt( 'height', -1 );
 
        // If a width is requested...
        if ( $width != -1 ) {
            $mto = $file->transform( array( 'width' => $width, 'height' => $height ) );
            // ... and we can
            if ( $mto && !$mto->isError() ) {
                // ... change the URL to point to a thumbnail.
                $url = $mto->getURL();
            }
        }
 
        return $url;
    }
 
    function dispatchRevision() {
        $oldid = $this->mValue;
        if ( !ctype_digit( $oldid ) ) {
            return null;
        }
        $oldid = (int)$oldid;
        if ( $oldid === 0 ) {
            return null;
        }
 
        return wfAppendQuery( wfScript( 'index' ), array(
            'oldid' => $oldid
        ) );
    }
 
    function dispatchPage() {
        $curid = $this->mValue;
        if ( !ctype_digit( $curid ) ) {
            return null;
        }
        $curid = (int)$curid;
        if ( $curid === 0 ) {
            return null;
        }
 
        return wfAppendQuery( wfScript( 'index' ), array(
            'curid' => $curid
        ) );
    }
 
    function dispatch() {
        // the various namespaces supported by Special:Redirect
        switch ( $this->mType ) {
            case 'user':
                $url = $this->dispatchUser();
                break;
            case 'file':
                $url = $this->dispatchFile();
                break;
            case 'revision':
                $url = $this->dispatchRevision();
                break;
            case 'page':
                $url = $this->dispatchPage();
                break;
            default:
                $this->getOutput()->setStatusCode( 404 );
                $url = null;
                break;
        }
        if ( $url ) {
            $this->getOutput()->redirect( $url );
 
            return true;
        }
        if ( !is_null( $this->mValue ) ) {
            $this->getOutput()->setStatusCode( 404 );
            // Message: redirect-not-exists
            $msg = $this->getMessagePrefix() . '-not-exists';
 
            return Status::newFatal( $msg );
        }
 
        return false;
    }
 
    protected function getFormFields() {
        $mp = $this->getMessagePrefix();
        $ns = array(
            // subpage => message
            // Messages: redirect-user, redirect-page, redirect-revision,
            // redirect-file
            'user' => $mp . '-user',
            'page' => $mp . '-page',
            'revision' => $mp . '-revision',
            'file' => $mp . '-file',
        );
        $a = array();
        $a['type'] = array(
            'type' => 'select',
            'label-message' => $mp . '-lookup', // Message: redirect-lookup
            'options' => array(),
            'default' => current( array_keys( $ns ) ),
        );
        foreach ( $ns as $n => $m ) {
            $m = $this->msg( $m )->text();
            $a['type']['options'][$m] = $n;
        }
        $a['value'] = array(
            'type' => 'text',
            'label-message' => $mp . '-value' // Message: redirect-value
        );
        // set the defaults according to the parsed subpage path
        if ( !empty( $this->mType ) ) {
            $a['type']['default'] = $this->mType;
        }
        if ( !empty( $this->mValue ) ) {
            $a['value']['default'] = $this->mValue;
        }
 
        return $a;
    }
 
    public function onSubmit( array $data ) {
        if ( !empty( $data['type'] ) && !empty( $data['value'] ) ) {
            $this->setParameter( $data['type'] . '/' . $data['value'] );
        }
 
        /* if this returns false, will show the form */
        return $this->dispatch();
    }
 
    public function onSuccess() {
        /* do nothing, we redirect in $this->dispatch if successful. */
    }
 
    protected function alterForm( HTMLForm $form ) {
        /* display summary at top of page */
        $this->outputHeader();
        // tweak label on submit button
        // Message: redirect-submit
        $form->setSubmitTextMsg( $this->getMessagePrefix() . '-submit' );
        /* submit form every time */
        $form->setMethod( 'get' );
    }
 
    protected function getGroupName() {
        return 'redirects';
    }
}