Add processing of target account property (#54)

* target account id for python

* added processing of target account ID

---------

Signed-off-by: Anna Hileta <[email protected]>
Co-authored-by: Paige Rossi <[email protected]>

Author: annahileta

OAuth/code_grant.php

@@ -75,14 +75,34 @@
 file_put_contents($outputFile, $accessToken);
 echo "\nAccess token has been written to " . $outputFile . "\n\n";
 
-$userInfo = http(
-    $authorizationEndpoint . 'userinfo', false, [
-    'Authorization: Bearer ' . $accessToken]
+$userInfo = http($authorizationEndpoint . 'userinfo', false, 
+    [
+        'Authorization: Bearer ' . $accessToken
+    ]
 );
-  // impersonation user guid
-  // var_dump($userInfo->sub);
-  $APIAccountId = $userInfo->accounts[0]->account_id;
-  file_put_contents($apiAccountIdFile, $APIAccountId);
-  echo "Account id: $APIAccountId\n";
-  echo "Account id has been written to config/API_ACCOUNT_ID file...\n\n";
+
+if ($TARGET_ACCOUNT_ID != "{TARGET_ACCOUNT_ID}") {
+    $targetAccountFound = false;
+    foreach ($userInfo->accounts as $account_info) {
+        if ($account_info->account_id == $TARGET_ACCOUNT_ID) {
+            $APIAccountId = $account_info->account_id;
+            $targetAccountFound = true;
+            break;
+        }
+    }
+    if (! $targetAccountFound) {
+        throw new Exception("Targeted Account with Id " . $TARGET_ACCOUNT_ID . " not found.");
+    }
+} else {
+    foreach ($userInfo->accounts as $account_info) {
+        if ($account_info->is_default) {
+            $APIAccountId = $account_info->account_id;
+            break;
+        }
+    }
+}
+
+file_put_contents($apiAccountIdFile, $APIAccountId);
+echo "Account id: $APIAccountId\n";
+echo "Account id has been written to config/API_ACCOUNT_ID file...\n\n";
 ?>

OAuth/jwt.php

@@ -136,7 +136,24 @@
     ]
 );
 
-$APIAccountId = $userInfo->accounts[0]->account_id;
-file_put_contents('config/API_ACCOUNT_ID', $APIAccountId);
-
+if ($TARGET_ACCOUNT_ID != "{TARGET_ACCOUNT_ID}") {
+    $targetAccountFound = false;
+    foreach ($userInfo->accounts as $account_info) {
+        if ($account_info->account_id == $TARGET_ACCOUNT_ID) {
+            file_put_contents('config/API_ACCOUNT_ID', $account_info->account_id);
+            $targetAccountFound = true;
+            break;
+        }
+    }
+    if (! $targetAccountFound) {
+        throw new Exception("Targeted Account with Id " . $TARGET_ACCOUNT_ID . " not found.");
+    }
+} else {
+    foreach ($userInfo->accounts as $account_info) {
+        if ($account_info->is_default) {
+            file_put_contents('config/API_ACCOUNT_ID', $account_info->account_id);
+            break;
+        }
+    }
+}
 ?>

OAuth/jwt_auth.py

@@ -109,9 +109,15 @@ def _write_token(cls, scopes):
 
         user_info = api_client.get_user_info(cls.ds_app.access_token)
         accounts = user_info.get_accounts()
-        api_account_id = open("./config/API_ACCOUNT_ID", "w")
-        api_account_id.write(accounts[0].account_id)
-        api_account_id.close()
+        target_account_id = os.environ.get('TARGET_ACCOUNT_ID')
+        if target_account_id != "{TARGET_ACCOUNT_ID}":
+            api_account_id = open("./config/API_ACCOUNT_ID", "w")
+            api_account_id.write(target_account_id)
+            api_account_id.close()
+        else:
+            api_account_id = open("./config/API_ACCOUNT_ID", "w")
+            api_account_id.write(accounts[0].account_id)
+            api_account_id.close()
 
 
     @staticmethod

OAuth/utils.php

@@ -11,6 +11,7 @@
 
 $INTEGRATION_KEY_JWT = getenv("INTEGRATION_KEY_JWT");
 $IMPERSONATION_USER_GUID = getenv("IMPERSONATION_USER_GUID");
+$TARGET_ACCOUNT_ID = getenv("TARGET_ACCOUNT_ID");
 
 $authorizationEndpoint = 'https://account-d.docusign.com/oauth/';
 

config/settings.example.txt

@@ -10,3 +10,5 @@ export INTEGRATION_KEY_AUTH_CODE="{INTEGRATION_KEY_AUTH_CODE}"
 export SECRET_KEY="{SECRET_KEY}"
 export GATEWAY_ACCOUNT_ID="{DS_PAYMENT_GATEWAY_ID}"
 export QUICKSTART="{QUICKSTART_VALUE}"
+export TARGET_ACCOUNT_ID="{TARGET_ACCOUNT_ID}"
+