Init V4 community edition (#2265)

* Init V4 community edition

* Init V4 community edition

pkg/thumb/vips.go

@@ -5,59 +5,90 @@ import (
 	"context"
 	"fmt"
 	"io"
+	"os"
 	"os/exec"
 	"path/filepath"
-	"strings"
+	"runtime"
+	"strconv"
 
-	model "github.com/cloudreve/Cloudreve/v3/models"
-	"github.com/cloudreve/Cloudreve/v3/pkg/util"
+	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/manager/entitysource"
+	"github.com/cloudreve/Cloudreve/v4/pkg/logging"
+	"github.com/cloudreve/Cloudreve/v4/pkg/setting"
+	"github.com/cloudreve/Cloudreve/v4/pkg/util"
 	"github.com/gofrs/uuid"
 )
 
-func init() {
-	RegisterGenerator(&VipsGenerator{})
+func NewVipsGenerator(l logging.Logger, settings setting.Provider) *VipsGenerator {
+	return &VipsGenerator{l: l, settings: settings}
 }
 
 type VipsGenerator struct {
-	exts        []string
-	lastRawExts string
+	l        logging.Logger
+	settings setting.Provider
 }
 
-func (v *VipsGenerator) Generate(ctx context.Context, file io.Reader, src, name string, options map[string]string) (*Result, error) {
-	const (
-		thumbVipsPath      = "thumb_vips_path"
-		thumbVipsExts      = "thumb_vips_exts"
-		thumbEncodeQuality = "thumb_encode_quality"
-		thumbEncodeMethod  = "thumb_encode_method"
-		tempPath           = "temp_path"
-	)
-	vipsOpts := model.GetSettingByNames(thumbVipsPath, thumbVipsExts, thumbEncodeQuality, thumbEncodeMethod, tempPath)
-
-	if v.lastRawExts != vipsOpts[thumbVipsExts] {
-		v.exts = strings.Split(vipsOpts[thumbVipsExts], ",")
-		v.lastRawExts = vipsOpts[thumbVipsExts]
+func (v *VipsGenerator) Generate(ctx context.Context, es entitysource.EntitySource, ext string, previous *Result) (*Result, error) {
+	if !util.IsInExtensionListExt(v.settings.VipsThumbExts(ctx), ext) {
+		return nil, fmt.Errorf("unsupported video format: %w", ErrPassThrough)
 	}
 
-	if !util.IsInExtensionList(v.exts, name) {
-		return nil, fmt.Errorf("unsupported image format: %w", ErrPassThrough)
+	if es.Entity().Size() > v.settings.VipsThumbMaxSize(ctx) {
+		return nil, fmt.Errorf("file is too big: %w", ErrPassThrough)
 	}
 
 	outputOpt := ".png"
-	if vipsOpts[thumbEncodeMethod] == "jpg" {
-		outputOpt = fmt.Sprintf(".jpg[Q=%s]", vipsOpts[thumbEncodeQuality])
+	encode := v.settings.ThumbEncode(ctx)
+	if encode.Format == "jpg" {
+		outputOpt = fmt.Sprintf(".jpg[Q=%d]", encode.Quality)
 	}
 
-	cmd := exec.CommandContext(ctx,
-		vipsOpts[thumbVipsPath], "thumbnail_source", "[descriptor=0]", outputOpt, options["thumb_width"],
-		"--height", options["thumb_height"])
+	input := "[descriptor=0]"
+	usePipe := true
+	if runtime.GOOS == "windows" {
+		// Pipe IO is not working on Windows for VIPS
+		if es.IsLocal() {
+			// escape [ and ] in file name
+			input = fmt.Sprintf("[filename=\"%s\"]", es.LocalPath(ctx))
+			usePipe = false
+		} else {
+			usePipe = false
+			// If not local policy files, download to temp folder
+			tempPath := filepath.Join(
+				util.DataPath(v.settings.TempPath(ctx)),
+				"thumb",
+				fmt.Sprintf("vips_%s.%s", uuid.Must(uuid.NewV4()).String(), ext),
+			)
+			input = fmt.Sprintf("[filename=\"%s\"]", tempPath)
 
-	outTempPath := filepath.Join(
-		util.RelativePath(vipsOpts[tempPath]),
-		"thumb",
+			// Due to limitations of ffmpeg, we need to write the input file to disk first
+			tempInputFile, err := util.CreatNestedFile(tempPath)
+			if err != nil {
+				return nil, fmt.Errorf("failed to create temp file: %w", err)
+			}
+
+			defer os.Remove(tempPath)
+			defer tempInputFile.Close()
+
+			if _, err = io.Copy(tempInputFile, es); err != nil {
+				return &Result{Path: tempPath}, fmt.Errorf("failed to write input file: %w", err)
+			}
+
+			tempInputFile.Close()
+		}
+	}
+
+	w, h := v.settings.ThumbSize(ctx)
+	cmd := exec.CommandContext(ctx,
+		v.settings.VipsPath(ctx), "thumbnail_source", input, outputOpt, strconv.Itoa(w),
+		"--height", strconv.Itoa(h))
+
+	tempPath := filepath.Join(
+		util.DataPath(v.settings.TempPath(ctx)),
+		thumbTempFolder,
 		fmt.Sprintf("thumb_%s", uuid.Must(uuid.NewV4()).String()),
 	)
 
-	thumbFile, err := util.CreatNestedFile(outTempPath)
+	thumbFile, err := util.CreatNestedFile(tempPath)
 	if err != nil {
 		return nil, fmt.Errorf("failed to create temp file: %w", err)
 	}
@@ -66,22 +97,24 @@ func (v *VipsGenerator) Generate(ctx context.Context, file io.Reader, src, name
 
 	// Redirect IO
 	var vipsErr bytes.Buffer
-	cmd.Stdin = file
+	if usePipe {
+		cmd.Stdin = es
+	}
 	cmd.Stdout = thumbFile
 	cmd.Stderr = &vipsErr
 
 	if err := cmd.Run(); err != nil {
-		util.Log().Warning("Failed to invoke vips: %s", vipsErr.String())
-		return nil, fmt.Errorf("failed to invoke vips: %w", err)
+		v.l.Warning("Failed to invoke vips: %s", vipsErr.String())
+		return &Result{Path: tempPath}, fmt.Errorf("failed to invoke vips: %w, raw output: %s", err, vipsErr.String())
 	}
 
-	return &Result{Path: outTempPath}, nil
+	return &Result{Path: tempPath}, nil
 }
 
 func (v *VipsGenerator) Priority() int {
 	return 100
 }
 
-func (v *VipsGenerator) EnableFlag() string {
-	return "thumb_vips_enabled"
+func (v *VipsGenerator) Enabled(ctx context.Context) bool {
+	return v.settings.VipsThumbGeneratorEnabled(ctx)
 }