fix(invoke): use sessionToken

use sessionToken
TencentCloud · Apr 30, 2019 · c31d3d7 · c31d3d7
1 parent 1ab4ee0
commit c31d3d7
Show file tree

Hide file tree

Showing 6 changed files with 28 additions and 22 deletions.
diff --git a/dist/index.d.ts b/dist/index.d.ts
@@ -2,6 +2,7 @@ import * as services from './services';
 interface InitConfig {
     secretId: any;
     secretKey: any;
+    token?: any;
     region: any;
 }
 declare class SDK {

diff --git a/dist/index.js b/dist/index.js
@@ -10,6 +10,7 @@ class SDK {
         const __config = Object.assign({}, {
             secretId: process.env.TENCENTCLOUD_SECRETID,
             secretKey: process.env.TENCENTCLOUD_SECRETKEY,
+            token: process.env.TENCENTCLOUD_SESSIONTOKEN,
             region: 'ap-guangzhou'
         }, config);
         if (!__config.secretId || !__config.secretKey)

diff --git a/dist/services/invoke.js b/dist/services/invoke.js
@@ -4,7 +4,7 @@ const utils_1 = require("../helper/utils");
 async function default_1(params) {
     const requestHelper = this.requestHelper;
     const region = this.config.region;
-    const __params = utils_1.caseForObject(Object.assign({
+    let __params = Object.assign({
         region,
         namespace: 'default',
         qualifier: '$LATEST'
@@ -13,8 +13,10 @@ async function default_1(params) {
         logType: 'Tail',
         version: '2018-04-16',
         action: 'Invoke'
-    }), 'upper');
-    return await requestHelper(__params, {
+    });
+    if (this.config.token)
+        __params.token = this.config.token;
+    return await requestHelper(utils_1.caseForObject(__params, 'upper'), {
         serviceType: 'scf'
     });
 }

diff --git a/node_modules/qcloudapi-sdk/index.js b/node_modules/qcloudapi-sdk/index.js
diff --git a/src/index.ts b/src/index.ts
@@ -5,6 +5,7 @@ import * as util from 'util'
 interface InitConfig {
   secretId
   secretKey
+  token?
   region
 }
 
@@ -18,6 +19,7 @@ class SDK {
       {
         secretId: process.env.TENCENTCLOUD_SECRETID,
         secretKey: process.env.TENCENTCLOUD_SECRETKEY,
+        token: process.env.TENCENTCLOUD_SESSIONTOKEN,
         region: 'ap-guangzhou'
       },
       config

diff --git a/src/services/invoke.ts b/src/services/invoke.ts
@@ -34,24 +34,24 @@ export default async function(params: Params): Promise<Res> {
   const requestHelper = this.requestHelper
   const region = this.config.region
 
-  const __params = caseForObject(
-    Object.assign(
-      {
-        region,
-        namespace: 'default',
-        qualifier: '$LATEST'
-      },
-      params,
-      {
-        invocationType: 'RequestResponse',
-        logType: 'Tail',
-        version: '2018-04-16',
-        action: 'Invoke'
-      }
-    ),
-    'upper'
-  )
-  return await requestHelper(__params, {
+  let __params = Object.assign(
+    {
+      region,
+      namespace: 'default',
+      qualifier: '$LATEST'
+    },
+    params,
+    {
+      invocationType: 'RequestResponse',
+      logType: 'Tail',
+      version: '2018-04-16',
+      action: 'Invoke'
+    }
+  ) as any
+
+  if (this.config.token) __params.token = this.config.token
+
+  return await requestHelper(caseForObject(__params, 'upper'), {
     serviceType: 'scf'
   })
 }