Mirrors/stash
1
0
Fork 0
mirror of https://github.com/stashapp/stash.git synced 2025-12-17 20:34:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add fingerprint resolver (#4287)

Browse Source 
* Refactor into internal/api/models.go
* Add file fingerprint resolver
This commit is contained in:
DingDongSoLong4
2023-11-20 04:09:12 +02:00
committed by GitHub
parent a0f33e3dab
commit 049a1b15c3
8 changed files with 118 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
internal/api/resolver_model_image.go
View File

@@ -2,21 +2,12 @@ package api
import (
"context"
"fmt"
"github.com/stashapp/stash/internal/api/loaders"
"github.com/stashapp/stash/internal/api/urlbuilders"
"github.com/stashapp/stash/pkg/models"
)
func convertVisualFile(f models.File) (models.VisualFile, error) {
vf, ok := f.(models.VisualFile)
if !ok {
return nil, fmt.Errorf("file %s is not a visual file", f.Base().Path)
}
return vf, nil
}
func (r *imageResolver) getFiles(ctx context.Context, obj *models.Image) ([]models.File, error) {
fileIDs, err := loaders.From(ctx).ImageFiles.Load(obj.ID)
if err != nil {
@@ -32,13 +23,13 @@ func (r *imageResolver) Title(ctx context.Context, obj *models.Image) (*string,
return &ret, nil
}
func (r *imageResolver) VisualFiles(ctx context.Context, obj *models.Image) ([]models.VisualFile, error) {
func (r *imageResolver) VisualFiles(ctx context.Context, obj *models.Image) ([]VisualFile, error) {
files, err := r.getFiles(ctx, obj)
if err != nil {
return nil, err
}
ret := make([]models.VisualFile, len(files))
ret := make([]VisualFile, len(files))
for i, f := range files {
ret[i], err = convertVisualFile(f)
if err != nil {
@@ -57,13 +48,13 @@ func (r *imageResolver) Date(ctx context.Context, obj *models.Image) (*string, e
return nil, nil
}
func (r *imageResolver) Files(ctx context.Context, obj *models.Image) ([]*models.ImageFile, error) {
func (r *imageResolver) Files(ctx context.Context, obj *models.Image) ([]*ImageFile, error) {
files, err := r.getFiles(ctx, obj)
if err != nil {
return nil, err
}
var ret []*models.ImageFile
var ret []*ImageFile
for _, f := range files {
// filter out non-image files
@@ -72,7 +63,9 @@ func (r *imageResolver) Files(ctx context.Context, obj *models.Image) ([]*models
continue
}
ret = append(ret, imageFile)
ret = append(ret, &ImageFile{
ImageFile: imageFile,
})
}
return ret, nil