mirror of
https://github.com/hrfee/jfa-go.git
synced 2024-12-22 09:00:10 +00:00
Harvey Tindall
1fa340f096
"import_existing" option in settings enables an every 5-minute daemon which loops through users and imports them to Jellyseerr and copies contact info, if necessary. Also sets new API client flag AutoImportUsers, which decides whether to automatically import non-existent users in it's various methods. also cleaned up the various daemons in the software, most now using the GenericDaemon struct and just providing a new constructor. broken page loop in jellyseerr client also fixed.
111 lines
3.0 KiB
Go
111 lines
3.0 KiB
Go
package main
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/hrfee/mediabrowser"
|
|
"github.com/lithammer/shortuuid/v3"
|
|
)
|
|
|
|
func newUserDaemon(interval time.Duration, app *appContext) *GenericDaemon {
|
|
d := NewGenericDaemon(interval, app,
|
|
func(app *appContext) {
|
|
app.checkUsers()
|
|
},
|
|
)
|
|
d.Name("User daemon")
|
|
return d
|
|
}
|
|
|
|
func (app *appContext) checkUsers() {
|
|
if len(app.storage.GetUserExpiries()) == 0 {
|
|
return
|
|
}
|
|
app.info.Println("Daemon: Checking for user expiry")
|
|
users, status, err := app.jf.GetUsers(false)
|
|
if err != nil || status != 200 {
|
|
app.err.Printf("Failed to get users (%d): %s", status, err)
|
|
return
|
|
}
|
|
mode := "disable"
|
|
term := "Disabling"
|
|
if app.config.Section("user_expiry").Key("behaviour").MustString("disable_user") == "delete_user" {
|
|
mode = "delete"
|
|
term = "Deleting"
|
|
}
|
|
contact := false
|
|
if messagesEnabled && app.config.Section("user_expiry").Key("send_email").MustBool(true) {
|
|
contact = true
|
|
}
|
|
// Use a map to speed up checking for deleted users later
|
|
userExists := map[string]bool{}
|
|
for _, user := range users {
|
|
userExists[user.ID] = true
|
|
}
|
|
for _, expiry := range app.storage.GetUserExpiries() {
|
|
id := expiry.JellyfinID
|
|
if _, ok := userExists[id]; !ok {
|
|
app.info.Printf("Deleting expiry for non-existent user \"%s\"", id)
|
|
app.storage.DeleteUserExpiryKey(expiry.JellyfinID)
|
|
} else if time.Now().After(expiry.Expiry) {
|
|
found := false
|
|
var user mediabrowser.User
|
|
for _, u := range users {
|
|
if u.ID == id {
|
|
found = true
|
|
user = u
|
|
break
|
|
}
|
|
}
|
|
if !found {
|
|
app.info.Printf("Expired user already deleted, ignoring.")
|
|
app.storage.DeleteUserExpiryKey(expiry.JellyfinID)
|
|
continue
|
|
}
|
|
app.info.Printf("%s expired user \"%s\"", term, user.Name)
|
|
|
|
// Record activity
|
|
activity := Activity{
|
|
UserID: id,
|
|
SourceType: ActivityDaemon,
|
|
Time: time.Now(),
|
|
}
|
|
|
|
if mode == "delete" {
|
|
status, err = app.jf.DeleteUser(id)
|
|
activity.Type = ActivityDeletion
|
|
activity.Value = user.Name
|
|
} else if mode == "disable" {
|
|
user.Policy.IsDisabled = true
|
|
// Admins can't be disabled
|
|
user.Policy.IsAdministrator = false
|
|
status, err = app.jf.SetPolicy(id, user.Policy)
|
|
activity.Type = ActivityDisabled
|
|
}
|
|
if !(status == 200 || status == 204) || err != nil {
|
|
app.err.Printf("Failed to %s \"%s\" (%d): %s", mode, user.Name, status, err)
|
|
continue
|
|
}
|
|
|
|
app.storage.SetActivityKey(shortuuid.New(), activity, nil, false)
|
|
|
|
app.storage.DeleteUserExpiryKey(expiry.JellyfinID)
|
|
app.jf.CacheExpiry = time.Now()
|
|
if contact {
|
|
if !ok {
|
|
continue
|
|
}
|
|
name := app.getAddressOrName(user.ID)
|
|
msg, err := app.email.constructUserExpired(app, false)
|
|
if err != nil {
|
|
app.err.Printf("Failed to construct expiry message for \"%s\": %s", user.Name, err)
|
|
} else if err := app.sendByID(msg, user.ID); err != nil {
|
|
app.err.Printf("Failed to send expiry message to \"%s\": %s", name, err)
|
|
} else {
|
|
app.info.Printf("Sent expiry notification to \"%s\"", name)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|