From c1810951975c6ed5275798323dade43ca7ab9c46 Mon Sep 17 00:00:00 2001
From: Nathan Coad <nathan@thecoads.com>
Date: Mon, 8 Jan 2024 10:27:04 +1100
Subject: [PATCH] handle existing secrets with no safe id

---
 models/setup.go | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/models/setup.go b/models/setup.go
index d6914e0..0b9f437 100644
--- a/models/setup.go
+++ b/models/setup.go
@@ -180,6 +180,20 @@ func CreateTables() {
 		log.Printf("Error checking safes table : '%s'", err)
 		os.Exit(1)
 	}
+	// Create an initial safe
+	rowCount, _ = CheckCount("safes")
+	if rowCount == 0 {
+		if _, err = db.Exec("INSERT INTO safes VALUES(1, 'Default Safe');"); err != nil {
+			log.Printf("Error adding initial safe entry : '%s'", err)
+			os.Exit(1)
+		}
+	}
+
+	// Set any unassigned secrets to the default safe id
+	if _, err = db.Exec("UPDATE secrets SET SafeId = REPLACE(SafeId,'NULL','1');"); err != nil {
+		log.Printf("Error setting safe ID of existing secrets : '%s'", err)
+		os.Exit(1)
+	}
 
 	// Secrets table
 	if _, err = db.Exec(createSecrets); err != nil {