Harvey Tindall
ebff016b5d
RecordCounter class created from that in activityList, and put in
accountsList. PageCount-type route standardized and made for /users
(/users/count). Created userCache, which regularly generates the
respUser list returned by /users. Added a currently dumb POST /users for
searching/pagination, GET /users is now just for getting -all- users.
go-getted expr, an expression language that seems like it'll be useful
for evaluating local searches. We don't store this data in the badger
DB, so we can't use the nice query form provided by badgerhold.
129 lines
3.1 KiB
Go
129 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"log"
|
|
"math/rand"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/hrfee/mediabrowser"
|
|
)
|
|
|
|
var (
|
|
names = []string{"Aaron", "Agnes", "Bridget", "Brandon", "Dolly", "Drake", "Elizabeth", "Erika", "Geoff", "Graham", "Haley", "Halsey", "Josie", "John", "Kayleigh", "Luka", "Melissa", "Nasreen", "Paul", "Ross", "Sam", "Talib", "Veronika", "Zaynab"}
|
|
COUNT = 3000
|
|
)
|
|
|
|
const (
|
|
PASSWORD = "test"
|
|
)
|
|
|
|
func main() {
|
|
fmt.Println("Usage: account-gen <server> <username> <password, or file://path to file containing password>")
|
|
var server, username, password string
|
|
reader := bufio.NewReader(os.Stdin)
|
|
if len(os.Args) > 1 {
|
|
server = os.Args[1]
|
|
} else {
|
|
fmt.Print("Server Address: ")
|
|
server, _ = reader.ReadString('\n')
|
|
server = strings.TrimSuffix(server, "\n")
|
|
}
|
|
|
|
if len(os.Args) > 2 {
|
|
username = os.Args[2]
|
|
} else {
|
|
fmt.Print("Username: ")
|
|
username, _ = reader.ReadString('\n')
|
|
username = strings.TrimSuffix(username, "\n")
|
|
}
|
|
|
|
if len(os.Args) > 3 {
|
|
password = os.Args[3]
|
|
if strings.HasPrefix(password, "file://") {
|
|
p, err := os.ReadFile(strings.TrimPrefix(password, "file://"))
|
|
if err != nil {
|
|
log.Fatalf("Failed to read password file \"%s\": %+v\n", password, err)
|
|
}
|
|
password = strings.TrimSuffix(string(p), "\n")
|
|
}
|
|
} else {
|
|
fmt.Print("Password: ")
|
|
password, _ = reader.ReadString('\n')
|
|
password = strings.TrimSuffix(password, "\n")
|
|
}
|
|
|
|
if countEnv := os.Getenv("COUNT"); countEnv != "" {
|
|
COUNT, _ = strconv.Atoi(countEnv)
|
|
}
|
|
|
|
fmt.Printf("Will generate %d users\n", COUNT)
|
|
|
|
jf, err := mediabrowser.NewServer(
|
|
mediabrowser.JellyfinServer,
|
|
server,
|
|
"jfa-go-account-gen-script",
|
|
"0.0.1",
|
|
"testing",
|
|
"my_left_foot",
|
|
mediabrowser.NewNamedTimeoutHandler("Jellyfin Account Gen", "\""+server+"\"", true),
|
|
30,
|
|
)
|
|
|
|
if err != nil {
|
|
log.Fatalf("Failed to connect to Jellyin @ \"%s\": %+v\n", server, err)
|
|
}
|
|
|
|
_, status, err := jf.Authenticate(username, password)
|
|
if status != 200 || err != nil {
|
|
log.Fatalf("Failed to authenticate: %+v\n", err)
|
|
}
|
|
|
|
jfTemp, err := mediabrowser.NewServer(
|
|
mediabrowser.JellyfinServer,
|
|
server,
|
|
"jfa-go-account-gen-script",
|
|
"0.0.1",
|
|
"fake-activity",
|
|
"my_left_foot",
|
|
mediabrowser.NewNamedTimeoutHandler("Jellyfin Account Gen", "\""+server+"\"", true),
|
|
30,
|
|
)
|
|
|
|
if err != nil {
|
|
log.Fatalf("Failed to connect to Jellyin @ \"%s\": %+v\n", server, err)
|
|
}
|
|
|
|
rand.Seed(time.Now().Unix())
|
|
|
|
for i := 0; i < COUNT; i++ {
|
|
name := names[rand.Intn(len(names))] + strconv.Itoa(rand.Intn(100))
|
|
|
|
user, status, err := jf.NewUser(name, PASSWORD)
|
|
if (status != 200 && status != 201 && status != 204) || err != nil {
|
|
log.Fatalf("Acc no %d: Failed to create user \"%s\" (%d): %+v\n", i, name, status, err)
|
|
}
|
|
|
|
if rand.Intn(100) > 65 {
|
|
user.Policy.IsAdministrator = true
|
|
}
|
|
|
|
if rand.Intn(100) > 80 {
|
|
user.Policy.IsDisabled = true
|
|
}
|
|
|
|
status, err = jf.SetPolicy(user.ID, user.Policy)
|
|
if (status != 200 && status != 201 && status != 204) || err != nil {
|
|
log.Fatalf("Acc no %d: Failed to set policy for user \"%s\" (%d): %+v\n", i, name, status, err)
|
|
}
|
|
|
|
if rand.Intn(100) > 20 {
|
|
jfTemp.Authenticate(name, PASSWORD)
|
|
}
|
|
}
|
|
}
|