3 use MediaWiki\MediaWikiServices
;
4 use Psr\Log\LoggerInterface
;
9 class ImportableUploadRevisionImporter
implements UploadRevisionImporter
{
12 * @var LoggerInterface
19 private $enableUploads;
24 private $shouldCreateNullRevision = true;
27 * @param bool $enableUploads
28 * @param LoggerInterface $logger
30 public function __construct(
32 LoggerInterface
$logger
34 $this->enableUploads
= $enableUploads;
35 $this->logger
= $logger;
39 * Setting this to false will deactivate the creation of a null revision as part of the upload
40 * process logging in LocalFile::recordUpload2, see T193621
42 * @param bool $shouldCreateNullRevision
44 public function setNullRevisionCreation( $shouldCreateNullRevision ) {
45 $this->shouldCreateNullRevision
= $shouldCreateNullRevision;
51 private function newNotOkStatus() {
52 $statusValue = new StatusValue();
53 $statusValue->setOK( false );
57 public function import( ImportableUploadRevision
$importableRevision ) {
59 $archiveName = $importableRevision->getArchiveName();
61 $this->logger
->debug( __METHOD__
. "Importing archived file as $archiveName\n" );
62 $file = OldLocalFile
::newFromArchiveName( $importableRevision->getTitle(),
63 RepoGroup
::singleton()->getLocalRepo(), $archiveName );
65 $file = wfLocalFile( $importableRevision->getTitle() );
66 $file->load( File
::READ_LATEST
);
67 $this->logger
->debug( __METHOD__
. 'Importing new file as ' . $file->getName() . "\n" );
68 if ( $file->exists() && $file->getTimestamp() > $importableRevision->getTimestamp() ) {
69 $archiveName = $importableRevision->getTimestamp() . '!' . $file->getName();
70 $file = OldLocalFile
::newFromArchiveName( $importableRevision->getTitle(),
71 RepoGroup
::singleton()->getLocalRepo(), $archiveName );
72 $this->logger
->debug( __METHOD__
. "File already exists; importing as $archiveName\n" );
76 $this->logger
->debug( __METHOD__
. ': Bad file for ' . $importableRevision->getTitle() . "\n" );
77 return $this->newNotOkStatus();
80 # Get the file source or download if necessary
81 $source = $importableRevision->getFileSrc();
82 $autoDeleteSource = $importableRevision->isTempSrc();
83 if ( !strlen( $source ) ) {
84 $source = $this->downloadSource( $importableRevision );
85 $autoDeleteSource = true;
87 if ( !strlen( $source ) ) {
88 $this->logger
->debug( __METHOD__
. ": Could not fetch remote file.\n" );
89 return $this->newNotOkStatus();
92 $tmpFile = new TempFSFile( $source );
93 if ( $autoDeleteSource ) {
94 $tmpFile->autocollect();
97 $sha1File = ltrim( sha1_file( $source ), '0' );
98 $sha1 = $importableRevision->getSha1();
99 if ( $sha1 && ( $sha1 !== $sha1File ) ) {
100 $this->logger
->debug( __METHOD__
. ": Corrupt file $source.\n" );
101 return $this->newNotOkStatus();
104 $user = $importableRevision->getUserObj()
105 ?
: User
::newFromName( $importableRevision->getUser(), false );
107 # Do the actual upload
108 if ( $file instanceof OldLocalFile
) {
109 $status = $file->uploadOld(
111 $importableRevision->getTimestamp(),
112 $importableRevision->getComment(),
117 $status = $file->upload(
119 $importableRevision->getComment(),
120 $importableRevision->getComment(),
123 $importableRevision->getTimestamp(),
126 $this->shouldCreateNullRevision
130 if ( $status->isGood() ) {
131 $this->logger
->debug( __METHOD__
. ": Successful\n" );
133 $this->logger
->debug( __METHOD__
. ': failed: ' . $status->getHTML() . "\n" );
140 * @deprecated DO NOT CALL ME.
141 * This method was introduced when factoring UploadImporter out of WikiRevision.
142 * It only has 1 use by the deprecated downloadSource method in WikiRevision.
143 * Do not use this in new code.
145 * @param ImportableUploadRevision $wikiRevision
147 * @return bool|string
149 public function downloadSource( ImportableUploadRevision
$wikiRevision ) {
150 if ( !$this->enableUploads
) {
154 $tempo = tempnam( wfTempDir(), 'download' );
155 $f = fopen( $tempo, 'wb' );
157 $this->logger
->debug( "IMPORT: couldn't write to temp file $tempo\n" );
162 $src = $wikiRevision->getSrc();
163 $data = MediaWikiServices
::getInstance()->getHttpRequestFactory()->
164 get( $src, [], __METHOD__
);
166 $this->logger
->debug( "IMPORT: couldn't fetch source $src\n" );