/
profile_provider.go
96 lines (85 loc) · 2.52 KB
/
profile_provider.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
package common
import (
"os"
"path/filepath"
"runtime"
tcerr "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common/errors"
)
const (
EnvCredentialFile = "TENCENTCLOUD_CREDENTIALS_FILE"
)
type ProfileProvider struct {
}
// DefaultProfileProvider return a default Profile provider
// profile path :
// 1. The value of the environment variable TENCENTCLOUD_CREDENTIALS_FILE
// 2. linux: ~/.tencentcloud/credentials
// windows: \c:\Users\NAME\.tencentcloud\credentials
func DefaultProfileProvider() *ProfileProvider {
return &ProfileProvider{}
}
// getHomePath return home directory according to the system.
// if the environmental variables does not exist, it will return empty string
func getHomePath() string {
// Windows
if runtime.GOOS == "windows" {
return os.Getenv("USERPROFILE")
}
// *nix
return os.Getenv("HOME")
}
func getCredentialsFilePath() string {
homePath := getHomePath()
if homePath == "" {
return homePath
}
return filepath.Join(homePath, ".tencentcloud", "credentials")
}
func checkDefaultFile() (path string, err error) {
path = getCredentialsFilePath()
if path == "" {
return path, nil
}
_, err = os.Stat(path)
if err != nil {
if os.IsNotExist(err) {
return "", nil
}
return "", err
}
return path, nil
}
func (p *ProfileProvider) GetCredential() (CredentialIface, error) {
path, ok := os.LookupEnv(EnvCredentialFile)
// if not set custom file path, will use the default path
if !ok {
var err error
path, err = checkDefaultFile()
// only when the file exist but failed read it the err is not nil
if err != nil {
return nil, tcerr.NewTencentCloudSDKError(creErr, "Failed to find profile file,"+err.Error(), "")
}
// when the path is "" means the file dose not exist
if path == "" {
return nil, fileDoseNotExist
}
// if the EnvCredentialFile is set to "", will return an error
} else if path == "" {
return nil, tcerr.NewTencentCloudSDKError(creErr, "Environment variable '"+EnvCredentialFile+"' cannot be empty", "")
}
cfg, err := parse(path)
if err != nil {
return nil, err
}
sId := cfg.section("default").key("secret_id").string()
sKey := cfg.section("default").key("secret_key").string()
// if sId and sKey is "", but the credential file exist, means an error
if sId == "" || sKey == "" {
return nil, tcerr.NewTencentCloudSDKError(creErr, "Failed to parse profile file,please confirm whether it contains \"secret_id\" and \"secret_key\" in section: \"default\" ", "")
}
return &Credential{
SecretId: sId,
SecretKey: sKey,
Token: "",
}, nil
}