feat: upload

backend/pkg/storage/storage.go

@@ -0,0 +1,70 @@
+package storage
+
+import (
+	"context"
+	"os"
+	"path/filepath"
+	"time"
+
+	"backend/database/fields"
+	"backend/database/models/qvyun_v2/public/model"
+)
+
+func Build(m *model.Storages) Storage {
+	switch m.Type {
+	case fields.StorageTypeLocal:
+		return &Local{
+			Host: m.Config.Local.Host,
+			Path: m.Config.Local.Path,
+		}
+	case fields.StorageTypeS3:
+		return &S3{
+			Endpoint:        m.Config.S3.Endpoint,
+			AccessKeyID:     m.Config.S3.AccessKeyID,
+			AccessKeySecret: m.Config.S3.AccessKeySecret,
+			BucketName:      m.Config.S3.BucketName,
+			Path:            m.Config.S3.Path,
+		}
+	default:
+		panic("invalid storage type")
+	}
+}
+
+type Storage interface {
+	Save(ctx context.Context, file *UploadedFile) (*UploadedFile, error)
+}
+
+type Local struct {
+	Host string
+	Path string
+}
+
+func (p *Local) Save(ctx context.Context, file *UploadedFile) (*UploadedFile, error) {
+	saveToPath := filepath.Join(file.Path, time.Now().Format("2006/01/02"))
+	if err := os.MkdirAll(saveToPath, 0o755); err != nil {
+		return nil, err
+	}
+
+	filename := file.Hash + filepath.Ext(file.Name)
+	finalPath := filepath.Join(os.TempDir(), filename)
+	saveTo := filepath.Join(saveToPath, file.Name)
+	if err := os.Rename(finalPath, saveTo); err != nil {
+		return nil, err
+	}
+
+	file.Path = filepath.Join(time.Now().Format("2006/01/02"), filename)
+	file.Preview = filepath.Join(p.Host, time.Now().Format("2006/01/02"), filename)
+	return file, nil
+}
+
+type S3 struct {
+	Endpoint        string
+	AccessKeyID     string
+	AccessKeySecret string
+	BucketName      string
+	Path            string
+}
+
+func (s *S3) Save(ctx context.Context, file *UploadedFile) (*UploadedFile, error) {
+	return nil, nil
+}