Synnefo » Pithos MacOS Client

//

//  PithosAccountNode.m

//  pithos-macos

//

// Copyright 2011-2012 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 "PithosAccountNode.h"

#import "PithosContainerNode.h"

#import "ASIPithos.h"

#import "ASIPithosAccountRequest.h"

#import "ASIPithosAccount.h"

#import "ASIPithosContainer.h"

#import "ASIDownloadCache.h"

#import "PithosUtilities.h"

#import "PithosActivityFacility.h"

static NSImage *sharedIcon = nil;

@implementation PithosAccountNode

@synthesize pithos, pithosAccount;

+ (void)initialize {

	if (self == [PithosAccountNode class])

        sharedIcon = [[NSWorkspace sharedWorkspace] iconForFileType:NSFileTypeForHFSTypeCode(kUserIcon)];

}

#pragma mark -

#pragma mark Object Lifecycle

- (id)initWithPithos:(ASIPithos *)aPithos {

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

        pithos = aPithos;

    }

    return self;

}

- (void)dealloc {

    [accountRequest clearDelegatesAndCancel];

    [refreshMetadataAccountRequest clearDelegatesAndCancel];

    [applyMetadataAccountRequest clearDelegatesAndCancel];

}

#pragma mark -

#pragma mark Properties

- (void)setPithos:(ASIPithos *)aPithos {

    if (aPithos && ![aPithos isEqualTo:pithos]) {

        pithos = aPithos;

        url = nil;

        [accountRequest clearDelegatesAndCancel];

        accountRequest = nil;

        [refreshMetadataAccountRequest clearDelegatesAndCancel];

        refreshMetadataAccountRequest = nil;

        [applyMetadataAccountRequest clearDelegatesAndCancel];

        applyMetadataAccountRequest = nil;

        reset = YES;

    }

}

- (NSString *)url {

    if (url == nil)

        url = [[NSString alloc] initWithFormat:@"%@%@", 

               (sharingAccount ? [pithos storageURLWithAuthUser:sharingAccount] : pithos.storageURL), 

               (shared ? @"?shared" : @"")];

    return url;

}

- (NSArray *)children {

    @synchronized(self) {

        if (reset) {

            [accountRequest clearDelegatesAndCancel];

            accountRequest = nil;

            [refreshMetadataAccountRequest clearDelegatesAndCancel];

            refreshMetadataAccountRequest = nil;

            [applyMetadataAccountRequest clearDelegatesAndCancel];

            applyMetadataAccountRequest = nil;

            children = nil;

            newChildren = nil;

            self.pithosAccount = nil;

            freshness = PithosNodeStateRefreshNeeded;

            forcedRefresh = YES;

            reset = NO;

            [self postChildrenUpdatedNotificationName];

        }

        switch (freshness) {

            case PithosNodeStateFresh:

                break;

            case PithosNodeStateRefreshNeeded:

                freshness = PithosNodeStateRefreshing;

                accountRequest = [ASIPithosAccountRequest listContainersRequestWithPithos:pithos 

                                                                                     limit:0 

                                                                                    marker:nil 

                                                                                    shared:shared 

                                                                                     until:nil];

                if (sharingAccount)

                    [accountRequest setRequestUserFromDefaultTo:sharingAccount withPithos:pithos];

                else if (!forcedRefresh)

                    accountRequest.downloadCache = [ASIDownloadCache sharedCache];

                accountRequest.delegate = self;

                accountRequest.didFinishSelector = @selector(performRequestFinishedDelegateInBackground:);

                accountRequest.didFailSelector = @selector(performRequestFailedDelegateInBackground:);

                accountRequest.userInfo = [NSMutableDictionary dictionaryWithObjectsAndKeys:

                                           [NSNumber numberWithInteger:NSOperationQueuePriorityVeryHigh], @"priority", 

                                           [NSNumber numberWithUnsignedInteger:10], @"retries", 

                                           NSStringFromSelector(@selector(accountRequestFinished:)), @"didFinishSelector", 

                                           NSStringFromSelector(@selector(accountRequestFailed:)), @"didFailSelector", 

                                           nil];

                [[PithosUtilities prepareRequest:accountRequest priority:NSOperationQueuePriorityVeryHigh] startAsynchronous];

                break;

            case PithosNodeStateRefreshing:

                break;

            case PithosNodeStateRefreshFinished:

                if (newChildren) {

                    children = newChildren;

                    newChildren = nil;

                }

                freshness = PithosNodeStateFresh;

            default:

                break;

        }

        return children;

    }

}

- (NSString *)displayName {

    if (displayName == nil)

        return [NSString stringWithString:(sharingAccount ? sharingAccount : @"account")];

    return [displayName copy];

}

- (NSImage *)icon {

    if (icon == nil)

        icon = sharedIcon;

    return icon;

}

#pragma mark -

#pragma mark ASIHTTPRequestDelegate

- (void)accountRequestFailed:(ASIPithosAccountRequest *)request {

    @autoreleasepool {

        NSString *message;

        NSError *error = [accountRequest error];

        if (error)

            message = [NSString stringWithFormat:@"Account listing %@ failed: %@", accountRequest.url, [error localizedDescription]];

        else

            message = [NSString stringWithFormat:@"Account listing %@ failed: (%d) %@", 

                       accountRequest.url, accountRequest.responseStatusCode, accountRequest.responseStatusMessage];

        dispatch_async(dispatch_get_main_queue(), ^{

            [[PithosActivityFacility defaultPithosActivityFacility] startAndEndActivityWithType:PithosActivityOther message:message];

        });

        NSUInteger retries = [[accountRequest.userInfo objectForKey:@"retries"] unsignedIntegerValue];

        if (retries > 0) {

            ASIPithosAccountRequest *newAccountRequest = (ASIPithosAccountRequest *)[PithosUtilities copyRequest:accountRequest];

            [(NSMutableDictionary *)(newAccountRequest.userInfo)setObject:[NSNumber numberWithUnsignedInteger:(--retries)] forKey:@"retries"];

            accountRequest = newAccountRequest;

            [[PithosUtilities prepareRequest:accountRequest priority:[[accountRequest.userInfo objectForKey:@"priority"] integerValue]] startAsynchronous];

        } else {

            newChildren = nil;

            accountRequest = nil;

            containers = nil;

            forcedRefresh = NO;

            @synchronized(self) {

                freshness = PithosNodeStateRefreshNeeded;

            }

        }

    }

}

- (void)accountRequestFinished:(ASIPithosAccountRequest *)request {

    @autoreleasepool {

        DLog(@"List account finished: %@", [accountRequest url]);

        DLog(@"Cached: %d", [accountRequest didUseCachedResponse]);

        if (accountRequest.responseStatusCode == 200) {

            self.pithosAccount = [accountRequest account];

            NSArray *someContainers = [accountRequest containers];

            if (containers == nil) {

                containers = [[NSMutableArray alloc] initWithArray:someContainers];

            } else {

                [containers addObjectsFromArray:someContainers];

            }

            if ([someContainers count] < 10000) {

                if (!accountRequest.didUseCachedResponse || ([containers count] != [someContainers count]) || !children) {

                    // Save new children

                    DLog(@"using newChildren");

                    newChildren = [[NSMutableArray alloc] init];

                    NSMutableIndexSet *keptNodes = [NSMutableIndexSet indexSet];

                    for (ASIPithosContainer *container in containers) {

                        PithosContainerNode *node = [[PithosContainerNode alloc] initWithPithos:pithos pithosContainer:container];

                        node.parent = self;

                        node.shared = shared;

                        node.sharingAccount = sharingAccount;

                        node.inheritChildrenUpdatedNotificationName = inheritChildrenUpdatedNotificationName;

                        if (children) {

                            NSUInteger oldIndex = [children indexOfObject:node];

                            if (oldIndex != NSNotFound) {

                                // Use the same pointer value, if possible

                                node = [children objectAtIndex:oldIndex];

    //                            node.pithosContainer = container;

                                [node setLimitedPithosContainer:container];

                                [keptNodes addIndex:oldIndex];

                            }

                        }

                        [newChildren addObject:node];

                    }

                    [[children objectsAtIndexes:

                      [[NSIndexSet indexSetWithIndexesInRange:NSMakeRange(0, [children count])] indexesPassingTest:^(NSUInteger idx, BOOL *stop){

                        if ([keptNodes containsIndex:idx])

                            return NO;

                        return YES;

                    }]] makeObjectsPerformSelector:@selector(pithosNodeWillBeRemoved)];

                }

                // Else cache was used and all results were fetched during this request, so existing children can be reused

                accountRequest = nil;

                containers = nil;

                forcedRefresh = NO;

                @synchronized(self) {

                    freshness = PithosNodeStateRefreshFinished;

                }

                [self postChildrenUpdatedNotificationName];

            } else {

                // Do an additional request to fetch more objects

                accountRequest = [ASIPithosAccountRequest listContainersRequestWithPithos:pithos 

                                                                                     limit:0 

                                                                                    marker:[[someContainers lastObject] name] 

                                                                                    shared:shared 

                                                                                     until:nil];

                if (sharingAccount)

                    [accountRequest setRequestUserFromDefaultTo:sharingAccount withPithos:pithos];

                else if (!forcedRefresh)

                    accountRequest.downloadCache = [ASIDownloadCache sharedCache];

                accountRequest.delegate = self;

                accountRequest.didFinishSelector = @selector(performRequestFinishedDelegateInBackground:);

                accountRequest.didFailSelector = @selector(performRequestFailedDelegateInBackground:);

                accountRequest.userInfo = [NSMutableDictionary dictionaryWithObjectsAndKeys:

                                           [NSNumber numberWithInteger:NSOperationQueuePriorityVeryHigh], @"priority", 

                                           [NSNumber numberWithUnsignedInteger:10], @"retries", 

                                           NSStringFromSelector(@selector(accountRequestFinished:)), @"didFinishSelector", 

                                           NSStringFromSelector(@selector(accountRequestFailed:)), @"didFailSelector", 

                                           nil];

                [[PithosUtilities prepareRequest:accountRequest priority:NSOperationQueuePriorityVeryHigh] startAsynchronous];

            }

        } else if (accountRequest.responseStatusCode == 304) {

            // Account is not modified, so existing children can be reused

            accountRequest = nil;

            containers = nil;

            forcedRefresh = NO;

            @synchronized(self) {

                freshness = PithosNodeStateRefreshFinished;

            }

            [self postChildrenUpdatedNotificationName];

        } else {

            [self accountRequestFailed:accountRequest];

        }

    }

}

- (void)accountMetadataRequestFinished:(ASIPithosAccountRequest *)request {

    @autoreleasepool {

        DLog(@"URL: %@", [request url]);

        DLog(@"cached: %d", [request didUseCachedResponse]);

        if ([request isEqualTo:applyMetadataAccountRequest]) {

            @synchronized(self) {

                applyMetadataAccountRequest = nil;

            }

            [self refreshInfo];

        } else if ([request isEqualTo:refreshMetadataAccountRequest]) {

            self.pithosAccount = [refreshMetadataAccountRequest account];

            @synchronized(self) {

                refreshMetadataAccountRequest = nil;

            }

        }

    }

}

- (void)accountMetadataRequestFailed:(ASIPithosAccountRequest *)request {

    @autoreleasepool {

        NSUInteger retries = [[request.userInfo objectForKey:@"retries"] unsignedIntegerValue];

        if (retries > 0) {

            ASIPithosAccountRequest *newRequest = (ASIPithosAccountRequest *)[PithosUtilities copyRequest:request];

            [(NSMutableDictionary *)(newRequest.userInfo)setObject:[NSNumber numberWithUnsignedInteger:(--retries)] forKey:@"retries"];

            if ([request isEqualTo:applyMetadataAccountRequest]) {

                @synchronized(self) {

                    applyMetadataAccountRequest = newRequest;

                }

            } else if ([request isEqualTo:refreshMetadataAccountRequest]) {

                @synchronized(self) {

                    refreshMetadataAccountRequest = newRequest;

                }

            }

            [[PithosUtilities prepareRequest:newRequest priority:[[newRequest.userInfo objectForKey:@"priority"] integerValue]] startAsynchronous];

        } else {

            if ([request isEqualTo:applyMetadataAccountRequest]) {

                [PithosUtilities httpRequestErrorAlertWithRequest:applyMetadataAccountRequest];

                @synchronized(self) {

                    applyMetadataAccountRequest = nil;

                }

            } else if ([request isEqualTo:refreshMetadataAccountRequest]) {

                [PithosUtilities httpRequestErrorAlertWithRequest:refreshMetadataAccountRequest];

                @synchronized(self) {

                    refreshMetadataAccountRequest = nil;

                }

            }

        }

    }

}

#pragma mark -

#pragma mark Info

- (void)applyInfo {

    @synchronized(self) {

        if (applyMetadataAccountRequest == nil) {

            NSMutableDictionary *groups = pithosAccount.groups;

            if ([groups count] == 0)

                groups = [NSMutableDictionary dictionaryWithObject:@"" forKey:@"group"];

            applyMetadataAccountRequest = [ASIPithosAccountRequest updateAccountMetadataRequestWithPithos:pithos 

                                                                                                    groups:groups 

                                                                                                  metadata:pithosAccount.metadata 

                                                                                                    update:NO];

            applyMetadataAccountRequest.delegate = self;

            applyMetadataAccountRequest.didFinishSelector = @selector(performRequestFinishedDelegateInBackground:);

            applyMetadataAccountRequest.didFailSelector = @selector(performRequestFailedDelegateInBackground:);

            applyMetadataAccountRequest.userInfo = [NSMutableDictionary dictionaryWithObjectsAndKeys:

                                                    [NSNumber numberWithInteger:NSOperationQueuePriorityHigh], @"priority", 

                                                    [NSNumber numberWithUnsignedInteger:10], @"retries", 

                                                    NSStringFromSelector(@selector(accountMetadataRequestFinished:)), @"didFinishSelector", 

                                                    NSStringFromSelector(@selector(accountMetadataRequestFailed:)), @"didFailSelector", 

                                                    nil];

            [[PithosUtilities prepareRequest:applyMetadataAccountRequest priority:NSOperationQueuePriorityHigh] startAsynchronous];

        }

    }

}

- (void)refreshInfo {

    @synchronized(self) {

        if (refreshMetadataAccountRequest == nil) {

            refreshMetadataAccountRequest = [ASIPithosAccountRequest accountMetadataRequestWithPithos:pithos];

            refreshMetadataAccountRequest.delegate = self;

            refreshMetadataAccountRequest.didFinishSelector = @selector(performRequestFinishedDelegateInBackground:);

            refreshMetadataAccountRequest.didFailSelector = @selector(performRequestFailedDelegateInBackground:);

            refreshMetadataAccountRequest.userInfo = [NSMutableDictionary dictionaryWithObjectsAndKeys:

                                                      [NSNumber numberWithInteger:NSOperationQueuePriorityHigh], @"priority", 

                                                      [NSNumber numberWithUnsignedInteger:10], @"retries", 

                                                      NSStringFromSelector(@selector(accountMetadataRequestFinished:)), @"didFinishSelector", 

                                                      NSStringFromSelector(@selector(accountMetadataRequestFailed:)), @"didFailSelector", 

                                                      nil];

            if (!sharingAccount)

                refreshMetadataAccountRequest.downloadCache = [ASIDownloadCache sharedCache];

            [[PithosUtilities prepareRequest:refreshMetadataAccountRequest priority:NSOperationQueuePriorityHigh] startAsynchronous];

        }

    }

}

@end