6 * First, destination checks are made, and, if ignorewarnings is not
7 * checked, errors / warning is returned.
9 * 1. We return the uploadUrl.
10 * 2. We then accept chunk uploads from the client.
11 * 3. Return chunk id on each POSTED chunk.
12 * 4. Once the client posts "done=1", the files are concatenated together.
14 * (More info at: http://firefogg.org/dev/chunk_post.html)
16 class UploadFromChunks
extends UploadBase
{
22 protected $chunkMode; // INIT, CHUNK, DONE
23 protected $sessionKey;
31 // Parent class requires this function even though it is only
32 // used from SpecialUpload.php and we don't do chunked uploading
33 // from SpecialUpload -- best to raise an exception for
35 public function initializeFromRequest( &$request ) {
36 throw new MWException( 'not implemented' );
39 public function initialize( $done, $filename, $sessionKey, $path, $fileSize, $sessionData ) {
40 global $wgTmpDirectory;
41 $this->status
= new Status
;
43 $this->initFromSessionKey( $sessionKey, $sessionData );
45 if ( !$this->sessionKey
&& !$done ) {
46 // session key not set, init the chunk upload system:
47 $this->chunkMode
= self
::INIT
;
48 $this->initializePathInfo( $filename, $path, 0, true);
49 } else if ( $this->sessionKey
&& !$done ) {
50 $this->chunkMode
= self
::CHUNK
;
51 } else if ( $this->sessionKey
&& $done ) {
52 $this->chunkMode
= self
::DONE
;
54 if ( $this->chunkMode
== self
::CHUNK ||
$this->chunkMode
== self
::DONE
) {
55 $this->mTempPath
= $path;
56 $this->mFileSize +
= $fileSize;
61 * Set session information for chunked uploads and allocate a unique key.
62 * @param $comment string
63 * @param $pageText string
64 * @param $watch boolean
66 * @returns string the session key for this chunked upload
68 protected function setupChunkSession( $comment, $pageText, $watch ) {
69 $this->sessionKey
= $this->getSessionKey();
70 $_SESSION['wsUploadData'][$this->sessionKey
] = array(
71 'comment' => $comment,
72 'pageText' => $pageText,
74 'mFilteredName' => $this->mFilteredName
,
76 'mDesiredDestName' => $this->mDesiredDestName
,
77 'version' => self
::SESSION_VERSION
,
79 return $this->sessionKey
;
83 * Initialize a continuation of a chunked upload from a session key
84 * @param $sessionKey string
85 * @param $request WebRequest
89 protected function initFromSessionKey( $sessionKey, $sessionData ) {
90 // testing against null because we don't want to cause obscure
91 // bugs when $sessionKey is full of "0"
92 if ( $sessionKey === null ) {
95 $this->sessionKey
= $sessionKey;
97 if ( isset( $sessionData[$this->sessionKey
]['version'] )
98 && $sessionData[$this->sessionKey
]['version'] == self
::SESSION_VERSION
)
100 $this->comment
= $sessionData[$this->sessionKey
]['comment'];
101 $this->pageText
= $sessionData[$this->sessionKey
]['pageText'];
102 $this->watch
= $sessionData[$this->sessionKey
]['watch'];
103 $this->mFilteredName
= $sessionData[$this->sessionKey
]['mFilteredName'];
104 $this->repoPath
= $sessionData[$this->sessionKey
]['repoPath'];
105 $this->mDesiredDestName
= $sessionData[$this->sessionKey
]['mDesiredDestName'];
107 $this->status
= Status
::newFatal( 'invalid-session-key' );
112 * Handle a chunk of the upload. Overrides the parent method
113 * because Chunked Uploading clients (i.e. Firefogg) require
114 * specific API responses.
115 * @see UploadBase::performUpload
117 public function performUpload( $comment, $pageText, $watch, $user ) {
118 wfDebug( "\n\n\performUpload(chunked): comment:" . $comment . ' pageText: ' . $pageText . ' watch:' . $watch );
119 global $wgUser, $wgOut;
121 if ( $this->chunkMode
== self
::INIT
) {
122 // firefogg expects a specific result per:
123 // http://www.firefogg.org/dev/chunk_post.html
125 // it's okay to return the token here because
126 // a) the user must have requested the token to get here and
127 // b) should only happen over POST
128 // c) we need the token to validate chunks are coming from a non-xss request
129 return Status
::newGood(
130 array('uploadUrl' => wfExpandUrl( wfScript( 'api' ) ) . "?" .
131 wfArrayToCGI(array('action' => 'upload',
132 'token' => $wgUser->editToken(),
134 'filename' => $pageText,
135 'enablechunks' => 'true',
136 'chunksession' => $this->setupChunkSession( $comment, $pageText, $watch ) ) ) ) );
137 } else if ( $this->chunkMode
== self
::CHUNK
) {
138 $this->appendChunk();
139 if ( !$this->status
->isOK() ) {
140 return $this->status
;
143 // firefogg expects a specific result
144 // http://www.firefogg.org/dev/chunk_post.html
145 return Status
::newGood(
146 array( 'result' => 1, 'filesize' => $this->mFileSize
)
148 } else if ( $this->chunkMode
== self
::DONE
) {
150 $comment = $this->comment
;
153 $pageText = $this->pageText
;
156 $watch = $this->watch
;
158 $status = parent
::performUpload( $comment, $pageText, $watch, $user );
159 if ( !$status->isGood() ) {
162 $file = $this->getLocalFile();
164 // firefogg expects a specific result
165 // http://www.firefogg.org/dev/chunk_post.html
166 return Status
::newGood(
167 array('result' => 1, 'done' => 1, 'resultUrl' => $file->getDescriptionUrl() )
171 return Status
::newGood();
175 * Append a chunk to the Repo file
177 * @param string $srcPath Path to file to append from
178 * @param string $toAppendPath Path to file to append to
179 * @return Status Status
181 protected function appendToUploadFile( $srcPath, $toAppendPath ) {
182 $repo = RepoGroup
::singleton()->getLocalRepo();
183 $status = $repo->append( $srcPath, $toAppendPath );
188 * Append a chunk to the temporary file.
192 protected function appendChunk() {
193 global $wgMaxUploadSize;
195 if ( !$this->repoPath
) {
196 $this->status
= $this->saveTempUploadedFile( $this->mDesiredDestName
, $this->mTempPath
);
198 if ( $this->status
->isOK() ) {
199 $this->repoPath
= $this->status
->value
;
200 $_SESSION['wsUploadData'][$this->sessionKey
]['repoPath'] = $this->repoPath
;
204 if ( $this->getRealPath( $this->repoPath
) ) {
205 $this->status
= $this->appendToUploadFile( $this->repoPath
, $this->mTempPath
);
207 $this->status
= Status
::newFatal( 'filenotfound', $this->repoPath
);
209 if ( $this->mFileSize
> $wgMaxUploadSize )
210 $this->status
= Status
::newFatal( 'largefileserver' );
213 public function verifyUpload() {
214 if ( $this->chunkMode
!= self
::DONE
) {
215 return array('status' => UploadBase
::OK
);
217 return parent
::verifyUpload();
220 public function checkWarnings() {
221 if ( $this->chunkMode
!= self
::DONE
) {
224 return parent
::checkWarnings();
227 public function getImageInfo( $result ) {
228 if ( $this->chunkMode
!= self
::DONE
) {
231 return parent
::getImageInfo( $result );