TempFSFile.php

Go to the documentation of this file.

<?php
class TempFSFile extends FSFile {
    protected $canDelete = false;

    protected static $pathsCollect = null;

    public function __construct( $path ) {
        parent::__construct( $path );

        if ( self::$pathsCollect === null ) {
            self::$pathsCollect = [];
            register_shutdown_function( [ __CLASS__, 'purgeAllOnShutdown' ] );
        }
    }

    public static function factory( $prefix, $extension = '' ) {
        $ext = ( $extension != '' ) ? ".{$extension}" : '';

        $attempts = 5;
        while ( $attempts-- ) {
            $path = wfTempDir() . '/' . $prefix . wfRandomString( 12 ) . $ext;
            MediaWiki\suppressWarnings();
            $newFileHandle = fopen( $path, 'x' );
            MediaWiki\restoreWarnings();
            if ( $newFileHandle ) {
                fclose( $newFileHandle );
                $tmpFile = new self( $path );
                $tmpFile->autocollect();
                // Safely instantiated, end loop.
                return $tmpFile;
            }
        }

        // Give up
        return null;
    }

    public function purge() {
        $this->canDelete = false; // done
        MediaWiki\suppressWarnings();
        $ok = unlink( $this->path );
        MediaWiki\restoreWarnings();

        unset( self::$pathsCollect[$this->path] );

        return $ok;
    }

    public function bind( $object ) {
        if ( is_object( $object ) ) {
            if ( !isset( $object->tempFSFileReferences ) ) {
                // Init first since $object might use __get() and return only a copy variable
                $object->tempFSFileReferences = [];
            }
            $object->tempFSFileReferences[] = $this;
        }

        return $this;
    }

    public function preserve() {
        $this->canDelete = false;

        unset( self::$pathsCollect[$this->path] );

        return $this;
    }

    public function autocollect() {
        $this->canDelete = true;

        self::$pathsCollect[$this->path] = 1;

        return $this;
    }

    public static function purgeAllOnShutdown() {
        foreach ( self::$pathsCollect as $path ) {
            MediaWiki\suppressWarnings();
            unlink( $path );
            MediaWiki\restoreWarnings();
        }
    }

    function __destruct() {
        if ( $this->canDelete ) {
            $this->purge();
        }
    }
}