Added userClientFromTokenUid and teamClientFromTokenUid to DBClientsM…

…anager.
dropbox · Mar 29, 2017 · 63268d8 · 63268d8
1 parent b1a7abf
commit 63268d8
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 0 deletions.
diff --git a/Source/ObjectiveDropboxOfficial/Shared/Handwritten/DBClientsManager.h b/Source/ObjectiveDropboxOfficial/Shared/Handwritten/DBClientsManager.h
@@ -8,6 +8,7 @@
 
 @class DBUserClient;
 @class DBTeamClient;
+@class DBTransportDefaultConfig;
 @class DBOAuthResult;
 
 ///
@@ -57,6 +58,26 @@
 ///
 + (void)setAuthorizedTeamClient:(DBTeamClient * _Nullable)client;
 
+///
+/// Creates a User Client from the supplied `tokenUid` associated with the stored access token.
+///
+/// @param tokenUid The uid used to lookup the stored access token.
+/// @param transportConfig The transport config used to configure the client. Will use default settings if `nil`.
+///
+/// @returns An authorized User Client, or `nil`.
+///
++ (DBUserClient * _Nullable)userClientFromTokenUid:(NSString * _Nonnull)tokenUid transportConfig:(DBTransportDefaultConfig * _Nullable)transportConfig;
+
+///
+/// Creates a Team Client from the supplied `tokenUid` associated with the stored access token.
+///
+/// @param tokenUid The uid used to lookup the stored access token.
+/// @param transportConfig The transport config used to configure the client. Will use default settings if `nil`.
+///
+/// @returns An authorized Team Client, or `nil`.
+///
++ (DBTeamClient * _Nullable)teamClientFromTokenUid:(NSString * _Nonnull)tokenUid transportConfig:(DBTransportDefaultConfig * _Nullable)transportConfig;
+
 ///
 /// Reauthorizes the shared authorized user client instance with the access token retrieved from storage via the
 /// supplied `tokenUid` key.

diff --git a/Source/ObjectiveDropboxOfficial/Shared/Handwritten/DBClientsManager.m b/Source/ObjectiveDropboxOfficial/Shared/Handwritten/DBClientsManager.m
@@ -55,13 +55,30 @@ + (void)setAuthorizedTeamClient:(DBTeamClient *)client {
   authorizedTeamClient = client;
 }
 
++ (DBUserClient *)userClientFromTokenUid:(NSString *)tokenUid transportConfig:(DBTransportDefaultConfig *)transportConfig {
+  NSString *accessToken = [[DBOAuthManager sharedOAuthManager] getAccessToken:tokenUid].accessToken;
+  if (accessToken) {
+    return [[DBUserClient alloc] initWithAccessToken:accessToken transportConfig:transportConfig];
+  }
+  return nil;
+}
+
++ (DBTeamClient *)teamClientFromTokenUid:(NSString *)tokenUid transportConfig:(DBTransportDefaultConfig *)transportConfig {
+  NSString *accessToken = [[DBOAuthManager sharedOAuthManager] getAccessToken:tokenUid].accessToken;
+  if (accessToken) {
+    return [[DBTeamClient alloc] initWithAccessToken:accessToken transportConfig:transportConfig];
+  }
+  return nil;
+}
+
 + (void)setupWithOAuthManager:(DBOAuthManager *)oAuthManager
               transportConfig:(DBTransportDefaultConfig *)transportConfig {
   NSAssert(![DBOAuthManager sharedOAuthManager], @"Only call `[DBClientsManager setupWith...]` once");
 
   [[self class] setupHelperWithOAuthManager:oAuthManager transportConfig:transportConfig];
 
   DBAccessToken *accessToken = [[DBOAuthManager sharedOAuthManager] getFirstAccessToken];
+  NSLog(@"\n\n%@\n\n", accessToken.uid);
   [[self class] setupAuthorizedClient:accessToken];
 }