From 6be4d06e15603622098a3cc056bacd49e61dd736 Mon Sep 17 00:00:00 2001
From: HimbeerserverDE <himbeerserverde@gmail.com>
Date: Sat, 3 Apr 2021 21:24:04 +0200
Subject: [PATCH] Fix priv modification

---
 privs.go | 16 +++++-----------
 1 file changed, 5 insertions(+), 11 deletions(-)

diff --git a/privs.go b/privs.go
index 9d61745..daf91b8 100644
--- a/privs.go
+++ b/privs.go
@@ -63,7 +63,7 @@ func addPrivItem(db *sql.DB, name string) error {
 
 // modPrivItem updates a priv DB entry
 func modPrivItem(db *sql.DB, name, privs string) error {
-	_, err := db.Exec(`UPDATE privileges SET privileges = ? WHERE name = ?;`, name)
+	_, err := db.Exec(`UPDATE privileges SET privileges = ? WHERE name = ?;`, privs, name)
 	return err
 }
 
@@ -143,23 +143,17 @@ func (c *Conn) CheckPrivs(req map[string]bool) (bool, error) {
 
 func init() {
 	if admin, ok := ConfKey("admin").(string); ok {
-		db, err := initAuthDB()
+		privs, err := Privs(admin)
 		if err != nil {
 			log.Print(err)
 			return
 		}
 
-		eprivs, err := readPrivItem(db, admin)
-		if err != nil {
+		privs["privs"] = true
+
+		if err = SetPrivs(admin, privs); err != nil {
 			log.Print(err)
 			return
 		}
-
-		privs := decodePrivs(eprivs)
-		privs["privs"] = true
-
-		newprivs := encodePrivs(privs)
-
-		modPrivItem(db, admin, newprivs)
 	}
 }