OpenFlare/backend/plugins/domain/upload/ingest/helpers.go

// Copyright 2026 Arctel.net
// SPDX-License-Identifier: Apache-2.0

package ingest

import (
	"Wavelet/pkg/idgen"
	"Wavelet/pkg/logger"
	"Wavelet/plugins/domain/upload/models"
	"Wavelet/plugins/domain/upload/repository"
	"Wavelet/plugins/domain/upload/shared"
	"context"
	"encoding/json"
	"errors"
	"fmt"
	"io"
	"strings"
	"time"

	"gorm.io/gorm"

	uploadcache "Wavelet/plugins/domain/upload/cache"

	uploadstats "Wavelet/plugins/domain/upload/stats"
	uploadstorage "Wavelet/plugins/domain/upload/storage"
)

func normalizeRequest(req *Request) {
	req.Extension = strings.ToLower(strings.TrimSpace(req.Extension))
	if req.Extension == "" {
		req.Extension = "bin"
	}
	if req.Type == "" {
		req.Type = "generic"
	}
	if req.Status == "" {
		req.Status = models.UploadStatusUsed
	}
}

func resolveAccessMode(uploadType string, explicit *int) int {
	if explicit != nil {
		return *explicit
	}
	if uploadType == shared.DefaultPublicUploadType {
		return 1
	}
	return 0
}

func validateAllowedExtension(ctx context.Context, ext string) error {
	var val string
	db := shared.GetDB(ctx)
	if db != nil {
		err := db.Table("w_system_configs").Where("key = ?", "upload_allowed_extensions").Pluck("value", &val).Error
		if err != nil {
			if errors.Is(err, gorm.ErrRecordNotFound) {
				return nil
			}
			logger.WarnF(ctx, "failed to query upload_allowed_extensions: %v", err)
			return nil
		}
	}
	if val == "" {
		return nil
	}

	var list []string
	if err := json.Unmarshal([]byte(val), &list); err == nil {
		for _, allowedExt := range list {
			if strings.EqualFold(strings.TrimSpace(allowedExt), ext) {
				return nil
			}
		}
		return errors.New(shared.ErrUnsupportedFormat)
	}

	allowedExts := strings.Split(strings.ToLower(val), ",")
	for _, allowedExt := range allowedExts {
		if strings.TrimSpace(allowedExt) == ext {
			return nil
		}
	}
	return errors.New(shared.ErrUnsupportedFormat)
}

func defaultObjectKey(id uint64, ext string) string {
	return fmt.Sprintf("uploads/%s/%d.%s", time.Now().Format("2006/01/02"), id, ext)
}

func buildObjectKey(req Request, id uint64) string {
	if req.ObjectKeyFn != nil {
		return req.ObjectKeyFn(id, req.Extension)
	}
	return defaultObjectKey(id, req.Extension)
}

func storeObject(ctx context.Context, objectKey string, reader io.Reader, size int64, mimeType string, meta *models.UploadMetadata) (string, error) {
	if uploadstorage.ReadOnly(ctx) {
		return "", ErrStorageReadOnly
	}

	storageSvc := shared.GetStorage(ctx)
	if storageSvc == nil {
		logger.ErrorF(ctx, "初始化活动存储失败: storage service is nil")
		return "", errors.New(shared.ErrSaveFileFailed)
	}

	result, err := storageSvc.Put(ctx, objectKey, reader, size, mimeType)
	if err != nil {
		logger.ErrorF(ctx, "写入存储失败: %v", err)
		return "", errors.New(shared.ErrSaveFileFailed)
	}

	meta.Bucket = result.Bucket
	return result.Key, nil
}

func persistUploadRecord(ctx context.Context, upload *models.Upload, objectKey string) error {
	if err := createUploadWithStats(ctx, upload); err != nil {
		if storageSvc := shared.GetStorage(ctx); storageSvc != nil {
			if deleteErr := storageSvc.Delete(ctx, objectKey); deleteErr != nil {
				logger.WarnF(ctx, "清理未写入数据库的上传对象失败: %v", deleteErr)
			}
		}
		return err
	}
	uploadcache.SetUploadMeta(ctx, *upload)
	return nil
}

func createUploadWithStats(ctx context.Context, upload *models.Upload) error {
	db := shared.GetDB(ctx)
	if db == nil {
		return errors.New("database service not available")
	}
	return db.Transaction(func(tx *gorm.DB) error {
		if err := repository.CreateUploadTx(tx, upload); err != nil {
			return err
		}
		return uploadstats.ApplyUploadStatsDeltaTx(tx, upload, 1)
	})
}

func createDedupRecord(ctx context.Context, existing models.Upload, req Request) (Result, error) {
	accessMode := resolveAccessMode(req.Type, req.AccessMode)
	newUpload := models.Upload{
		ID:         idgen.NextUint64ID(),
		UserID:     req.UserID,
		FileName:   req.FileName,
		FilePath:   existing.FilePath,
		FileSize:   req.Size,
		MimeType:   req.MimeType,
		Extension:  req.Extension,
		Hash:       req.Hash,
		Type:       req.Type,
		Status:     req.Status,
		AccessMode: accessMode,
		Metadata:   existing.Metadata,
	}
	if err := persistUploadRecord(ctx, &newUpload, existing.FilePath); err != nil {
		return Result{}, err
	}
	logger.InfoF(ctx, "文件触发秒传成功! ID: %d, Path: %s", newUpload.ID, existing.FilePath)
	return Result{
		Upload:  newUpload,
		Created: true,
		Stored:  false,
	}, nil
}

func uploadstorageReadOnly(ctx context.Context) bool {
	return uploadstorage.ReadOnly(ctx)
}

func createNewUpload(ctx context.Context, req Request) (Result, error) {
	if uploadstorageReadOnly(ctx) {
		return Result{}, ErrStorageReadOnly
	}
	if !req.SkipExtensionCheck {
		if err := validateAllowedExtension(ctx, req.Extension); err != nil {
			return Result{}, err
		}
	}

	id := idgen.NextUint64ID()
	objectKey := buildObjectKey(req, id)
	storedKey, err := storeObject(ctx, objectKey, req.Reader, req.Size, req.MimeType, &req.Metadata)
	if err != nil {
		return Result{}, err
	}

	accessMode := resolveAccessMode(req.Type, req.AccessMode)
	upload := models.Upload{
		ID:         id,
		UserID:     req.UserID,
		FileName:   req.FileName,
		FilePath:   storedKey,
		FileSize:   req.Size,
		MimeType:   req.MimeType,
		Extension:  req.Extension,
		Hash:       req.Hash,
		Type:       req.Type,
		Status:     req.Status,
		AccessMode: accessMode,
		Metadata:   req.Metadata,
	}
	if err := persistUploadRecord(ctx, &upload, storedKey); err != nil {
		return Result{}, err
	}

	return Result{
		Upload:  upload,
		Created: true,
		Stored:  true,
	}, nil
}