/Users/awniemel/Notepad-SVN/svn/trunk/CatNiP/CatNiP/MediaWikiDownload.m

Go to the documentation of this file.

#import "MediaWikiDownload.h"

@implementation MediaWikiDownload
@synthesize mediaWikiURLString;
@synthesize mediaWikiFilename;

-(NSInteger)expectedSize {
    return expectedSize;
}

-(NSInteger)currentBytes {
    return currentBytes;
}

-(id)init {
    if((self = [super init])) {
        expectedSize = NSIntegerMin; // MAAAGIC NUMBER
        currentBytes = 0;
        progressListeners = [[NSMutableSet alloc] init];
    }
    return self;
}


-(id)initWithDelegate:(id <MediaWikiDownloadDelegate>)mediaWikiDownloadDelegate {
    if(!mediaWikiDownloadDelegate) {
        return nil;
    }
    if((self = [self init])) {
        // Won't work without a delegate
        //if(mediaWikiDownloadDelegate == nil) {
        //    return nil;
        //}
        queryDelegate = mediaWikiDownloadDelegate;
        working = NO;
    }
    return self;  
}
-(id)initWithDelegate:(id <MediaWikiDownloadDelegate>)mediaWikiDownloadDelegate mediaWikiURL:(NSString *)mwURLString {
    if(!mwURLString) {
        return nil;
    }
    self = [self initWithDelegate:mediaWikiDownloadDelegate];
    if(self != nil) {
        self.mediaWikiURLString = mwURLString;
    }
    return self;
}
-(NSString*)buildQueryURL {
    return [self.mediaWikiURLString stringByAppendingString:self.mediaWikiFilename];
}
-(BOOL)startQuery {
    NSString* queryURL = [self buildQueryURL];
    if(!self.mediaWikiFilename || [self.mediaWikiFilename length] < 1) {
        return NO;
    }
    if(queryURL == nil || [queryURL length] < 3) {
        return NO;
    }
    if(working) {
        return NO;
    }
    // We probably need to ensure that we're not in the middle
    // of a request!
    // queryURL = [queryURL stringByAddingPercentEscapesUsingEncoding:NSISOLatin1StringEncoding];
    queryURL = [queryURL stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
    // The request is mutable so that we can set the cookie header ourselves
    mediaWikiRequest = [[NSMutableURLRequest alloc] initWithURL:[[[NSURL alloc] initWithString:queryURL] autorelease]];
    //[queryURL release];
    [mediaWikiRequest setHTTPShouldHandleCookies:NO];
    // IMSLP won't let us download a file unless we have this cookie
    [mediaWikiRequest setValue:@"imslpdisclaimeraccepted=yes" forHTTPHeaderField:@"Cookie"];
    
    //NSLog(@"urlstring: %@ requestobj: %@", queryURL, mediaWikiRequest);
    //Is this the right spot for a retain?
    if(![NSURLConnection canHandleRequest:mediaWikiRequest]) {
        //[mediaWikiRequest release];
        // TODO: Add an error message or something!
        return NO;
    }
    mediaWikiConnection = [[NSURLConnection alloc] initWithRequest:mediaWikiRequest delegate:self];
    if(mediaWikiConnection) {
      //receivedData = [[NSMutableData data] retain];
        [mediaWikiConnection start];
        //[mediaWikiRequest release];
        working = YES;
    } else {
        NSDictionary *errordata = 
        [NSDictionary dictionaryWithObjectsAndKeys:
         @"Unknown error while opening HTTP connection", NSLocalizedFailureReasonErrorKey, 
         [NSString stringWithFormat:@"Could not establish a connection to \"%@\".",queryURL],
         NSLocalizedDescriptionKey, nil];
        [[CatNiPErrorManager sharedManager] reportError:[NSError errorWithDomain:NSPOSIXErrorDomain code:EIO userInfo:errordata]];
        NSLog(@"NSURLConnection failed for some reason.");
        //[mediaWikiRequest release];
        return NO;
    }
    return YES;
    
}
- (void)connection:(NSURLConnection *)connection didReceiveResponse:(NSURLResponse *)response {
    NSLog(@"received response with MIMEType:%@, resetting...", [response MIMEType]);
    // This method is called when the server has determined that it
    // has enough information to create the NSURLResponse.
    if([response expectedContentLength]) {
        expectedSize = [response expectedContentLength];
    } else {
        expectedSize = -1; // Maaaagic number 2.
    }
    // It can be called multiple times, for example in the case of a
    // redirect, so each time we reset the data.
    //[receivedData setLength:0];
    if(outHandle) {
        NSLog(@"Output file handle exists already, closing time.");
        [outHandle closeFile];
        [outHandle release];
    }
    // Create the output file path.
    if(!outputPath) {
        NSLog(@"No outputPath defined yet, creating...");
        outputPath = [NSTemporaryDirectory() stringByAppendingPathComponent:self.mediaWikiFilename];
        [outputPath retain];
        NSLog(@"OutputPath is now %@",outputPath);
    }
    if([[NSFileManager defaultManager] createFileAtPath:outputPath contents:nil attributes:nil]) {
        NSLog(@"Succesfully created empty file");
        outHandle = [[NSFileHandle fileHandleForWritingAtPath:outputPath] retain];
        if(!outHandle) {
            NSLog(@"Got a null outHandle, oh god!");
        }

    }
    else {
        // Could not create temp file!
        NSLog(@"Could not create temp file, this is going to be ugly.");
    }
    
    
}
- (void)connection:(NSURLConnection *)connection didReceiveData:(NSData *)data {
    // Append the new data to receivedData.
    // receivedData is an instance variable declared in the interface.
    //[receivedData appendData:data];
    //NSLog(@"received data, %d bytes",[data length]);
    [outHandle writeData:data];
    currentBytes += [data length];
    for (id<MediaWikiProgressListener> listener in progressListeners) {
        [listener progressUpdate:self];
    }
}
- (void)connection:(NSURLConnection *)connection
  didFailWithError:(NSError *)error {
    // release the connection, and the data object
    [mediaWikiConnection release];
    //[receivedData release];
    if(outHandle) {
        [outHandle closeFile];
        [outHandle release];
    }
    working = NO;
    if(outputPath) {
        [outputPath release];
    }
    [[CatNiPErrorManager sharedManager] reportError:error];
    

    // inform the user
    NSLog(@"Connection failed! Error - %@ %@",
          [error localizedDescription],
          [[error userInfo] objectForKey:NSURLErrorFailingURLStringErrorKey]);
}
- (void)connectionDidFinishLoading:(NSURLConnection *)connection {
    // do something with the data
    // receivedData is declared as a method instance elsewhere
    //NSData* tmp = [NSData dataWithData:receivedData];
    [mediaWikiConnection release];
    //[receivedData release];
    if(outHandle) {
        [outHandle closeFile];
        [outHandle release];
    }
    working = NO;
    NSLog(@"connection ended succesfully, hooray");
    NSArray *paths = NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES);
    NSString *documentsDirectory = [paths objectAtIndex:0];
    documentsDirectory = [documentsDirectory stringByAppendingPathComponent:self.mediaWikiFilename];
    NSError* copyError;
    if ([[NSFileManager defaultManager] fileExistsAtPath:documentsDirectory]) {   
        [[NSFileManager defaultManager] removeItemAtPath:documentsDirectory error:&copyError];
    }
    if(![[NSFileManager defaultManager] moveItemAtPath:outputPath toPath:documentsDirectory error:&copyError]) {
        //[outputPath autorelease];
        [[CatNiPErrorManager sharedManager] reportError:copyError];
        
    }
    else {
        [queryDelegate didFinishDownload:[NSString stringWithString:documentsDirectory]];
    }
    [outputPath release];
}

-(void)addProgressListener:(id <MediaWikiProgressListener>)listener {
    if([progressListeners containsObject:listener]) {
        return;
    }
    else {
        [progressListeners addObject:listener];
    }
}

-(void)removeProgressListener:(id <MediaWikiProgressListener>)listener {
    if([progressListeners containsObject:listener]) {
        [progressListeners removeObject:listener];
    }
}

-(void)abortRead {
    if(working) {
        [mediaWikiConnection cancel];
        [mediaWikiConnection release];
        //[receivedData release];
        if(outHandle) {
            [outHandle closeFile];
            [outHandle release];
        }
        working = NO;
        if(outputPath) {
            [outputPath release];
        }        
    }
}


@end