MLMediaLibrary.m 34.3 KB
Newer Older
1 2 3 4 5
/*****************************************************************************
 * MLMediaLibrary.m
 * MobileMediaLibraryKit
 *****************************************************************************
 * Copyright (C) 2010 Pierre d'Herbemont
6
 * Copyright (C) 2010-2014 VLC authors and VideoLAN
7 8 9 10 11
 * $Id$
 *
 * Authors: Pierre d'Herbemont <pdherbemont # videolan.org>
 *          Felix Paul Kühne <fkuehne # videolan.org>
 *
12 13 14
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or
15 16 17 18
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19 20
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Lesser General Public License for more details.
21
 *
22 23 24
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
25
 *****************************************************************************/
Pierre's avatar
Pierre committed
26 27

#import "MLMediaLibrary.h"
Pierre's avatar
Pierre committed
28 29
#import "MLTitleDecrapifier.h"
#import "MLMovieInfoGrabber.h"
Pierre's avatar
Pierre committed
30 31
#import "MLTVShowInfoGrabber.h"
#import "MLTVShowEpisodesInfoGrabber.h"
Pierre's avatar
Pierre committed
32 33 34 35
#import "MLFile.h"
#import "MLLabel.h"
#import "MLShowEpisode.h"
#import "MLShow.h"
36 37 38
#import "MLThumbnailerQueue.h"
#import "MLAlbumTrack.h"
#import "MLAlbum.h"
39
#import "MLFileParserQueue.h"
40
#import "MLCrashPreventer.h"
Pierre's avatar
Pierre committed
41

42 43
@interface MLMediaLibrary ()
{
44 45 46 47 48
    NSManagedObjectContext *_managedObjectContext;
    NSManagedObjectModel   *_managedObjectModel;

    BOOL _allowNetworkAccess;

49 50
    NSString *_thumbnailFolderPath;
    NSString *_databaseFolderPath;
51
    NSString *_documentFolderPath;
52 53 54
}
@end

Pierre's avatar
Pierre committed
55
#define DEBUG 1
56 57
// To debug
#define DELETE_LIBRARY_ON_EACH_LAUNCH 0
Pierre's avatar
Pierre committed
58

Pierre's avatar
Pierre committed
59 60
// Pref key
static NSString *kLastTVDBUpdateServerTime = @"MLLastTVDBUpdateServerTime";
61
static NSString *kUpdatedToTheGreatSharkHuntDatabaseFormat = @"upgradedToDatabaseFormat 2.3";
62
static NSString *kDecrapifyTitles = @"MLDecrapifyTitles";
Pierre's avatar
Pierre committed
63

Pierre's avatar
Pierre committed
64
#if HAVE_BLOCK
Pierre's avatar
Pierre committed
65
@interface MLMediaLibrary ()
Pierre's avatar
Pierre committed
66 67 68
#else
@interface MLMediaLibrary () <MLMovieInfoGrabberDelegate, MLTVShowEpisodesInfoGrabberDelegate, MLTVShowInfoGrabberDelegate>
#endif
Pierre's avatar
Pierre committed
69
- (NSManagedObjectContext *)managedObjectContext;
Pierre's avatar
Pierre committed
70
- (NSString *)databaseFolderPath;
Pierre's avatar
Pierre committed
71 72
@end

Pierre's avatar
Pierre committed
73
@implementation MLMediaLibrary
74 75
+ (void)initialize
{
76
    NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
Felix Paul Kühne's avatar
Felix Paul Kühne committed
77
    [defaults registerDefaults:@{kUpdatedToTheGreatSharkHuntDatabaseFormat : @NO, kDecrapifyTitles : @YES}];
78 79
}

Pierre's avatar
Pierre committed
80 81 82
+ (id)sharedMediaLibrary
{
    static id sharedMediaLibrary = nil;
Pierre's avatar
Pierre committed
83
    if (!sharedMediaLibrary) {
Pierre's avatar
Pierre committed
84
        sharedMediaLibrary = [[[self class] alloc] init];
85
        APLog(@"Initializing db in %@", [sharedMediaLibrary databaseFolderPath]);
86 87 88 89

        // Also force to init the crash preventer
        // Because it will correctly set up the parser and thumbnail queue
        [MLCrashPreventer sharedPreventer];
Pierre's avatar
Pierre committed
90
    }
Pierre's avatar
Pierre committed
91 92 93
    return sharedMediaLibrary;
}

94 95 96 97 98 99
- (void)dealloc
{
    if (_managedObjectContext)
        [_managedObjectContext removeObserver:self forKeyPath:@"hasChanges"];
}

Pierre's avatar
Pierre committed
100 101 102 103 104
- (NSFetchRequest *)fetchRequestForEntity:(NSString *)entity
{
    NSFetchRequest *request = [[NSFetchRequest alloc] init];
    NSManagedObjectContext *moc = [self managedObjectContext];
    NSEntityDescription *entityDescription = [NSEntityDescription entityForName:entity inManagedObjectContext:moc];
Pierre's avatar
Pierre committed
105
    NSAssert(entityDescription, @"No entity");
Pierre's avatar
Pierre committed
106
    [request setEntity:entityDescription];
107
    return request;
Pierre's avatar
Pierre committed
108 109 110 111 112 113 114 115
}

- (id)createObjectForEntity:(NSString *)entity
{
    NSManagedObjectContext *moc = [self managedObjectContext];
    return [NSEntityDescription insertNewObjectForEntityForName:entity inManagedObjectContext:moc];
}

116 117 118 119 120
- (void)removeObject:(NSManagedObject *)object
{
    [[self managedObjectContext] deleteObject:object];
}

Pierre's avatar
Pierre committed
121 122 123 124 125 126
#pragma mark -
#pragma mark Media Library
- (NSManagedObjectModel *)managedObjectModel
{
    if (_managedObjectModel)
        return _managedObjectModel;
127
    _managedObjectModel = [NSManagedObjectModel mergedModelFromBundles:nil];
Pierre's avatar
Pierre committed
128 129 130
    return _managedObjectModel;
}

Pierre's avatar
Pierre committed
131
- (NSString *)databaseFolderPath
Pierre's avatar
Pierre committed
132
{
133 134 135 136
    if (_databaseFolderPath) {
        if (_databaseFolderPath.length > 0)
            return _databaseFolderPath;
    }
Pierre's avatar
Pierre committed
137 138
    int directory = NSLibraryDirectory;
    NSArray *paths = NSSearchPathForDirectoriesInDomains(directory, NSUserDomainMask, YES);
139
    NSString *directoryPath = paths[0];
140
#if DELETE_LIBRARY_ON_EACH_LAUNCH
Pierre's avatar
Pierre committed
141
    [[NSFileManager defaultManager] removeItemAtPath:directoryPath error:nil];
142
#endif
143 144
    _databaseFolderPath = directoryPath;
    return _databaseFolderPath;
Pierre's avatar
Pierre committed
145 146
}

147 148
- (NSString *)thumbnailFolderPath
{
149 150 151 152
    if (_thumbnailFolderPath) {
        if (_thumbnailFolderPath.length > 0)
            return _thumbnailFolderPath;
    }
153 154
    int directory = NSLibraryDirectory;
    NSArray *paths = NSSearchPathForDirectoriesInDomains(directory, NSUserDomainMask, YES);
155
    NSString *directoryPath = paths[0];
156 157 158
#if DELETE_LIBRARY_ON_EACH_LAUNCH
    [[NSFileManager defaultManager] removeItemAtPath:directoryPath error:nil];
#endif
159 160
    _thumbnailFolderPath = [directoryPath stringByAppendingPathComponent:@"Thumbnails"];
    return _thumbnailFolderPath;
161 162
}

163 164 165 166 167 168 169 170 171
- (NSString *)documentFolderPath
{
    if (_documentFolderPath) {
        if (_documentFolderPath.length > 0)
            return _documentFolderPath;
    }
    int directory = NSDocumentDirectory;
    NSArray *paths = NSSearchPathForDirectoriesInDomains(directory, NSUserDomainMask, YES);

172
    _documentFolderPath = [NSString stringWithFormat:@"file://%@", paths[0]];
173 174 175
    return _documentFolderPath;
}

Pierre's avatar
Pierre committed
176 177 178 179 180
- (NSManagedObjectContext *)managedObjectContext
{
    if (_managedObjectContext)
        return _managedObjectContext;

Pierre's avatar
Pierre committed
181
    NSString *databaseFolderPath = [self databaseFolderPath];
Pierre's avatar
Pierre committed
182

Pierre's avatar
Pierre committed
183 184
    NSString *path = [databaseFolderPath stringByAppendingPathComponent: @"MediaLibrary.sqlite"];
    NSURL *url = [NSURL fileURLWithPath:path];
Pierre's avatar
Pierre committed
185 186
    NSPersistentStoreCoordinator *coordinator = [[NSPersistentStoreCoordinator alloc] initWithManagedObjectModel:[self managedObjectModel]];

187
    NSNumber *yes = @YES;
188 189
    NSDictionary *options = @{NSMigratePersistentStoresAutomaticallyOption : yes,
                             NSInferMappingModelAutomaticallyOption : yes};
Pierre's avatar
Pierre committed
190 191

    NSError *error;
Pierre's avatar
Pierre committed
192 193 194
    NSPersistentStore *persistentStore = [coordinator addPersistentStoreWithType:NSSQLiteStoreType configuration:nil URL:url options:options error:&error];

    if (!persistentStore) {
Pierre's avatar
Pierre committed
195
#if! TARGET_OS_IPHONE
Pierre's avatar
Pierre committed
196 197 198 199
        // FIXME: Deal with versioning
        NSInteger ret = NSRunAlertPanel(@"Error", @"The Media Library you have on your disk is not compatible with the one Lunettes can read. Do you want to create a new one?", @"No", @"Yes", nil);
        if (ret == NSOKButton)
            [NSApp terminate:nil];
Pierre's avatar
Pierre committed
200
        [[NSFileManager defaultManager] removeItemAtPath:path error:nil];
Pierre's avatar
Pierre committed
201
#else
Pierre's avatar
Pierre committed
202
        [[NSFileManager defaultManager] removeItemAtPath:path error:nil];
Pierre's avatar
Pierre committed
203
#endif
Pierre's avatar
Pierre committed
204 205 206 207 208 209 210 211 212 213 214
        persistentStore = [coordinator addPersistentStoreWithType:NSSQLiteStoreType configuration:nil URL:url options:options error:&error];
        if (!persistentStore) {
#if! TARGET_OS_IPHONE
            NSRunInformationalAlertPanel(@"Corrupted Media Library", @"There is nothing we can apparently do about it...", @"OK", nil, nil);
#else
            UIAlertView *alert = [[UIAlertView alloc] initWithTitle:@"Corrupted Media Library" message:@"There is nothing we can apparently do about it..." delegate:nil cancelButtonTitle:@"OK" otherButtonTitles:nil];
            [alert show];
#endif
            // Probably assert instead.
            return nil;
        }
Pierre's avatar
Pierre committed
215
    }
Pierre's avatar
Pierre committed
216 217 218

    _managedObjectContext = [[NSManagedObjectContext alloc] init];
    [_managedObjectContext setPersistentStoreCoordinator:coordinator];
Pierre's avatar
Pierre committed
219 220 221 222 223 224 225 226
    [_managedObjectContext setUndoManager:nil];
    [_managedObjectContext addObserver:self forKeyPath:@"hasChanges" options:NSKeyValueObservingOptionInitial context:nil];
    return _managedObjectContext;
}

- (void)savePendingChanges
{
    [NSObject cancelPreviousPerformRequestsWithTarget:self selector:@selector(savePendingChanges) object:nil];
Pierre's avatar
Pierre committed
227 228 229
    NSError *error = nil;
    BOOL success = [[self managedObjectContext] save:&error];
    NSAssert1(success, @"Can't save: %@", error);
Pierre's avatar
Pierre committed
230
#if !TARGET_OS_IPHONE && MAC_OS_X_VERSION_MAX_ALLOWED > MAC_OS_X_VERSION_10_5
Pierre's avatar
Pierre committed
231 232 233 234 235 236
    NSProcessInfo *process = [NSProcessInfo processInfo];
    if ([process respondsToSelector:@selector(enableSuddenTermination)])
        [process enableSuddenTermination];
#endif
}

237 238 239 240 241 242 243
- (void)save
{
    NSError *error = nil;
    BOOL success = [[self managedObjectContext] save:&error];
    NSAssert1(success, @"Can't save: %@", error);
}

Pierre's avatar
Pierre committed
244
- (void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void *)context
Pierre's avatar
Pierre committed
245 246
{
    if ([keyPath isEqualToString:@"hasChanges"] && object == _managedObjectContext) {
Pierre's avatar
Pierre committed
247
#if !TARGET_OS_IPHONE && MAC_OS_X_VERSION_MAX_ALLOWED > MAC_OS_X_VERSION_10_5
Pierre's avatar
Pierre committed
248 249 250 251 252
        NSProcessInfo *process = [NSProcessInfo processInfo];
        if ([process respondsToSelector:@selector(disableSuddenTermination)])
            [process disableSuddenTermination];
#endif

Pierre's avatar
Pierre committed
253 254 255 256
        if ([[self managedObjectContext] hasChanges]) {
            [NSObject cancelPreviousPerformRequestsWithTarget:self selector:@selector(savePendingChanges) object:nil];
            [self performSelector:@selector(savePendingChanges) withObject:nil afterDelay:1.];
        }
Pierre's avatar
Pierre committed
257 258 259 260 261
        return;
    }
    [super observeValueForKeyPath:keyPath ofObject:object change:change context:context];
}

Pierre's avatar
Pierre committed
262 263
#pragma mark -
#pragma mark No meta data fallbacks
Pierre's avatar
Pierre committed
264

Pierre's avatar
Pierre committed
265
- (void)computeThumbnailForFile:(MLFile *)file
Pierre's avatar
Pierre committed
266
{
Pierre's avatar
Pierre committed
267
    if (!file.computedThumbnail) {
268
        APLog(@"Computing thumbnail for %@", file.title);
Pierre's avatar
Pierre committed
269 270
        [[MLThumbnailerQueue sharedThumbnailerQueue] addFile:file];
    }
Pierre's avatar
Pierre committed
271
}
Pierre's avatar
Pierre committed
272
- (void)errorWhenFetchingMetaDataForFile:(MLFile *)file
Pierre's avatar
Pierre committed
273
{
274
    APLog(@"Error when fetching for '%@'", file.title);
Pierre's avatar
Pierre committed
275

Pierre's avatar
Pierre committed
276 277
    [self computeThumbnailForFile:file];
}
Pierre's avatar
Pierre committed
278

Pierre's avatar
Pierre committed
279 280 281 282 283 284 285
- (void)errorWhenFetchingMetaDataForShow:(MLShow *)show
{
    for (MLShowEpisode *episode in show.episodes) {
        for (MLFile *file in episode.files)
            [self errorWhenFetchingMetaDataForFile:file];
    }
}
Pierre's avatar
Pierre committed
286

Pierre's avatar
Pierre committed
287 288
- (void)noMetaDataInRemoteDBForFile:(MLFile *)file
{
289
    file.noOnlineMetaData = @YES;
Pierre's avatar
Pierre committed
290
    [self computeThumbnailForFile:file];
Pierre's avatar
Pierre committed
291 292
}

Pierre's avatar
Pierre committed
293
- (void)noMetaDataInRemoteDBForShow:(MLShow *)show
Pierre's avatar
Pierre committed
294
{
Pierre's avatar
Pierre committed
295 296 297
    for (MLShowEpisode *episode in show.episodes) {
        for (MLFile *file in episode.files)
            [self noMetaDataInRemoteDBForFile:file];
Pierre's avatar
Pierre committed
298 299 300
    }
}

Pierre's avatar
Pierre committed
301 302 303 304
#pragma mark -
#pragma mark Getter

- (void)addNewLabelWithName:(NSString *)name
Pierre's avatar
Pierre committed
305
{
Pierre's avatar
Pierre committed
306 307
    MLLabel *label = [self createObjectForEntity:@"Label"];
    label.name = name;
Pierre's avatar
Pierre committed
308 309
}

Pierre's avatar
Pierre committed
310 311 312 313 314 315 316
/**
 * TV MLShow Episodes
 */

#pragma mark -
#pragma mark Online meta data grabbing

Pierre's avatar
Pierre committed
317 318 319
#if !HAVE_BLOCK
- (void)tvShowEpisodesInfoGrabberDidFinishGrabbing:(MLTVShowEpisodesInfoGrabber *)grabber
{
Pierre's avatar
Pierre committed
320
    MLShow *show = grabber.userData;
Pierre's avatar
Pierre committed
321 322

    NSArray *results = grabber.episodesResults;
323
    [show setValue:(grabber.results)[@"serieArtworkURL"] forKey:@"artworkURL"];
Pierre's avatar
Pierre committed
324
    for (id result in results) {
325
        if ([result[@"serie"] boolValue]) {
Pierre's avatar
Pierre committed
326 327
            continue;
        }
328 329 330 331 332
        MLShowEpisode *showEpisode = [MLShowEpisode episodeWithShow:show episodeNumber:result[@"episodeNumber"] seasonNumber:result[@"seasonNumber"] createIfNeeded:YES];
        showEpisode.name = result[@"title"];
        showEpisode.theTVDBID = result[@"id"];
        showEpisode.shortSummary = result[@"shortSummary"];
        showEpisode.artworkURL = result[@"artworkURL"];
Pierre's avatar
Pierre committed
333 334 335 336 337
        if (!showEpisode.artworkURL) {
            for (MLFile *file in showEpisode.files)
                [self computeThumbnailForFile:file];
        }

Pierre's avatar
Pierre committed
338 339 340 341 342
        showEpisode.lastSyncDate = [MLTVShowInfoGrabber serverTime];
    }
    show.lastSyncDate = [MLTVShowInfoGrabber serverTime];
}

Pierre's avatar
Pierre committed
343 344 345 346 347 348
- (void)tvShowEpisodesInfoGrabber:(MLTVShowEpisodesInfoGrabber *)grabber didFailWithError:(NSError *)error
{
    MLShow *show = grabber.userData;
    [self errorWhenFetchingMetaDataForShow:show];
}

Pierre's avatar
Pierre committed
349 350
- (void)tvShowInfoGrabberDidFinishGrabbing:(MLTVShowInfoGrabber *)grabber
{
Pierre's avatar
Pierre committed
351
    MLShow *show = grabber.userData;
Pierre's avatar
Pierre committed
352 353
    NSArray *results = grabber.results;
    if ([results count] > 0) {
354 355
        NSDictionary *result = results[0];
        NSString *showId = result[@"id"];
Pierre's avatar
Pierre committed
356 357

        show.theTVDBID = showId;
358 359 360
        show.name = result[@"title"];
        show.shortSummary = result[@"shortSummary"];
        show.releaseYear = result[@"releaseYear"];
Pierre's avatar
Pierre committed
361 362

        // Fetch episodes info
363
        MLTVShowEpisodesInfoGrabber *grabber = [[MLTVShowEpisodesInfoGrabber alloc] init];
Pierre's avatar
Pierre committed
364 365 366 367 368 369
        grabber.delegate = self;
        grabber.userData = show;
        [grabber lookUpForShowID:showId];
    }
    else {
        // Not found.
Pierre's avatar
Pierre committed
370
        [self noMetaDataInRemoteDBForShow:show];
Pierre's avatar
Pierre committed
371 372 373 374
        show.lastSyncDate = [MLTVShowInfoGrabber serverTime];
    }
}

Pierre's avatar
Pierre committed
375 376 377 378 379 380
- (void)tvShowInfoGrabber:(MLTVShowInfoGrabber *)grabber didFailWithError:(NSError *)error
{
    MLShow *show = grabber.userData;
    [self errorWhenFetchingMetaDataForShow:show];
}

Pierre's avatar
Pierre committed
381 382
- (void)tvShowInfoGrabberDidFetchServerTime:(MLTVShowInfoGrabber *)grabber
{
Pierre's avatar
Pierre committed
383
    MLShow *show = grabber.userData;
Pierre's avatar
Pierre committed
384 385 386

    [[NSUserDefaults standardUserDefaults] setInteger:[[MLTVShowInfoGrabber serverTime] integerValue] forKey:kLastTVDBUpdateServerTime];

Pierre's avatar
Pierre committed
387
    // First fetch the MLShow ID
388
    MLTVShowInfoGrabber *showInfoGrabber = [[MLTVShowInfoGrabber alloc] init];
Pierre's avatar
Pierre committed
389 390 391
    showInfoGrabber.delegate = self;
    showInfoGrabber.userData = show;

392
    APLog(@"Fetching show information on %@", show.name);
Pierre's avatar
Pierre committed
393

Pierre's avatar
Pierre committed
394 395 396 397
    [showInfoGrabber lookUpForTitle:show.name];
}
#endif

Pierre's avatar
Pierre committed
398
- (void)fetchMetaDataForShow:(MLShow *)show
Pierre's avatar
Pierre committed
399
{
400 401
    if (!_allowNetworkAccess)
        return;
402
    APLog(@"Fetching show server time");
Pierre's avatar
Pierre committed
403

Pierre's avatar
Pierre committed
404
    // First fetch the serverTime, so that we can update each entry.
Pierre's avatar
Pierre committed
405 406
#if HAVE_BLOCK
    [MLTVShowInfoGrabber fetchServerTimeAndExecuteBlock:^(NSNumber *serverDate) {
Pierre's avatar
Pierre committed
407 408 409

        [[NSUserDefaults standardUserDefaults] setInteger:[serverDate integerValue] forKey:kLastTVDBUpdateServerTime];

410
        APLog(@"Fetching show information on %@", show.name);
Pierre's avatar
Pierre committed
411 412 413

        // First fetch the MLShow ID
        MLTVShowInfoGrabber *grabber = [[[MLTVShowInfoGrabber alloc] init] autorelease];
Pierre's avatar
Pierre committed
414 415 416 417 418 419 420 421 422 423 424
        [grabber lookUpForTitle:show.name andExecuteBlock:^{
            NSArray *results = grabber.results;
            if ([results count] > 0) {
                NSDictionary *result = [results objectAtIndex:0];
                NSString *showId = [result objectForKey:@"id"];

                show.theTVDBID = showId;
                show.name = [result objectForKey:@"title"];
                show.shortSummary = [result objectForKey:@"shortSummary"];
                show.releaseYear = [result objectForKey:@"releaseYear"];

425
                APLog(@"Fetching show episode information on %@", showId);
Pierre's avatar
Pierre committed
426

Pierre's avatar
Pierre committed
427
                // Fetch episode info
Pierre's avatar
Pierre committed
428
                MLTVShowEpisodesInfoGrabber *grabber = [[[MLTVShowEpisodesInfoGrabber alloc] init] autorelease];
Pierre's avatar
Pierre committed
429 430 431 432 433 434 435
                [grabber lookUpForShowID:showId andExecuteBlock:^{
                    NSArray *results = grabber.episodesResults;
                    [show setValue:[grabber.results objectForKey:@"serieArtworkURL"] forKey:@"artworkURL"];
                    for (id result in results) {
                        if ([[result objectForKey:@"serie"] boolValue]) {
                            continue;
                        }
436
                        MLShowEpisode *showEpisode = [MLShowEpisode episodeWithShow:show episodeNumber:[result objectForKey:@"episodeNumber"] seasonNumber:[result objectForKey:@"seasonNumber"] createIfNeeded:YES];
Pierre's avatar
Pierre committed
437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452
                        showEpisode.name = [result objectForKey:@"title"];
                        showEpisode.theTVDBID = [result objectForKey:@"id"];
                        showEpisode.shortSummary = [result objectForKey:@"shortSummary"];
                        showEpisode.artworkURL = [result objectForKey:@"artworkURL"];
                        showEpisode.lastSyncDate = serverDate;
                    }
                    show.lastSyncDate = serverDate;
                }];
            }
            else {
                // Not found.
                show.lastSyncDate = serverDate;
            }

        }];
    }];
Pierre's avatar
Pierre committed
453
#else
454
    MLTVShowInfoGrabber *grabber = [[MLTVShowInfoGrabber alloc] init];
Pierre's avatar
Pierre committed
455 456 457
    grabber.delegate = self;
    grabber.userData = show;
    [grabber fetchServerTime];
Pierre's avatar
Pierre committed
458
#endif
Pierre's avatar
Pierre committed
459 460
}

Pierre's avatar
Pierre committed
461
- (void)addTVShowEpisodeWithInfo:(NSDictionary *)tvShowEpisodeInfo andFile:(MLFile *)file
Pierre's avatar
Pierre committed
462
{
Pierre's avatar
Pierre committed
463
    file.type = kMLFileTypeTVShowEpisode;
Pierre's avatar
Pierre committed
464

465 466 467
    NSNumber *seasonNumber = tvShowEpisodeInfo[@"season"];
    NSNumber *episodeNumber = tvShowEpisodeInfo[@"episode"];
    NSString *tvShowName = tvShowEpisodeInfo[@"tvShowName"];
468
    NSString *tvEpisodeName = tvShowEpisodeInfo[@"tvEpisodeName"];
Pierre's avatar
Pierre committed
469 470
    BOOL hasNoTvShow = NO;
    if (!tvShowName) {
471
        tvShowName = @"";
Pierre's avatar
Pierre committed
472 473 474
        hasNoTvShow = YES;
    }
    BOOL wasInserted = NO;
Pierre's avatar
Pierre committed
475 476 477 478
    MLShow *show = nil;
    MLShowEpisode *episode = [MLShowEpisode episodeWithShowName:tvShowName episodeNumber:episodeNumber seasonNumber:seasonNumber createIfNeeded:YES wasCreated:&wasInserted];
    if (episode)
        show = episode.show;
Pierre's avatar
Pierre committed
479 480 481 482
    if (wasInserted && !hasNoTvShow) {
        show.name = tvShowName;
        [self fetchMetaDataForShow:show];
    }
483
    episode.name = tvEpisodeName;
Pierre's avatar
Pierre committed
484

485
    if (episode.name.length < 1)
Pierre's avatar
Pierre committed
486 487 488
        episode.name = file.title;
    file.seasonNumber = seasonNumber;
    file.episodeNumber = episodeNumber;
489
    episode.shouldBeDisplayed = @YES;
Pierre's avatar
Pierre committed
490 491

    [episode addFilesObject:file];
Pierre's avatar
Pierre committed
492
    file.showEpisode = episode;
Pierre's avatar
Pierre committed
493

Pierre's avatar
Pierre committed
494
    // The rest of the meta data will be fetched using the MLShow
495
    file.hasFetchedInfo = @YES;
Pierre's avatar
Pierre committed
496 497
}

498 499 500 501
- (void)addAudioContentWithInfo:(NSDictionary *)audioContentInfo andFile:(MLFile *)file
{
    file.type = kMLFileTypeAudio;

502 503 504
    file.title = audioContentInfo[VLCMetaInformationTitle];

    /* all further meta data is set by the FileParserQueue */
505 506 507 508

    file.hasFetchedInfo = @YES;
}

Pierre's avatar
Pierre committed
509
/**
Pierre's avatar
Pierre committed
510
 * MLFile auto detection
Pierre's avatar
Pierre committed
511 512
 */

Pierre's avatar
Pierre committed
513
#if !HAVE_BLOCK
Pierre's avatar
Pierre committed
514 515 516 517 518 519
- (void)movieInfoGrabber:(MLMovieInfoGrabber *)grabber didFailWithError:(NSError *)error
{
    MLFile *file = grabber.userData;
    [self errorWhenFetchingMetaDataForFile:file];
}

Pierre's avatar
Pierre committed
520 521
- (void)movieInfoGrabberDidFinishGrabbing:(MLMovieInfoGrabber *)grabber
{
522
    NSNumber *yes = @YES;
Pierre's avatar
Pierre committed
523 524

    NSArray *results = grabber.results;
Pierre's avatar
Pierre committed
525
    MLFile *file = grabber.userData;
Pierre's avatar
Pierre committed
526
    if ([results count] > 0) {
527 528 529 530 531
        NSDictionary *result = results[0];
        file.artworkURL = result[@"artworkURL"];
        file.title = result[@"title"];
        file.shortSummary = result[@"shortSummary"];
        file.releaseYear = result[@"releaseYear"];
Pierre's avatar
Pierre committed
532
    }
Pierre's avatar
Pierre committed
533 534 535 536
    else {
        [self noMetaDataInRemoteDBForFile:file];
    }

Pierre's avatar
Pierre committed
537 538 539 540
    file.hasFetchedInfo = yes;
}
#endif

Pierre's avatar
Pierre committed
541
- (void)fetchMetaDataForFile:(MLFile *)file
Pierre's avatar
Pierre committed
542
{
543
    APLog(@"Fetching meta data for %@", file.title);
Pierre's avatar
Pierre committed
544

545
    [[MLFileParserQueue sharedFileParserQueue] addFile:file];
546

547 548 549 550 551
    if (!_allowNetworkAccess) {
        // Automatically compute the thumbnail
        [self computeThumbnailForFile:file];
    }

Pierre's avatar
Pierre committed
552
    NSDictionary *tvShowEpisodeInfo = [MLTitleDecrapifier tvShowEpisodeInfoFromString:file.title];
Pierre's avatar
Pierre committed
553 554 555 556 557
    if (tvShowEpisodeInfo) {
        [self addTVShowEpisodeWithInfo:tvShowEpisodeInfo andFile:file];
        return;
    }

558 559 560 561 562 563 564 565
    if ([file isSupportedAudioFile]) {
        NSDictionary *audioContentInfo = [MLTitleDecrapifier audioContentInfoFromFile:file];
        if (audioContentInfo && ![file videoTrack]) {
            [self addAudioContentWithInfo:audioContentInfo andFile:file];
            return;
        }
    }

566 567 568
    if (!_allowNetworkAccess)
        return;

Pierre's avatar
Pierre committed
569
    // Go online and fetch info.
Pierre's avatar
Pierre committed
570 571 572

    // We don't care about keeping a reference to track the item during its life span
    // because we are a singleton
573
    MLMovieInfoGrabber *grabber = [[MLMovieInfoGrabber alloc] init];
Pierre's avatar
Pierre committed
574

575
    APLog(@"Looking up for Movie '%@'", file.title);
Pierre's avatar
Pierre committed
576

Pierre's avatar
Pierre committed
577
#if HAVE_BLOCK
Pierre's avatar
Pierre committed
578
    [grabber lookUpForTitle:file.title andExecuteBlock:^(NSError *err){
Pierre's avatar
Pierre committed
579 580
        if (err) {
            [self errorWhenFetchingMetaDataForFile:file];
Pierre's avatar
Pierre committed
581
            return;
Pierre's avatar
Pierre committed
582
        }
Pierre's avatar
Pierre committed
583 584 585 586 587

        NSArray *results = grabber.results;
        if ([results count] > 0) {
            NSDictionary *result = [results objectAtIndex:0];
            file.artworkURL = [result objectForKey:@"artworkURL"];
Pierre's avatar
Pierre committed
588 589
            if (!file.artworkURL)
                [self computeThumbnailForFile:file];
Pierre's avatar
Pierre committed
590 591 592
            file.title = [result objectForKey:@"title"];
            file.shortSummary = [result objectForKey:@"shortSummary"];
            file.releaseYear = [result objectForKey:@"releaseYear"];
Pierre's avatar
Pierre committed
593 594 595
        } else
            [self noMetaDataInRemoteDBForFile:file];
        file.hasFetchedInfo = [NSNumber numberWithBool:YES];
Pierre's avatar
Pierre committed
596
    }];
Pierre's avatar
Pierre committed
597
#else
Pierre's avatar
Pierre committed
598 599 600
    grabber.userData = file;
    grabber.delegate = self;
    [grabber lookUpForTitle:file.title];
Pierre's avatar
Pierre committed
601
#endif
Pierre's avatar
Pierre committed
602 603
}

Pierre's avatar
Pierre committed
604 605 606
#pragma mark -
#pragma mark Adding file to the DB

Pierre's avatar
Pierre committed
607
- (void)addFilePath:(NSString *)filePath
Pierre's avatar
Pierre committed
608
{
609
    APLog(@"Adding Path %@", filePath);
Pierre's avatar
Pierre committed
610

Pierre's avatar
Pierre committed
611
    NSURL *url = [NSURL fileURLWithPath:filePath];
Pierre's avatar
Pierre committed
612 613
    NSDictionary *attributes = [[NSFileManager defaultManager] attributesOfItemAtPath:filePath error:nil];
    NSString *title = [filePath lastPathComponent];
Pierre's avatar
Pierre committed
614
#if !TARGET_OS_IPHONE
Pierre's avatar
Pierre committed
615 616
    NSDate *openedDate = nil; // FIXME kMDItemLastUsedDate
    NSDate *modifiedDate = nil; // FIXME [result valueForAttribute:@"kMDItemFSContentChangeDate"];
Pierre's avatar
Pierre committed
617
#endif
618
    NSNumber *size = attributes[NSFileSize]; // FIXME [result valueForAttribute:@"kMDItemFSSize"];
Pierre's avatar
Pierre committed
619

Pierre's avatar
Pierre committed
620
    MLFile *file = [self createObjectForEntity:@"File"];
Pierre's avatar
Pierre committed
621
    file.url = [url absoluteString];
Pierre's avatar
Pierre committed
622 623 624 625

    // Yes, this is a negative number. VLCTime nicely display negative time
    // with "XX minutes remaining". And we are using this facility.

626 627
    NSNumber *no = @NO;
    NSNumber *yes = @YES;
Pierre's avatar
Pierre committed
628 629

    file.currentlyWatching = no;
630 631
    file.lastPosition = @0.0;
    file.remainingTime = @0.0;
Pierre's avatar
Pierre committed
632 633
    file.unread = yes;

Pierre's avatar
Pierre committed
634
#if !TARGET_OS_IPHONE
Pierre's avatar
Pierre committed
635 636 637 638
    if ([openedDate isGreaterThan:modifiedDate]) {
        file.playCount = [NSNumber numberWithDouble:1];
        file.unread = no;
    }
Pierre's avatar
Pierre committed
639 640
#endif

641 642 643 644
    if ([[[NSUserDefaults standardUserDefaults] objectForKey:kDecrapifyTitles] boolValue] == YES)
        file.title = [MLTitleDecrapifier decrapify:[title stringByDeletingPathExtension]];
    else
        file.title = [title stringByDeletingPathExtension];
Pierre's avatar
Pierre committed
645 646

    if ([size longLongValue] < 150000000) /* 150 MB */
Pierre's avatar
Pierre committed
647
        file.type = kMLFileTypeClip;
Pierre's avatar
Pierre committed
648
    else
Pierre's avatar
Pierre committed
649
        file.type = kMLFileTypeMovie;
Pierre's avatar
Pierre committed
650 651 652 653

    [self fetchMetaDataForFile:file];
}

Pierre's avatar
Pierre committed
654
- (void)addFilePaths:(NSArray *)filepaths
Pierre's avatar
Pierre committed
655
{
Pierre's avatar
Pierre committed
656
    NSUInteger count = [filepaths count];
Pierre's avatar
Pierre committed
657 658 659 660
    NSMutableArray *fetchPredicates = [NSMutableArray arrayWithCapacity:count];
    NSMutableDictionary *urlToObject = [NSMutableDictionary dictionaryWithCapacity:count];

    // Prepare a fetch request for all items
Pierre's avatar
Pierre committed
661
    for (NSString *path in filepaths) {
Pierre's avatar
Pierre committed
662
        NSURL *url = [NSURL fileURLWithPath:path];
663
        NSString *urlString = [url absoluteString];
664 665 666 667 668 669
#if TARGET_OS_IPHONE
        /* check for the end of a path only, since parts of the path components may change
         * while it is still the same file
         * of course, this is only true for the flat file structure we use on iOS */
        [fetchPredicates addObject:[NSPredicate predicateWithFormat:@"url ENDSWITH %@", [urlString lastPathComponent]]];
#else
Pierre's avatar
Pierre committed
670
        [fetchPredicates addObject:[NSPredicate predicateWithFormat:@"url == %@", urlString]];
671
#endif
672
        urlToObject[urlString] = path;
Pierre's avatar
Pierre committed
673 674 675 676 677 678
    }

    NSFetchRequest *request = [self fetchRequestForEntity:@"File"];

    [request setPredicate:[NSCompoundPredicate orPredicateWithSubpredicates:fetchPredicates]];

679
    APLog(@"Fetching");
Pierre's avatar
Pierre committed
680
    NSArray *dbResults = [[self managedObjectContext] executeFetchRequest:request error:nil];
681
    APLog(@"Done");
Pierre's avatar
Pierre committed
682

Pierre's avatar
Pierre committed
683
    NSMutableArray *filePathsToAdd = [NSMutableArray arrayWithArray:filepaths];
Pierre's avatar
Pierre committed
684 685

    // Remove objects that are already in db.
Pierre's avatar
Pierre committed
686 687
    for (MLFile *dbResult in dbResults) {
        NSString *urlString = dbResult.url;
688
        [filePathsToAdd removeObject:urlToObject[urlString]];
Pierre's avatar
Pierre committed
689 690 691
    }

    // Add only the newly added items
Pierre's avatar
Pierre committed
692 693
    for (NSString* path in filePathsToAdd)
        [self addFilePath:path];
Pierre's avatar
Pierre committed
694 695
}

Pierre's avatar
Pierre committed
696 697 698 699

#pragma mark -
#pragma mark DB Updates

Pierre's avatar
Pierre committed
700 701 702 703 704 705
#if !HAVE_BLOCK
- (void)tvShowInfoGrabber:(MLTVShowInfoGrabber *)grabber didFetchUpdates:(NSArray *)updates
{
    NSFetchRequest *request = [self fetchRequestForEntity:@"Show"];
    [request setPredicate:[NSComparisonPredicate predicateWithLeftExpression:[NSExpression expressionForKeyPath:@"theTVDBID"] rightExpression:[NSExpression expressionForConstantValue:updates] modifier:NSDirectPredicateModifier type:NSInPredicateOperatorType options:0]];
    NSArray *results = [[self managedObjectContext] executeFetchRequest:request error:nil];
Pierre's avatar
Pierre committed
706
    for (MLShow *show in results)
Pierre's avatar
Pierre committed
707 708 709 710
        [self fetchMetaDataForShow:show];
}
#endif

711 712
- (BOOL)libraryNeedsUpgrade
{
713
    if (![[[NSUserDefaults standardUserDefaults] objectForKey:kUpdatedToTheGreatSharkHuntDatabaseFormat] boolValue])
714 715 716 717 718
        return YES;
    return NO;
}

- (void)upgradeLibrary
Felix Paul Kühne's avatar
Felix Paul Kühne committed
719 720 721 722 723 724
{
    if (![[[NSUserDefaults standardUserDefaults] objectForKey:kUpdatedToTheGreatSharkHuntDatabaseFormat] boolValue])
        [self _upgradeLibraryToGreatSharkHuntDatabaseFormat];
}

- (void)_upgradeLibraryToGreatSharkHuntDatabaseFormat
725
{
Felix Paul Kühne's avatar
Felix Paul Kühne committed
726
    [self libraryDidDisappear];
727 728 729
    NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
    NSFileManager *fileManager = [NSFileManager defaultManager];

730
    /* remove potential empty albums left over by previous releases */
731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757
    NSArray *collection = [MLAlbum allAlbums];
    NSUInteger count = collection.count;
    MLAlbum *album;
    MLAlbumTrack *track;
    NSArray *secondaryCollection;
    NSURL *fileURL;
    NSUInteger secondaryCount = 0;
    NSArray *tertiaryCollection;
    NSUInteger tertiaryCount = 0;
    NSUInteger emptyAlbumCounter = 0;
    for (NSUInteger x = 0; x < count; x++) {
        album = collection[x];
        if (album.tracks.count < 1)
            [[self managedObjectContext] deleteObject:album];
        else {
            secondaryCollection = album.tracks.allObjects;
            secondaryCount = secondaryCollection.count;
            emptyAlbumCounter = 0;
            for (NSUInteger y = 0; y < secondaryCount; y++) {
                track = secondaryCollection[y];
                tertiaryCollection = track.files.allObjects;
                tertiaryCount = tertiaryCollection.count;
                for (NSUInteger z = 0; z < tertiaryCount; z++) {
                    fileURL = [NSURL URLWithString:[(MLFile *)tertiaryCollection[z] url]];
                    BOOL exists = [fileManager fileExistsAtPath:[fileURL path]];
                    if (exists)
                        emptyAlbumCounter++;
758 759
                    else
                        [album removeTrack:track];
760 761 762 763 764 765 766
                }
            }
            if (emptyAlbumCounter == 0)
                [[self managedObjectContext] deleteObject:album];
        }
    }
    album = nil;
767 768

    /* remove potential empty shows left over by previous releases */
769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789
    collection = [MLShow allShows];
    MLShow *show;
    MLShowEpisode *showEpisode;
    count = collection.count;
    for (NSUInteger x = 0; x < count; x++) {
        show = collection[x];
        if (show.episodes.count < 1)
            [[self managedObjectContext] deleteObject:show];
        else {
            secondaryCollection = show.episodes.allObjects;
            secondaryCount = secondaryCollection.count;
            emptyAlbumCounter = 0;
            for (NSUInteger y = 0; y < secondaryCount; y++) {
                showEpisode = secondaryCollection[y];
                tertiaryCollection = showEpisode.files.allObjects;
                tertiaryCount = tertiaryCollection.count;
                for (NSUInteger z = 0; z < tertiaryCount; z++) {
                    fileURL = [NSURL URLWithString:[(MLFile *)tertiaryCollection[z] url]];
                    BOOL exists = [fileManager fileExistsAtPath:[fileURL path]];
                    if (exists)
                        emptyAlbumCounter++;
790 791
                    else
                        [show removeEpisode:showEpisode];
792 793 794 795 796 797 798
                }
            }
            if (emptyAlbumCounter == 0)
                [[self managedObjectContext] deleteObject:show];
        }
    }

799 800 801 802 803
    /* remove duplicates */
    NSArray *allFiles = [MLFile allFiles];
    NSUInteger allFilesCount = allFiles.count;
    NSMutableArray *seenFiles = [[NSMutableArray alloc] initWithCapacity:allFilesCount];
    MLFile *currentFile;
Felix Paul Kühne's avatar
Felix Paul Kühne committed
804 805
    NSString *currentFilePath;
    for (NSUInteger x = 0; x < allFilesCount; x++) {
806
        currentFile = allFiles[x];
Felix Paul Kühne's avatar
Felix Paul Kühne committed
807 808
        currentFilePath = [currentFile.url stringByReplacingOccurrencesOfString:@"/localhost/" withString:@"//"];
        if ([seenFiles containsObject:currentFilePath])
809 810
            [[self managedObjectContext] deleteObject:currentFile];
        else
Felix Paul Kühne's avatar
Felix Paul Kühne committed
811
            [seenFiles addObject:currentFilePath];
812 813
    }

814
    [defaults setBool:YES forKey:kUpdatedToTheGreatSharkHuntDatabaseFormat];
Felix Paul Kühne's avatar
Felix Paul Kühne committed
815 816 817
    [defaults synchronize];

    [self libraryDidAppear];
818 819 820 821
    if ([self.delegate respondsToSelector:@selector(libraryUpgradeComplete)])
        [self.delegate libraryUpgradeComplete];
}

822
- (void)updateMediaDatabase
Pierre's avatar
Pierre committed
823
{
Felix Paul Kühne's avatar
Felix Paul Kühne committed
824
    [self libraryDidDisappear];
Pierre's avatar
Pierre committed
825
    // Remove no more present files
Pierre's avatar
Pierre committed
826 827
    NSFetchRequest *request = [self fetchRequestForEntity:@"File"];
    NSArray *results = [[self managedObjectContext] executeFetchRequest:request error:nil];
Pierre's avatar
Pierre committed
828
    NSFileManager *fileManager = [NSFileManager defaultManager];
829

830
    unsigned int count = (unsigned int)results.count;
831 832
    for (unsigned int x = 0; x < count; x++) {
        MLFile *file = results[x];
Pierre's avatar
Pierre committed
833 834
        NSString *urlString = [file url];
        NSURL *fileURL = [NSURL URLWithString:urlString];
835
        BOOL exists = [fileManager fileExistsAtPath:[fileURL path]];
836
        if (!exists) {
837
            APLog(@"Marking - %@", [fileURL absoluteString]);
838
            file.isSafe = YES; // It doesn't exist, it's safe.
839 840
            if (file.isAlbumTrack) {
                MLAlbum *album = file.albumTrack.album;
841 842 843 844 845 846 847
                if (album.tracks.count <= 1) {
                    @try {
                        [[self managedObjectContext] deleteObject:album];
                    }
                    @catch (NSException *exception) {
                        APLog(@"failed to nuke object because it disappeared in front of us");
                    }
848
                } else
849
                    [album removeTrack:file.albumTrack];
850 851 852
            }
            if (file.isShowEpisode) {
                MLShow *show = file.showEpisode.show;
853 854 855 856 857 858 859
                if (show.episodes.count <= 1) {
                    @try {
                        [[self managedObjectContext] deleteObject:show];
                    }
                    @catch (NSException *exception) {
                        APLog(@"failed to nuke object because it disappeared in front of us");
                    }
860 861
                } else
                    [show removeEpisode:file.showEpisode];
862
            }
863
#if TARGET_OS_IPHONE
864
            NSString *thumbPath = [file thumbnailPath];
865 866
            bool thumbExists = [fileManager fileExistsAtPath:thumbPath];
            if (thumbExists)
867
                [fileManager removeItemAtPath:thumbPath error:nil];
868 869
            [[self managedObjectContext] deleteObject:file];
#endif
870
        }
871 872 873
#if !TARGET_OS_IPHONE
    file.isOnDisk = @(exists);
#endif
Pierre's avatar
Pierre committed
874
    }
Felix Paul Kühne's avatar
Felix Paul Kühne committed
875
    [self libraryDidAppear];
Pierre's avatar
Pierre committed
876

877 878 879 880 881 882 883
    // Get the file to parse
    request = [self fetchRequestForEntity:@"File"];
    [request setPredicate:[NSPredicate predicateWithFormat:@"isOnDisk == YES && tracks.@count == 0"]];
    results = [[self managedObjectContext] executeFetchRequest:request error:nil];
    for (MLFile *file in results)
        [[MLFileParserQueue sharedFileParserQueue] addFile:file];

884 885 886
    if (!_allowNetworkAccess) {
        // Always attempt to fetch
        request = [self fetchRequestForEntity:@"File"];
887
        [request setPredicate:[NSPredicate predicateWithFormat:@"isOnDisk == YES"]];
888
        results = [[self managedObjectContext] executeFetchRequest:request error:nil];
889 890 891 892
        for (MLFile *file in results) {
            if (!file.computedThumbnail)
                [self computeThumbnailForFile:file];
        }
893 894 895
        return;
    }

Pierre's avatar
Pierre committed
896 897
    // Get the thumbnails to compute
    request = [self fetchRequestForEntity:@"File"];
898
    [request setPredicate:[NSPredicate predicateWithFormat:@"isOnDisk == YES && hasFetchedInfo == 1 && artworkURL == nil"]];
Pierre's avatar
Pierre committed
899 900
    results = [[self managedObjectContext] executeFetchRequest:request error:nil];
    for (MLFile *file in results)
901
        if (!file.computedThumbnail && ![file isAlbumTrack])
902
            [self computeThumbnailForFile:file];
Pierre's avatar
Pierre committed
903 904 905 906 907 908

    // Get to fetch meta data
    request = [self fetchRequestForEntity:@"File"];
    [request setPredicate:[NSPredicate predicateWithFormat:@"isOnDisk == YES && hasFetchedInfo == 0"]];
    results = [[self managedObjectContext] executeFetchRequest:request error:nil];
    for (MLFile *file in results)
Pierre's avatar
Pierre committed
909 910
        [self fetchMetaDataForFile:file];

Pierre's avatar
Pierre committed
911
    // Get to fetch show info
Pierre's avatar
Pierre committed
912 913 914
    request = [self fetchRequestForEntity:@"Show"];
    [request setPredicate:[NSPredicate predicateWithFormat:@"lastSyncDate == 0"]];
    results = [[self managedObjectContext] executeFetchRequest:request error:nil];
Pierre's avatar
Pierre committed
915
    for (MLShow *show in results)
Pierre's avatar
Pierre committed
916 917
        [self fetchMetaDataForShow:show];

Pierre's avatar
Pierre committed
918
    // Get updated TV Shows
919
    NSNumber *lastServerTime = @([[NSUserDefaults standardUserDefaults] integerForKey:kLastTVDBUpdateServerTime]);
Pierre's avatar
Pierre committed
920
#if HAVE_BLOCK
Pierre's avatar
Pierre committed
921
    [MLTVShowInfoGrabber fetchUpdatesSinceServerTime:lastServerTime andExecuteBlock:^(NSArray *updates){
Pierre's avatar
Pierre committed
922 923 924
        NSFetchRequest *request = [self fetchRequestForEntity:@"Show"];
        [request setPredicate:[NSComparisonPredicate predicateWithLeftExpression:[NSExpression expressionForKeyPath:@"theTVDBID"] rightExpression:[NSExpression expressionForConstantValue:updates] modifier:NSDirectPredicateModifier type:NSInPredicateOperatorType options:0]];
        NSArray *results = [[self managedObjectContext] executeFetchRequest:request error:nil];
Pierre's avatar
Pierre committed
925
        for (MLShow *show in results)
Pierre's avatar
Pierre committed
926 927
            [self fetchMetaDataForShow:show];
    }];
Pierre's avatar
Pierre committed
928
#else
929
    MLTVShowInfoGrabber *grabber = [[MLTVShowInfoGrabber alloc] init];
Pierre's avatar
Pierre committed
930 931
    grabber.delegate = self;
    [grabber fetchUpdatesSinceServerTime:lastServerTime];
Pierre's avatar
Pierre committed
932
#endif
Pierre's avatar
Pierre committed
933
    /* Update every hour - FIXME: Preferences key */
934
    [self performSelector:@selector(updateMediaDatabase) withObject:nil afterDelay:60 * 60];
Pierre's avatar
Pierre committed
935
}
Pierre's avatar
Pierre committed
936

937 938 939 940 941
- (void)applicationWillExit
{
    [[MLCrashPreventer sharedPreventer] cancelAllFileParse];
}

Pierre's avatar
Pierre committed
942 943 944 945 946
- (void)applicationWillStart
{
    [[MLCrashPreventer sharedPreventer] markCrasherFiles];
}

Pierre's avatar
Pierre committed
947 948 949 950 951 952 953 954 955 956 957
- (void)libraryDidDisappear
{
    // Stop expansive work
    [[MLThumbnailerQueue sharedThumbnailerQueue] stop];
}

- (void)libraryDidAppear
{
    // Resume our work
    [[MLThumbnailerQueue sharedThumbnailerQueue] resume];
}
Pierre's avatar
Pierre committed
958
@end