SvgHandler.php

Go to the documentation of this file.

<?php
use MediaWiki\MainConfigNames;
use MediaWiki\MediaWikiServices;
use MediaWiki\Shell\Shell;
use Wikimedia\AtEase\AtEase;
use Wikimedia\RequestTimeout\TimeoutException;
use Wikimedia\ScopedCallback;
 
class SvgHandler extends ImageHandler {
    public const SVG_METADATA_VERSION = 2;
 
    private static $metaConversion = [
        'originalwidth' => 'ImageWidth',
        'originalheight' => 'ImageLength',
        'description' => 'ImageDescription',
        'title' => 'ObjectName',
    ];
 
    public function isEnabled() {
        $svgConverters = MediaWikiServices::getInstance()->getMainConfig()->get( MainConfigNames::SVGConverters );
        $svgConverter = MediaWikiServices::getInstance()->getMainConfig()->get( MainConfigNames::SVGConverter );
        if ( !isset( $svgConverters[$svgConverter] ) ) {
            wfDebug( "\$wgSVGConverter is invalid, disabling SVG rendering." );
 
            return false;
        } else {
            return true;
        }
    }
 
    public function mustRender( $file ) {
        return true;
    }
 
    public function isVectorized( $file ) {
        return true;
    }
 
    public function isAnimatedImage( $file ) {
        # @todo Detect animated SVGs
        $metadata = $this->validateMetadata( $file->getMetadataArray() );
        if ( isset( $metadata['animated'] ) ) {
            return $metadata['animated'];
        }
 
        return false;
    }
 
    public function getAvailableLanguages( File $file ) {
        $langList = [];
        $metadata = $this->validateMetadata( $file->getMetadataArray() );
        if ( isset( $metadata['translations'] ) ) {
            foreach ( $metadata['translations'] as $lang => $langType ) {
                if ( $langType === SVGReader::LANG_FULL_MATCH ) {
                    $langList[] = strtolower( $lang );
                }
            }
        }
        return array_unique( $langList );
    }
 
    public function getMatchedLanguage( $userPreferredLanguage, array $svgLanguages ) {
        foreach ( $svgLanguages as $svgLang ) {
            if ( strcasecmp( $svgLang, $userPreferredLanguage ) === 0 ) {
                return $svgLang;
            }
            $trimmedSvgLang = $svgLang;
            while ( strpos( $trimmedSvgLang, '-' ) !== false ) {
                $trimmedSvgLang = substr( $trimmedSvgLang, 0, strrpos( $trimmedSvgLang, '-' ) );
                if ( strcasecmp( $trimmedSvgLang, $userPreferredLanguage ) === 0 ) {
                    return $svgLang;
                }
            }
        }
        return null;
    }
 
    protected function getLanguageFromParams( array $params ) {
        return $params['lang'] ?? $params['targetlang'] ?? 'en';
    }
 
    public function getDefaultRenderLanguage( File $file ) {
        return 'en';
    }
 
    public function canAnimateThumbnail( $file ) {
        return false;
    }
 
    public function normaliseParams( $image, &$params ) {
        if ( parent::normaliseParams( $image, $params ) ) {
            $params = $this->normaliseParamsInternal( $image, $params );
            return true;
        }
 
        return false;
    }
 
    protected function normaliseParamsInternal( $image, $params ) {
        $svgMaxSize = MediaWikiServices::getInstance()->getMainConfig()->get( MainConfigNames::SVGMaxSize );
 
        # Don't make an image bigger than wgMaxSVGSize on the smaller side
        if ( $params['physicalWidth'] <= $params['physicalHeight'] ) {
            if ( $params['physicalWidth'] > $svgMaxSize ) {
                $srcWidth = $image->getWidth( $params['page'] );
                $srcHeight = $image->getHeight( $params['page'] );
                $params['physicalWidth'] = $svgMaxSize;
                $params['physicalHeight'] = File::scaleHeight( $srcWidth, $srcHeight, $svgMaxSize );
            }
        } elseif ( $params['physicalHeight'] > $svgMaxSize ) {
            $srcWidth = $image->getWidth( $params['page'] );
            $srcHeight = $image->getHeight( $params['page'] );
            $params['physicalWidth'] = File::scaleHeight( $srcHeight, $srcWidth, $svgMaxSize );
            $params['physicalHeight'] = $svgMaxSize;
        }
        // To prevent the proliferation of thumbnails in languages not present in SVGs, unless
        // explicitly forced by user.
        if ( isset( $params['targetlang'] ) && !$image->getMatchedLanguage( $params['targetlang'] ) ) {
            unset( $params['targetlang'] );
        }
 
        return $params;
    }
 
    public function doTransform( $image, $dstPath, $dstUrl, $params, $flags = 0 ) {
        if ( !$this->normaliseParams( $image, $params ) ) {
            return new TransformParameterError( $params );
        }
        $clientWidth = $params['width'];
        $clientHeight = $params['height'];
        $physicalWidth = $params['physicalWidth'];
        $physicalHeight = $params['physicalHeight'];
        $lang = $this->getLanguageFromParams( $params );
 
        if ( $flags & self::TRANSFORM_LATER ) {
            return new ThumbnailImage( $image, $dstUrl, $dstPath, $params );
        }
 
        $metadata = $this->validateMetadata( $image->getMetadataArray() );
        if ( isset( $metadata['error'] ) ) {
            $err = wfMessage( 'svg-long-error', $metadata['error']['message'] );
 
            return new MediaTransformError( 'thumbnail_error', $clientWidth, $clientHeight, $err );
        }
 
        if ( !wfMkdirParents( dirname( $dstPath ), null, __METHOD__ ) ) {
            return new MediaTransformError( 'thumbnail_error', $clientWidth, $clientHeight,
                wfMessage( 'thumbnail_dest_directory' ) );
        }
 
        $srcPath = $image->getLocalRefPath();
        if ( $srcPath === false ) { // Failed to get local copy
            wfDebugLog( 'thumbnail',
                sprintf( 'Thumbnail failed on %s: could not get local copy of "%s"',
                    wfHostname(), $image->getName() ) );
 
            return new MediaTransformError( 'thumbnail_error',
                $params['width'], $params['height'],
                wfMessage( 'filemissing' )
            );
        }
 
        // Make a temp dir with a symlink to the local copy in it.
        // This plays well with rsvg-convert policy for external entities.
        // https://git.gnome.org/browse/librsvg/commit/?id=f01aded72c38f0e18bc7ff67dee800e380251c8e
        $tmpDir = wfTempDir() . '/svg_' . wfRandomString( 24 );
        $lnPath = "$tmpDir/" . basename( $srcPath );
        $ok = mkdir( $tmpDir, 0771 );
        if ( !$ok ) {
            wfDebugLog( 'thumbnail',
                sprintf( 'Thumbnail failed on %s: could not create temporary directory %s',
                    wfHostname(), $tmpDir ) );
            return new MediaTransformError( 'thumbnail_error',
                $params['width'], $params['height'],
                wfMessage( 'thumbnail-temp-create' )->text()
            );
        }
        $ok = symlink( $srcPath, $lnPath );
        $cleaner = new ScopedCallback( static function () use ( $tmpDir, $lnPath ) {
            AtEase::suppressWarnings();
            unlink( $lnPath );
            rmdir( $tmpDir );
            AtEase::restoreWarnings();
        } );
        if ( !$ok ) {
            // Fallback because symlink often fails on Windows
            $ok = copy( $srcPath, $lnPath );
        }
        if ( !$ok ) {
            wfDebugLog( 'thumbnail',
                sprintf( 'Thumbnail failed on %s: could not link %s to %s',
                    wfHostname(), $lnPath, $srcPath ) );
            return new MediaTransformError( 'thumbnail_error',
                $params['width'], $params['height'],
                wfMessage( 'thumbnail-temp-create' )
            );
        }
 
        $status = $this->rasterize( $lnPath, $dstPath, $physicalWidth, $physicalHeight, $lang );
        if ( $status === true ) {
            return new ThumbnailImage( $image, $dstUrl, $dstPath, $params );
        } else {
            return $status; // MediaTransformError
        }
    }
 
    public function rasterize( $srcPath, $dstPath, $width, $height, $lang = false ) {
        $mainConfig = MediaWikiServices::getInstance()->getMainConfig();
        $svgConverters = $mainConfig->get( MainConfigNames::SVGConverters );
        $svgConverter = $mainConfig->get( MainConfigNames::SVGConverter );
        $svgConverterPath = $mainConfig->get( MainConfigNames::SVGConverterPath );
        $err = false;
        $retval = '';
        if ( isset( $svgConverters[$svgConverter] ) ) {
            if ( is_array( $svgConverters[$svgConverter] ) ) {
                // This is a PHP callable
                $func = $svgConverters[$svgConverter][0];
                if ( !is_callable( $func ) ) {
                    throw new MWException( "$func is not callable" );
                }
                $err = $func( $srcPath,
                    $dstPath,
                    $width,
                    $height,
                    $lang,
                    ...array_slice( $svgConverters[$svgConverter], 1 )
                );
                $retval = (bool)$err;
            } else {
                // External command
                $cmd = str_replace(
                    [ '$path/', '$width', '$height', '$input', '$output' ],
                    [ $svgConverterPath ? Shell::escape( "{$svgConverterPath}/" ) : "",
                        intval( $width ),
                        intval( $height ),
                        Shell::escape( $srcPath ),
                        Shell::escape( $dstPath ) ],
                    $svgConverters[$svgConverter]
                );
 
                $env = [];
                if ( $lang !== false ) {
                    $env['LANG'] = $lang;
                }
 
                wfDebug( __METHOD__ . ": $cmd" );
                $err = wfShellExecWithStderr( $cmd, $retval, $env );
            }
        }
        // @phan-suppress-next-line PhanTypeMismatchArgumentNullable False positive
        $removed = $this->removeBadFile( $dstPath, $retval );
        if ( $retval != 0 || $removed ) {
            // @phan-suppress-next-next-line PhanPossiblyUndeclaredVariable cmd is set when used
            // @phan-suppress-next-line PhanTypeMismatchArgumentNullable cmd is set when used
            $this->logErrorForExternalProcess( $retval, $err, $cmd );
            return new MediaTransformError( 'thumbnail_error', $width, $height, $err );
        }
 
        return true;
    }
 
    public static function rasterizeImagickExt( $srcPath, $dstPath, $width, $height ) {
        $im = new Imagick( $srcPath );
        $im->setBackgroundColor( 'transparent' );
        $im->readImage( $srcPath );
        $im->setImageFormat( 'png' );
        $im->setImageDepth( 8 );
 
        if ( !$im->thumbnailImage( intval( $width ), intval( $height ), /* fit */ false ) ) {
            return 'Could not resize image';
        }
        if ( !$im->writeImage( $dstPath ) ) {
            return "Could not write to $dstPath";
        }
    }
 
    public function getThumbType( $ext, $mime, $params = null ) {
        return [ 'png', 'image/png' ];
    }
 
    public function getLongDesc( $file ) {
        $metadata = $this->validateMetadata( $file->getMetadataArray() );
        if ( isset( $metadata['error'] ) ) {
            return wfMessage( 'svg-long-error', $metadata['error']['message'] )->text();
        }
 
        if ( $this->isAnimatedImage( $file ) ) {
            $msg = wfMessage( 'svg-long-desc-animated' );
        } else {
            $msg = wfMessage( 'svg-long-desc' );
        }
 
        return $msg->numParams( $file->getWidth(), $file->getHeight() )->sizeParams( $file->getSize() )->parse();
    }
 
    public function getSizeAndMetadata( $state, $filename ) {
        $metadata = [ 'version' => self::SVG_METADATA_VERSION ];
 
        try {
            $svgReader = new SVGReader( $filename );
            $metadata += $svgReader->getMetadata();
        } catch ( TimeoutException $e ) {
            throw $e;
        } catch ( Exception $e ) { // @todo SVG specific exceptions
            // File not found, broken, etc.
            $metadata['error'] = [
                'message' => $e->getMessage(),
                'code' => $e->getCode()
            ];
            wfDebug( __METHOD__ . ': ' . $e->getMessage() );
        }
 
        return [
            'width' => $metadata['width'] ?? 0,
            'height' => $metadata['height'] ?? 0,
            'metadata' => $metadata
        ];
    }
 
    protected function validateMetadata( $unser ) {
        if ( isset( $unser['version'] ) && $unser['version'] == self::SVG_METADATA_VERSION ) {
            return $unser;
        } else {
            return null;
        }
    }
 
    public function getMetadataType( $image ) {
        return 'parsed-svg';
    }
 
    public function isFileMetadataValid( $image ) {
        $meta = $this->validateMetadata( $image->getMetadataArray() );
        if ( !$meta ) {
            return self::METADATA_BAD;
        }
        if ( !isset( $meta['originalWidth'] ) ) {
            // Old but compatible
            return self::METADATA_COMPATIBLE;
        }
 
        return self::METADATA_GOOD;
    }
 
    protected function visibleMetadataFields() {
        $fields = [ 'objectname', 'imagedescription' ];
 
        return $fields;
    }
 
    public function formatMetadata( $file, $context = false ) {
        $result = [
            'visible' => [],
            'collapsed' => []
        ];
        $metadata = $this->validateMetadata( $file->getMetadataArray() );
        if ( !$metadata || isset( $metadata['error'] ) ) {
            return false;
        }
 
        /* @todo Add a formatter
        $format = new FormatSVG( $metadata );
        $formatted = $format->getFormattedData();
        */
 
        // Sort fields into visible and collapsed
        $visibleFields = $this->visibleMetadataFields();
 
        $showMeta = false;
        foreach ( $metadata as $name => $value ) {
            $tag = strtolower( $name );
            if ( isset( self::$metaConversion[$tag] ) ) {
                $tag = strtolower( self::$metaConversion[$tag] );
            } else {
                // Do not output other metadata not in list
                continue;
            }
            $showMeta = true;
            self::addMeta( $result,
                in_array( $tag, $visibleFields ) ? 'visible' : 'collapsed',
                'exif',
                $tag,
                $value
            );
        }
 
        return $showMeta ? $result : false;
    }
 
    public function validateParam( $name, $value ) {
        if ( in_array( $name, [ 'width', 'height' ] ) ) {
            // Reject negative heights, widths
            return ( $value > 0 );
        } elseif ( $name == 'lang' ) {
            // Validate $code
            if ( $value === ''
                || !MediaWikiServices::getInstance()->getLanguageNameUtils()
                    ->isValidCode( $value )
            ) {
                return false;
            }
 
            return true;
        }
 
        // Only lang, width and height are acceptable keys
        return false;
    }
 
    public function makeParamString( $params ) {
        $lang = '';
        $code = $this->getLanguageFromParams( $params );
        if ( $code !== 'en' ) {
            $lang = 'lang' . strtolower( $code ) . '-';
        }
        if ( !isset( $params['width'] ) ) {
            return false;
        }
 
        return "$lang{$params['width']}px";
    }
 
    public function parseParamString( $str ) {
        $m = false;
        if ( preg_match( '/^lang([a-z]+(?:-[a-z]+)*)-(\d+)px$/i', $str, $m ) ) {
            return [ 'width' => array_pop( $m ), 'lang' => $m[1] ];
        } elseif ( preg_match( '/^(\d+)px$/', $str, $m ) ) {
            return [ 'width' => $m[1], 'lang' => 'en' ];
        } else {
            return false;
        }
    }
 
    public function getParamMap() {
        return [ 'img_lang' => 'lang', 'img_width' => 'width' ];
    }
 
    protected function getScriptParams( $params ) {
        $scriptParams = [ 'width' => $params['width'] ];
        if ( isset( $params['lang'] ) ) {
            $scriptParams['lang'] = $params['lang'];
        }
 
        return $scriptParams;
    }
 
    public function getCommonMetaArray( File $file ) {
        $metadata = $this->validateMetadata( $file->getMetadataArray() );
        if ( !$metadata || isset( $metadata['error'] ) ) {
            return [];
        }
        $stdMetadata = [];
        foreach ( $metadata as $name => $value ) {
            $tag = strtolower( $name );
            if ( $tag === 'originalwidth' || $tag === 'originalheight' ) {
                // Skip these. In the exif metadata stuff, it is assumed these
                // are measured in px, which is not the case here.
                continue;
            }
            if ( isset( self::$metaConversion[$tag] ) ) {
                $tag = self::$metaConversion[$tag];
                $stdMetadata[$tag] = $value;
            }
        }
 
        return $stdMetadata;
    }
}