From ce3878e18bd5442c30109eb6797713e69eb66ca2 Mon Sep 17 00:00:00 2001
From: ztimson <zaktimson@gmail.com>
Date: Mon, 7 Jul 2025 14:53:14 -0400
Subject: [PATCH] Added db safe guards

---
 package.json    |  2 +-
 src/database.ts | 12 ++++++++----
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/package.json b/package.json
index ea9e30d..0cfae4d 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
 	"name": "@ztimson/utils",
-	"version": "0.26.2",
+	"version": "0.26.3",
 	"description": "Utility library",
 	"author": "Zak Timson",
 	"license": "MIT",
diff --git a/src/database.ts b/src/database.ts
index 7437c2b..f7eaa80 100644
--- a/src/database.ts
+++ b/src/database.ts
@@ -166,8 +166,11 @@ export class Table<K extends IDBValidKey = any, T = any> {
 		return this.tx(this.name, store => store.getAllKeys(), true);
 	}
 
-	put(key: K, value: T): Promise<void> {
-		return this.tx(this.name, store => store.put(value, key));
+	put(value: T, key?: string): Promise<void> {
+		return this.tx(this.name, store => {
+			if (store.keyPath) return store.put(value);
+			return store.put(value, key);
+		});
 	}
 
 	read(): Promise<T[]>;
@@ -177,8 +180,9 @@ export class Table<K extends IDBValidKey = any, T = any> {
 	}
 
 	set(value: T, key?: K): Promise<void> {
-		if(!key && !(<any>value)[this.key]) return this.add(value);
-		return this.put(key || (<any>value)[this.key], value);
+		if(key) (<any>value)[this.key] = key;
+		if(!(<any>value)[this.key]) return this.add(value);
+		return this.put(value);
 	}
 
 	update = this.set;