migrateFileRepoLayout.php

Go to the documentation of this file.

<?php
use MediaWiki\FileRepo\File\File;
use MediaWiki\FileRepo\File\LocalFile;
use MediaWiki\FileRepo\FileBackendDBRepoWrapper;
use MediaWiki\FileRepo\LocalRepo;
use MediaWiki\Maintenance\Maintenance;
use Wikimedia\FileBackend\FileBackend;
 
// @codeCoverageIgnoreStart
require_once __DIR__ . '/Maintenance.php';
// @codeCoverageIgnoreEnd
 
class MigrateFileRepoLayout extends Maintenance {
    public function __construct() {
        parent::__construct();
        $this->addDescription( 'Copy files in repo to a different layout.' );
        $this->addOption( 'oldlayout', "Old layout; one of 'name' or 'sha1'", true, true );
        $this->addOption( 'newlayout', "New layout; one of 'name' or 'sha1'", true, true );
        $this->addOption( 'since', "Copy only files from after this timestamp", false, true );
        $this->setBatchSize( 50 );
    }
 
    public function execute() {
        $oldLayout = $this->getOption( 'oldlayout' );
        if ( !in_array( $oldLayout, [ 'name', 'sha1' ] ) ) {
            $this->fatalError( "Invalid old layout." );
        }
        $newLayout = $this->getOption( 'newlayout' );
        if ( !in_array( $newLayout, [ 'name', 'sha1' ] ) ) {
            $this->fatalError( "Invalid new layout." );
        }
        $since = $this->getOption( 'since' );
 
        $repo = $this->getRepo();
 
        $be = $repo->getBackend();
        if ( $be instanceof FileBackendDBRepoWrapper ) {
            // avoid path translations for this script
            $be = $be->getInternalBackend();
        }
 
        $dbw = $repo->getPrimaryDB();
 
        $origBase = $be->getContainerStoragePath( "{$repo->getName()}-original" );
        $startTime = wfTimestampNow();
 
        // Do current and archived versions...
        $conds = [];
        if ( $since ) {
            $conds[] = $dbw->expr( 'img_timestamp', '>=', $dbw->timestamp( $since ) );
        }
 
        $batchSize = $this->getBatchSize();
        $batch = [];
        $lastName = '';
        do {
            $res = $dbw->newSelectQueryBuilder()
                ->select( [ 'img_name', 'img_sha1' ] )
                ->from( 'image' )
                ->where( $dbw->expr( 'img_name', '>', $lastName ) )
                ->andWhere( $conds )
                ->orderBy( 'img_name' )
                ->limit( $batchSize )
                ->caller( __METHOD__ )->fetchResultSet();
 
            foreach ( $res as $row ) {
                $lastName = $row->img_name;
                $file = $repo->newFile( $row->img_name );
                // Check in case SHA1 rows are not populated for some files
                $sha1 = $row->img_sha1 !== '' ? $row->img_sha1 : $file->getSha1();
 
                if ( $sha1 === '' ) {
                    $this->error( "Image SHA-1 not known for {$row->img_name}." );
                } else {
                    if ( $oldLayout === 'sha1' ) {
                        $spath = "{$origBase}/{$sha1[0]}/{$sha1[1]}/{$sha1[2]}/{$sha1}";
                    } else {
                        $spath = $file->getPath();
                    }
 
                    if ( $newLayout === 'sha1' ) {
                        $dpath = "{$origBase}/{$sha1[0]}/{$sha1[1]}/{$sha1[2]}/{$sha1}";
                    } else {
                        $dpath = $file->getPath();
                    }
 
                    $status = $be->prepare( [
                        'dir' => dirname( $dpath ), 'bypassReadOnly' => true ] );
                    if ( !$status->isOK() ) {
                        $this->error( $status );
                    }
 
                    $batch[] = [ 'op' => 'copy', 'overwrite' => true,
                        'src' => $spath, 'dst' => $dpath, 'img' => $row->img_name ];
                }
 
                foreach ( $file->getHistory() as $ofile ) {
                    $sha1 = $ofile->getSha1();
                    if ( $sha1 === '' ) {
                        $this->error( "Image SHA-1 not set for {$ofile->getArchiveName()}." );
                        continue;
                    }
 
                    if ( $oldLayout === 'sha1' ) {
                        $spath = "{$origBase}/{$sha1[0]}/{$sha1[1]}/{$sha1[2]}/{$sha1}";
                    } elseif ( $ofile->isDeleted( File::DELETED_FILE ) ) {
                        $spath = $be->getContainerStoragePath( "{$repo->getName()}-deleted" ) .
                            '/' . $repo->getDeletedHashPath( $sha1 ) .
                            $sha1 . '.' . $ofile->getExtension();
                    } else {
                        $spath = $ofile->getPath();
                    }
 
                    if ( $newLayout === 'sha1' ) {
                        $dpath = "{$origBase}/{$sha1[0]}/{$sha1[1]}/{$sha1[2]}/{$sha1}";
                    } else {
                        $dpath = $ofile->getPath();
                    }
 
                    $status = $be->prepare( [
                        'dir' => dirname( $dpath ), 'bypassReadOnly' => true ] );
                    if ( !$status->isOK() ) {
                        $this->error( $status );
                    }
                    $batch[] = [ 'op' => 'copy', 'overwrite' => true,
                        'src' => $spath, 'dst' => $dpath, 'img' => $ofile->getArchiveName() ];
                }
 
                if ( count( $batch ) >= $batchSize ) {
                    $this->runBatch( $batch, $be );
                    $batch = [];
                }
            }
        } while ( $res->numRows() );
 
        if ( count( $batch ) ) {
            $this->runBatch( $batch, $be );
        }
 
        // Do deleted versions...
        $conds = [];
        if ( $since ) {
            $conds[] = $dbw->expr( 'fa_deleted_timestamp', '>=', $dbw->timestamp( $since ) );
        }
 
        $batch = [];
        $lastId = 0;
        do {
            $res = $dbw->newSelectQueryBuilder()
                ->select( [ 'fa_storage_key', 'fa_id', 'fa_name' ] )
                ->from( 'filearchive' )
                ->where( $dbw->expr( 'fa_id', '>', $lastId ) )
                ->andWhere( $conds )
                ->orderBy( 'fa_id' )
                ->limit( $batchSize )
                ->caller( __METHOD__ )->fetchResultSet();
 
            foreach ( $res as $row ) {
                $lastId = $row->fa_id;
                $sha1Key = $row->fa_storage_key;
                if ( $sha1Key === '' ) {
                    $this->error( "Image SHA-1 not set for file #{$row->fa_id} (deleted)." );
                    continue;
                }
                $sha1 = substr( $sha1Key, 0, strpos( $sha1Key, '.' ) );
 
                if ( $oldLayout === 'sha1' ) {
                    $spath = "{$origBase}/{$sha1[0]}/{$sha1[1]}/{$sha1[2]}/{$sha1}";
                } else {
                    $spath = $be->getContainerStoragePath( "{$repo->getName()}-deleted" ) .
                        '/' . $repo->getDeletedHashPath( $sha1Key ) . $sha1Key;
                }
 
                if ( $newLayout === 'sha1' ) {
                    $dpath = "{$origBase}/{$sha1[0]}/{$sha1[1]}/{$sha1[2]}/{$sha1}";
                } else {
                    $dpath = $be->getContainerStoragePath( "{$repo->getName()}-deleted" ) .
                        '/' . $repo->getDeletedHashPath( $sha1Key ) . $sha1Key;
                }
 
                $status = $be->prepare( [
                    'dir' => dirname( $dpath ), 'bypassReadOnly' => true ] );
                if ( !$status->isOK() ) {
                    $this->error( $status );
                }
 
                $batch[] = [ 'op' => 'copy', 'src' => $spath, 'dst' => $dpath,
                    'overwriteSame' => true, 'img' => "(ID {$row->fa_id}) {$row->fa_name}" ];
 
                if ( count( $batch ) >= $batchSize ) {
                    $this->runBatch( $batch, $be );
                    $batch = [];
                }
            }
        } while ( $res->numRows() );
 
        if ( count( $batch ) ) {
            $this->runBatch( $batch, $be );
        }
 
        $this->output( "Done (started $startTime)\n" );
    }
 
    protected function getRepo(): LocalRepo {
        return $this->getServiceContainer()->getRepoGroup()->getLocalRepo();
    }
 
    protected function runBatch( array $ops, FileBackend $be ) {
        $this->output( "Migrating file batch:\n" );
        foreach ( $ops as $op ) {
            $this->output( "\"{$op['img']}\" (dest: {$op['dst']})\n" );
        }
 
        $status = $be->doOperations( $ops, [ 'bypassReadOnly' => true ] );
        if ( !$status->isOK() ) {
            $this->error( $status );
        }
 
        $this->output( "Batch done\n\n" );
    }
}
 
// @codeCoverageIgnoreStart
$maintClass = MigrateFileRepoLayout::class;
require_once RUN_MAINTENANCE_IF_MAIN;
// @codeCoverageIgnoreEnd