SpecialMediaStatistics.php

Go to the documentation of this file.

<?php
class MediaStatisticsPage extends QueryPage {
    protected $totalCount = 0, $totalBytes = 0;

    function __construct( $name = 'MediaStatistics' ) {
        parent::__construct( $name );
        // Generally speaking there is only a small number of file types,
        // so just show all of them.
        $this->limit = 5000;
        $this->shownavigation = false;
    }

    function isExpensive() {
        return true;
    }

    public function getQueryInfo() {
        $dbr = wfGetDB( DB_SLAVE );
        $fakeTitle = $dbr->buildConcat( array(
            'img_media_type',
            $dbr->addQuotes( ';' ),
            'img_major_mime',
            $dbr->addQuotes( '/' ),
            'img_minor_mime',
            $dbr->addQuotes( ';' ),
            'COUNT(*)',
            $dbr->addQuotes( ';' ),
            'SUM( img_size )'
        ) );
        return array(
            'tables' => array( 'image' ),
            'fields' => array(
                'title' => $fakeTitle,
                'namespace' => NS_MEDIA, /* needs to be something */
                'value' => '1'
            ),
            'options' => array(
                'GROUP BY' => array(
                    'img_media_type',
                    'img_major_mime',
                    'img_minor_mime',
                )
            )
        );
    }

    function getOrderFields() {
        return array( 'img_media_type', 'count(*)', 'img_major_mime', 'img_minor_mime' );
    }

    protected function outputResults( $out, $skin, $dbr, $res, $num, $offset ) {
        $prevMediaType = null;
        foreach ( $res as $row ) {
            list( $mediaType, $mime, $totalCount, $totalBytes ) = $this->splitFakeTitle( $row->title );
            if ( $prevMediaType !== $mediaType ) {
                if ( $prevMediaType !== null ) {
                    // We're not at beginning, so we have to
                    // close the previous table.
                    $this->outputTableEnd();
                }
                $this->outputMediaType( $mediaType );
                $this->outputTableStart( $mediaType );
                $prevMediaType = $mediaType;
            }
            $this->outputTableRow( $mime, intval( $totalCount ), intval( $totalBytes ) );
        }
        if ( $prevMediaType !== null ) {
            $this->outputTableEnd();
        }
    }

    protected function outputTableEnd() {
        $this->getOutput()->addHtml( Html::closeElement( 'table' ) );
    }

    protected function outputTableRow( $mime, $count, $bytes ) {
        $mimeSearch = SpecialPage::getTitleFor( 'MIMEsearch', $mime );
        $row = Html::rawElement(
            'td',
            array(),
            Linker::link( $mimeSearch, htmlspecialchars( $mime ) )
        );
        $row .= Html::element(
            'td',
            array(),
            $this->getExtensionList( $mime )
        );
        $row .= Html::rawElement(
            'td',
            array(),
            $this->msg( 'mediastatistics-nfiles' )
                ->numParams( $count )
                ->numParams( $this->makePercentPretty( $count / $this->totalCount ) )
                ->parse()
        );
        $row .= Html::rawElement(
            'td',
            // Make sure js sorts it in numeric order
            array( 'data-sort-value' =>  $bytes ),
            $this->msg( 'mediastatistics-nbytes' )
                ->numParams( $bytes )
                ->sizeParams( $bytes )
                ->numParams( $this->makePercentPretty( $bytes / $this->totalBytes ) )
                ->parse()
        );

        $this->getOutput()->addHTML( Html::rawElement( 'tr', array(), $row ) );
    }

    protected function makePercentPretty( $decimal ) {
        $decimal *= 100;
        // Always show three useful digits
        if ( $decimal == 0 ) {
            return '0';
        }
        $percent = sprintf( "%." . max( 0, 2 - floor( log10( $decimal ) ) ) . "f", $decimal );
        // Then remove any trailing 0's
        return preg_replace( '/\.?0*$/', '', $percent );
    }

    private function getExtensionList( $mime ) {
        $exts = MimeMagic::singleton()->getExtensionsForType( $mime );
        if ( $exts === null ) {
            return '';
        }
        $extArray = explode( ' ', $exts );
        $extArray = array_unique( $extArray );
        foreach ( $extArray as &$ext ) {
            $ext = '.' . $ext;
        }

        return $this->getLanguage()->commaList( $extArray );
    }

    protected function outputTableStart( $mediaType ) {
        $this->getOutput()->addHTML(
            Html::openElement(
                'table',
                array( 'class' => array(
                    'mw-mediastats-table',
                    'mw-mediastats-table-' . strtolower( $mediaType ),
                    'sortable',
                    'wikitable'
                ))
            )
        );
        $this->getOutput()->addHTML( $this->getTableHeaderRow() );
    }

    protected function getTableHeaderRow() {
        $headers = array( 'mimetype', 'extensions', 'count', 'totalbytes' );
        $ths = '';
        foreach ( $headers as $header ) {
            $ths .= Html::rawElement(
                'th',
                array(),
                // for grep:
                // mediastatistics-table-mimetype, mediastatistics-table-extensions
                // tatistics-table-count, mediastatistics-table-totalbytes
                $this->msg( 'mediastatistics-table-' . $header )->parse()
            );
        }
        return Html::rawElement( 'tr', array(), $ths );
    }

    protected function outputMediaType( $mediaType ) {
        $this->getOutput()->addHTML(
            Html::element(
                'h2',
                array( 'class' => array(
                    'mw-mediastats-mediatype',
                    'mw-mediastats-mediatype-' . strtolower( $mediaType )
                )),
                // for grep
                // mediastatistics-header-unknown, mediastatistics-header-bitmap,
                // mediastatistics-header-drawing, mediastatistics-header-audio,
                // mediastatistics-header-video, mediastatistics-header-multimedia,
                // mediastatistics-header-office, mediastatistics-header-text,
                // mediastatistics-header-executable, mediastatistics-header-archive,
                $this->msg( 'mediastatistics-header-' . strtolower( $mediaType ) )->text()
            )
        );
    }

    private function splitFakeTitle( $fakeTitle ) {
        return explode( ';', $fakeTitle, 4 );
    }

    protected function getGroupName() {
        return 'media';
    }

    public function formatResult( $skin, $result ) {
        throw new MWException( "unimplemented" );
    }

    public function preprocessResults( $dbr, $res ) {
        $this->totalCount = $this->totalBytes = 0;
        foreach ( $res as $row ) {
            list( , , $count, $bytes ) = $this->splitFakeTitle( $row->title );
            $this->totalCount += $count;
            $this->totalBytes += $bytes;
        }
        $res->seek( 0 );
    }
}