OpenFlare/backend/plugins/domain/upload/plugin.go

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

// Package upload provides file uploading, storage abstraction, image transcoding, and caching domain plugin for Cordis.
package upload

import (
	"Wavelet/core"
	"Wavelet/core/contracts"
	"Wavelet/core/extpoints"
	"Wavelet/plugins/domain/upload/filesrv"
	"Wavelet/plugins/domain/upload/handler"
	"Wavelet/plugins/domain/upload/shared"
	"Wavelet/plugins/domain/upload/task"
	"context"
	"embed"
	"reflect"

	"github.com/gin-gonic/gin"
)

//go:embed migrations/*/*.sql
var uploadMigrations embed.FS

// Plugin implements core.Plugin to provide file upload and media serving domain services.
type Plugin struct{}

// New creates a new upload domain plugin.
func New() *Plugin {
	return &Plugin{}
}

// Name returns the unique identifier for the upload domain plugin.
func (p *Plugin) Name() string {
	return "upload"
}

// Inject declares required dependencies for the upload domain plugin.
func (p *Plugin) Inject() []reflect.Type {
	return []reflect.Type{
		reflect.TypeFor[contracts.DBService](),
		reflect.TypeFor[contracts.StorageService](),
		reflect.TypeFor[contracts.AuthService](),
	}
}

// Manifest returns the plugin metadata.
func (p *Plugin) Manifest() core.Manifest {
	return core.Manifest{
		Name:        "upload",
		Version:     "1.0.0",
		Description: "File upload, secure delivery, image transcoding, and storage management domain plugin",
		Author:      "Wavelet Team",
	}
}

// Apply registers upload routes, tasks, and settings into the Context.
func (p *Plugin) Apply(ctx *core.Context) error {
	// Bind DBService
	if db, err := core.Inject[contracts.DBService](ctx); err == nil && db != nil {
		shared.SetDBService(db)
	} else {
		core.When[contracts.DBService](ctx, func(db contracts.DBService) {
			shared.SetDBService(db)
		})
	}

	// Bind CacheService
	if cache, err := core.Inject[contracts.CacheService](ctx); err == nil && cache != nil {
		shared.SetCacheService(cache)
	} else {
		core.When[contracts.CacheService](ctx, func(cache contracts.CacheService) {
			shared.SetCacheService(cache)
		})
	}

	// Bind StorageService
	if storage, err := core.Inject[contracts.StorageService](ctx); err == nil && storage != nil {
		shared.SetStorageService(storage)
	} else {
		core.When[contracts.StorageService](ctx, func(storage contracts.StorageService) {
			shared.SetStorageService(storage)
		})
	}

	// Bind TaskService
	if taskSvc, err := core.Inject[contracts.TaskService](ctx); err == nil && taskSvc != nil {
		shared.SetTaskService(taskSvc)
	} else {
		core.When[contracts.TaskService](ctx, func(taskSvc contracts.TaskService) {
			shared.SetTaskService(taskSvc)
		})
	}

	// Bind AuthService
	if authSvc, err := core.Inject[contracts.AuthService](ctx); err == nil && authSvc != nil {
		shared.SetAuthService(authSvc)
	} else {
		core.When[contracts.AuthService](ctx, func(authSvc contracts.AuthService) {
			shared.SetAuthService(authSvc)
		})
	}

	ctx.OnDispose(func() error {
		shared.ResetServices()
		return nil
	})

	// 0. Resolve auth service for middleware
	var authSvc contracts.AuthService
	if err := core.Using[contracts.AuthService](ctx, func(svc contracts.AuthService) { authSvc = svc }); err != nil {
		return err
	}
	loginMW := authSvc.RequireAuthMiddleware().(gin.HandlerFunc)

	// 0a. Register migrations
	ctx.Migrations().Register("upload", uploadMigrations)

	// 1. Register File Server Routes
	ctx.Router().GET("/f/:id", filesrv.ServeFileByID)

	// 2. Register User/Admin Upload HTTP Routes
	uploadGroup := ctx.Router().Group("/api/v1/upload", loginMW)
	{
		uploadGroup.POST("", handler.UploadFile)
		uploadGroup.GET("", handler.ListFiles)
		uploadGroup.DELETE("/:id", handler.DeleteFile)
		uploadGroup.POST("/batch-download", handler.BatchDownloadFiles)
	}

	adminUploadGroup := ctx.Router().Group("/api/v1/admin/uploads", loginMW)
	{
		adminUploadGroup.GET("", handler.ListFiles)
		adminUploadGroup.GET("/stats", handler.GetFileStats)
		adminUploadGroup.DELETE("/:id", handler.DeleteFile)
		adminUploadGroup.GET("/download/:id", handler.DownloadFile)
		adminUploadGroup.POST("/download/batch", handler.BatchDownloadFiles)
		adminUploadGroup.GET("/types", handler.GetDistinctUploadTypes)
	}

	const (
		defaultCleanupRetry = 3
		defaultStatsRetry   = 2
		defaultSingleRetry  = 1
	)

	// 3. Register tasks. Handlers take raw payload bytes rather than a driver
	// specific task type so they run under both the asynq and in-process workers.
	cleanupHandler := &task.SystemCleanupHandler{}
	ctx.Task().Register(task.SystemCleanupTask, func(c context.Context, payload []byte) error {
		_, err := cleanupHandler.Execute(c, payload)
		return err
	}, extpoints.WithTaskRetry(defaultCleanupRetry))

	rebuildStatsHandler := &task.RebuildUploadStatsHandler{}
	ctx.Task().Register(task.RebuildUploadStatsTask, func(c context.Context, payload []byte) error {
		_, err := rebuildStatsHandler.Execute(c, payload)
		return err
	}, extpoints.WithTaskRetry(defaultStatsRetry))

	migrationHandler := &task.MigrationHandler{}
	ctx.Task().Register(task.StorageMigrationTask, func(c context.Context, payload []byte) error {
		_, err := migrationHandler.Execute(c, payload)
		return err
	}, extpoints.WithTaskRetry(defaultSingleRetry))

	warmHandler := &task.WarmImageCacheHandler{}
	ctx.Task().Register(task.WarmImageCacheTask, func(c context.Context, payload []byte) error {
		_, err := warmHandler.Execute(c, payload)
		return err
	}, extpoints.WithTaskRetry(1))

	// 4. Register Cron Schedule
	ctx.Schedule().RegisterCron("0 3 * * *", task.SystemCleanupTask, nil)

	// 5. Register Settings Schemas
	ctx.Settings().Register(extpoints.SettingSchema{
		Key:         "upload.max_file_size_mb",
		Default:     100,
		Description: "Maximum upload file size limit in MB",
		Type:        "integer",
		Category:    "storage",
	})
	ctx.Settings().Register(extpoints.SettingSchema{
		Key:         "upload.allowed_extensions",
		Default:     "png,jpg,jpeg,gif,webp,svg,pdf,zip,tar,gz",
		Description: "Allowed upload file extensions separated by commas",
		Type:        "string",
		Category:    "storage",
	})

	return nil
}