-
Notifications
You must be signed in to change notification settings - Fork 167
/
OATokenManager.m
401 lines (348 loc) · 10.8 KB
/
OATokenManager.m
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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
//
// OATokenManager.m
// OAuthConsumer
//
// Created by Alberto García Hierro on 01/09/08.
// Copyright 2008 Alberto García Hierro. All rights reserved.
// bynotes.com
#import "OAConsumer.h"
#import "OAToken.h"
#import "OAProblem.h"
#import "OACall.h"
#import "OATokenManager.h"
@interface OATokenManager (Private)
- (void)callProblem:(OACall *)call problem:(OAProblem *)problem;
- (void)callError:(OACall *)call error:(NSError *)error;
- (void)callFinished:(OACall *)call body:(NSString *)body;
- (void)dispatch;
- (void)performCall:(OACall *)aCall;
- (void)requestToken;
- (void)requestTokenReceived;
- (void)exchangeToken;
- (void)renewToken;
- (void)accessTokenReceived;
- (void)setAccessToken:(OAToken *)token;
- (void)deleteSavedRequestToken;
- (OACall *)queue;
- (void)enqueue:(OACall *)call selector:(SEL)selector;
- (void)dequeue:(OACall *)call;
- (SEL)getSelector:(OACall *)call;
@end
@implementation OATokenManager
- (id)init {
return [self initWithConsumer:nil
token:nil
oauthBase:nil
realm:nil
callback:nil
delegate:nil];
}
- (id)initWithConsumer:(OAConsumer *)aConsumer token:(OAToken *)aToken oauthBase:(const NSString *)base
realm:(const NSString *)aRealm callback:(const NSString *)aCallback
delegate:(NSObject <OATokenManagerDelegate> *)aDelegate {
if ((self = [super init])) {
consumer = [aConsumer retain];
acToken = nil;
reqToken = nil;
initialToken = [aToken retain];
authorizedTokenKey = nil;
oauthBase = [base copy];
realm = [aRealm copy];
callback = [aCallback copy];
delegate = aDelegate;
calls = [[NSMutableArray alloc] init];
selectors = [[NSMutableArray alloc] init];
delegates = [[NSMutableDictionary alloc] init];
isDispatching = NO;
}
return self;
}
- (void)dealloc {
[consumer release];
[acToken release];
[reqToken release];
[initialToken release];
[authorizedTokenKey release];
[oauthBase release];
[realm release];
[callback release];
[calls release];
[selectors release];
[delegates release];
[super dealloc];
}
// The application got a new authorized
// request token and is notifying us
- (void)authorizedToken:(NSString *)aKey
{
if (reqToken && [aKey isEqualToString:reqToken.key]) {
[self exchangeToken];
} else {
[authorizedTokenKey release];
authorizedTokenKey = [aKey retain];
}
}
// Private functions
// Deal with problems and errors in calls
- (void)call:(OACall *)call failedWithProblem:(OAProblem *)problem
{
/* Always clear the saved request token, just in case */
[self deleteSavedRequestToken];
if ([problem isEqualToProblem:[OAProblem TokenExpired]]) {
/* renewToken checks if it's renewable */
[self renewToken];
} else if ([problem isEqualToProblem:[OAProblem TokenNotRenewable]] ||
[problem isEqualToProblem:[OAProblem TokenRejected]]) {
/* This token may have been revoked by the user, get a new one
after removing the stored requestToken, since the problem may be in
it */
[self setAccessToken:nil];
[self requestToken];
} else if ([problem isEqualToProblem:[OAProblem NonceUsed]]) {
/* Just repeat this request */
[self performCall:call];
} else {
/* Non-recoverable error, tell the delegate and dequeue the call
if appropiate */
if([delegate tokenManager:self failedCall:call withProblem:problem]) {
[self dequeue:call];
}
@synchronized(self) {
isDispatching = NO;
}
}
}
- (void)call:(OACall *)call failedWithError:(NSError *)error
{
if([delegate tokenManager:self failedCall:call withError:error]) {
[self dequeue:call];
}
@synchronized(self) {
isDispatching = NO;
}
}
// When a call finish, notify the delegate
- (void)callFinished:(OACall *)call body:(NSString *)body
{
SEL selector = [self getSelector:call];
id deleg = [delegates objectForKey:[NSString stringWithFormat:@"%p", call]];
if (deleg) {
[deleg performSelector:selector withObject:body];
[delegates removeObjectForKey:call];
} else {
[delegate performSelector:selector withObject:body];
}
@synchronized(self) {
isDispatching = NO;
}
[self dequeue:call];
[self dispatch];
}
- (OACall *)queue {
id obj = nil;
@synchronized(calls) {
if ([calls count]) {
obj = [calls objectAtIndex:0];
}
}
return obj;
}
- (void)enqueue:(OACall *)call selector:(SEL)selector {
NSUInteger idx = [calls indexOfObject:call];
if (idx == NSNotFound) {
@synchronized(calls) {
[calls addObject:call];
[call release];
[selectors addObject:NSStringFromSelector(selector)];
}
}
}
- (void)dequeue:(OACall *)call {
NSUInteger idx = [calls indexOfObject:call];
if (idx != NSNotFound) {
@synchronized(calls) {
[calls removeObjectAtIndex:idx];
[selectors removeObjectAtIndex:idx];
}
}
}
- (SEL)getSelector:(OACall *)call
{
NSUInteger idx = [calls indexOfObject:call];
if (idx != NSNotFound) {
return NSSelectorFromString([selectors objectAtIndex:idx]);
}
return 0;
}
// Token management functions
// Requesting a new token
// Gets a new token and opens the default
// browser for authorizing it. The application
// is expected to call authorizedToken when it
// gets the authorized token back
- (void)requestToken
{
/* Try to load an access token from settings */
OAToken *atoken = [[[OAToken alloc] initWithUserDefaultsUsingServiceProviderName:oauthBase prefix:[@"access:" stringByAppendingString:realm]] autorelease];
if (atoken && [atoken isValid]) {
[self setAccessToken:atoken];
return;
}
/* Try to load a stored requestToken from
settings (useful for iPhone) */
OAToken *token = [[[OAToken alloc] initWithUserDefaultsUsingServiceProviderName:oauthBase prefix:[@"request:" stringByAppendingString:realm]] autorelease];
/* iPhone specific, the manager must have got the authorized token before reaching this point */
NSLog(@"request token in settings %@", token);
if (token && token.key && [authorizedTokenKey isEqualToString:token.key]) {
reqToken = [token retain];
[self exchangeToken];
return;
}
if ([delegate respondsToSelector:@selector(tokenManagerNeedsToken:)]) {
if (![delegate tokenManagerNeedsToken:self]) {
return;
}
}
OACall *call = [[OACall alloc] initWithURL:[NSURL URLWithString:[oauthBase stringByAppendingString:@"request_token"]] method:@"POST"];
[call perform:consumer
token:initialToken
realm:realm
delegate:self
didFinish:@selector(requestTokenReceived:body:)];
}
- (void)requestTokenReceived:(OACall *)call body:(NSString *)body
{
/* XXX: Check if token != nil */
NSLog(@"Received request token %@", body);
OAToken *token = [[[OAToken alloc] initWithHTTPResponseBody:body] autorelease];
if (token) {
[reqToken release];
reqToken = [token retain];
[reqToken storeInUserDefaultsWithServiceProviderName:oauthBase prefix:[@"request:" stringByAppendingString:realm]];
/* Save the token in case we exit and start again
before the token is authorized (useful for iPhone) */
NSURL *url = [NSURL URLWithString:[NSString stringWithFormat:@"%@authorize?oauth_token=%@&oauth_callback=%@",
oauthBase, token.key, callback]];
[[UIApplication sharedApplication] openURL:url];
}
[call release];
}
// Exchaing a request token for an access token
// Exchanges the current authorized
// request token for an access token
- (void)exchangeToken
{
if (!reqToken) {
[self requestToken];
return;
}
NSURL *url = [NSURL URLWithString:[oauthBase stringByAppendingString:@"access_token"]];
OACall *call = [[OACall alloc] initWithURL:url method:@"POST"];
[call perform:consumer
token:reqToken
realm:realm
delegate:self
didFinish:@selector(accessTokenReceived:body:)];
}
- (void)accessTokenReceived:(OACall *)call body:(NSString *)body
{
OAToken *token = [[[OAToken alloc] initWithHTTPResponseBody:body] autorelease];
[self setAccessToken:token];
}
- (void)renewToken {
NSLog(@"Renewing token");
if (!acToken || ![acToken isRenewable]) {
[self requestToken];
return;
}
acToken.forRenewal = YES;
NSURL *url = [NSURL URLWithString:[oauthBase stringByAppendingString:@"access_token"]];
OACall *call = [[OACall alloc] initWithURL:url method:@"POST"];
[call perform:consumer
token:acToken
realm:realm
delegate:self
didFinish:@selector(accessTokenReceived:body:)];
}
- (void)setAccessToken:(OAToken *)token {
/* Remove the stored requestToken which generated
this access token */
[self deleteSavedRequestToken];
if (token) {
[acToken release];
acToken = [token retain];
[acToken storeInUserDefaultsWithServiceProviderName:oauthBase prefix:[@"access:" stringByAppendingString:realm]];
@synchronized(self) {
isDispatching = NO;
}
[self dispatch];
} else {
/* Clear the in-memory and saved access tokens */
[acToken release];
acToken = nil;
[OAToken removeFromUserDefaultsWithServiceProviderName:oauthBase prefix:[@"access:" stringByAppendingString:realm]];
}
}
- (void)deleteSavedRequestToken {
[OAToken removeFromUserDefaultsWithServiceProviderName:oauthBase prefix:[@"request:" stringByAppendingString:realm]];
[reqToken release];
reqToken = nil;
}
- (void)performCall:(OACall *)aCall {
NSLog(@"Performing call");
[aCall perform:consumer
token:acToken
realm:realm
delegate:self
didFinish:@selector(callFinished:body:)];
}
- (void)dispatch {
OACall *call = [self queue];
if (!call) {
return;
}
@synchronized(self) {
if (isDispatching) {
return;
}
isDispatching = YES;
}
NSLog(@"Started dispatching");
if(acToken) {
[self performCall:call];
} else if(reqToken) {
[self exchangeToken];
} else {
[self requestToken];
}
}
- (void)fetchData:(NSString *)aURL method:(NSString *)aMethod parameters:(NSArray *)theParameters
files:(NSDictionary *)theFiles finished:(SEL)didFinish delegate:(NSObject*)aDelegate {
OACall *call = [[[OACall alloc] initWithURL:[NSURL URLWithString:aURL]
method:aMethod
parameters:theParameters
files:theFiles] autorelease];
NSLog(@"Received request for: %@", aURL);
[self enqueue:call selector:didFinish];
if (aDelegate) {
[delegates setObject:aDelegate forKey:[NSString stringWithFormat:@"%p", call]];
}
[self dispatch];
}
- (void)fetchData:(NSString *)aURL method:(NSString *)aMethod parameters:(NSArray *)theParameters
files:(NSDictionary *)theFiles finished:(SEL)didFinish {
[self fetchData:aURL method:aMethod parameters:theParameters files:theFiles
finished:didFinish delegate:nil];
}
- (void)fetchData:(NSString *)aURL method:(NSString *)aMethod parameters:(NSArray *)theParameters
finished:(SEL)didFinish {
[self fetchData:aURL method:aMethod parameters:theParameters files:nil finished:didFinish];
}
- (void)fetchData:(NSString *)aURL parameters:(NSArray *)theParameters files:(NSDictionary *)theFiles
finished:(SEL)didFinish {
[self fetchData:aURL method:@"POST" parameters:theParameters files:theFiles finished:didFinish];
}
- (void)fetchData:(NSString *)aURL finished:(SEL)didFinish {
[self fetchData:aURL method:nil parameters:nil files:nil finished:didFinish];
}
@end