JCS-pub/common/pkgs/storage/s3/s3.go

package s3

import (
	"fmt"

	"github.com/aws/aws-sdk-go-v2/aws"
	"github.com/aws/aws-sdk-go-v2/credentials"
	"github.com/aws/aws-sdk-go-v2/service/s3"
	"gitlink.org.cn/cloudream/jcs-pub/common/pkgs/storage/factory/reg"
	stgtypes "gitlink.org.cn/cloudream/jcs-pub/common/pkgs/storage/types"
	jcstypes "gitlink.org.cn/cloudream/jcs-pub/common/types"
)

func init() {
	reg.RegisterBuilder[*jcstypes.S3Type](newBuilder)
}

type builder struct {
	stgtypes.EmptyBuilder
	detail *jcstypes.UserSpaceDetail
}

func newBuilder(detail *jcstypes.UserSpaceDetail) stgtypes.StorageBuilder {
	return &builder{
		detail: detail,
	}
}

func (b *builder) FeatureDesc() stgtypes.FeatureDesc {
	return stgtypes.FeatureDesc{}
}

func (b *builder) CreateShardStore(typeOnly bool) (stgtypes.ShardStore, error) {
	stgType := b.detail.UserSpace.Storage.(*jcstypes.S3Type)
	s3Cred, ok := b.detail.UserSpace.Credential.(*jcstypes.S3Cred)
	if !ok {
		return nil, fmt.Errorf("invalid storage credential type %T for s3 storage", b.detail.UserSpace.Credential)
	}

	if typeOnly {
		return (*ShardStore)(nil), nil
	}

	cli, bkt, err := createClient(stgType, s3Cred)
	if err != nil {
		return nil, err
	}

	return NewShardStore(b.detail, cli, bkt, ShardStoreOption{UseAWSSha256: true})
}

func (b *builder) CreateBaseStore(typeOnly bool) (stgtypes.BaseStore, error) {
	stgType := b.detail.UserSpace.Storage.(*jcstypes.S3Type)
	s3Cred, ok := b.detail.UserSpace.Credential.(*jcstypes.S3Cred)
	if !ok {
		return nil, fmt.Errorf("invalid storage credential type %T for s3 storage", b.detail.UserSpace.Credential)
	}

	if typeOnly {
		return (*BaseStore)(nil), nil
	}

	cli, bkt, err := createClient(stgType, s3Cred)
	if err != nil {
		return nil, err
	}

	return NewBaseStore(b.detail, cli, bkt, BaseStoreOption{UseAWSSha256: false})
}

func createClient(stgType *jcstypes.S3Type, cred *jcstypes.S3Cred) (*s3.Client, string, error) {
	awsConfig := aws.Config{}

	if cred.AK != "" && cred.SK != "" {
		cre := aws.Credentials{
			AccessKeyID:     cred.AK,
			SecretAccessKey: cred.SK,
		}
		awsConfig.Credentials = &credentials.StaticCredentialsProvider{Value: cre}
	}

	awsConfig.Region = stgType.Region

	options := []func(*s3.Options){}
	options = append(options, func(s3Opt *s3.Options) {
		s3Opt.BaseEndpoint = &stgType.Endpoint
	})

	cli := s3.NewFromConfig(awsConfig, options...)
	return cli, stgType.Bucket, nil
}

func (b *builder) CreateMultiparter(typeOnly bool) (stgtypes.Multiparter, error) {
	stgType := b.detail.UserSpace.Storage.(*jcstypes.S3Type)
	feat := stgtypes.FindFeature[*jcstypes.MultipartUploadFeature](b.detail)
	if feat == nil {
		return nil, fmt.Errorf("feature %T not found", jcstypes.MultipartUploadFeature{})
	}

	s3Cred, ok := b.detail.UserSpace.Credential.(*jcstypes.S3Cred)
	if !ok {
		return nil, fmt.Errorf("invalid storage credential type %T for s3 base store", b.detail.UserSpace.Credential)
	}

	if typeOnly {
		return (*Multiparter)(nil), nil
	}

	cli, bucket, err := createClient(stgType, s3Cred)
	if err != nil {
		return nil, err
	}

	return NewMultiparter(
		b.detail,
		feat,
		bucket,
		cli,
	), nil
}