//
//  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)] retain];
}

#pragma mark -
#pragma mark Object Lifecycle

- (id)initWithPithos:(ASIPithos *)aPithos {
    if ((self = [super init])) {
        self.pithos = aPithos;
    }
    return self;
}

- (void)dealloc {
    [accountRequest clearDelegatesAndCancel];
    [accountRequest release];
    [refreshMetadataAccountRequest clearDelegatesAndCancel];
    [refreshMetadataAccountRequest release];
    [applyMetadataAccountRequest clearDelegatesAndCancel];
    [applyMetadataAccountRequest release];
    [containers release];
    [pithosAccount release];
    [pithos release];
    [super dealloc];
}

#pragma mark -
#pragma mark Properties

- (void)setPithos:(ASIPithos *)aPithos {
    if (aPithos && ![aPithos isEqualTo:pithos]) {
        [pithos release];
        pithos = [aPithos retain];
        [url release];
        url = nil;
    }
}

- (NSString *)url {
    if (url == nil)
        url = [[NSString alloc] initWithFormat:@"%@%@", 
               (sharingAccount ? [pithos storageURLWithAuthUser:sharingAccount] : pithos.storageURL), 
               (shared ? @"?shared" : @"")];
    return url;
}

- (NSArray *)children {
    @synchronized(self) {
        switch (freshness) {
            case PithosNodeStateFresh:
                break;
            case PithosNodeStateRefreshNeeded:
                freshness = PithosNodeStateRefreshing;
                accountRequest = [[ASIPithosAccountRequest listContainersRequestWithPithos:pithos 
                                                                                     limit:0 
                                                                                    marker:nil 
                                                                                    shared:shared 
                                                                                     until:nil] retain];
                if (sharingAccount)
                    [accountRequest setRequestUserFromDefaultTo:sharingAccount withPithos:pithos];
                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];
                if (!forcedRefresh)
                    accountRequest.downloadCache = [ASIDownloadCache sharedCache];
                [[PithosUtilities prepareRequest:accountRequest priority:NSOperationQueuePriorityVeryHigh] startAsynchronous];
                break;
            case PithosNodeStateRefreshing:
                break;
            case PithosNodeStateRefreshFinished:
                if (newChildren) {
                    [children release];
                    children = newChildren;
                    newChildren = nil;
                }
                freshness = PithosNodeStateFresh;
            default:
                break;
        }
        return children;
    }
}

- (NSString *)displayName {
    if (displayName == nil)
        return [NSString stringWithString:(sharingAccount ? sharingAccount : @"account")];
    return [[displayName copy] autorelease];
}

- (NSImage *)icon {
    if (icon == nil)
        icon = [sharedIcon retain];
    return icon;
}

#pragma mark -
#pragma mark ASIHTTPRequestDelegate

- (void)accountRequestFailed:(ASIPithosAccountRequest *)request {
    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
    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 release];
        accountRequest = newAccountRequest;
        [[PithosUtilities prepareRequest:accountRequest priority:[[accountRequest.userInfo objectForKey:@"priority"] integerValue]] startAsynchronous];
    } else {
        NSString *message;
        NSError *error = [accountRequest error];
        if (error)
            message = [NSString stringWithFormat:@"Account listing failed: %@", error];
        else
            message = [NSString stringWithFormat:@"Account listing failed: (%d) %@", 
                       accountRequest.responseStatusCode, accountRequest.responseStatusMessage];
        dispatch_async(dispatch_get_main_queue(), ^{
            [[PithosActivityFacility defaultPithosActivityFacility] startAndEndActivityWithType:PithosActivityOther message:message];
        });
        [newChildren release];
        newChildren = nil;
        [accountRequest release];
        accountRequest = nil;
        [containers release];
        containers = nil;
        forcedRefresh = NO;
        @synchronized(self) {
            freshness = PithosNodeStateRefreshNeeded;
        }
    }
    [pool drain];
}

- (void)accountRequestFinished:(ASIPithosAccountRequest *)request {
    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
    NSLog(@"List account finished: %@", [accountRequest url]);
    NSLog(@"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
                NSLog(@"using newChildren");
                newChildren = [[NSMutableArray alloc] init];
                NSMutableIndexSet *keptNodes = [NSMutableIndexSet indexSet];
                for (ASIPithosContainer *container in containers) {
                    PithosContainerNode *node = [[[PithosContainerNode alloc] initWithPithos:pithos pithosContainer:container] autorelease];
                    node.parent = self;
                    node.shared = shared;
                    node.sharingAccount = sharingAccount;
                    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 release];
            accountRequest = nil;
            [containers release];
            containers = nil;
            forcedRefresh = NO;
            @synchronized(self) {
                freshness = PithosNodeStateRefreshFinished;
            }
            // Notify observers that children are updated
            [[NSNotificationCenter defaultCenter] postNotificationName:@"PithosAccountNodeChildrenUpdated" object:self];
        } else {
            [accountRequest release];
            // Do an additional request to fetch more objects
            accountRequest = [[ASIPithosAccountRequest listContainersRequestWithPithos:pithos 
                                                                                 limit:0 
                                                                                marker:[[someContainers lastObject] name] 
                                                                                shared:shared 
                                                                                 until:nil] retain];
            if (sharingAccount)
                [accountRequest setRequestUserFromDefaultTo:sharingAccount withPithos:pithos];
            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];
            if (!forcedRefresh)
                accountRequest.downloadCache = [ASIDownloadCache sharedCache];
            [[PithosUtilities prepareRequest:accountRequest priority:NSOperationQueuePriorityVeryHigh] startAsynchronous];
        }
    } else if (accountRequest.responseStatusCode == 304) {
        // Account is not modified, so existing children can be reused
        [accountRequest release];
        accountRequest = nil;
        [containers release];
        containers = nil;
        forcedRefresh = NO;
        @synchronized(self) {
            freshness = PithosNodeStateRefreshFinished;
        }
        // Notify observers that children are updated
        [[NSNotificationCenter defaultCenter] postNotificationName:@"PithosAccountNodeChildrenUpdated" object:self];
    } else {
        [self accountRequestFailed:accountRequest];
    }
    [pool drain];
}

- (void)accountMetadataRequestFinished:(ASIPithosAccountRequest *)request {
    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
    NSLog(@"URL: %@", [request url]);
    NSLog(@"cached: %d", [request didUseCachedResponse]);
    
    if ([request isEqualTo:applyMetadataAccountRequest]) {
        @synchronized(self) {
            [applyMetadataAccountRequest release];
            applyMetadataAccountRequest = nil;
        }
        [self refreshInfo];
    } else if ([request isEqualTo:refreshMetadataAccountRequest]) {
        self.pithosAccount = [refreshMetadataAccountRequest account];
        @synchronized(self) {
            [refreshMetadataAccountRequest release];
            refreshMetadataAccountRequest = nil;
        }
    }
    [pool drain];
}

- (void)accountMetadataRequestFailed:(ASIPithosAccountRequest *)request {
    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
    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 release];
                applyMetadataAccountRequest = newRequest;
            }
        } else if ([request isEqualTo:refreshMetadataAccountRequest]) {
            @synchronized(self) {
                [refreshMetadataAccountRequest release];
                refreshMetadataAccountRequest = newRequest;
            }
        }
        [[PithosUtilities prepareRequest:newRequest priority:[[newRequest.userInfo objectForKey:@"priority"] integerValue]] startAsynchronous];
    } else {
        if ([request isEqualTo:applyMetadataAccountRequest]) {
            dispatch_async(dispatch_get_main_queue(), ^{
                [PithosUtilities httpRequestErrorAlertWithRequest:applyMetadataAccountRequest];
            });
            @synchronized(self) {
                [applyMetadataAccountRequest release];
                applyMetadataAccountRequest = nil;
            }
        } else if ([request isEqualTo:refreshMetadataAccountRequest]) {
            dispatch_async(dispatch_get_main_queue(), ^{
                [PithosUtilities httpRequestErrorAlertWithRequest:refreshMetadataAccountRequest];
            });
            @synchronized(self) {
                [refreshMetadataAccountRequest release];
                refreshMetadataAccountRequest = nil;
            }
        }
    }
    [pool drain];
}

#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] retain];
            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] retain];
            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];
            refreshMetadataAccountRequest.downloadCache = [ASIDownloadCache sharedCache];
            [[PithosUtilities prepareRequest:refreshMetadataAccountRequest priority:NSOperationQueuePriorityHigh] startAsynchronous];
        }
    }
}

@end