IRWebAPIKit/IRWebAPIEngine+FormURLEncoding.m at develop · waveface/IRWebAPIKit · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
//
//  IRWebAPIEngine+FormURLEncoding.m
//  IRWebAPIKit
//
//  Created by Evadne Wu on 11/7/11.
//  Copyright (c) 2011 __MyCompanyName__. All rights reserved.
//

#import "IRWebAPIHelpers.h"
#import "IRWebAPIEngine+FormURLEncoding.h"

extern NSData * IRWebAPIEngineFormURLEncodedDataWithDictionary (NSDictionary *dictionary);

static NSString * const kFormURLEncodingFields = @"-[IRWebAPIRequestContext(FormURLEncoding) formURLEncodingFields]";

@implementation IRWebAPIRequestContext (FormURLEncoding)

- (NSDictionary *) formURLEncodingFields {

	NSMutableDictionary *formURLEncodingFields = objc_getAssociatedObject(self, &kFormURLEncodingFields);

	if (!formURLEncodingFields) {
		formURLEncodingFields = [NSMutableDictionary dictionary];
		objc_setAssociatedObject(self, &kFormURLEncodingFields, formURLEncodingFields, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
	}

	return [formURLEncodingFields copy];

}

- (void) removeAllFormURLEncodingFieldValues {

	[self willChangeValueForKey:@"formURLEncodingFields"];
	objc_setAssociatedObject(self, &kFormURLEncodingFields, nil, OBJC_ASSOCIATION_ASSIGN);
	[self didChangeValueForKey:@"formURLEncodingFields"];

}

- (void) setValue:(id)obj forFormURLEncodingField:(NSString *)key {

	NSMutableDictionary *formURLEncodingFields = objc_getAssociatedObject(self, &kFormURLEncodingFields);

	if (!formURLEncodingFields) {
		formURLEncodingFields = [NSMutableDictionary dictionary];
		objc_setAssociatedObject(self, &kFormURLEncodingFields, formURLEncodingFields, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
	}

	[self willChangeValueForKey:@"formURLEncodingFields"];

	if (obj) {
		[formURLEncodingFields setObject:obj forKey:key];
	} else {
		[formURLEncodingFields removeObjectForKey:key];
	}

	[self didChangeValueForKey:@"formURLEncodingFields"];

}

@end


@implementation IRWebAPIEngine (FormURLEncoding)

+ (IRWebAPIRequestContextTransformer) defaultFormURLEncodingTransformer {

	return [(^ (IRWebAPIRequestContext *context) {

		NSDictionary *formURLEncodingFields = context.formURLEncodingFields;

		if (![formURLEncodingFields count])
			return context;

		[context setValue:@"8bit" forHeaderField:@"Content-Transfer-Encoding"];
		[context setValue:@"application/x-www-form-urlencoded" forHeaderField:@"Content-Type"];

		[context setBody:IRWebAPIEngineFormURLEncodedDataWithDictionary(formURLEncodingFields)];
		[context removeAllFormURLEncodingFieldValues];

		[context setMethod:@"POST"];

		return context;

	}) copy];

}

@end


NSData * IRWebAPIEngineFormURLEncodedDataWithDictionary (NSDictionary *formNamesToContents) {

	NSMutableData *sentData = [NSMutableData data];

	[formNamesToContents enumerateKeysAndObjectsUsingBlock: ^ (id key, id obj, BOOL *stop) {

		if ([sentData length])
			[sentData appendData:[@"&" dataUsingEncoding:NSUTF8StringEncoding]];

		[sentData appendData:[IRWebAPIKitRFC3986EncodedStringMake(key) dataUsingEncoding:NSUTF8StringEncoding]];
		[sentData appendData:[@"=" dataUsingEncoding:NSUTF8StringEncoding]];
		[sentData appendData:[IRWebAPIKitRFC3986EncodedStringMake(obj) dataUsingEncoding:NSUTF8StringEncoding]];

	}];

	return sentData;

}