84 lines
1.5 KiB
Go
84 lines
1.5 KiB
Go
package handle
|
|
|
|
import (
|
|
"allapp-go/internal/errors"
|
|
"allapp-go/internal/httpx"
|
|
"allapp-go/pkg/s3store"
|
|
"fmt"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/service/s3"
|
|
"github.com/gofiber/fiber/v3"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
func S3ListBuckets(c fiber.Ctx) error {
|
|
result, err := s3store.S3Client.ListBuckets(c.Context(), &s3.ListBucketsInput{})
|
|
if err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
buckets := make([]string, 0, len(result.Buckets))
|
|
for _, b := range result.Buckets {
|
|
if b.Name != nil {
|
|
buckets = append(buckets, *b.Name)
|
|
}
|
|
}
|
|
|
|
return httpx.OK(c, buckets)
|
|
}
|
|
|
|
func S3UploadFile(c fiber.Ctx) error {
|
|
form, err := c.MultipartForm()
|
|
if err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
files := form.File["files"]
|
|
|
|
if len(files) == 0 {
|
|
return httpx.OK(c, nil)
|
|
}
|
|
|
|
now := time.Now()
|
|
folderPath := fmt.Sprintf("%d/%02d/%02d",
|
|
now.Year(),
|
|
now.Month(),
|
|
now.Day(),
|
|
)
|
|
|
|
var result []string
|
|
|
|
for _, fileHeader := range files {
|
|
file, err := fileHeader.Open()
|
|
if err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
ext := filepath.Ext(fileHeader.Filename)
|
|
newName := fmt.Sprintf("%s%s", uuid.NewString(), ext)
|
|
key := fmt.Sprintf("%s/%s", folderPath, newName)
|
|
|
|
contentType := fileHeader.Header.Get("Content-Type")
|
|
|
|
err = s3store.UploadToRustFS(
|
|
c.Context(),
|
|
key,
|
|
file,
|
|
fileHeader.Size,
|
|
contentType,
|
|
)
|
|
|
|
_ = file.Close()
|
|
|
|
if err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
result = append(result, key)
|
|
}
|
|
|
|
return httpx.OK(c, result)
|
|
}
|