MemoryFileBackend.php

Go to the documentation of this file.

<?php
class MemoryFileBackend extends FileBackendStore {
    protected $files = array();

    public function isPathUsableInternal( $storagePath ) {
        return true;
    }

    protected function doCreateInternal( array $params ) {
        $status = Status::newGood();

        $dst = $this->resolveHashKey( $params['dst'] );
        if ( $dst === null ) {
            $status->fatal( 'backend-fail-invalidpath', $params['dst'] );

            return $status;
        }

        $this->files[$dst] = array(
            'data' => $params['content'],
            'mtime' => wfTimestamp( TS_MW, time() )
        );

        return $status;
    }

    protected function doStoreInternal( array $params ) {
        $status = Status::newGood();

        $dst = $this->resolveHashKey( $params['dst'] );
        if ( $dst === null ) {
            $status->fatal( 'backend-fail-invalidpath', $params['dst'] );

            return $status;
        }

        wfSuppressWarnings();
        $data = file_get_contents( $params['src'] );
        wfRestoreWarnings();
        if ( $data === false ) { // source doesn't exist?
            $status->fatal( 'backend-fail-store', $params['src'], $params['dst'] );

            return $status;
        }

        $this->files[$dst] = array(
            'data' => $data,
            'mtime' => wfTimestamp( TS_MW, time() )
        );

        return $status;
    }

    protected function doCopyInternal( array $params ) {
        $status = Status::newGood();

        $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] = array(
            'data' => $this->files[$src]['data'],
            'mtime' => wfTimestamp( TS_MW, time() )
        );

        return $status;
    }

    protected function doDeleteInternal( array $params ) {
        $status = Status::newGood();

        $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 null;
        }

        if ( isset( $this->files[$src] ) ) {
            return array(
                'mtime' => $this->files[$src]['mtime'],
                'size' => strlen( $this->files[$src]['data'] ),
            );
        }

        return false;
    }

    protected function doGetLocalCopyMulti( array $params ) {
        $tmpFiles = array(); // (path => TempFSFile)
        foreach ( $params['srcs'] as $srcPath ) {
            $src = $this->resolveHashKey( $srcPath );
            if ( $src === null || !isset( $this->files[$src] ) ) {
                $fsFile = null;
            } else {
                // Create a new temporary file with the same extension...
                $ext = FileBackend::extensionFromPath( $src );
                $fsFile = TempFSFile::factory( 'localcopy_', $ext );
                if ( $fsFile ) {
                    $bytes = file_put_contents( $fsFile->getPath(), $this->files[$src]['data'] );
                    if ( $bytes !== strlen( $this->files[$src]['data'] ) ) {
                        $fsFile = null;
                    }
                }
            }
            $tmpFiles[$srcPath] = $fsFile;
        }

        return $tmpFiles;
    }

    protected function doStreamFile( array $params ) {
        $status = Status::newGood();

        $src = $this->resolveHashKey( $params['src'] );
        if ( $src === null || !isset( $this->files[$src] ) ) {
            $status->fatal( 'backend-fail-stream', $params['src'] );

            return $status;
        }

        print $this->files[$src]['data'];

        return $status;
    }

    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 = array();
        $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 = array();
        $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 ) {
        list( $fullCont, $relPath ) = $this->resolveStoragePathReal( $storagePath );
        if ( $relPath === null ) {
            return null; // invalid
        }

        return ( $relPath !== '' ) ? "$fullCont/$relPath" : $fullCont;
    }
}