TeePee.m

//
//  TeePee.m
//  ChimpKit
//
//  Created by Christopher Burnett on 2/3/10.
//  Copyright 2010 twoism. All rights reserved.
//

#import "TeePee.h"
#import "NSDictionary+Parameterize.h"

#define TEE_PEE_KEYS [NSArray arrayWithObjects:@"delegate",@"onSuccess",@"onFailure",@"get",@"post",@"put",@"delete",nil]
#define LOTS_OF_ARGS "@^v@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"

@implementation TeePee

@synthesize delegate, userInfo, requestQueue, onSuccess, onFailure, request, baseUri, requestParams;

- (id)initWithDelegate:(id)aDelegate
{
  self = [super init];
  if (self != nil) {
    self.requestParams  = [[NSMutableDictionary alloc] init];
    self.delegate       = aDelegate;
  }
  return self;
}

- (void)parseParamDefaults:(NSDictionary*)params
{
  if ([params objectForKey:@"delegate"])
    self.delegate = [params objectForKey:@"delegate"];
  if ([params objectForKey:@"onSuccess"])
    self.onSuccess = NSSelectorFromString([params objectForKey:@"onSuccess"]);
  if ([params objectForKey:@"onFailure"])
    self.onFailure = NSSelectorFromString([params objectForKey:@"onFailure"]);
}

- (void)dispatchRequest:(NSString*)signature 
             withParams:(NSDictionary*)params
{
  
  NSString *httpMethod  = [[signature componentsSeparatedByString:@":"] objectAtIndex:0];
  NSString *path        = [params objectForKey:httpMethod];
  
  [self requestForPath:path withParams:params];
  
  if (![httpMethod isEqualToString:@"get"]) {
    [self paramsForRequest:[params mutableCopy]];
  }
  
  [self.request setRequestMethod:[httpMethod uppercaseString]];
  [[self requestQueue] addOperation:self.request];
  
}

- (void)addParamsToRequest
{
  for (id key in self.requestParams) {
    [self.request setPostValue:[self.requestParams objectForKey:key] 
                        forKey:key];
  }
}

- (void)stripParams:(NSMutableDictionary*)params
{
  [params removeObjectsForKeys:TEE_PEE_KEYS];
}

- (void)paramsForRequest:(NSMutableDictionary*)params
{
  [self stripParams:params];
  for (id key in params) {
    id param = [params valueForKey:key];
    if ([param respondsToSelector:@selector(objectForKey:)] || [param respondsToSelector:@selector(objectAtIndex:)]) {
      [self.requestParams addEntriesFromDictionary:[param parameterizeWithScope:key]];
    }else{
      [self.requestParams setObject:[params valueForKey:key] 
                             forKey:key];
    }
  }
  [self addParamsToRequest];
}

- (NSString*)paramStringForRequest:(NSMutableDictionary*)params
{
  NSString *queryString = @"";
  [self stripParams:params];
  
  for (id key in params) {
    id param = [params valueForKey:key];
    if ([param respondsToSelector:@selector(objectForKey:)] || [param respondsToSelector:@selector(objectAtIndex:)]) {
      // handle dictionary & array values
      queryString = [queryString stringByAppendingString:[param toQueryStringWithScope:key]];
    } else {
        // handle strings
        NSString *value;
        if ([[params objectForKey:key] respondsToSelector:@selector(stringValue)]) {
            value = [[params objectForKey:key] stringValue];
        } else { 
            value = [params objectForKey:key];
        }
        value = [value stringByAddingPercentEscapesUsingEncoding:NSASCIIStringEncoding];
        value = [value stringByReplacingOccurrencesOfString:@"+" withString:@"%2B"];
        queryString = [queryString stringByAppendingString:[NSString stringWithFormat:@"%@=%@&",key,value]];
    }
  }
    return queryString;
}

- (void)requestForPath:(NSString*)path 
            withParams:(NSDictionary*)params
{
  
  if (![self requestQueue]) {
    [self setRequestQueue:[[[NSOperationQueue alloc] init] autorelease]];
  }
  
  [self paramsForRequest:[params mutableCopy]];
  
  NSString  *urlString  = [NSString stringWithFormat:@"%@%@?%@",self.baseUri,path,[self paramStringForRequest:self.requestParams]];
  NSURL     *url        = [NSURL URLWithString:urlString];
  self.request          = [ASIFormDataRequest requestWithURL:url];
  
  [self.request setDelegate:self];
  [self.request setDidFinishSelector:@selector(tp_requestDidLoad:)];
  [self.request setDidFailSelector:@selector(tp_requestDidFail:)];
  [self.request setUserInfo:self.userInfo];
  self.userInfo = nil;
}

- (void)tp_requestDidLoad:(ASIFormDataRequest*)aRequest
{
  [self.delegate performSelector:self.onSuccess 
                      withObject:aRequest];
}
- (void)tp_requestDidFail:(ASIFormDataRequest*)aRequest
{
  [self.delegate performSelector:self.onFailure 
                      withObject:aRequest];
}

- (void)methodMissing:(NSString*)method 
           withParams:(NSDictionary*)params
{
  [self parseParamDefaults:params];
  [self dispatchRequest:method 
             withParams:params];
}




+ (void)forwardInvocation:(NSInvocation *)invocation
{
  [self handleInvocation:invocation];
}

- (void)forwardInvocation:(NSInvocation *)invocation
{
  [self handleInvocation:invocation];
}

+ (NSMethodSignature *)methodSignatureForSelector:(SEL)selector
{
  return [self formatDynamicSignature];
}
- (NSMethodSignature *)methodSignatureForSelector:(SEL)selector
{
  return [[self class] formatDynamicSignature];
}

+ (NSMethodSignature*)formatDynamicSignature
{
  return [NSMethodSignature signatureWithObjCTypes:LOTS_OF_ARGS];
}

+ (void)handleInvocation:(NSInvocation*)invocation
{
  NSString        *selectorString = NSStringFromSelector([invocation selector]);
  NSArray         *parts          = [selectorString componentsSeparatedByString: @":"];
  NSArray         *keys           = [parts objectsAtIndexes:[NSIndexSet indexSetWithIndexesInRange:NSMakeRange(0, [parts count]-1)]];
  NSMutableArray  *values         = [[NSMutableArray alloc] init];
  
  for (int i=0; i<[keys count]; i++) {
    id arg;
    [invocation getArgument:&arg atIndex:i+2];
    [values addObject:arg];
  }
  
  NSDictionary *params = [NSMutableDictionary dictionaryWithObjects:values 
                                                            forKeys:keys];
  
  if ([self respondsToSelector:@selector(methodMissing:withParams:)]) {
    
    [self performSelector:@selector(methodMissing:withParams:) 
               withObject:selectorString 
               withObject:params];
    
  }
  [values release];
}


- (void)handleInvocation:(NSInvocation*)invocation
{
  NSString        *selectorString = NSStringFromSelector([invocation selector]);
  NSArray         *parts          = [selectorString componentsSeparatedByString: @":"];
  NSArray         *keys           = [parts objectsAtIndexes:[NSIndexSet indexSetWithIndexesInRange:NSMakeRange(0, [parts count]-1)]];
  NSMutableArray  *values         = [[NSMutableArray alloc] init];
  
  for (int i=0; i<[keys count]; i++) {
    id arg;
    [invocation getArgument:&arg atIndex:i+2];
    if (arg != nil) {
      [values addObject:arg];
    }else {
      [values addObject:@""];
    }
  
      
  }
  
  NSDictionary *params = [NSMutableDictionary dictionaryWithObjects:values 
                                                            forKeys:keys];
  
  if ([self respondsToSelector:@selector(methodMissing:withParams:)]) {
    
    [self performSelector:@selector(methodMissing:withParams:) 
               withObject:selectorString 
               withObject:params];
    
  }
  [values release];
}


@end