Use random performer image if none provided

This commit is contained in:
WithoutPants
2019-08-24 17:17:25 +10:00
parent 20662532d8
commit 9c92e64106
44 changed files with 24 additions and 4 deletions

19
pkg/api/images.go Normal file
View File

@@ -0,0 +1,19 @@
package api
import (
"math/rand"
"github.com/gobuffalo/packr/v2"
)
var performerBox *packr.Box
func initialiseImages() {
performerBox = packr.New("Performer Box", "../../static/performer")
}
func getRandomPerformerImage() ([]byte, error) {
imageFiles := performerBox.List()
index := rand.Intn(len(imageFiles))
return performerBox.Find(imageFiles[index])
}

View File

@@ -19,10 +19,11 @@ func (r *mutationResolver) PerformerCreate(ctx context.Context, input models.Per
var err error
if input.Image == nil {
input.Image = &models.DefaultPerformerImage
imageData, err = getRandomPerformerImage()
} else {
_, imageData, err = utils.ProcessBase64Image(*input.Image)
}
_, imageData, err = utils.ProcessBase64Image(*input.Image)
if err != nil {
return nil, err
}

View File

@@ -36,6 +36,8 @@ func Start() {
//legacyUiBox = packr.New("UI Box", "../../ui/v1/dist/stash-frontend")
setupUIBox = packr.New("Setup UI Box", "../../ui/setup")
initialiseImages()
r := chi.NewRouter()
r.Use(middleware.Recoverer)