mirror of
https://github.com/stashapp/stash.git
synced 2025-12-07 17:02:38 +01:00
32 lines
653 B
Go
32 lines
653 B
Go
package api
|
|
|
|
import (
|
|
"math/rand"
|
|
"strings"
|
|
|
|
"github.com/gobuffalo/packr/v2"
|
|
)
|
|
|
|
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])
|
|
}
|