VLCHTTPConnection.m 27.8 KB
Newer Older
1 2 3 4
/*****************************************************************************
 * VLCHTTPConnection.m
 * VLC for iOS
 *****************************************************************************
5
 * Copyright (c) 2013-2015 VideoLAN. All rights reserved.
6 7 8
 * $Id$
 *
 * Authors: Felix Paul Kühne <fkuehne # videolan.org>
Felix Paul Kühne's avatar
Felix Paul Kühne committed
9 10
 *          Pierre Sagaspe <pierre.sagaspe # me.com>
 *          Carola Nitz <caro # videolan.org>
11 12 13 14
 *          Jean-Baptiste Kempf <jb # videolan.org>
 *
 * Refer to the COPYING file of the official project for license.
 *****************************************************************************/
15

Felix Paul Kühne's avatar
Felix Paul Kühne committed
16
#import "VLCActivityManager.h"
17 18 19 20 21 22
#import "VLCHTTPConnection.h"
#import "MultipartFormDataParser.h"
#import "HTTPMessage.h"
#import "HTTPDataResponse.h"
#import "HTTPFileResponse.h"
#import "MultipartMessageHeaderField.h"
23
#import "HTTPDynamicFileResponse.h"
24
#import "VLCThumbnailsCache.h"
25
#import "NSString+SupportedMedia.h"
26
#import "UIDevice+VLC.h"
Felix Paul Kühne's avatar
Felix Paul Kühne committed
27
#import "VLCHTTPUploaderController.h"
28 29 30

@interface VLCHTTPConnection()
{
31 32 33
    MultipartFormDataParser *_parser;
    NSFileHandle *_storeFile;
    NSString *_filepath;
34 35
    UInt64 _contentLength;
    UInt64 _receivedContent;
36 37 38 39 40 41 42 43
}
@end

@implementation VLCHTTPConnection

- (BOOL)supportsMethod:(NSString *)method atPath:(NSString *)path
{
    // Add support for POST
Felix Paul Kühne's avatar
Felix Paul Kühne committed
44
    if ([method isEqualToString:@"POST"] && [path isEqualToString:@"/upload.json"])
45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
            return YES;

    return [super supportsMethod:method atPath:path];
}

- (BOOL)expectsRequestBodyFromMethod:(NSString *)method atPath:(NSString *)path
{
    // Inform HTTP server that we expect a body to accompany a POST request
    if ([method isEqualToString:@"POST"] && [path isEqualToString:@"/upload.json"]) {
        // here we need to make sure, boundary is set in header
        NSString* contentType = [request headerField:@"Content-Type"];
        NSUInteger paramsSeparator = [contentType rangeOfString:@";"].location;
        if (NSNotFound == paramsSeparator)
            return NO;

        if (paramsSeparator >= contentType.length - 1)
            return NO;

        NSString* type = [contentType substringToIndex:paramsSeparator];
        if (![type isEqualToString:@"multipart/form-data"]) {
            // we expect multipart/form-data content type
            return NO;
        }

        // enumerate all params in content-type, and find boundary there
        NSArray* params = [[contentType substringFromIndex:paramsSeparator + 1] componentsSeparatedByString:@";"];
Felix Paul Kühne's avatar
Felix Paul Kühne committed
71 72 73
        NSUInteger count = params.count;
        for (NSUInteger i = 0; i < count; i++) {
            NSString *param = params[i];
74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95
            paramsSeparator = [param rangeOfString:@"="].location;
            if ((NSNotFound == paramsSeparator) || paramsSeparator >= param.length - 1)
                continue;

            NSString* paramName = [param substringWithRange:NSMakeRange(1, paramsSeparator-1)];
            NSString* paramValue = [param substringFromIndex:paramsSeparator+1];

            if ([paramName isEqualToString: @"boundary"])
                // let's separate the boundary from content-type, to make it more handy to handle
                [request setHeaderField:@"boundary" value:paramValue];
        }
        // check if boundary specified
        if (nil == [request headerField:@"boundary"])
            return NO;

        return YES;
    }
    return [super expectsRequestBodyFromMethod:method atPath:path];
}

- (NSObject<HTTPResponse> *)httpResponseForMethod:(NSString *)method URI:(NSString *)path
{
Felix Paul Kühne's avatar
Felix Paul Kühne committed
96
    if ([method isEqualToString:@"POST"] && [path isEqualToString:@"/upload.json"])
97
        return [[HTTPDataResponse alloc] initWithData:[@"\"OK\"" dataUsingEncoding:NSUTF8StringEncoding]];
Felix Paul Kühne's avatar
Felix Paul Kühne committed
98

99 100
    if ([path hasPrefix:@"/download/"]) {
        NSString *filePath = [[path stringByReplacingOccurrencesOfString:@"/download/" withString:@""]stringByReplacingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
101 102 103
        HTTPFileResponse *fileResponse = [[HTTPFileResponse alloc] initWithFilePath:filePath forConnection:self];
        fileResponse.contentType = @"application/octet-stream";
        return fileResponse;
104
    }
105 106 107 108 109
    if ([path hasPrefix:@"/thumbnail"]) {
        NSString *filePath = [[path stringByReplacingOccurrencesOfString:@"/thumbnail/" withString:@""]stringByReplacingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
        filePath = [filePath stringByReplacingOccurrencesOfString:@".png" withString:@""];

        NSManagedObjectContext *moc = [[MLMediaLibrary sharedMediaLibrary] managedObjectContext];
110 111 112 113
        if (moc) {
            NSPersistentStoreCoordinator *psc = [moc persistentStoreCoordinator];
            if (psc) {
                NSManagedObject *mo = [moc existingObjectWithID:[psc managedObjectIDForURIRepresentation:[NSURL URLWithString:filePath]] error:nil];
Felix Paul Kühne's avatar
Felix Paul Kühne committed
114 115 116 117 118 119 120 121 122 123 124 125
                NSData *theData;
                NSString *contentType;

                /* devices category 3 and faster include HW accelerated JPEG encoding
                 * so we can make our transfers faster by using waaay smaller images */
                if ([[UIDevice currentDevice] speedCategory] < 3) {
                    theData = UIImagePNGRepresentation([VLCThumbnailsCache thumbnailForManagedObject:mo]);
                    contentType = @"image/png";
                } else {
                    theData = UIImageJPEGRepresentation([VLCThumbnailsCache thumbnailForManagedObject:mo], .9);
                    contentType = @"image/jpg";
                }
126 127 128

                if (theData) {
                    HTTPDataResponse *dataResponse = [[HTTPDataResponse alloc] initWithData:theData];
Felix Paul Kühne's avatar
Felix Paul Kühne committed
129
                    dataResponse.contentType = contentType;
130 131 132
                    return dataResponse;
                }
            }
133
        }
134
    }
135 136 137
    NSString *filePath = [self filePathForURI:path];
    NSString *documentRoot = [config documentRoot];
    NSString *relativePath = [filePath substringFromIndex:[documentRoot length]];
138
    BOOL shouldReturnLibVLCXML = [relativePath isEqualToString:@"/libMediaVLC.xml"];
139

140
    if ([relativePath isEqualToString:@"/index.html"] || shouldReturnLibVLCXML) {
141 142 143
        NSMutableArray *allMedia = [[NSMutableArray alloc] init];

        /* add all albums */
144 145
        NSArray *allAlbums = [MLAlbum allAlbums];
        for (MLAlbum *album in allAlbums) {
146 147
            if (album.name.length > 0 && album.tracks.count > 1)
                [allMedia addObject:album];
148
        }
149 150

        /* add all shows */
151 152
        NSArray *allShows = [MLShow allShows];
        for (MLShow *show in allShows) {
153 154
            if (show.name.length > 0 && show.episodes.count > 1)
                [allMedia addObject:show];
155
        }
156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177

        /* add all folders*/
        NSArray *allFolders = [MLLabel allLabels];
        for (MLLabel *folder in allFolders)
            [allMedia addObject:folder];

        /* add all remaining files */
        NSArray *allFiles = [MLFile allFiles];
        for (MLFile *file in allFiles) {
            if (file.labels.count > 0) continue;

            if (!file.isShowEpisode && !file.isAlbumTrack)
                [allMedia addObject:file];
            else if (file.isShowEpisode) {
                if (file.showEpisode.show.episodes.count < 2)
                    [allMedia addObject:file];
            } else if (file.isAlbumTrack) {
                if (file.albumTrack.album.tracks.count < 2)
                    [allMedia addObject:file];
            }
        }

Felix Paul Kühne's avatar
Felix Paul Kühne committed
178 179 180
        NSUInteger mediaCount = allMedia.count;
        NSMutableArray *mediaInHtml = [[NSMutableArray alloc] initWithCapacity:mediaCount];
        NSMutableArray *mediaInXml = [[NSMutableArray alloc] initWithCapacity:mediaCount];
Felix Paul Kühne's avatar
Felix Paul Kühne committed
181
        NSString *hostName = [[VLCHTTPUploaderController sharedInstance] hostname];
182
        NSString *duration;
183 184

        for (NSManagedObject *mo in allMedia) {
185
            if ([mo isKindOfClass:[MLFile class]]) {
186 187
                MLFile *file = (MLFile *)mo;
                duration = [[VLCTime timeWithNumber:file.duration] stringValue];
188 189
                [mediaInHtml addObject:[NSString stringWithFormat:
                                        @"<div style=\"background-image:url('thumbnail/%@.png')\"> \
190
                                        <a href=\"download/%@\" class=\"inner\"> \
191 192 193
                                        <div class=\"down icon\"></div> \
                                        <div class=\"infos\"> \
                                        <span class=\"first-line\">%@</span> \
194
                                        <span class=\"second-line\">%@ - %0.2f MB</span> \
195 196 197
                                        </div> \
                                        </a> \
                                        </div>",
198
                                        file.objectID.URIRepresentation,
199
                                        [file.url.path stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding],
200 201
                                        file.title,
                                        duration, (float)(file.fileSizeInBytes / 1e6)]];
202
                if (shouldReturnLibVLCXML) {
203
                    NSString *pathSub = [self _checkIfSubtitleWasFound:file.path];
Felix Paul Kühne's avatar
Felix Paul Kühne committed
204
                    if (pathSub)
205
                        pathSub = [NSString stringWithFormat:@"http://%@/download/%@", hostName, pathSub];
206
                    [mediaInXml addObject:[NSString stringWithFormat:@"<Media title=\"%@\" thumb=\"http://%@/thumbnail/%@.png\" duration=\"%@\" size=\"%li\" pathfile=\"http://%@/download/%@\" pathSubtitle=\"%@\"/>", file.title, hostName, file.objectID.URIRepresentation, duration, file.fileSizeInBytes, hostName, [file.url.path stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding], pathSub]];
207
                }
208
            }
209
            else if ([mo isKindOfClass:[MLShow class]]) {
210 211
                MLShow *show = (MLShow *)mo;
                NSArray *episodes = [show sortedEpisodes];
212 213
                [mediaInHtml addObject:[NSString stringWithFormat:
                                        @"<div style=\"background-image:url('thumbnail/%@.png')\"> \
214
                                        <a href=\"#\" class=\"inner folder\"> \
215 216 217
                                        <div class=\"open icon\"></div> \
                                        <div class=\"infos\"> \
                                        <span class=\"first-line\">%@</span> \
Felix Paul Kühne's avatar
Felix Paul Kühne committed
218
                                        <span class=\"second-line\">%lu items</span> \
219 220 221 222
                                        </div> \
                                        </a> \
                                        <div class=\"content\">",
                                        mo.objectID.URIRepresentation,
223
                                        show.name,
Felix Paul Kühne's avatar
Felix Paul Kühne committed
224
                                        (unsigned long)[episodes count]]];
225
                for (MLShowEpisode *showEp in episodes) {
Felix Paul Kühne's avatar
Felix Paul Kühne committed
226 227
                    MLFile *anyFileFromEpisode = (MLFile *)[[showEp files] anyObject];
                    duration = [[VLCTime timeWithNumber:[anyFileFromEpisode duration]] stringValue];
228 229
                    [mediaInHtml addObject:[NSString stringWithFormat:
                                            @"<div style=\"background-image:url('thumbnail/%@.png')\"> \
230
                                            <a href=\"download/%@\" class=\"inner\"> \
231 232 233
                                            <div class=\"down icon\"></div> \
                                            <div class=\"infos\"> \
                                            <span class=\"first-line\">S%@E%@ - %@</span> \
234
                                            <span class=\"second-line\">%@ - %0.2f MB</span> \
235 236 237 238
                                            </div> \
                                            </a> \
                                            </div>",
                                            showEp.objectID.URIRepresentation,
239
                                            [anyFileFromEpisode.url.path stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding],
240 241
                                            showEp.seasonNumber,
                                            showEp.episodeNumber,
242
                                            showEp.name,
Felix Paul Kühne's avatar
Felix Paul Kühne committed
243
                                            duration, (float)([anyFileFromEpisode fileSizeInBytes] / 1e6)]];
244
                    if (shouldReturnLibVLCXML) {
245
                        NSString *pathSub = [self _checkIfSubtitleWasFound:[anyFileFromEpisode path]];
246 247
                        if (![pathSub isEqualToString:@""])
                            pathSub = [NSString stringWithFormat:@"http://%@/download/%@", hostName, pathSub];
248
                        [mediaInXml addObject:[NSString stringWithFormat:@"<Media title=\"%@ - S%@E%@\" thumb=\"http://%@/thumbnail/%@.png\" duration=\"%@\" size=\"%li\" pathfile=\"http://%@/download/%@\" pathSubtitle=\"%@\"/>", show.name, showEp.seasonNumber, showEp.episodeNumber, hostName, showEp.objectID.URIRepresentation, duration, [anyFileFromEpisode fileSizeInBytes], hostName, [anyFileFromEpisode.url.path stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding], pathSub]];
249
                    }
250
                }
251
                [mediaInHtml addObject:@"</div></div>"];
252
            } else if ([mo isKindOfClass:[MLLabel class]]) {
253 254
                MLLabel *label = (MLLabel *)mo;
                NSArray *folderItems = [label sortedFolderItems];
255 256
                [mediaInHtml addObject:[NSString stringWithFormat:
                                        @"<div style=\"background-image:url('thumbnail/%@.png')\"> \
257
                                        <a href=\"#\" class=\"inner folder\"> \
258 259 260
                                        <div class=\"open icon\"></div> \
                                        <div class=\"infos\"> \
                                        <span class=\"first-line\">%@</span> \
Felix Paul Kühne's avatar
Felix Paul Kühne committed
261
                                        <span class=\"second-line\">%lu items</span> \
262 263 264
                                        </div> \
                                        </a> \
                                        <div class=\"content\">",
265 266 267
                                        label.objectID.URIRepresentation,
                                        label.name,
                                        (unsigned long)folderItems.count]];
268
                for (MLFile *file in folderItems) {
269
                    duration = [[VLCTime timeWithNumber:[file duration]] stringValue];
270 271
                    [mediaInHtml addObject:[NSString stringWithFormat:
                                            @"<div style=\"background-image:url('thumbnail/%@.png')\"> \
272
                                            <a href=\"download/%@\" class=\"inner\"> \
273 274 275
                                            <div class=\"down icon\"></div> \
                                            <div class=\"infos\"> \
                                            <span class=\"first-line\">%@</span> \
276
                                            <span class=\"second-line\">%@ - %0.2f MB</span> \
277 278 279 280
                                            </div> \
                                            </a> \
                                            </div>",
                                            file.objectID.URIRepresentation,
281
                                            [file.url.path stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding],
282
                                            file.title,
283
                                            duration, (float)(file.fileSizeInBytes / 1e6)]];
284
                    if (shouldReturnLibVLCXML) {
285
                        NSString *pathSub = [self _checkIfSubtitleWasFound:file.path];
Felix Paul Kühne's avatar
Felix Paul Kühne committed
286
                        if (pathSub)
287
                            pathSub = [NSString stringWithFormat:@"http://%@/download/%@", hostName, pathSub];
288
                        [mediaInXml addObject:[NSString stringWithFormat:@"<Media title=\"%@\" thumb=\"http://%@/thumbnail/%@.png\" duration=\"%@\" size=\"%li\" pathfile=\"http://%@/download/%@\" pathSubtitle=\"%@\"/>", file.title, hostName, file.objectID.URIRepresentation, duration, file.fileSizeInBytes, hostName, [file.url.path stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding], pathSub]];
289
                    }
290
                }
291
                [mediaInHtml addObject:@"</div></div>"];
292
            } else if ([mo isKindOfClass:[MLAlbum class]]) {
293 294
                MLAlbum *album = (MLAlbum *)mo;
                NSArray *albumTracks = [album sortedTracks];
295 296
                [mediaInHtml addObject:[NSString stringWithFormat:
                                        @"<div style=\"background-image:url('thumbnail/%@.png')\"> \
297
                                        <a href=\"#\" class=\"inner folder\"> \
298 299 300
                                        <div class=\"open icon\"></div> \
                                        <div class=\"infos\"> \
                                        <span class=\"first-line\">%@</span> \
Felix Paul Kühne's avatar
Felix Paul Kühne committed
301
                                        <span class=\"second-line\">%lu items</span> \
302 303 304
                                        </div> \
                                        </a> \
                                        <div class=\"content\">",
305 306 307
                                        album.objectID.URIRepresentation,
                                        album.name,
                                        (unsigned long)albumTracks.count]];
308
                for (MLAlbumTrack *track in albumTracks) {
309
                    MLFile *anyFileFromTrack = [track anyFileFromTrack];
Felix Paul Kühne's avatar
Felix Paul Kühne committed
310
                    duration = [[VLCTime timeWithNumber:[anyFileFromTrack duration]] stringValue];
311 312
                    [mediaInHtml addObject:[NSString stringWithFormat:
                                            @"<div style=\"background-image:url('thumbnail/%@.png')\"> \
313
                                            <a href=\"download/%@\" class=\"inner\"> \
314 315 316
                                            <div class=\"down icon\"></div> \
                                            <div class=\"infos\"> \
                                            <span class=\"first-line\">%@</span> \
317
                                            <span class=\"second-line\">%@ - %0.2f MB</span> \
318 319 320 321
                                            </div> \
                                            </a> \
                                            </div>",
                                            track.objectID.URIRepresentation,
322
                                            [anyFileFromTrack.url.path stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding],
323
                                            track.title,
Felix Paul Kühne's avatar
Felix Paul Kühne committed
324
                                            duration, (float)([anyFileFromTrack fileSizeInBytes] / 1e6)]];
325
                    if (shouldReturnLibVLCXML)
326
                        [mediaInXml addObject:[NSString stringWithFormat:@"<Media title=\"%@\" thumb=\"http://%@/thumbnail/%@.png\" duration=\"%@\" size=\"%li\" pathfile=\"http://%@/download/%@\" pathSubtitle=\"\"/>", track.title, hostName, track.objectID.URIRepresentation, duration, [anyFileFromTrack fileSizeInBytes], hostName, [anyFileFromTrack.url.path stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding]]];
327
                }
328
                [mediaInHtml addObject:@"</div></div>"];
329
            }
330
        }
331

Felix Paul Kühne's avatar
Felix Paul Kühne committed
332 333
        UIDevice *currentDevice = [UIDevice currentDevice];
        NSString *deviceModel = [currentDevice model];
334
        NSDictionary *replacementDict;
335
        HTTPDynamicFileResponse *fileResponse;
336

337
        if (shouldReturnLibVLCXML) {
338 339
            replacementDict = @{@"FILES" : [mediaInXml componentsJoinedByString:@" "],
                                @"NB_FILE" : [NSString stringWithFormat:@"%li", (unsigned long)mediaInXml.count],
Felix Paul Kühne's avatar
Felix Paul Kühne committed
340
                                @"LIB_TITLE" : [currentDevice name]};
341 342 343 344 345 346 347

            fileResponse = [[HTTPDynamicFileResponse alloc] initWithFilePath:[self filePathForURI:path]
                                                               forConnection:self
                                                                   separator:@"%%"
                                                       replacementDictionary:replacementDict];
            fileResponse.contentType = @"application/xml";
        } else {
348 349 350 351 352 353
            replacementDict = @{@"FILES" : [mediaInHtml componentsJoinedByString:@" "],
                                @"WEBINTF_TITLE" : NSLocalizedString(@"WEBINTF_TITLE", nil),
                                @"WEBINTF_DROPFILES" : NSLocalizedString(@"WEBINTF_DROPFILES", nil),
                                @"WEBINTF_DROPFILES_LONG" : [NSString stringWithFormat:NSLocalizedString(@"WEBINTF_DROPFILES_LONG", nil), deviceModel],
                                @"WEBINTF_DOWNLOADFILES" : NSLocalizedString(@"WEBINTF_DOWNLOADFILES", nil),
                                @"WEBINTF_DOWNLOADFILES_LONG" : [NSString stringWithFormat: NSLocalizedString(@"WEBINTF_DOWNLOADFILES_LONG", nil), deviceModel]};
354 355 356 357 358 359
            fileResponse = [[HTTPDynamicFileResponse alloc] initWithFilePath:[self filePathForURI:path]
                                                               forConnection:self
                                                                   separator:@"%%"
                                                       replacementDictionary:replacementDict];
            fileResponse.contentType = @"text/html";
        }
360

361
        return fileResponse;
362
    } else if ([relativePath isEqualToString:@"/style.css"]) {
363
        NSDictionary *replacementDict = @{@"WEBINTF_TITLE" : NSLocalizedString(@"WEBINTF_TITLE", nil)};
364 365 366 367 368 369
        HTTPDynamicFileResponse *fileResponse = [[HTTPDynamicFileResponse alloc] initWithFilePath:[self filePathForURI:path]
                                                                                    forConnection:self
                                                                                        separator:@"%%"
                                                                            replacementDictionary:replacementDict];
        fileResponse.contentType = @"text/css";
        return fileResponse;
370
    }
371

372 373 374 375 376 377 378 379 380 381
    return [super httpResponseForMethod:method URI:path];
}

- (void)prepareForBodyWithSize:(UInt64)contentLength
{
    // set up mime parser
    NSString* boundary = [request headerField:@"boundary"];
    _parser = [[MultipartFormDataParser alloc] initWithBoundary:boundary formEncoding:NSUTF8StringEncoding];
    _parser.delegate = self;

382 383
    APLog(@"expecting file of size %lli kB", contentLength / 1024);
    _contentLength = contentLength;
384 385 386 387 388 389 390
}

- (void)processBodyData:(NSData *)postDataChunk
{
    /* append data to the parser. It will invoke callbacks to let us handle
     * parsed data. */
    [_parser appendData:postDataChunk];
391

392
    _receivedContent += postDataChunk.length;
393

394
    APLog(@"received %lli kB (%lli %%)", _receivedContent / 1024, ((_receivedContent * 100) / _contentLength));
395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414
}

//-----------------------------------------------------------------
#pragma mark multipart form data parser delegate


- (void)processStartOfPartWithHeader:(MultipartMessageHeader*) header
{
    /* in this sample, we are not interested in parts, other then file parts.
     * check content disposition to find out filename */

    MultipartMessageHeaderField* disposition = (header.fields)[@"Content-Disposition"];
    NSString* filename = [(disposition.params)[@"filename"] lastPathComponent];

    if ((nil == filename) || [filename isEqualToString: @""]) {
        // it's either not a file part, or
        // an empty form sent. we won't handle it.
        return;
    }

415
    // create the path where to store the media temporarily
416
    NSArray *searchPaths = NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES);
Felix Paul Kühne's avatar
Felix Paul Kühne committed
417
    NSString *uploadDirPath = [searchPaths[0] stringByAppendingPathComponent:@"Upload"];
418
    NSFileManager *fileManager = [NSFileManager defaultManager];
419 420

    BOOL isDir = YES;
Felix Paul Kühne's avatar
Felix Paul Kühne committed
421
    if (![fileManager fileExistsAtPath:uploadDirPath isDirectory:&isDir])
422
        [fileManager createDirectoryAtPath:uploadDirPath withIntermediateDirectories:YES attributes:nil error:nil];
423

424
    _filepath = [uploadDirPath stringByAppendingPathComponent: filename];
425

426 427 428 429 430 431 432 433 434
    NSNumber *freeSpace = [[UIDevice currentDevice] freeDiskspace];
    if (_contentLength >= freeSpace.longLongValue) {
        /* avoid deadlock since we are on a background thread */
        [self performSelectorOnMainThread:@selector(notifyUserAboutEndOfFreeStorage:) withObject:filename waitUntilDone:NO];
        [self handleResourceNotFound];
        [self stop];
        return;
    }

435
    APLog(@"Saving file to %@", _filepath);
436
    if (![fileManager createDirectoryAtPath:uploadDirPath withIntermediateDirectories:true attributes:nil error:nil])
437
        APLog(@"Could not create directory at path: %@", _filepath);
438

439
    if (![fileManager createFileAtPath:_filepath contents:nil attributes:nil])
440
        APLog(@"Could not create file at path: %@", _filepath);
441

442
    _storeFile = [NSFileHandle fileHandleForWritingAtPath:_filepath];
Felix Paul Kühne's avatar
Felix Paul Kühne committed
443 444 445 446

    VLCActivityManager *activityManager = [VLCActivityManager defaultManager];
    [activityManager networkActivityStarted];
    [activityManager disableIdleTimer];
447 448
}

449 450
- (void)notifyUserAboutEndOfFreeStorage:(NSString *)filename
{
451 452 453 454 455 456 457 458
    VLCAlertView *alert = [[VLCAlertView alloc] initWithTitle:NSLocalizedString(@"DISK_FULL", nil)
                                                      message:[NSString stringWithFormat:
                                                               NSLocalizedString(@"DISK_FULL_FORMAT", nil),
                                                               filename,
                                                               [[UIDevice currentDevice] model]]
                                                     delegate:self
                                            cancelButtonTitle:NSLocalizedString(@"BUTTON_OK", nil)
                                            otherButtonTitles:nil];
459 460 461
    [alert show];
}

462 463 464
- (void)processContent:(NSData*)data WithHeader:(MultipartMessageHeader*) header
{
    // here we just write the output from parser to the file.
465 466 467 468 469 470 471 472 473 474 475 476 477
    if (_storeFile) {
        @try {
            [_storeFile writeData:data];
        }
        @catch (NSException *exception) {
            APLog(@"File to write further data because storage is full.");
            [_storeFile closeFile];
            _storeFile = nil;
            /* don't block */
            [self performSelector:@selector(stop) withObject:nil afterDelay:0.1];
        }
    }

478 479 480 481 482
}

- (void)processEndOfPartWithHeader:(MultipartMessageHeader*)header
{
    // as the file part is over, we close the file.
483
    APLog(@"closing file");
484 485 486 487
    [_storeFile closeFile];
    _storeFile = nil;
}

488
- (BOOL)shouldDie
489
{
490 491
    if (_filepath) {
        if (_filepath.length > 0)
Felix Paul Kühne's avatar
Felix Paul Kühne committed
492
            [[VLCHTTPUploaderController sharedInstance] moveFileFrom:_filepath];
493
    }
494
    return [super shouldDie];
495 496
}

497 498
#pragma mark subtitle

Felix Paul Kühne's avatar
Felix Paul Kühne committed
499
- (NSMutableArray *)_listOfSubtitles
500
{
Felix Paul Kühne's avatar
Felix Paul Kühne committed
501
    NSMutableArray *listOfSubtitles = [[NSMutableArray alloc] init];
502
    NSString *documentsDirectory = [NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES) objectAtIndex:0];
Felix Paul Kühne's avatar
Felix Paul Kühne committed
503
    NSArray *allFiles = [[NSFileManager defaultManager] contentsOfDirectoryAtPath:documentsDirectory error:nil];
504
    NSString *filePath;
Felix Paul Kühne's avatar
Felix Paul Kühne committed
505 506 507
    NSUInteger count = allFiles.count;
    for (NSUInteger i = 0; i < count; i++) {
        filePath = [[NSString stringWithFormat:@"%@/%@", documentsDirectory, allFiles[i]] stringByReplacingOccurrencesOfString:@"file://"withString:@""];
508
        if ([filePath isSupportedSubtitleFormat])
Felix Paul Kühne's avatar
Felix Paul Kühne committed
509
            [listOfSubtitles addObject:filePath];
510
    }
Felix Paul Kühne's avatar
Felix Paul Kühne committed
511
    return listOfSubtitles;
512 513
}

514
- (NSString *)_checkIfSubtitleWasFound:(NSString *)filePath
515
{
Felix Paul Kühne's avatar
Felix Paul Kühne committed
516
    NSString *subtitlePath;
517
    NSString *fileName = [[filePath lastPathComponent] stringByDeletingPathExtension];
Felix Paul Kühne's avatar
Felix Paul Kühne committed
518
    NSMutableArray *listOfSubtitles = [self _listOfSubtitles];
519
    NSString *fileSub;
Felix Paul Kühne's avatar
Felix Paul Kühne committed
520 521 522 523 524
    NSUInteger count = listOfSubtitles.count;
    NSString *currentPath;
    for (NSUInteger i = 0; i < count; i++) {
        currentPath = listOfSubtitles[i];
        fileSub = [NSString stringWithFormat:@"%@", currentPath];
525
        if ([fileSub rangeOfString:fileName].location != NSNotFound)
Felix Paul Kühne's avatar
Felix Paul Kühne committed
526
            subtitlePath = currentPath;
527 528 529 530
    }
    return subtitlePath;
}

531
@end