mockgen

internal/client/client.go

@@ -0,0 +1,59 @@
+package client
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/aws/aws-sdk-go-v2/config"
+	"github.com/aws/aws-sdk-go-v2/service/iam"
+	"github.com/aws/aws-sdk-go-v2/service/s3"
+	"github.com/aws/aws-sdk-go-v2/service/sts"
+)
+
+type AwsClient interface {
+	IamClient() *AwsIamClient
+	StsClient() *AwsStsClient
+	S3Client(region, bucketName string) *AwsS3Client
+}
+
+type AwsIamClient struct {
+	Client AwsIamAPI
+}
+type AwsStsClient struct {
+	Client AwsStsAPI
+}
+type AwsS3Client struct {
+	Client     AwsS3API
+	region     string
+	bucketName string
+}
+
+func NewAwsClientFactory(ctx context.Context) (*AwsClientFactory, error) {
+	cfg, err := config.LoadDefaultConfig(
+		ctx,
+	)
+	if err != nil {
+		return nil, fmt.Errorf("unable to load SDK config, %w", err)
+	}
+	return &AwsClientFactory{config: cfg}, nil
+}
+
+func (a *AwsClientFactory) IamClient() *AwsIamClient {
+	return &AwsIamClient{
+		iam.NewFromConfig(a.config),
+	}
+}
+
+func (a *AwsClientFactory) StsClient() *AwsStsClient {
+	return &AwsStsClient{
+		sts.NewFromConfig(a.config),
+	}
+}
+
+func (a *AwsClientFactory) S3Client(bucketName, region string) *AwsS3Client {
+	return &AwsS3Client{
+		s3.NewFromConfig(a.config),
+		region,
+		bucketName,
+	}
+}