mirror of
https://github.com/stashapp/stash.git
synced 2025-12-07 08:54:10 +01:00
e16118f551
* Allow clearing of tag images * Allow clearing of studio images * Allow clearing of performer images * Allow clearing of movie images * Add filtering for missing images
49 lines
1 KiB
Go
49 lines
1 KiB
Go
package api
|
|
|
|
import (
|
|
"math/rand"
|
|
"strings"
|
|
|
|
"github.com/gobuffalo/packr/v2"
|
|
"github.com/stashapp/stash/pkg/utils"
|
|
)
|
|
|
|
var performerBox *packr.Box
|
|
var performerBoxMale *packr.Box
|
|
|
|
func initialiseImages() {
|
|
performerBox = packr.New("Performer Box", "../../static/performer")
|
|
performerBoxMale = packr.New("Male Performer Box", "../../static/performer_male")
|
|
}
|
|
|
|
func getRandomPerformerImage(gender string) ([]byte, error) {
|
|
var box *packr.Box
|
|
switch strings.ToUpper(gender) {
|
|
case "FEMALE":
|
|
box = performerBox
|
|
case "MALE":
|
|
box = performerBoxMale
|
|
default:
|
|
box = performerBox
|
|
|
|
}
|
|
imageFiles := box.List()
|
|
index := rand.Intn(len(imageFiles))
|
|
return box.Find(imageFiles[index])
|
|
}
|
|
|
|
func getRandomPerformerImageUsingName(name, gender string) ([]byte, error) {
|
|
var box *packr.Box
|
|
switch strings.ToUpper(gender) {
|
|
case "FEMALE":
|
|
box = performerBox
|
|
case "MALE":
|
|
box = performerBoxMale
|
|
default:
|
|
box = performerBox
|
|
|
|
}
|
|
imageFiles := box.List()
|
|
index := utils.IntFromString(name) % uint64(len(imageFiles))
|
|
return box.Find(imageFiles[index])
|
|
}
|