//
// Copyright (c) Zach Wily
// All rights reserved.
// 
// Redistribution and use in source and binary forms, with or without modification, 
// are permitted provided that the following conditions are met:
// 
// - Redistributions of source code must retain the above copyright notice, this 
//     list of conditions and the following disclaimer.
// 
// - 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.
// 
// - Neither the name of Zach Wily nor the names of its contributors may be used to 
//     endorse or promote products derived from this software without specific prior 
//     written permission.
// 
//  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "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 COPYRIGHT OWNER 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 "ZWGallery.h"
#import "ZWGalleryAlbum.h"
#import "NSString+misc.h"
#import "ZWURLConnection.h"
#import "InterThreadMessaging.h"
#import "ZWMutableURLRequest.h"
 
@interface ZWGallery (PrivateAPI)
- (void)loginThread:(NSDictionary *)threadDispatchInfo;
- (ZWGalleryRemoteStatusCode)doLogin;
 
- (void)getAlbumsThread:(NSDictionary *)threadDispatchInfo;
- (ZWGalleryRemoteStatusCode)doGetAlbums;
 
- (void)createAlbumThread:(NSDictionary *)threadDispatchInfo;
- (ZWGalleryRemoteStatusCode)doCreateAlbumWithName:(NSString *)name title:(NSString *)title summary:(NSString *)summary parent:(ZWGalleryAlbum *)parent;
 
@end
 
@implementation ZWGallery
 
#pragma mark Object Life Cycle
 
- (id)init {
    return self;
}
 
- (id)initWithURL:(NSURL*)newUrl username:(NSString*)newUsername {
    url = [newUrl retain];
    fullURL = [[NSURL alloc] initWithString:[[url absoluteString] stringByAppendingString:@"gallery_remote2.php"]];
    username = [newUsername retain];
    delegate = self;
    loggedIn = FALSE;
    majorVersion = 0;
    minorVersion = 0;
    type = GalleryTypeG1;
    
    return self;
}
 
- (id)initWithDictionary:(NSDictionary*)dictionary {
    [self initWithURL:[NSURL URLWithString:[dictionary objectForKey:@"url"]] 
             username:[dictionary objectForKey:@"username"]];
    
    NSNumber *typeNumber = [dictionary objectForKey:@"type"];
    if (typeNumber)
        type = [typeNumber intValue];
    
    return self;
}
 
+ (ZWGallery*)galleryWithURL:(NSURL*)newUrl username:(NSString*)newUsername {
    return [[[ZWGallery alloc] initWithURL:newUrl username:newUsername] autorelease];
}
 
+ (ZWGallery*)galleryWithDictionary:(NSDictionary*)dictionary {
    return [[[ZWGallery alloc] initWithDictionary:dictionary] autorelease];
}
 
- (void)dealloc
{
    [url release];
    [username release];
    [password release];
    [albums release];
    [lastCreatedAlbumName release];
    
    [super dealloc];
}
 
#pragma mark NSComparisonMethods
 
- (BOOL)isEqual:(id)gal
{
    return ([username isEqual:[gal username]] && [[url absoluteString] isEqual:[[gal url] absoluteString]]);
}
 
- (NSComparisonResult)compare:(id)gal
{
    return [[self identifier] caseInsensitiveCompare:[gal identifier]];
}
 
#pragma mark Accessors
 
- (void)setDelegate:(id)newDelegate {
    delegate = newDelegate;
}
 
- (id)delegate {
    return delegate;
}
 
- (void)setPassword:(NSString*)newPassword {
    [newPassword retain];
    [password release];
    password = newPassword;
}
 
- (NSURL*)url {
    return url;
}
 
- (NSURL*)fullURL {
    return fullURL;
}
 
- (NSString*)identifier {
    return [NSString stringWithFormat:@"%@%@ (%@)", [url host], [url path], username];
}
 
- (NSString*)urlString {
    return [url absoluteString];
}
 
- (NSString*)username {
    return username;
}
 
- (int)majorVersion {
    return majorVersion;
}
 
- (int)minorVersion {
    return minorVersion;
}
 
- (BOOL)loggedIn {
    return loggedIn;
}
 
- (NSArray*)albums {
    return albums;
}
 
- (NSDictionary*)infoDictionary {
    return [NSDictionary dictionaryWithObjectsAndKeys:
        username, @"username",
        [url absoluteString], @"url",
        [NSNumber numberWithInt:(int)type], @"type",
        nil];
}
 
- (BOOL)isGalleryV2 {
  return ([self type] == GalleryTypeG2 || [self type] == GalleryTypeG2XMLRPC);
}
 
- (ZWGalleryType)type {
    return type;
}
 
- (NSString *)lastCreatedAlbumName
{
    return lastCreatedAlbumName;
}
 
- (NSStringEncoding)sniffedEncoding
{
    return sniffedEncoding;
}
 
#pragma mark Actions
 
- (void)cancelOperation
{
    if (currentConnection && ![currentConnection isCancelled]) {
        [currentConnection cancel];
    }
}
 
- (void)login {
    NSDictionary *threadDispatchInfo = [NSDictionary dictionaryWithObjectsAndKeys:
        [NSThread currentThread], @"CallingThread",
        nil];
    [NSThread detachNewThreadSelector:@selector(loginThread:) toTarget:self withObject:threadDispatchInfo];
}
 
- (void)logout {
    loggedIn = FALSE;
}
 
- (void)createAlbumWithName:(NSString *)name title:(NSString *)title summary:(NSString *)summary parent:(ZWGallery *)parent
{
    if (parent == nil) 
        (id)parent = (id)[NSNull null];
        
    NSDictionary *threadDispatchInfo = [NSDictionary dictionaryWithObjectsAndKeys:
        name, @"AlbumName",
        title, @"AlbumTitle",
        summary, @"AlbumSummary",
        parent, @"AlbumParent",
        [NSThread currentThread], @"CallingThread",
        nil];
 
    [NSThread detachNewThreadSelector:@selector(createAlbumThread:) toTarget:self withObject:threadDispatchInfo];
}
 
- (void)getAlbums {
    NSDictionary *threadDispatchInfo = [NSDictionary dictionaryWithObjectsAndKeys:
        [NSThread currentThread], @"CallingThread",
        nil];
    [NSThread detachNewThreadSelector:@selector(getAlbumsThread:) toTarget:self withObject:threadDispatchInfo];
}
 
#pragma mark Helpers
 
- (NSDictionary*)parseResponseData:(NSData*)responseData {
    NSString *response = [[[NSString alloc] initWithData:responseData encoding:[self sniffedEncoding]] autorelease];
    
    if (response == nil) {
        NSLog(@"Could not convert response data into a string with encoding: %i", [self sniffedEncoding]);
        return nil;
    }
    
    NSMutableDictionary *dict = [NSMutableDictionary dictionary];
    NSScanner *scanner = [NSScanner scannerWithString:response];
    // first scan up to the #__GR2PROTO__ line
    [scanner scanUpToString:@"#__GR2PROTO__\n" intoString:nil];
    if (![scanner scanString:@"#__GR2PROTO__\n" intoString:nil]) {
        return nil;
    }
    while ([scanner isAtEnd] == NO) {
        NSString *line;
        if ([scanner scanUpToString:@"\n" intoString:&line]) {
            NSArray *pair = [line componentsSeparatedByString:@"="];
            if ([pair count] > 1) 
                [dict setObject:[pair objectAtIndex:1] forKey:[pair objectAtIndex:0]];
        }
    }
    
    // "status" is required - let's help ourselves out and make it an int right now
    NSString *statusStr = [dict objectForKey:@"status"];
    if (statusStr == nil) {
        return nil;
    }
    [dict setObject:[NSNumber numberWithInt:[statusStr intValue]] forKey:@"statusCode"];
    
    return [NSDictionary dictionaryWithDictionary:dict];
}
 
- (NSString *)formNameWithName:(NSString *)paramName
{
    // Gallery 1 names don't need mangling
    if (![self isGalleryV2]) 
        return paramName;
    
    // For some reason userfile is just changed to g2_userfile
    if ([paramName isEqualToString:@"userfile"])
        return @"g2_userfile";
    
    // All other G2 params are mangled like this:
    return [NSString stringWithFormat:@"g2_form[%@]", paramName];
}
 
#pragma mark Threads
 
- (void)loginThread:(NSDictionary *)threadDispatchInfo {
    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
    
    [NSThread prepareForInterThreadMessages];
 
    NSThread *callingThread = [threadDispatchInfo objectForKey:@"CallingThread"];
    
    ZWGalleryRemoteStatusCode status = [self doLogin];
    
    if (status == GR_STAT_SUCCESS)
        [delegate performSelector:@selector(galleryDidLogin:) 
                       withObject:self 
                         inThread:callingThread];
    else
        [delegate performSelector:@selector(gallery:loginFailedWithCode:) 
                       withObject:self 
                       withObject:[NSNumber numberWithInt:status] 
                         inThread:callingThread];
    
    [pool release];
}
    
- (ZWGalleryRemoteStatusCode)doLogin
{
    // remove the cookies sent to the gallery (the login function ain't so smart)
    NSHTTPCookieStorage *cookieStore = [NSHTTPCookieStorage sharedHTTPCookieStorage];
    NSArray *cookies = [cookieStore cookiesForURL:fullURL];
    id cookie;
    NSEnumerator *enumerator = [cookies objectEnumerator];
    while (cookie = [enumerator nextObject]) {
        [cookieStore deleteCookie:cookie];
    }
    
    // do an initial connection to get the session key
    NSMutableURLRequest *setupRequest = [NSMutableURLRequest requestWithURL:url
                                                                cachePolicy:NSURLRequestReloadIgnoringCacheData
                                                            timeoutInterval:60.0];
    [setupRequest setHTTPMethod:@"GET"];
    
    currentConnection = [ZWURLConnection connectionWithRequest:setupRequest];
    while ([currentConnection isRunning]) 
        [[NSRunLoop currentRunLoop] runMode:NSDefaultRunLoopMode beforeDate:[NSDate dateWithTimeIntervalSinceNow:0.2]];
    
    if ([currentConnection isCancelled]) 
        return ZW_GALLERY_OPERATION_DID_CANCEL;
    
    // Default to UTF-8
    sniffedEncoding = NSUTF8StringEncoding;
    NSURLResponse *response = [currentConnection response];
    NSString *encodingString = [response textEncodingName];
    if (encodingString) {
        CFStringEncoding cfStrEncoding = CFStringConvertIANACharSetNameToEncoding((CFStringRef)encodingString);
        if (cfStrEncoding)
            sniffedEncoding = CFStringConvertEncodingToNSStringEncoding(cfStrEncoding);
    }
    if (!sniffedEncoding)
        sniffedEncoding = NSUTF8StringEncoding;
    
    // Now try to log in (try twice - switch to other type of gallery if first try fails)
    BOOL tryGalleryV2 = [self isGalleryV2];
    NSDictionary *galleryResponse = nil;
    int tries;
    for (tries = 0; tries < 2; tries++) {
        if (!tryGalleryV2) {
            // try logging into Gallery v1
            fullURL = [[NSURL alloc] initWithString:[[url absoluteString] stringByAppendingString:@"gallery_remote2.php"]];
            NSMutableURLRequest *theRequest = [NSMutableURLRequest requestWithURL:fullURL
                                                                      cachePolicy:NSURLRequestReloadIgnoringCacheData
                                                                  timeoutInterval:60.0];
            [theRequest setValue:@"iPhotoToGallery 0.63" forHTTPHeaderField:@"User-Agent"];
            
            [theRequest setHTTPMethod:@"POST"];
            
            NSString *requestString = [NSString stringWithFormat:@"cmd=login&protocol_version=2.1&uname=%s&password=%s",
                [[username stringByEscapingURL] UTF8String], [[password stringByEscapingURL] UTF8String]];
            NSData *requestData = [requestString dataUsingEncoding:NSUTF8StringEncoding allowLossyConversion:YES];
            [theRequest setHTTPBody:requestData];
            
            currentConnection = [ZWURLConnection connectionWithRequest:theRequest];
            while ([currentConnection isRunning]) {
                [[NSRunLoop currentRunLoop] runMode:NSDefaultRunLoopMode beforeDate:[NSDate dateWithTimeIntervalSinceNow:0.2]];
            }
            
            if ([currentConnection isCancelled]) 
                return ZW_GALLERY_OPERATION_DID_CANCEL;
            
            NSData *data = [currentConnection data];
            response = [currentConnection response];
 
            if (data == nil) {
                if (tryGalleryV2) 
                    return ZW_GALLERY_COULD_NOT_CONNECT;
                
                // There is at least one instance (reported by adeh@desandies.com) where 
                // data will be nil, even though a G2 installation does exist. In that case,
                // let's try G2 if we haven't already.
                tryGalleryV2 = YES;
                continue;
            }
            
            galleryResponse = [self parseResponseData:data];
            
            if ([(NSHTTPURLResponse *)response statusCode] == 404 || galleryResponse == nil) 
                tryGalleryV2 = YES;
            else {
                // we successfully logged into a G1
                type = GalleryTypeG1;
                break;
            }
        }
        else {
            // try logging into Gallery v2
            fullURL = [[NSURL alloc] initWithString:[[url absoluteString] stringByAppendingString:@"main.php"]];
            
            NSMutableURLRequest *theRequest = [NSMutableURLRequest requestWithURL:fullURL
                                                                      cachePolicy:NSURLRequestReloadIgnoringCacheData
                                                                  timeoutInterval:60.0];
            [theRequest setValue:@"iPhotoToGallery" forHTTPHeaderField:@"User-Agent"];
            [theRequest setHTTPMethod:@"POST"];
            
            NSString *requestString = [NSString stringWithFormat:@"g2_controller=remote:GalleryRemote&g2_form[cmd]=login&g2_form[protocol_version]=2.2&g2_form[uname]=%s&g2_form[password]=%s",
                [[username stringByEscapingURL] UTF8String], [[password stringByEscapingURL] UTF8String]];
            NSData *requestData = [requestString dataUsingEncoding:NSUTF8StringEncoding allowLossyConversion:YES];
            [theRequest setHTTPBody:requestData];
            
            currentConnection = [ZWURLConnection connectionWithRequest:theRequest];
            while ([currentConnection isRunning]) {
                [[NSRunLoop currentRunLoop] runMode:NSDefaultRunLoopMode beforeDate:[NSDate dateWithTimeIntervalSinceNow:0.2]];
            }
            
            if ([currentConnection isCancelled]) 
                return ZW_GALLERY_OPERATION_DID_CANCEL;
            
            NSData *data = [currentConnection data];
            response = [currentConnection response];
            
            if (data == nil) 
                return ZW_GALLERY_COULD_NOT_CONNECT;
            
            galleryResponse = [self parseResponseData:data];
            if ([(NSHTTPURLResponse *)response statusCode] == 404 || galleryResponse == nil) 
                tryGalleryV2 = NO;
            else {
                // we successfully logged into a G2
                type = GalleryTypeG2;
                break;
            }
        }
    }
    
    if (galleryResponse == nil) 
        return ZW_GALLERY_PROTOCOL_ERROR;
    
    ZWGalleryRemoteStatusCode status = (ZWGalleryRemoteStatusCode)[[galleryResponse objectForKey:@"statusCode"] intValue];
    
    if (status == GR_STAT_PASSWD_WRONG) 
        return GR_STAT_PASSWD_WRONG;
    
    // lookup version
    NSString *version = [galleryResponse objectForKey:@"server_version"];
    if (version == nil) 
        return ZW_GALLERY_PROTOCOL_ERROR;
    NSArray *versionArray = [version componentsSeparatedByString:@"."];
    majorVersion = [[versionArray objectAtIndex:0] intValue];
    minorVersion = [[versionArray objectAtIndex:1] intValue];
    
    if (status == GR_STAT_SUCCESS) {
        loggedIn = YES;
        
        NSArray *cookies = [NSHTTPCookie cookiesWithResponseHeaderFields:[(NSHTTPURLResponse*)response allHeaderFields] forURL:fullURL];
        NSEnumerator *c = [cookies objectEnumerator];
        id cookie;
        while (cookie = [c nextObject]) 
            [cookieStore setCookie:cookie];
 
        return GR_STAT_SUCCESS;
    }
    
    return ZW_GALLERY_UNKNOWN_ERROR;
}
 
- (void)getAlbumsThread:(NSDictionary *)threadDispatchInfo {
    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
 
    [NSThread prepareForInterThreadMessages];
 
    NSThread *callingThread = [threadDispatchInfo objectForKey:@"CallingThread"];
    
    ZWGalleryRemoteStatusCode status = [self doGetAlbums];
    
    if (status == GR_STAT_SUCCESS)
        [delegate performSelector:@selector(galleryDidGetAlbums:) 
                       withObject:self
                         inThread:callingThread];
    else
        [delegate performSelector:@selector(gallery:getAlbumsFailedWithCode:) 
                       withObject:self 
                       withObject:[NSNumber numberWithInt:status] 
                         inThread:callingThread];
 
    [pool release];
}
 
- (ZWGalleryRemoteStatusCode)doGetAlbums
{
    NSMutableURLRequest *theRequest = [NSMutableURLRequest requestWithURL:fullURL
                                                              cachePolicy:NSURLRequestReloadIgnoringCacheData
                                                          timeoutInterval:60.0];
    [theRequest