lupin/LupinFramework/LPFile.m

/*
 * Copyright (c) 2004-2005 The Lupin Project. All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without modification, are permitted 
 * provided that the following conditions are met:
 * 
 * 1. Redistributions of source code must retain the above copyright notice, this list of conditions 
 *    and the following disclaimer.
 * 
 * 2. Redistributions in binary form must reproduce the above copyright notice, this list of 
 *    conditions and the following disclaimer in the documentation and/or other materials provided 
 *    with the distribution.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE LUPIN PROJECT ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, 
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 
 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE LUPIN PROJECT OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, 
 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 
 * POSSIBILITY OF SUCH DAMAGE.
 */

#import "LPFile.h"

static NSComparisonResult compareFile(LPFile* in1, LPFile* in2, NSString* inKind)
{
        if ([inKind isEqualToString:@"name"])
                return [[in1 name] compare:[in2 name]];
        else if ([inKind isEqualToString:@"kind"])
                return [[in1 kind] compare:[in2 kind]];
        else if ([inKind isEqualToString:@"modifieddate"])
                return [[in1 modifiedDate] compare:[in2 modifiedDate]];
        else if ([inKind isEqualToString:@"createddate"])
                return [[in1 createdDate] compare:[in2 createdDate]];
        else if ([inKind isEqualToString:@"size"])
                return [[in1 size] compare:[in2 size]];
        else if ([inKind isEqualToString:@"kind"])
                return [[in1 kind] compare:[in2 kind]];
        else
                return NSOrderedSame;
}

static NSComparisonResult compare(id in1, id in2, void* inKind)
{
        return compareFile(in1,in2,inKind);
}

static NSComparisonResult compareReverce(id in1, id in2, void* inKind)
{
        return compareFile(in2,in1,inKind);
}


@implementation LPFile

//! root�����������������g�p���������I
- (LPFile*)init
{
        self = [super init];
        _infoDict = [NSMutableDictionary dictionary];
        [_infoDict retain];
        [_infoDict setObject:[NSMutableArray array] forKey:@"children"];
        _isDir = YES; //! root������
        return self;
}

- (LPFile*)init:(NSString*)filePath name:(NSString*)name modifiedDate:(NSDate*)modifiedDate createdDate:(NSDate*)createdDate size:(NSNumber*)size kind:(NSString*)kind isDir:(BOOL)isDir
{
        self = [super init];
        _infoDict = [NSMutableDictionary dictionary];
        [_infoDict retain];
        [_infoDict setObject:[NSMutableArray array] forKey:@"children"];
        /*
        if (filePath) { //! �t�@�C���p�X���������X���b�V�����t������������������
                NSMutableArray* array = [NSMutableArray arrayWithArray:[filePath pathComponents]];
                if ([[array lastObject] isEqualToString:@"/"]) {
                        [array removeLastObject];
                }
                [_infoDict setObject:[NSString pathWithComponents:array] forKey:@"path"];
        }
        */
        if (filePath) [_infoDict setObject:filePath forKey:@"path"];
        if (name) [_infoDict setObject:name forKey:@"name"];
        else if (filePath) [_infoDict setObject:[filePath lastPathComponent] forKey:@"name"];
        if (modifiedDate) [_infoDict setObject:modifiedDate forKey:@"modifieddate"];
        if (createdDate) [_infoDict setObject:createdDate forKey:@"createddate"];
        if (size) [_infoDict setObject:size forKey:@"size"];
        _isDir = isDir; //!�f�B���N�g��������YES���n���B�t�@�C�����A�t�@�C�����t�H���_�����������������ANO���n���B
        if (kind) [_infoDict setObject:kind forKey:@"kind"];
        else if (filePath) [_infoDict setObject:[self kindForFile:filePath] forKey:@"kind"];
        return self;
}

- (void)appendFile:(LPFile*)file hierarchy:(BOOL)hierarchy
{
        if (hierarchy)
                [self appendFile:file index:0];
        else
                [self addChild:file];
}

- (void)appendFile:(LPFile*)file index:(int)index
{
        if (!file) return;
        NSString* path = [file path];
        NSMutableArray *arr = [NSMutableArray arrayWithArray:[path pathComponents]];
        if ([[arr lastObject] isEqualToString:@"/"]) [arr removeLastObject];
        
        //NSLog(@"str=%@,lastpath=%@,count=%d,index=%d",path,lastPath,[arr count],index);
        
        if ([arr count] <= index) return;
        NSEnumerator* enumerator = [[self children] objectEnumerator];
        LPFile* childFile;
        int childIndex = 0; //For replacing a child when another child who has same name is added.
        while (childFile = [enumerator nextObject]) {
                //NSLog(@"path=%@,childFilePath=%@",path,[childFile path]);
                NSMutableArray* childPathArray = [NSMutableArray arrayWithArray:[[childFile path] pathComponents]];
                if ([[childPathArray lastObject] isEqualToString:@"/"]) [childPathArray removeLastObject];
                if ([[NSString pathWithComponents:arr] isEqualToString:[NSString pathWithComponents:childPathArray]] && ([childFile isDir] == [file isDir])) {
                        //already appended to root.
                        [[file children] setArray:[childFile children]];
                        [[self children] replaceObjectAtIndex:childIndex withObject:file];
                        return;
                }
                //NSLog(@"path=%@,childFilePath=%@",path,[NSString pathWithComponents:childPathArray]);
                if ([path hasPrefix:[NSString stringWithFormat:@"%@/",[NSString pathWithComponents:childPathArray]]]) { //! foo/var(file)��foo/(childFile)��������
                        [childFile appendFile:file index:index+1];
                        return;
                }
                childIndex++;
        }
        //NSLog(@"index=%d,%@",index,[NSString pathWithComponents:[arr subarrayWithRange:NSMakeRange(0,index+1)]]);
        if ([arr count] - index == 1)
                [self addChild:file];
        else {
                [self addChild:[[LPFile alloc] init:[NSString pathWithComponents:[arr subarrayWithRange:NSMakeRange(0,index+1)]] name:nil modifiedDate:nil createdDate:nil size:nil kind:nil isDir:YES]];
                [[[self children] lastObject] appendFile:file index:index+1];
        }
}

#pragma mark -

- (NSString*)path
{
        return [_infoDict objectForKey:@"path"];
}

- (NSString*)name
{
        return [_infoDict objectForKey:@"name"];
}

- (BOOL)isDir
{
        return _isDir;
}

- (BOOL)isExpanded
{
        return _isExpanded;
}

- (void)setExpanded:(BOOL)expanded
{
        _isExpanded = expanded;
}

- (NSString*)kind
{
        return [_infoDict objectForKey:@"kind"];
}

- (NSString*)kindForFile:(NSString*)path
{
        NSString *res;
        OSType type,creator;
        if (_isDir || [[self children] count] != 0) {
                type = 'fold';
                creator = 'MACS';
        } else {
                type = kLSUnknownType;
                creator = kLSUnknownCreator;
        }
        
        LSCopyKindStringForTypeInfo(type,creator,(CFStringRef)[path pathExtension],(CFStringRef*)&res);
        return (res) ? [res autorelease] : nil;
}

- (NSImage*)icon
{
        return [_infoDict objectForKey:@"icon"];
}

- (void)setIcon:(NSImage*)icon
{
        [_infoDict setObject:icon forKey:@"icon"];
}

- (NSNumber*)size
{
        return [_infoDict objectForKey:@"size"];
}

- (NSNumber*)totalSize
{
        NSNumber* _totalSize = [_infoDict objectForKey:@"totalsize"];
        if (_totalSize) return _totalSize;
        unsigned long long childrenSize = ([self size]) ? [[self size] unsignedLongLongValue] : 0;
        NSEnumerator* enumerator = [[self children] objectEnumerator];
        LPFile* childFile;
        while (childFile = [enumerator nextObject]) {
                childrenSize += [[childFile totalSize] unsignedLongLongValue];
        }
        return [NSNumber numberWithUnsignedLongLong:childrenSize];
}

- (void)setTotalSize:(NSNumber*)_totalSize
{
        [_infoDict setObject:_totalSize forKey:@"totalsize"];
}

//�t�@�C���������q������������
- (NSNumber*)totalNumOfChildren
{
        NSNumber* _totalNumOfChildren = [_infoDict objectForKey:@"totalchildren"];
        if (_totalNumOfChildren) return _totalNumOfChildren;
        unsigned int numOfChildren = 0;
        NSEnumerator* enumerator = [[self children] objectEnumerator];
        LPFile* childFile;
        while (childFile = [enumerator nextObject]) {
                numOfChildren += [[childFile totalNumOfChildren] unsignedIntValue];
                numOfChildren++;
        }
        return [NSNumber numberWithUnsignedInt:numOfChildren];
}

- (void)setTotalNumOfChildren:(NSNumber*)_totalNumOfChildren
{
        [_infoDict setObject:_totalNumOfChildren forKey:@"totalchildren"];
}

- (NSString*)convertSize
{
        NSNumber* _fileSize = [self size];
        if (!_fileSize) return nil;
        unsigned long long fileSize = [_fileSize unsignedLongLongValue];
        NSString* sizeStr;
        if (fileSize < 1024)
                sizeStr = [NSString stringWithFormat:@"%lld %@",fileSize,
                        [[NSBundle bundleForClass:[self class]] localizedStringForKey:@"SizeByteFormat" value:nil table:nil]];
        else if (fileSize < 1024*1024) {
                fileSize >>= 10;
                sizeStr = [NSString stringWithFormat:@"%lld %@",fileSize,@"KB"];
        } else {
                NSNumberFormatter *numberFormatter = [[[NSNumberFormatter alloc] init] autorelease];
                [numberFormatter setFormat:@"###,###,###.#"];
                
                double moreFileSize = fileSize / 1048576;
                sizeStr = [NSString stringWithFormat:@"%@ %@",[numberFormatter stringForObjectValue:[NSNumber numberWithDouble:moreFileSize]],@"MB"];
        }
        return sizeStr;
}

- (NSDate*)modifiedDate
{
        return [_infoDict objectForKey:@"modifieddate"];
}

- (NSDate*)createdDate
{
        return [_infoDict objectForKey:@"createddate"];
}

- (NSString*)convertModifiedDateWithWidth:(float)width
{
        return [self convertDate:[_infoDict objectForKey:@"modifieddate"] width:width];
}

- (NSString*)convertCreatedDateWithWidth:(float)width
{
        return [self convertDate:[_infoDict objectForKey:@"createddate"] width:width];
}

- (NSString*)convertDate:(NSDate*)_date width:(float)width
{
        if (!_date) return nil;
        NSDictionary* locale = [[NSUserDefaults standardUserDefaults] dictionaryRepresentation];
        NSString* dateTypeStr;
        if (width < 120) dateTypeStr = @"DateFormatFlat";
        else if (width < 150) dateTypeStr = @"DateFormatShort";
        else if (width < 200) dateTypeStr = @"DateFormatMiddle";
        else dateTypeStr = @"DateFormatLong";
        return [_date descriptionWithCalendarFormat:[[NSBundle bundleForClass:[self class]] localizedStringForKey:dateTypeStr value:@"" table:nil] 
                        timeZone:[NSTimeZone systemTimeZone] 
                        locale:locale];
}

#pragma mark -

- (void)sortBy:(NSString*)identifier order:(BOOL)order
{
        NSMutableArray* children = [self children];
        if (order)
                [children sortUsingFunction:compare context:identifier];
        else
                [children sortUsingFunction:compareReverce context:identifier];
        NSEnumerator* enumerator = [children objectEnumerator];
        LPFile* file;
        while (file = [enumerator nextObject]) {
                [file sortBy:identifier order:order];
        }
}

#pragma mark -

- (NSMutableArray*)children
{
        return [_infoDict objectForKey:@"children"];
}

- (void)addChild:(LPFile*)child
{
        [[_infoDict objectForKey:@"children"] addObject:child];
        if (!_isDir) {
                _isDir = YES;
                [_infoDict setObject:[self kindForFile:[self path]] forKey:@"kind"];
        }
}

#pragma mark -

- (void)dealloc
{
        [_infoDict release];
        [super dealloc];
}

@end