EGOHTTPFormRequest.m

//
//  EGOHTTPFormRequest.m
//  EGOHTTPRequest
//
//  Created by Shaun Harrison on 3/28/10.
//  Copyright (c) 2010 enormego
//
//  Permission is hereby granted, free of charge, to any person obtaining a copy
//  of this software and associated documentation files (the "Software"), to deal
//  in the Software without restriction, including without limitation the rights
//  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
//  copies of the Software, and to permit persons to whom the Software is
//  furnished to do so, subject to the following conditions:
//
//  The above copyright notice and this permission notice shall be included in
//  all copies or substantial portions of the Software.
//
//  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
//  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
//  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
//  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
//  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
//  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
//  THE SOFTWARE.
//

#import "EGOHTTPFormRequest.h"

@interface EGOHTTPFormRequest ()
- (void)buildFormDataPostBody;
@end

@implementation EGOHTTPFormRequest
@synthesize stringEncoding=_stringEncoding;

- (void)setPostValue:(id<NSObject>)value forKey:(NSString*)key {
	if(!_formData) {
		_formData = [[NSMutableDictionary alloc] init];
	}
	
	[_formData setObject:value forKey:key];
}

- (void)buildFormDataPostBody {
	self.requestMethod = @"POST";
	if(!_formData) return;
	
	NSMutableString* requestString = [[NSMutableString alloc] init];
	
	if(_stringEncoding == 0) {
		_stringEncoding = NSUTF8StringEncoding;
	}
	
	for(NSString* key in _formData) {
		NSString* value = [[_formData objectForKey:key] description];

		if(!value || value.length == 0) {
			[requestString appendFormat:@"%@=&", [key urlEncodedStringWithEncoding:_stringEncoding]];
		} else {
			[requestString appendFormat:@"%@=%@&", [key urlEncodedStringWithEncoding:_stringEncoding], [value urlEncodedStringWithEncoding:_stringEncoding]];
		}
	}
	
	if(requestString.length > 0) {
		[requestString replaceCharactersInRange:NSMakeRange(requestString.length-1, 1) withString:@""]; // Removes trailing &
		self.requestBody = [requestString dataUsingEncoding:_stringEncoding];
		
		NSString* charset = (NSString*)CFStringConvertEncodingToIANACharSetName(CFStringConvertNSStringEncodingToEncoding(_stringEncoding));
		[self addRequestHeader:@"Content-Type" value:[NSString stringWithFormat:@"application/x-www-form-urlencoded; charset=%@", charset]];
	}

	[requestString release];
}

- (void)startAsynchronous {
	[self buildFormDataPostBody];
	[super startAsynchronous];	
}

- (void)startSynchronous {
	[self buildFormDataPostBody];
	[super startSynchronous];
}

- (void)dealloc {
	[_formData release];
	[super dealloc];
}

@end

@implementation NSString (EGOHTTPFormRequest)

/**
 * @see http://github.com/pokeb/asi-http-request/raw/master/Classes/ASIFormDataRequest.m
 */

- (NSString*)urlEncodedString { 
	return [self urlEncodedStringWithEncoding:NSUTF8StringEncoding];
}

- (NSString*)urlEncodedStringWithEncoding:(NSStringEncoding)encoding { 
	NSString* newString = [(NSString *)CFURLCreateStringByAddingPercentEscapes(kCFAllocatorDefault, (CFStringRef)self, NULL, CFSTR(":/?#[]@!$ &'()*+,;=\"<>%{}|\\^~`"), CFStringConvertNSStringEncodingToEncoding(encoding)) autorelease];
	
	if (newString) {
		return newString;
	}
	
	return @"";
}

@end