rename daemon files

2024-07-31 16:44:59 +01:00
parent 2d066ea7cd
commit c442ff5f98
4 changed files with 0 additions and 0 deletions
@@ -0,0 +1,110 @@
+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)
+				}
+			}
+		}
+	}
+}