root/branches/CoreData/SapphireFrappliance/MetaData/SapphireMObjects/SapphireTVShow.m @ 840

#import "SapphireTVShow.h"
#import "SapphireSeason.h"
#import "SapphireEpisode.h"
#import "SapphireSubEpisode.h"
#import "CoreDataSupportFunctions.h"
#import "SapphireMetaDataSupport.h"
#import "NSString-Extensions.h"
#import "SapphireFileSorter.h"
#import "SapphireTVTranslation.h"

@implementation SapphireTVShow

static NSArray *allowedSorts = nil;

+ (void)load
{
        allowedSorts = [[NSArray alloc] initWithObjects:[SapphireTVEpisodeSorter sharedInstance], [SapphireDateSorter sharedInstance], nil];
}

+ (SapphireTVShow *)show:(NSString *)show withPath:(NSString *)showPath inContext:(NSManagedObjectContext *)moc
{
        NSPredicate *predicate = [NSPredicate predicateWithFormat:@"name == %@", show];
        SapphireTVShow *ret = (SapphireTVShow *)doSingleFetchRequest(SapphireTVShowName, moc, predicate);
        if(ret != nil)
        {
                if(ret.showPath == nil)
                        ret.showPath = showPath;
                return ret;
        }
        
        ret = [NSEntityDescription insertNewObjectForEntityForName:SapphireTVShowName inManagedObjectContext:moc];
        ret.name = show;
        ret.showPath = showPath;
        return ret;
}

+ (SapphireTVShow *)showWithPath:(NSString *)path inContext:(NSManagedObjectContext *)moc
{
        NSPredicate *predicate = [NSPredicate predicateWithFormat:@"showPath == %@", path];
        return (SapphireTVShow *)doSingleFetchRequest(SapphireTVShowName, moc, predicate);
}

+ (void)upgradeV1ShowsFromContext:(NSManagedObjectContext *)oldMoc toContext:(NSManagedObjectContext *)newMoc
{
        NSArray *oldShows = doFetchRequest(SapphireTVShowName, oldMoc, nil);
        NSEnumerator *showEnum = [oldShows objectEnumerator];
        NSManagedObject *oldShow;
        while((oldShow = [showEnum nextObject]) != nil)
        {
                SapphireTVShow *newShow = [NSEntityDescription insertNewObjectForEntityForName:SapphireTVShowName inManagedObjectContext:newMoc];
                newShow.name = [oldShow valueForKey:@"name"];
                newShow.showDescription = [oldShow valueForKey:@"showDescription"];
                newShow.showID = [oldShow valueForKey:@"showID"];
                newShow.showPath = [oldShow valueForKey:@"showPath"];
                
                NSEnumerator *translationEnum = [[oldShow valueForKey:@"translations"] objectEnumerator];
                NSManagedObject *translation;
                while((translation = [translationEnum nextObject]) != nil)
                {
                        [SapphireTVTranslation upgradeV1TVTranslation:translation toShow:newShow];
                }
                
                NSEnumerator *seasonEnum = [[oldShow valueForKey:@"seasons"] objectEnumerator];
                NSManagedObject *season;
                while((season = [seasonEnum nextObject]) != nil)
                {
                        [SapphireSeason upgradeV1Season:season toShow:newShow]; 
                }
        }
}

- (NSComparisonResult)compare:(SapphireTVShow *)other
{
        return [self.name nameCompare:other.name];
}

- (NSString *)dirNameValue
{
        return @"seasonName";
}

- (NSPredicate *)metaFileFetchPredicate
{
        return [NSPredicate predicateWithFormat:@"tvEpisode.tvShow == %@", self];
}

- (NSArray *)fileSorters
{
        return allowedSorts;
}

- (void)defaultDirectorySort:(NSMutableArray *)dirs
{
        [dirs sortUsingSelector:@selector(compare:)];
}

- (NSString *)path
{
        return [NSString stringWithFormat:@"@TV/%@", self.name];
}

static inline NSArray *getEpsFromFiles(NSManagedObjectContext *moc, NSArray *files)
{
        NSSet *epIds = [NSSet setWithArray:[files valueForKeyPath:@"tvEpisode.objectID"]];
        
        NSPredicate *epPredicate = [NSPredicate predicateWithFormat:@"SELF IN %@", epIds];
        return doFetchRequest(SapphireEpisodeName, moc, epPredicate);
}

- (NSMutableArray *)internalDirectoryFetchFromFiles:(NSArray *)files
{
        NSManagedObjectContext *moc = [self managedObjectContext];
        NSArray *eps = getEpsFromFiles(moc, files);
        
        NSSet *seasonIds = [NSSet setWithArray:[eps valueForKeyPath:@"season.objectID"]];
        NSPredicate *seasonPred = [NSPredicate predicateWithFormat:@"SELF IN %@", seasonIds];
        NSMutableArray *ret = [doFetchRequest(SapphireSeasonName, moc, seasonPred) mutableCopy];
        return [ret autorelease];
}

- (void)prefetch:(NSArray *)files
{
        NSManagedObjectContext *moc = [self managedObjectContext];
        NSArray *eps = getEpsFromFiles(moc, files);
        
        NSPredicate *subEpsPred = [NSPredicate predicateWithFormat:@"episode IN %@", eps];
        doFetchRequest(SapphireSubEpisodeName, moc, subEpsPred);
}

- (BOOL)shouldDelete
{
        return [self.episodesSet count] == 0;
}

-(void)removeEpisodes:(NSSet*)removedEpisodes
{
        [super removeEpisodes:removedEpisodes];
        if([self.episodesSet count] == 0)
                [SapphireMetaDataSupport setObjectForPendingDelete:self];
}

- (void)removeEpisodesObject:(SapphireEpisode*)removedEpisode
{
        [super removeEpisodesObject:removedEpisode];
        if([self.episodesSet count] == 0)
                [SapphireMetaDataSupport setObjectForPendingDelete:self];
}

@end