Introduce TencentCloud.Credential

Author: stevapple

Sources/TencentCloudCore/Credential.swift

@@ -0,0 +1,37 @@
+//===------------------------------------------------------------------------------------===//
+//
+// This source file is part of the SwiftTencentSCFRuntime open source project
+//
+// Copyright (c) 2020 stevapple and the SwiftTencentSCFRuntime project authors
+// Licensed under Apache License v2.0
+//
+// See LICENSE.txt for license information
+// See CONTRIBUTORS.txt for the list of SwiftTencentSCFRuntime project authors
+//
+// SPDX-License-Identifier: Apache-2.0
+//
+//===------------------------------------------------------------------------------------===//
+
+/// A `struct` representing a Tencent Cloud credential.
+extension TencentCloud {
+    public struct Credential {
+        let secretId: String
+        let secretKey: String
+        let sessionToken: String?
+
+        enum CodingKeys: String, CodingKey {
+            case secretId = "SecretId"
+            case secretKey = "SecretKey"
+            case sessionToken = "SessionToken"
+        }
+
+        public init(secretId: String,
+                    secretKey: String,
+                    sessionToken: String? = nil)
+        {
+            self.secretId = secretId
+            self.secretKey = secretKey
+            self.sessionToken = sessionToken
+        }
+    }
+}

Sources/TencentSCFRuntimeCore/SCFContext.swift

@@ -106,6 +106,12 @@ extension SCF {
             .init(stringLiteral: SCF.Env["SCF_FUNCTIONVERSION"] ?? "")
         }
 
+        public var credential: TencentCloud.Credential {
+            .init(secretId: SCF.Env["TENCENTCLOUD_SECRETID"] ?? "",
+                  secretKey: SCF.Env["TENCENTCLOUD_SECRETKEY"] ?? "",
+                  sessionToken: SCF.Env["TENCENTCLOUD_SESSIONTOKEN"] ?? "")
+        }
+
         /// `Logger` to log with.
         ///
         /// - Note: The `LogLevel` can be configured using the `LOG_LEVEL` environment variable.

Tests/TencentSCFRuntimeCoreTests/SCFContextTest.swift

@@ -34,6 +34,9 @@ class SCFContextTest: XCTestCase {
         XCTAssertEqual(context.name, "my-swift-function")
         XCTAssertEqual(context.namespace, "default")
         XCTAssertEqual(context.version, .latest)
+        XCTAssertEqual(context.credential.secretId, "")
+        XCTAssertEqual(context.credential.secretKey, "")
+        XCTAssertEqual(context.credential.sessionToken, "")
     }
 
     func testEnvUpdateWithDict() {
@@ -44,6 +47,9 @@ class SCFContextTest: XCTestCase {
             "SCF_FUNCTIONNAME": "another-swift-function",
             "SCF_NAMESPACE": "custom",
             "SCF_FUNCTIONVERSION": "2",
+            "TENCENTCLOUD_SECRETID": "SECRET_ID",
+            "TENCENTCLOUD_SECRETKEY": "SECRET_KEY",
+            "TENCENTCLOUD_SESSIONTOKEN": "SESSION_TOKEN",
         ]
 
         SCF.Env.update(with: customEnvironment)
@@ -62,6 +68,9 @@ class SCFContextTest: XCTestCase {
         XCTAssertEqual(context.name, "another-swift-function")
         XCTAssertEqual(context.namespace, "custom")
         XCTAssertEqual(context.version, .version(2))
+        XCTAssertEqual(context.credential.secretId, "SECRET_ID")
+        XCTAssertEqual(context.credential.secretKey, "SECRET_KEY")
+        XCTAssertEqual(context.credential.sessionToken, "SESSION_TOKEN")
 
         SCF.Env.reset()
     }
@@ -74,6 +83,9 @@ class SCFContextTest: XCTestCase {
             "SCF_FUNCTIONNAME": "another-swift-function",
             "SCF_NAMESPACE": "custom",
             "SCF_FUNCTIONVERSION": "2",
+            "TENCENTCLOUD_SECRETID": "SECRET_ID",
+            "TENCENTCLOUD_SECRETKEY": "SECRET_KEY",
+            "TENCENTCLOUD_SESSIONTOKEN": "SESSION_TOKEN",
         ]
 
         for (key, value) in customEnvironment {
@@ -94,6 +106,9 @@ class SCFContextTest: XCTestCase {
         XCTAssertEqual(context.name, "another-swift-function")
         XCTAssertEqual(context.namespace, "custom")
         XCTAssertEqual(context.version, .version(2))
+        XCTAssertEqual(context.credential.secretId, "SECRET_ID")
+        XCTAssertEqual(context.credential.secretKey, "SECRET_KEY")
+        XCTAssertEqual(context.credential.sessionToken, "SESSION_TOKEN")
 
         SCF.Env.reset()
     }