/
rsa_client_deps.go
56 lines (45 loc) · 2.14 KB
/
rsa_client_deps.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
package rsaauth
import (
"github.com/aws/amazon-ssm-agent/agent/appconfig"
"github.com/aws/amazon-ssm-agent/agent/log"
"github.com/aws/amazon-ssm-agent/agent/ssm/authtokenrequest"
"github.com/aws/amazon-ssm-agent/agent/ssm/util"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/client"
"github.com/aws/aws-sdk-go/aws/credentials"
"github.com/aws/aws-sdk-go/aws/request"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/ssm"
)
type IRsaClientDeps interface {
NewStaticCredentials(id string, secret string, token string) *credentials.Credentials
NewSession(config *aws.Config) (*session.Session, error)
AwsConfig(log log.T, appConfig appconfig.SsmagentConfig, service string, region string) *aws.Config
NewSsmSdk(p client.ConfigProvider, cfgs ...*aws.Config) *ssm.SSM
NewAuthTokenClient(sdk *ssm.SSM) authtokenrequest.IClient
MakeAddToUserAgentHandler(name string, version string, extra ...string) func(*request.Request)
NewCredentials(provider credentials.Provider) *credentials.Credentials
}
type rsaClientDeps struct{}
var deps IRsaClientDeps = &rsaClientDeps{}
func (r *rsaClientDeps) NewStaticCredentials(id string, secret string, token string) *credentials.Credentials {
return credentials.NewStaticCredentials(id, secret, token)
}
func (r *rsaClientDeps) NewSession(config *aws.Config) (*session.Session, error) {
return session.NewSession(config)
}
func (r *rsaClientDeps) AwsConfig(log log.T, appConfig appconfig.SsmagentConfig, service string, region string) *aws.Config {
return util.AwsConfig(log, appConfig, service, region)
}
func (r *rsaClientDeps) NewSsmSdk(p client.ConfigProvider, cfgs ...*aws.Config) *ssm.SSM {
return ssm.New(p, cfgs...)
}
func (r *rsaClientDeps) NewAuthTokenClient(sdk *ssm.SSM) authtokenrequest.IClient {
return authtokenrequest.NewClient(sdk)
}
func (r *rsaClientDeps) MakeAddToUserAgentHandler(name string, version string, extra ...string) func(*request.Request) {
return request.MakeAddToUserAgentHandler(name, version, extra...)
}
func (r *rsaClientDeps) NewCredentials(provider credentials.Provider) *credentials.Credentials {
return credentials.NewCredentials(provider)
}