Synnefo » Pithos iOS Client

//

//  GetObjectsRequest.m

//  OpenStack

//

//  Created by Mike Mayo on 12/24/10.

//  The OpenStack project is provided under the Apache 2.0 license.

//

#import "GetObjectsRequest.h"

#import "OpenStackAccount.h"

#import "Container.h"

#import "StorageObject.h"

#import "AccountManager.h"

#import "Folder.h"

#import "SBJSON.h"

#import "NSString+Conveniences.h"

@implementation GetObjectsRequest

@synthesize container;

+ (id)request:(OpenStackAccount *)account method:(NSString *)method url:(NSURL *)url {

	GetObjectsRequest *request = [[[GetObjectsRequest alloc] initWithURL:url] autorelease];

    request.account = account;

	[request setRequestMethod:method];

	[request addRequestHeader:@"X-Auth-Token" value:[account authToken]];

    [request addRequestHeader:@"Content-Type" value:@"application/json"];

	return request;

}

+ (id)filesRequest:(OpenStackAccount *)account method:(NSString *)method path:(NSString *)path marker:(NSString *)marker {

    NSString *now = [[[NSDate date] description] stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding];

    NSString *filesUrl;

    NSString *urlString = [account.filesURL description];

    if (account.sharingAccount) {

        NSRange authUserRange = [urlString rangeOfString:account.username];

        filesUrl = [NSString stringWithFormat:@"%@%@",

                    [urlString substringToIndex:authUserRange.location],

                    account.sharingAccount];

    } else

        filesUrl = urlString;

	NSURL *url = [NSURL URLWithString:[NSString stringWithFormat:@"%@%@?format=json%@%@&now=%@",

                                       filesUrl,

                                       path,

                                       account.shared ? @"&shared=" : @"",

                                       marker ? [NSString stringWithFormat:@"&marker=%@", marker] : @"",

                                       now]];

    return [GetObjectsRequest request:account method:method url:url];

}

+ (id)filesRequest:(OpenStackAccount *)account method:(NSString *)method path:(NSString *)path {

    return [GetObjectsRequest filesRequest:account method:method path:path marker:nil];

}

+ (GetObjectsRequest *)request:(OpenStackAccount *)account container:(Container *)container {

    return [GetObjectsRequest request:account container:container marker:nil objectsBuffer:nil];

}

+ (GetObjectsRequest *)request:(OpenStackAccount *)account

                     container:(Container *)container

                        marker:(NSString *)marker

                 objectsBuffer:(NSMutableDictionary *)objectsBuffer

{

    GetObjectsRequest *request = [GetObjectsRequest filesRequest:account method:@"GET"

                                                            path:[NSString stringWithFormat:@"/%@", [NSString encodeToPercentEscape:container.name]] 

                                                          marker:[NSString encodeToPercentEscape:marker]];    

    request.account = account;

    request.container = container;

    if (objectsBuffer == nil)

        objectsBuffer = [NSMutableDictionary dictionary];

    [request setUserInfo:[NSMutableDictionary dictionaryWithObject:objectsBuffer forKey:@"objectsBuffer"]];

    [request setTimeOutSeconds:60];

    [request setNumberOfTimesToRetryOnTimeout:5];

    return request;

}

- (void)requestFinished {

    if ([self isSuccess]) {

        Container *aContainer = self.container;  //[self.userInfo objectForKey:@"container"];

        NSMutableDictionary *objects = [self objects];

        NSMutableDictionary *objectsBuffer = [self.userInfo objectForKey:@"objectsBuffer"];

        SBJSON *parser = [[SBJSON alloc] init];

        NSArray *jsonObjects = [parser objectWithString:[self responseString]];

        NSUInteger numberOfObjectsInResponse = [jsonObjects count];

        [parser release];

        [objectsBuffer addEntriesFromDictionary:objects];

        if (numberOfObjectsInResponse < 10000) {

            aContainer.rootFolder = [Folder folder];

            aContainer.rootFolder.objects = objectsBuffer;

            [self.account persist];

            NSNotification *notification = [NSNotification notificationWithName:@"getObjectsSucceeded" object:self.container userInfo:

                                            [NSDictionary dictionaryWithObjectsAndKeys:aContainer,@"container",

                                             self, @"request",nil]];

            [[NSNotificationCenter defaultCenter] postNotification:notification];

        } else {

            NSString *marker = [[jsonObjects lastObject] objectForKey:@"name"];

            [self.account.manager getObjects:self.container afterMarker:marker objectsBuffer:objectsBuffer];

        }        

    } else {

        NSNotification *notification = [NSNotification notificationWithName:@"getObjectsFailed" object:self.container userInfo:[NSDictionary dictionaryWithObject:self forKey:@"request"]];

        [[NSNotificationCenter defaultCenter] postNotification:notification];

    }

    [super requestFinished];

}

- (void)failWithError:(NSError *)theError {

    NSNotification *notification = [NSNotification notificationWithName:@"getObjectsFailed" object:self.container userInfo:[NSDictionary dictionaryWithObject:self forKey:@"request"]];

    [[NSNotificationCenter defaultCenter] postNotification:notification];

    [super failWithError:theError];

}

- (void)dealloc {

    [container release];

    [super dealloc];

}

@end