root/branches/CoreData/SapphireFrappliance/Extension/NSFileManager-Extensions.m @ 827

/*
 * NSFileManager-Extensions.m
 * Sapphire
 *
 * Created by Patrick Merrill on Dec. 09, 2007.
 * Copyright 2007 Sapphire Development Team and/or www.nanopi.net
 * All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify it under the terms of the GNU
 * General Public License as published by the Free Software Foundation; either version 3 of the License,
 * or (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 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
 * Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License along with this program; if not,
 * write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

#import "NSFileManager-Extensions.h"
#import "SapphireMetaDataSupport.h"

@implementation NSFileManager (SapphireExtensions)
- (BOOL)constructPath:(NSString *)proposedPath
{
        BOOL isDir ;
        if(!([self fileExistsAtPath:proposedPath isDirectory:&isDir]&& isDir))
                if(![self createDirectoryAtPath:proposedPath attributes:nil])
                        if([self constructPath:[proposedPath stringByDeletingLastPathComponent]])
                                return [self createDirectoryAtPath:proposedPath attributes:nil];
                        else
                                return NO;
        return YES;     
}

// Static set of file extensions to filter
static NSSet *videoExtensions = nil;
static NSSet *audioExtensions = nil;
static NSSet *allExtensions = nil;

+ (void)load
{
        videoExtensions = [[NSSet alloc] initWithObjects:
                                           @"avi", @"divx", @"xvid",
                                           @"mov",
                                           @"mpg", @"mpeg", @"m2v", @"ts",
                                           @"wmv", @"asx", @"asf",
                                           @"mkv",
                                           @"flv",
                                           @"mp4", @"m4v",
                                           @"3gp",
                                           @"pls",
                                           @"avc",
                                           @"ogm",
                                           @"dv",
                                           @"fli",
                                           nil];
        audioExtensions = [[NSSet alloc] initWithObjects:
                                           @"m4b", @"m4a",
                                           @"mp3", @"mp2",
                                           @"wma",
                                           @"wav",
                                           @"aif", @"aiff",
                                           @"flac",
                                           @"alac",
                                           @"m3u",
                                           @"ac3",
                                           nil];
        NSMutableSet *mutSet = [videoExtensions mutableCopy];
        [mutSet unionSet:audioExtensions];
        allExtensions = [[NSSet alloc] initWithSet:mutSet];
        [mutSet release];
}

+ (NSSet *)videoExtensions
{
        return videoExtensions;
}

+ (NSSet *)audioExtensions
{
        return audioExtensions;
}

- (BOOL)hasVIDEO_TS:(NSString *)path
{
        BOOL isDir = NO;
        NSFileManager *fm = [NSFileManager defaultManager];
        if([fm fileExistsAtPath:[path stringByAppendingPathComponent:@"VIDEO_TS"] isDirectory:&isDir] && isDir)
                return YES;
        return NO;
}

- (BOOL)isDirectory:(NSString *)path
{
        BOOL isDir = NO;
        BOOL exists = [self fileExistsAtPath:path isDirectory:&isDir];
        BOOL ret = exists && isDir;
        if(ret)
        {
                if([self hasVIDEO_TS:path])
                        return NO;
        }
        return ret;
}

- (BOOL)acceptFilePath:(NSString *)path
{
        if(path == nil)
                return NO;
        NSString *name = [path lastPathComponent];
        
        /*Skip hidden files*/
        if([name hasPrefix:@"."])
                return NO;
        
        /*Skip the Cover Art directory*/
        if([name isEqualToString:@"Cover Art"])
                return NO;
        
        path = [path stringByResolvingSymlinksInPath];
        return ([self isDirectory:path] || [self hasVIDEO_TS:path] || [allExtensions containsObject:[path pathExtension]]);
}

+ (NSString *)previewArtPathForTV:(NSString *)show season:(unsigned int)seasonNum
{
        return [NSString stringWithFormat:@"%@/@TV/%@/%@",      [SapphireMetaDataSupport collectionArtPath],
                                                                                                                show,
                                                                                                                [NSString stringWithFormat:@"Season %d", seasonNum]];
}
@end