MemoryFileBackend.php

Go to the documentation of this file.

<?php
use Wikimedia\AtEase\AtEase;
use Wikimedia\FileBackend\FileBackend;
use Wikimedia\Timestamp\ConvertibleTimestamp;
 
class MemoryFileBackend extends FileBackendStore {
    protected $files = [];
 
    public function getFeatures() {
        return self::ATTR_UNICODE_PATHS;
    }
 
    public function isPathUsableInternal( $storagePath ) {
        return ( $this->resolveHashKey( $storagePath ) !== null );
    }
 
    protected function doCreateInternal( array $params ) {
        $status = $this->newStatus();
 
        $dst = $this->resolveHashKey( $params['dst'] );
        if ( $dst === null ) {
            $status->fatal( 'backend-fail-invalidpath', $params['dst'] );
 
            return $status;
        }
 
        $this->files[$dst] = [
            'data' => $params['content'],
            'mtime' => ConvertibleTimestamp::convert( TS_MW, time() )
        ];
 
        return $status;
    }
 
    protected function doStoreInternal( array $params ) {
        $status = $this->newStatus();
 
        $dst = $this->resolveHashKey( $params['dst'] );
        if ( $dst === null ) {
            $status->fatal( 'backend-fail-invalidpath', $params['dst'] );
 
            return $status;
        }
 
        AtEase::suppressWarnings();
        $data = file_get_contents( $params['src'] );
        AtEase::restoreWarnings();
        if ( $data === false ) { // source doesn't exist?
            $status->fatal( 'backend-fail-store', $params['src'], $params['dst'] );
 
            return $status;
        }
 
        $this->files[$dst] = [
            'data' => $data,
            'mtime' => ConvertibleTimestamp::convert( TS_MW, time() )
        ];
 
        return $status;
    }
 
    protected function doCopyInternal( array $params ) {
        $status = $this->newStatus();
 
        $src = $this->resolveHashKey( $params['src'] );
        if ( $src === null ) {
            $status->fatal( 'backend-fail-invalidpath', $params['src'] );
 
            return $status;
        }
 
        $dst = $this->resolveHashKey( $params['dst'] );
        if ( $dst === null ) {
            $status->fatal( 'backend-fail-invalidpath', $params['dst'] );
 
            return $status;
        }
 
        if ( !isset( $this->files[$src] ) ) {
            if ( empty( $params['ignoreMissingSource'] ) ) {
                $status->fatal( 'backend-fail-copy', $params['src'], $params['dst'] );
            }
 
            return $status;
        }
 
        $this->files[$dst] = [
            'data' => $this->files[$src]['data'],
            'mtime' => ConvertibleTimestamp::convert( TS_MW, time() )
        ];
 
        return $status;
    }
 
    protected function doMoveInternal( array $params ) {
        $status = $this->newStatus();
 
        $src = $this->resolveHashKey( $params['src'] );
        if ( $src === null ) {
            $status->fatal( 'backend-fail-invalidpath', $params['src'] );
 
            return $status;
        }
 
        $dst = $this->resolveHashKey( $params['dst'] );
        if ( $dst === null ) {
            $status->fatal( 'backend-fail-invalidpath', $params['dst'] );
 
            return $status;
        }
 
        if ( !isset( $this->files[$src] ) ) {
            if ( empty( $params['ignoreMissingSource'] ) ) {
                $status->fatal( 'backend-fail-move', $params['src'], $params['dst'] );
            }
 
            return $status;
        }
 
        $this->files[$dst] = $this->files[$src];
        unset( $this->files[$src] );
        $this->files[$dst]['mtime'] = ConvertibleTimestamp::convert( TS_MW, time() );
 
        return $status;
    }
 
    protected function doDeleteInternal( array $params ) {
        $status = $this->newStatus();
 
        $src = $this->resolveHashKey( $params['src'] );
        if ( $src === null ) {
            $status->fatal( 'backend-fail-invalidpath', $params['src'] );
 
            return $status;
        }
 
        if ( !isset( $this->files[$src] ) ) {
            if ( empty( $params['ignoreMissingSource'] ) ) {
                $status->fatal( 'backend-fail-delete', $params['src'] );
            }
 
            return $status;
        }
 
        unset( $this->files[$src] );
 
        return $status;
    }
 
    protected function doGetFileStat( array $params ) {
        $src = $this->resolveHashKey( $params['src'] );
        if ( $src === null ) {
            return self::RES_ERROR; // invalid path
        }
 
        if ( isset( $this->files[$src] ) ) {
            return [
                'mtime' => $this->files[$src]['mtime'],
                'size' => strlen( $this->files[$src]['data'] ),
            ];
        }
 
        return self::RES_ABSENT;
    }
 
    protected function doGetLocalCopyMulti( array $params ) {
        $tmpFiles = []; // (path => TempFSFile)
        foreach ( $params['srcs'] as $srcPath ) {
            $src = $this->resolveHashKey( $srcPath );
            if ( $src === null ) {
                $fsFile = self::RES_ERROR;
            } elseif ( !isset( $this->files[$src] ) ) {
                $fsFile = self::RES_ABSENT;
            } else {
                // Create a new temporary file with the same extension...
                $ext = FileBackend::extensionFromPath( $src );
                $fsFile = $this->tmpFileFactory->newTempFSFile( 'localcopy_', $ext );
                if ( $fsFile ) {
                    $bytes = file_put_contents( $fsFile->getPath(), $this->files[$src]['data'] );
                    if ( $bytes !== strlen( $this->files[$src]['data'] ) ) {
                        $fsFile = self::RES_ERROR;
                    }
                }
            }
            $tmpFiles[$srcPath] = $fsFile;
        }
 
        return $tmpFiles;
    }
 
    protected function doDirectoryExists( $container, $dir, array $params ) {
        $prefix = rtrim( "$container/$dir", '/' ) . '/';
        foreach ( $this->files as $path => $data ) {
            if ( strpos( $path, $prefix ) === 0 ) {
                return true;
            }
        }
 
        return false;
    }
 
    public function getDirectoryListInternal( $container, $dir, array $params ) {
        $dirs = [];
        $prefix = rtrim( "$container/$dir", '/' ) . '/';
        $prefixLen = strlen( $prefix );
        foreach ( $this->files as $path => $data ) {
            if ( strpos( $path, $prefix ) === 0 ) {
                $relPath = substr( $path, $prefixLen );
                if ( $relPath === false ) {
                    continue;
                } elseif ( strpos( $relPath, '/' ) === false ) {
                    continue; // just a file
                }
                $parts = array_slice( explode( '/', $relPath ), 0, -1 ); // last part is file name
                if ( !empty( $params['topOnly'] ) ) {
                    $dirs[$parts[0]] = 1; // top directory
                } else {
                    $current = '';
                    foreach ( $parts as $part ) { // all directories
                        $dir = ( $current === '' ) ? $part : "$current/$part";
                        $dirs[$dir] = 1;
                        $current = $dir;
                    }
                }
            }
        }
 
        return array_keys( $dirs );
    }
 
    public function getFileListInternal( $container, $dir, array $params ) {
        $files = [];
        $prefix = rtrim( "$container/$dir", '/' ) . '/';
        $prefixLen = strlen( $prefix );
        foreach ( $this->files as $path => $data ) {
            if ( strpos( $path, $prefix ) === 0 ) {
                $relPath = substr( $path, $prefixLen );
                if ( $relPath === false ) {
                    continue;
                } elseif ( !empty( $params['topOnly'] ) && strpos( $relPath, '/' ) !== false ) {
                    continue;
                }
                $files[] = $relPath;
            }
        }
 
        return $files;
    }
 
    protected function directoriesAreVirtual() {
        return true;
    }
 
    protected function resolveHashKey( $storagePath ) {
        [ $fullCont, $relPath ] = $this->resolveStoragePathReal( $storagePath );
        if ( $relPath === null ) {
            return null; // invalid
        }
 
        return ( $relPath !== '' ) ? "$fullCont/$relPath" : $fullCont;
    }
}