xref: /aosp_15_r20/external/grpc-grpc/src/objective-c/GRPCClient/GRPCCall+OAuth2.mm (revision cc02d7e222339f7a4f6ba5f422e6413f4bd931f2)
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