aws S3Client

internal/client/aws.go

@@ -0,0 +1,53 @@
+package client
+
+import (
+	"bytes"
+	"context"
+	"fmt"
+
+	"github.com/aws/aws-sdk-go-v2/aws"
+	"github.com/aws/aws-sdk-go-v2/config"
+	"github.com/aws/aws-sdk-go-v2/service/s3"
+)
+
+type AwsConfig struct {
+	config aws.Config
+}
+
+func NewAwsClient(ctx context.Context, region string) (*AwsConfig, error) {
+	cfg, err := config.LoadDefaultConfig(ctx,
+		config.WithRegion(region),
+	)
+	if err != nil {
+		return nil, fmt.Errorf("unable to load SDK config, %w", err)
+	}
+	return &AwsConfig{config: cfg}, nil
+}
+
+func (a *AwsConfig) S3Cient(bucketName string) *AwsS3Client {
+	return &AwsS3Client{
+		bucketName,
+		s3.NewFromConfig(a.config),
+	}
+}
+
+type AwsS3Client struct {
+	bucketName string
+	client     *s3.Client
+}
+
+func (a *AwsS3Client) PutObject(ctx context.Context, key string, body []byte) error {
+	_, err := a.client.PutObject(ctx, &s3.PutObjectInput{
+		Bucket: aws.String(a.bucketName),
+		Key:    aws.String(key),
+		Body:   bytes.NewReader(body),
+	})
+	return err
+}
+
+func (a *AwsS3Client) CreateBucket(ctx context.Context) error {
+	_, err := a.client.CreateBucket(ctx, &s3.CreateBucketInput{
+		Bucket: aws.String(a.bucketName),
+	})
+	return err
+}