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

Go to the documentation of this file.

#import "MediaWikiQuery.h"


@implementation MediaWikiQuery
@synthesize mediaWikiURLString;
-(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 <MediaWikiQueryDelegate>)mediaWikiQueryDelegate {
    if((self = [self init])) {
        // Won't work without a delegate
        if(mediaWikiQueryDelegate == nil) {
            return nil;
        }
        queryDelegate = mediaWikiQueryDelegate;
        working = NO;
    }
    return self;  
}
-(id)initWithDelegate:(id <MediaWikiQueryDelegate>)mediaWikiQueryDelegate mediaWikiURL:(NSString *)mwURLString {
    if(!mwURLString) {
        return nil;
    }
    self = [self initWithDelegate:mediaWikiQueryDelegate];
    if(self != nil) {
        self.mediaWikiURLString = [mwURLString copy];
    }
    return self;
}
-(NSString*)buildQueryURL {
    return self.mediaWikiURLString;
}
-(BOOL)startQuery {
    NSString* queryURL = [self buildQueryURL];
    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];
    mediaWikiRequest = [[[NSMutableURLRequest alloc] initWithURL:[[[NSURL alloc] initWithString:queryURL] autorelease]] autorelease];
    //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 {
    // 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];
    
    
}
- (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];
    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];
    working = NO;
    [[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];
    working = NO;    
    [queryDelegate didReceiveMediaWikiData:tmp];
}

-(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)dealloc {
    queryDelegate = nil;
    [mediaWikiURLString release];
    mediaWikiURLString = nil;
    mediaWikiConnection = nil;
    receivedData = nil;
    [progressListeners removeAllObjects];
    [progressListeners release];
    progressListeners = nil;
    [super dealloc];
}
@end