Synnefo » Pithos MacOS Client

//

//  PithosLocalObjectState.m

//  pithos-macos

//

// Copyright 2011 GRNET S.A. 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 GRNET S.A. ``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 GRNET S.A 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.

// 

// The views and conclusions contained in the software and

// documentation are those of the authors and should not be

// interpreted as representing official policies, either expressed

// or implied, of GRNET S.A.

#import "PithosLocalObjectState.h"

#import "PithosUtilities.h"

#import "HashMapHash.h"

@implementation PithosLocalObjectState

@synthesize filePath, isDirectory, hash, tmpFilePath, fileModificationDate, fileSize;

#pragma mark -

#pragma mark Object Lifecycle

- (id)initWithFile:(NSString *)aFilePath blockHash:(NSString *)blockHash blockSize:(NSUInteger)blockSize {

    if ((self = [self init])) {

        self.filePath = aFilePath;

        if ([[NSFileManager defaultManager] fileExistsAtPath:filePath isDirectory:&isDirectory] && !isDirectory) {

            self.hash = [HashMapHash calculateHashMapHash:[HashMapHash calculateObjectHashMap:filePath 

                                                                                withBlockHash:blockHash 

                                                                                 andBlockSize:blockSize]];

        }

    }

    return self;

}

- (id)initWithHash:(NSString *)aHash directory:(BOOL)anIsDirectory {

    if ((self = [self init])) {

        if (anIsDirectory)

            isDirectory = YES;

        else if ([aHash length] == 64)

            self.hash = aHash;

    }

    return self;

}

+ (id)localObjectState {

    return [[[self alloc] init] autorelease];

}

+ (id)localObjectStateWithFile:(NSString *)aFilePath blockHash:(NSString *)blockHash blockSize:(NSUInteger)blockSize {

    return [[[self alloc] initWithFile:aFilePath blockHash:blockHash blockSize:blockSize] autorelease];

}

+ (id)localObjectStateWithHash:(NSString *)aHash directory:(BOOL)anIsDirectory {

    return [[[self alloc] initWithHash:aHash directory:anIsDirectory] autorelease];

}

- (void)dealloc {

    [fileSize release];

    [fileModificationDate release];

    self.tmpFilePath = nil;

    [filePath release];

    [hash release];

    [super dealloc];    

}

- (NSString *)description {

    return [NSString stringWithFormat:@"hash: %@, filePath: %@, isDirectory: %d, fileModificationDate: %@, fileSize: %@", 

            hash, filePath, isDirectory, fileModificationDate, fileSize];

}

- (BOOL)isEqualToState:(PithosLocalObjectState *)aState {

    if (self.isDirectory)

        // Object is a directory, check the other

        return aState.isDirectory;

    else if (aState.isDirectory)

        // Object is not a directory, while the other is

        return NO;

    else if (![self exists])

        // Object doesn't exist, check the other

        return (![aState exists]);

    else if (![aState exists])

        // Object exists, while the other doesn't

        return NO;

    else

        // Both objects exist, check that they have the same hash

        return [self.hash isEqualToString:aState.hash];

}

#pragma mark -

#pragma mark Properties

- (void)setIsDirectory:(BOOL)anIsDirectory {

    isDirectory = anIsDirectory;

    if (isDirectory) {

        self.hash = nil;

        self.tmpFilePath = nil;

    }

}

- (void)setTmpFilePath:(NSString *)aTmpFilePath {

    if (![tmpFilePath isEqualToString:aTmpFilePath]) {

        if (!aTmpFilePath) {

            NSFileManager *fileManager = [NSFileManager defaultManager];

            if ([fileManager fileExistsAtPath:tmpFilePath]) {

                NSError *error = nil;

                if (![fileManager removeItemAtPath:tmpFilePath error:&error] || error)

                    [PithosUtilities fileActionFailedAlertWithTitle:@"Remove File Error" 

                                                            message:[NSString stringWithFormat:@"Cannot remove file at '%@'", tmpFilePath] 

                                                              error:error];

            }

        }

        [tmpFilePath release];

        tmpFilePath = [aTmpFilePath retain];

    }

}

- (void)setHash:(NSString *)aHash {

    [hash release];

    if ([aHash length] == 64)

        hash = [aHash retain];

    else

        hash = nil;

    if (filePath) {

        NSDictionary *attributes = [[NSFileManager defaultManager] attributesOfItemAtPath:filePath error:nil];

        self.fileModificationDate = [attributes objectForKey:NSFileModificationDate];

        self.fileSize = [attributes objectForKey:NSFileSize];

    } else {

        self.fileModificationDate = nil;

        self.fileSize = nil;

    }

}

#pragma mark -

#pragma mark Methods

- (BOOL)exists {

    return (isDirectory || hash);

}

- (BOOL)isModified {

    NSDictionary *attributes = [[NSFileManager defaultManager] attributesOfItemAtPath:filePath error:nil];

    return (![fileSize isEqualToNumber:[attributes objectForKey:NSFileSize]] || 

            ![fileModificationDate isEqualToDate:[attributes objectForKey:NSFileModificationDate]]);

}

#pragma mark -

#pragma mark NSCoding

- (id)initWithCoder:(NSCoder *)decoder {

    if ((self = [super init])) {

        self.hash = [decoder decodeObjectForKey:@"hash"];

        self.filePath = [decoder decodeObjectForKey:@"filePath"];

        self.isDirectory = [decoder decodeBoolForKey:@"isDirectory"];

        self.tmpFilePath = [decoder decodeObjectForKey:@"tmpFilePath"];

        self.fileModificationDate = [decoder decodeObjectForKey:@"fileModificationDate"];

        self.fileSize = [decoder decodeObjectForKey:@"fileSize"];

    }

    return self;

}

- (void)encodeWithCoder:(NSCoder *)encoder {

    [encoder encodeObject:hash forKey:@"hash"];

    [encoder encodeObject:filePath forKey:@"filePath"];

    [encoder encodeBool:isDirectory forKey:@"isDirectory"];

    [encoder encodeObject:tmpFilePath forKey:@"tmpFilePath"];

    [encoder encodeObject:fileModificationDate forKey:@"fileModificationDate"];

    [encoder encodeObject:fileSize forKey:@"fileSize"];

}

@end