mirror of https://github.com/cloudreve/Cloudreve
80 lines
2.0 KiB
Go
80 lines
2.0 KiB
Go
package thumb
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"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/dhowden/tag"
|
|
"github.com/gofrs/uuid"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
func NewMusicCoverGenerator(l logging.Logger, settings setting.Provider) *MusicCoverGenerator {
|
|
return &MusicCoverGenerator{l: l, settings: settings}
|
|
}
|
|
|
|
type MusicCoverGenerator struct {
|
|
l logging.Logger
|
|
settings setting.Provider
|
|
}
|
|
|
|
func (v *MusicCoverGenerator) Generate(ctx context.Context, es entitysource.EntitySource, ext string, previous *Result) (*Result, error) {
|
|
if !util.IsInExtensionListExt(v.settings.MusicCoverThumbExts(ctx), ext) {
|
|
return nil, fmt.Errorf("unsupported music format: %w", ErrPassThrough)
|
|
}
|
|
|
|
if es.Entity().Size() > v.settings.MusicCoverThumbMaxSize(ctx) {
|
|
return nil, fmt.Errorf("file is too big: %w", ErrPassThrough)
|
|
}
|
|
|
|
m, err := tag.ReadFrom(es)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("faield to read audio tags from file: %w", err)
|
|
}
|
|
|
|
p := m.Picture()
|
|
if p == nil || len(p.Data) == 0 {
|
|
return nil, fmt.Errorf("no cover found in given file")
|
|
}
|
|
|
|
thumbExt := ".jpg"
|
|
if p.Ext != "" {
|
|
thumbExt = p.Ext
|
|
}
|
|
|
|
tempPath := filepath.Join(
|
|
util.DataPath(v.settings.TempPath(ctx)),
|
|
thumbTempFolder,
|
|
fmt.Sprintf("thumb_%s.%s", uuid.Must(uuid.NewV4()).String(), thumbExt),
|
|
)
|
|
|
|
thumbFile, err := util.CreatNestedFile(tempPath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to create temp file: %w", err)
|
|
}
|
|
|
|
defer thumbFile.Close()
|
|
|
|
if _, err := thumbFile.Write(p.Data); err != nil {
|
|
return &Result{Path: tempPath}, fmt.Errorf("failed to write cover to file: %w", err)
|
|
}
|
|
|
|
return &Result{
|
|
Path: tempPath,
|
|
Continue: true,
|
|
Cleanup: []func(){func() { _ = os.Remove(tempPath) }},
|
|
}, nil
|
|
}
|
|
|
|
func (v *MusicCoverGenerator) Priority() int {
|
|
return 50
|
|
}
|
|
|
|
func (v *MusicCoverGenerator) Enabled(ctx context.Context) bool {
|
|
return v.settings.MusicCoverThumbGeneratorEnabled(ctx)
|
|
}
|