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

Go to the documentation of this file.

#import "MediaWikiCategoryQuery.h"

@implementation MediaWikiCategoryQuery
//@synthesize wikiAction=_wikiAction;
@synthesize wikiQueryType=_wikiQueryType;
@synthesize wikiQuery=_wikiQuery;
@synthesize wikiQueryTitles=_wikiQueryTitles;
@synthesize wikiQueryPageIds=_wikiQueryPageIds;
@synthesize categoryDelegate=_categoryDelegate;
@synthesize continueFrom=_continueFrom;
@synthesize parameters;

NSString* _queryContinueFrom;
NSXMLParser* categoryParser;
NSMutableArray* categories;


-(void)setWikiAction:(NSString *)wikiAction {
    [self.parameters setValue:[NSString stringWithString:wikiAction] forKey:@"action"];
}
-(NSString *)wikiAction {
    return [self.parameters valueForKey:@"action"];
}

-(void)setReturnFormat:(NSString *)returnFormat {
    [self.parameters setValue:[NSString stringWithString:returnFormat] forKey:@"format"];    
}
-(NSString *)returnFormat {
    return [self.parameters valueForKey:@"format"];
}

-(void)setCategoryName:(NSString *)categoryName {
    [self.parameters setValue:[NSString stringWithString:categoryName] forKey:@"cmtitle"];        
    
}
-(NSString *)categoryName {
    return [self.parameters valueForKey:@"cmtitle"];    
}

-(void)setChunkSize:(NSString *)chunkSize {
    [self.parameters setValue:[NSString stringWithString:chunkSize] forKey:@"cmlimit"];        
}
-(NSString *)chunkSize{
    return [self.parameters valueForKey:@"cmlimit"];    
}


+ (MediaWikiCategoryQuery*) newCategoryQuery:(NSString*)URL delegate:(id <MediaWikiCategoryQueryDelegate>)qDelegate {
    
    if(qDelegate == nil) {
        return nil;
    }
    MediaWikiCategoryQuery* cQuery = [MediaWikiCategoryQuery alloc];
    cQuery = [cQuery initWithDelegate:cQuery mediaWikiURL:URL];
    cQuery.wikiAction = @"query";
    cQuery.wikiQueryType = @"list";
    cQuery.wikiQuery = @"categorymembers";
    cQuery.returnFormat = @"xml";
    cQuery.categoryDelegate = qDelegate;
    return cQuery;
}
-init {
    if((self = [super init])) {
        self.parameters = [[NSMutableDictionary alloc] init];
        self.returnFormat = @"xml";
        return self;
    }
    return nil;
}
- (NSString*) buildQueryURL {
    NSString* myParams = [self _buildGetParams];
    NSString* retVal = [NSString stringWithFormat: @"%@?%@", self.mediaWikiURLString, myParams];
    return retVal;
}
-(void)didReceiveMediaWikiData:(NSData *)data {
    [self parseCategoryXML:data];
}
-(void)parseCategoryXML:(NSData *)XMLData {
//    NSLog(@"Starting parse (hopefully...)");
    if (categoryParser) // categoryParser is an NSXMLParser instance variable
        [categoryParser release];
    categoryParser = [[NSXMLParser alloc] initWithData:XMLData];
    [categoryParser setDelegate:self];
    [categoryParser setShouldResolveExternalEntities:YES];
    [categoryParser parse]; // return value not used
    // if not successful, delegate is informed of error    
}
-(void)parser:(NSXMLParser *)parser didStartElement:(NSString *)elementName namespaceURI:(NSString *)namespaceURI qualifiedName:(NSString *)qName attributes:(NSDictionary *)attributeDict {
    
    if( [elementName isEqualToString:@"cm"] ) {
        if(!categories) {
            categories = [[NSMutableArray alloc] init];
        }
        [categories addObject: [NSString stringWithString:[attributeDict valueForKey:@"title"]]];
    }
    else if( [elementName isEqualToString:@"categorymembers"] ) {
        if([attributeDict valueForKey:@"cmcontinue"]) {
            self.continueFrom = [attributeDict valueForKey:@"cmcontinue"];
            //            NSLog(@"%@",self.continueFrom);
        }
    }    
}
-(void)parserDidEndDocument:(NSXMLParser *)parser {
    if(categories) {
        [self.categoryDelegate didReceiveCategoryArray:[[[NSArray alloc] initWithArray:categories] autorelease]];
    }
}
-(void)parserDidStartDocument:(NSXMLParser *)parser {
//    NSLog(@"Parsing supposedly started.");
    if(categories) {
        [categories release];
        categories = nil;
    }
    self.continueFrom = nil;
}
-(void)parser:(NSXMLParser *)parser validationErrorOccurred:(NSError *)validationError {
    NSLog(@"Oh no, validation error!");
    [self.categoryDelegate categoryParseFailed:validationError];
}
-(void)parser:(NSXMLParser *)parser parseErrorOccurred:(NSError *)parseError {
    NSLog(@"Oh no, parse error! %@",[parseError localizedDescription]);
    [self.categoryDelegate categoryParseFailed:parseError];
}

@end

@implementation MediaWikiCategoryQuery(Private)

- (NSString*)_buildGetParams {
    NSMutableString* qString=[[NSMutableString alloc] init];
    NSEnumerator *enumerator = [self.parameters keyEnumerator];
    id key;
    
    while ((key = [enumerator nextObject])) {
        [qString appendString: key];
        [qString appendString:@"="];
        [qString appendString: [self.parameters valueForKey:key]];
        [qString appendString:@"&"];
    }
    
    [qString appendString:_wikiQueryType];
    [qString appendString:@"="];
    [qString appendString:_wikiQuery];
    
    if(self.continueFrom) {
        [qString appendString:@"&cmcontinue="];
        [qString appendString:self.continueFrom];        
    }
    NSString *retVal = [[NSString alloc] initWithString:qString];
    [qString release];
    return [retVal autorelease];
}

@end