1*cc02d7e2SAndroid Build Coastguard Worker/* 2*cc02d7e2SAndroid Build Coastguard Worker * 3*cc02d7e2SAndroid Build Coastguard Worker * Copyright 2015 gRPC authors. 4*cc02d7e2SAndroid Build Coastguard Worker * 5*cc02d7e2SAndroid Build Coastguard Worker * Licensed under the Apache License, Version 2.0 (the "License"); 6*cc02d7e2SAndroid Build Coastguard Worker * you may not use this file except in compliance with the License. 7*cc02d7e2SAndroid Build Coastguard Worker * You may obtain a copy of the License at 8*cc02d7e2SAndroid Build Coastguard Worker * 9*cc02d7e2SAndroid Build Coastguard Worker * http://www.apache.org/licenses/LICENSE-2.0 10*cc02d7e2SAndroid Build Coastguard Worker * 11*cc02d7e2SAndroid Build Coastguard Worker * Unless required by applicable law or agreed to in writing, software 12*cc02d7e2SAndroid Build Coastguard Worker * distributed under the License is distributed on an "AS IS" BASIS, 13*cc02d7e2SAndroid Build Coastguard Worker * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14*cc02d7e2SAndroid Build Coastguard Worker * See the License for the specific language governing permissions and 15*cc02d7e2SAndroid Build Coastguard Worker * limitations under the License. 16*cc02d7e2SAndroid Build Coastguard Worker * 17*cc02d7e2SAndroid Build Coastguard Worker */ 18*cc02d7e2SAndroid Build Coastguard Worker 19*cc02d7e2SAndroid Build Coastguard Worker#import <objc/runtime.h> 20*cc02d7e2SAndroid Build Coastguard Worker 21*cc02d7e2SAndroid Build Coastguard Worker#import "GRPCCall+OAuth2.h" 22*cc02d7e2SAndroid Build Coastguard Worker 23*cc02d7e2SAndroid Build Coastguard Workerstatic NSString *const kAuthorizationHeader = @"authorization"; 24*cc02d7e2SAndroid Build Coastguard Workerstatic NSString *const kBearerPrefix = @"Bearer "; 25*cc02d7e2SAndroid Build Coastguard Workerstatic NSString *const kChallengeHeader = @"www-authenticate"; 26*cc02d7e2SAndroid Build Coastguard Worker 27*cc02d7e2SAndroid Build Coastguard Worker@implementation GRPCCall (OAuth2) 28*cc02d7e2SAndroid Build Coastguard Worker@dynamic tokenProvider; 29*cc02d7e2SAndroid Build Coastguard Worker 30*cc02d7e2SAndroid Build Coastguard Worker- (NSString *)oauth2AccessToken { 31*cc02d7e2SAndroid Build Coastguard Worker NSString *headerValue = self.requestHeaders[kAuthorizationHeader]; 32*cc02d7e2SAndroid Build Coastguard Worker if ([headerValue hasPrefix:kBearerPrefix]) { 33*cc02d7e2SAndroid Build Coastguard Worker return [headerValue substringFromIndex:kBearerPrefix.length]; 34*cc02d7e2SAndroid Build Coastguard Worker } else { 35*cc02d7e2SAndroid Build Coastguard Worker return nil; 36*cc02d7e2SAndroid Build Coastguard Worker } 37*cc02d7e2SAndroid Build Coastguard Worker} 38*cc02d7e2SAndroid Build Coastguard Worker 39*cc02d7e2SAndroid Build Coastguard Worker- (void)setOauth2AccessToken:(NSString *)token { 40*cc02d7e2SAndroid Build Coastguard Worker if (token) { 41*cc02d7e2SAndroid Build Coastguard Worker self.requestHeaders[kAuthorizationHeader] = [kBearerPrefix stringByAppendingString:token]; 42*cc02d7e2SAndroid Build Coastguard Worker } else { 43*cc02d7e2SAndroid Build Coastguard Worker [self.requestHeaders removeObjectForKey:kAuthorizationHeader]; 44*cc02d7e2SAndroid Build Coastguard Worker } 45*cc02d7e2SAndroid Build Coastguard Worker} 46*cc02d7e2SAndroid Build Coastguard Worker 47*cc02d7e2SAndroid Build Coastguard Worker- (NSString *)oauth2ChallengeHeader { 48*cc02d7e2SAndroid Build Coastguard Worker return self.responseHeaders[kChallengeHeader]; 49*cc02d7e2SAndroid Build Coastguard Worker} 50*cc02d7e2SAndroid Build Coastguard Worker 51*cc02d7e2SAndroid Build Coastguard Worker- (void)setTokenProvider:(id<GRPCAuthorizationProtocol>)tokenProvider { 52*cc02d7e2SAndroid Build Coastguard Worker objc_setAssociatedObject(self, @selector(tokenProvider), tokenProvider, OBJC_ASSOCIATION_RETAIN); 53*cc02d7e2SAndroid Build Coastguard Worker} 54*cc02d7e2SAndroid Build Coastguard Worker 55*cc02d7e2SAndroid Build Coastguard Worker- (id<GRPCAuthorizationProtocol>)tokenProvider { 56*cc02d7e2SAndroid Build Coastguard Worker return objc_getAssociatedObject(self, @selector(tokenProvider)); 57*cc02d7e2SAndroid Build Coastguard Worker} 58*cc02d7e2SAndroid Build Coastguard Worker 59*cc02d7e2SAndroid Build Coastguard Worker@end 60