From ba988e9897ac7b8542e5571436df0072d601caa3 Mon Sep 17 00:00:00 2001
From: Frederik Schwan <frederik.schwan@linux.com>
Date: Tue, 16 Mar 2021 01:17:13 +0100
Subject: [PATCH] cleanup empty users in database

---
 internal/database/database.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/internal/database/database.go b/internal/database/database.go
index af9bc94..c40a51a 100644
--- a/internal/database/database.go
+++ b/internal/database/database.go
@@ -240,12 +240,16 @@ func (db *Database) AddRegistration(username, accountId, deviceToken string, mai
 }
 
 func (db *Database) DeleteIfExistRegistration(reg Registration) bool {
-	for _, user := range db.Users {
+	for username, user := range db.Users {
 		for accountId, account := range user.Accounts {
 			if accountId == reg.AccountId {
 				dbMutex.Lock()
 				log.Infoln("Deleting " + account.DeviceToken)
 				delete(user.Accounts, accountId)
+				// clean up empty users
+				if len(user.Accounts) == 0 {
+					delete(db.Users, username)
+				}
 				db.write()
 				dbMutex.Unlock()
 				return true