FilePlugin.go

package files

import (
	"github.com/go-catupiry/catu"
	files_processor "github.com/go-catupiry/files/processor"
	"github.com/gookit/event"
	"github.com/sirupsen/logrus"
)

type FilePlugin struct {
	catu.Pluginer

	Name            string
	FileController  *FileController
	ImageController *ImageController

	Storages         map[string]Storager
	FileStorageName  string
	ImageStorageName string

	MaxImageWidth  uint
	MaxImageHeight uint

	Processor files_processor.FileProcessor

	ImageStyles map[string]ImageStyleCfg
}

func (p *FilePlugin) GetName() string {
	return p.Name
}

func (p *FilePlugin) GetStorage(fileTypeName string) Storager {
	return p.Storages[fileTypeName]
}

func (p *FilePlugin) SetStorage(fileTypeName string, s Storager) error {
	p.Storages[fileTypeName] = s
	return nil
}

func (p *FilePlugin) Init(app catu.App) error {
	logrus.Debug(p.GetName() + " Init")

	p.FileController = NewFileController(&FileControllerConfiguration{
		App: app,
	})
	p.ImageController = NewImageController(&ImageControllerConfiguration{
		App: app,
	})

	app.GetEvents().On("bindRoutes", event.ListenerFunc(func(e event.Event) error {
		return p.BindRoutes(app)
	}), event.Normal)

	app.GetEvents().On("setTemplateFunctions", event.ListenerFunc(func(e event.Event) error {
		return p.setTemplateFunctions(app)
	}), event.Normal)

	return nil
}

func (p *FilePlugin) BindRoutes(app catu.App) error {
	logrus.Debug(p.GetName() + " BindRoutes")

	ctl := p.ImageController
	ctlFile := p.FileController

	routerAvatar := app.SetRouterGroup("avatar", "/avatar")
	routerAvatar.GET("/:userID", ctl.GetAvatar)

	routerAPI := app.SetRouterGroup("image-api", "/api/v1/image")
	routerAPI.GET("", ctl.Query)
	routerAPI.GET("/:id", ctl.FindOne)
	routerAPI.POST("/:id", ctl.Update)
	routerAPI.POST("/:id/reprocess", ctl.UpdateImageToReprocess)
	// routerAPI.GET("/:id", ctl.Delete)
	routerAPI.GET("/:style/:id", ctl.FindOne)
	routerAPI.GET("/:id/data", ctl.FindOneData)
	routerAPI.POST("", ctl.UploadFile)

	routerFileAPI := app.SetRouterGroup("file-api", "/api/v1/file")
	routerFileAPI.GET("", ctlFile.Query)
	routerFileAPI.GET("/:id", ctlFile.FindOne)
	routerFileAPI.POST("/:id", ctlFile.Update)
	// routerAPI.GET("/:id", ctl.Delete)
	routerFileAPI.GET("/:style/:id", ctlFile.FindOne)
	routerFileAPI.GET("/:id/data", ctlFile.FindOneData)
	routerFileAPI.POST("", ctlFile.UploadFile)

	return nil
}

func (p *FilePlugin) setTemplateFunctions(app catu.App) error {
	app.SetTemplateFunction("image", imageTPLHelper)
	app.SetTemplateFunction("images", imagesTPLHelper)

	return nil
}

type FilePluginCfgs struct {
	Storages         map[string]Storager
	ImageStyles      map[string]ImageStyleCfg
	FileStorageName  string
	ImageStorageName string
	MaxImageWidth    uint
	MaxImageHeight   uint
	Processor        files_processor.FileProcessor
}

type ImageStyleCfg struct {
	Width  int
	Height int
	// image format to convert with lowercase like jpg. Default: webp
	Format string
}

func NewPlugin(cfgs *FilePluginCfgs) *FilePlugin {
	p := FilePlugin{
		Name:             "files",
		FileStorageName:  "local",
		ImageStorageName: "local",
		ImageStyles:      cfgs.ImageStyles,
		MaxImageWidth:    2560,
		MaxImageHeight:   1700,
		Processor:        cfgs.Processor,
	}

	if cfgs.Storages != nil {
		p.Storages = cfgs.Storages
	}

	if cfgs.FileStorageName != "" {
		p.FileStorageName = cfgs.FileStorageName
	}

	if cfgs.ImageStorageName != "" {
		p.ImageStorageName = cfgs.ImageStorageName
	}

	if cfgs.MaxImageWidth != 0 {
		p.MaxImageWidth = cfgs.MaxImageWidth
	}

	if cfgs.MaxImageHeight != 0 {
		p.MaxImageHeight = cfgs.MaxImageHeight
	}

	return &p
}