Merge branch 'main' of github.com:ardeman/project-legalgo-go

internal/accessor/module.go

@@ -3,6 +3,7 @@ package repository
 import (
 	categoryrepository "legalgo-BE-go/internal/accessor/category"
 	newsrepository "legalgo-BE-go/internal/accessor/news"
+	"legalgo-BE-go/internal/accessor/oss"
 	redisaccessor "legalgo-BE-go/internal/accessor/redis"
 	staffrepository "legalgo-BE-go/internal/accessor/staff"
 	subscriberepository "legalgo-BE-go/internal/accessor/subscribe"
@@ -22,4 +23,5 @@ var Module = fx.Module("repository", fx.Provide(
 	tagrepository.New,
 	categoryrepository.New,
 	newsrepository.New,
+	oss.New,
 ))

internal/accessor/oss/oss.go

@@ -0,0 +1,70 @@
+package oss
+
+import (
+	"bytes"
+	"context"
+	"fmt"
+	"legalgo-BE-go/config"
+
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/credentials"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/s3"
+)
+
+type OSSConfig interface {
+	GetAccessKeyID() string
+	GetAccessKeySecret() string
+	GetEndpoint() string
+	GetBucketName() string
+	GetHostURL() string
+	GetPublicURL() string
+}
+
+const _awsRegion = "us-east-1"
+const _s3ACL = "public-read"
+
+type OssRepositoryImpl struct {
+	s3  *s3.S3
+	cfg OSSConfig
+}
+
+func New(cfg *config.Config) OSSRepository {
+	ossCfg := cfg.OSSConfig
+	sess, err := session.NewSession(&aws.Config{
+		S3ForcePathStyle: aws.Bool(true),
+		Endpoint:         aws.String(ossCfg.GetEndpoint()),
+		Region:           aws.String(_awsRegion),
+		Credentials:      credentials.NewStaticCredentials(ossCfg.GetAccessKeyID(), ossCfg.GetAccessKeySecret(), ""),
+	})
+
+	if err != nil {
+		fmt.Println("Failed to create AWS session:", err)
+		return nil
+	}
+
+	return &OssRepositoryImpl{
+		s3:  s3.New(sess),
+		cfg: ossCfg,
+	}
+}
+
+func (r *OssRepositoryImpl) UploadFile(ctx context.Context, fileName string, fileContent []byte) (fileUrl string, err error) {
+	reader := bytes.NewReader(fileContent)
+
+	_, err = r.s3.PutObject(&s3.PutObjectInput{
+		Bucket: aws.String(r.cfg.GetBucketName()),
+		Key:    aws.String(fileName),
+		Body:   reader,
+		ACL:    aws.String(_s3ACL),
+	})
+
+	return r.GetPublicURL(fileName), err
+}
+
+func (r *OssRepositoryImpl) GetPublicURL(fileName string) string {
+	if fileName == "" {
+		return ""
+	}
+	return fmt.Sprintf("%s:%s%s", r.cfg.GetPublicURL(), r.cfg.GetBucketName(), fileName)
+}

internal/accessor/oss/upload.go

@@ -0,0 +1,10 @@
+package oss
+
+import (
+	"context"
+)
+
+type OSSRepository interface {
+	UploadFile(ctx context.Context, fileName string, fileContent []byte) (fileUrl string, err error)
+	GetPublicURL(fileName string) string
+}