addOption( "wordlist", 'A list of words', true, true ); $this->addOption( "font", "The font to use", true, true ); $this->addOption( "font-size", "The font size ", false, true ); $this->addOption( "blacklist", "A blacklist of words that should not be used", false, true ); $this->addOption( "fill", "Fill the captcha container to N files", true, true ); $this->addOption( "verbose", "Show debugging information when running the captcha python script" ); $this->addOption( "oldcaptcha", "Whether to use captcha-old.py which doesn't have OCR fighting improvements" ); $this->addOption( "delete", "Deletes all the old captchas" ); $this->addOption( "threads", "The number of threads to use to generate the images", false, true ); $this->addDescription( "Generate new fancy captchas and move them into storage" ); $this->requireExtension( "FancyCaptcha" ); } public function execute() { global $wgCaptchaSecret, $wgCaptchaDirectoryLevels; $totalTime = -microtime( true ); $instance = ConfirmEditHooks::getInstance(); if ( !( $instance instanceof FancyCaptcha ) ) { $this->fatalError( "\$wgCaptchaClass is not FancyCaptcha.\n", 1 ); } $backend = $instance->getBackend(); $deleteOldCaptchas = $this->getOption( 'delete' ); $countGen = (int)$this->getOption( 'fill' ); if ( !$deleteOldCaptchas ) { $countAct = $instance->getCaptchaCount(); $this->output( "Current number of captchas is $countAct.\n" ); $countGen -= $countAct; } if ( $countGen <= 0 ) { $this->output( "No need to generate anymore captchas.\n" ); return; } $tmpDir = wfTempDir() . '/mw-fancycaptcha-' . time() . '-' . wfRandomString( 6 ); if ( !wfMkdirParents( $tmpDir ) ) { $this->fatalError( "Could not create temp directory.\n", 1 ); } $captchaScript = 'captcha.py'; if ( $this->hasOption( 'oldcaptcha' ) ) { $captchaScript = 'captcha-old.py'; } $cmd = sprintf( "python %s --key %s --output %s --count %s --dirs %s", wfEscapeShellArg( dirname( __DIR__ ) . '/' . $captchaScript ), wfEscapeShellArg( $wgCaptchaSecret ), wfEscapeShellArg( $tmpDir ), wfEscapeShellArg( (string)$countGen ), wfEscapeShellArg( $wgCaptchaDirectoryLevels ) ); foreach ( [ 'wordlist', 'font', 'font-size', 'blacklist', 'verbose', 'threads' ] as $par ) { if ( $this->hasOption( $par ) ) { $cmd .= " --$par " . wfEscapeShellArg( $this->getOption( $par ) ); } } $this->output( "Generating $countGen new captchas.." ); $retVal = 1; $captchaTime = -microtime( true ); wfShellExec( $cmd, $retVal, [], [ 'time' => 0 ] ); if ( $retVal != 0 ) { $this->output( " Failed.\n" ); wfRecursiveRemoveDir( $tmpDir ); $this->fatalError( "An error occured when running $captchaScript.\n", 1 ); } $captchaTime += microtime( true ); $this->output( " Done.\n" ); $this->output( sprintf( "\nGenerated %d captchas in %.1f seconds\n", $countGen, $captchaTime ) ); $filesToDelete = []; if ( $deleteOldCaptchas ) { $this->output( "Getting a list of old captchas to delete..." ); $path = $backend->getRootStoragePath() . '/captcha-render'; foreach ( $backend->getFileList( [ 'dir' => $path ] ) as $file ) { $filesToDelete[] = [ 'op' => 'delete', 'src' => $path . '/' . $file, ]; } $this->output( " Done.\n" ); } $this->output( "Copying the new captchas to storage..." ); $storeTime = -microtime( true ); $iter = new RecursiveIteratorIterator( new RecursiveDirectoryIterator( $tmpDir, FilesystemIterator::SKIP_DOTS ), RecursiveIteratorIterator::LEAVES_ONLY ); $captchasGenerated = iterator_count( $iter ); $filesToStore = []; /** * @var $fileInfo SplFileInfo */ foreach ( $iter as $fileInfo ) { if ( !$fileInfo->isFile() ) { continue; } list( $salt, $hash ) = $instance->hashFromImageName( $fileInfo->getBasename() ); $dest = $instance->imagePath( $salt, $hash ); $backend->prepare( [ 'dir' => dirname( $dest ) ] ); $filesToStore[] = [ 'op' => 'store', 'src' => $fileInfo->getPathname(), 'dst' => $dest, ]; } $ret = $backend->doQuickOperations( $filesToStore ); $storeTime += microtime( true ); $storeSucceeded = true; if ( $ret->isOK() ) { $this->output( " Done.\n" ); $this->output( sprintf( "\nCopied %d captchas to storage in %.1f seconds\n", $ret->successCount, $storeTime ) ); if ( !$ret->isGood() ) { $this->output( "Non fatal errors:\n" . Status::wrap( $ret )->getWikiText( false, false, 'en' ) . "\n" ); } if ( $ret->failCount ) { $storeSucceeded = false; $this->error( sprintf( "\nFailed to copy %d captchas\n", $ret->failCount ) ); } if ( $ret->successCount + $ret->failCount !== $captchasGenerated ) { $storeSucceeded = false; $this->error( sprintf( "Internal error: captchasGenerated: %d, successCount: %d, failCount: %d\n", $captchasGenerated, $ret->successCount, $ret->failCount ) ); } } else { $storeSucceeded = false; $this->output( "Errored.\n" ); $this->error( Status::wrap( $ret )->getWikiText( false, false, 'en' ) . "\n" ); } if ( $storeSucceeded && $deleteOldCaptchas ) { $numOriginalFiles = count( $filesToDelete ); $this->output( "Deleting {$numOriginalFiles} old captchas...\n" ); $deleteTime = -microtime( true ); $ret = $backend->doQuickOperations( $filesToDelete ); $deleteTime += microtime( true ); if ( $ret->isOK() ) { $this->output( "Done.\n" ); $this->output( sprintf( "\nDeleted %d old captchas in %.1f seconds\n", $numOriginalFiles, $deleteTime ) ); if ( !$ret->isGood() ) { $this->output( "Non fatal errors:\n" . Status::wrap( $ret )->getWikiText( false, false, 'en' ) . "\n" ); } } else { $this->output( "Errored.\n" ); $this->error( Status::wrap( $ret )->getWikiText( false, false, 'en' ) . "\n" ); } } $this->output( "Removing temporary files..." ); wfRecursiveRemoveDir( $tmpDir ); $this->output( " Done.\n" ); $totalTime += microtime( true ); $this->output( sprintf( "\nWhole captchas generation process took %.1f seconds\n", $totalTime ) ); } } $maintClass = GenerateFancyCaptchas::class; require_once RUN_MAINTENANCE_IF_MAIN;