1 files changed, 5 insertions, 0 deletions
diff --git a/nodejs/database/agents.js b/nodejs/database/agents.js
index 4dd7f4b..8b7203b 100644
--- a/nodejs/database/agents.js
+++ b/nodejs/database/agents.js
@@ -2,6 +2,7 @@ import db from './db.js';
 
 const addAgentStatement = db.prepare(`INSERT INTO agents(data) VALUES (json(?));`);
 const getAgentStatement = db.prepare(`SELECT data FROM agents;`);
+const setAgentStatement = db.prepare(`UPDATE agents SET data = json(?);`);
 
 export function addAgent(agent) {
 	return addAgentStatement.run(JSON.stringify(agent)).lastInsertRowid;
@@ -14,3 +15,7 @@ export function getAgent() {
 	}
 	return JSON.parse(data.data);
 }
+
+export function setAgent(agent) {
+	return setAgentStatement.run(JSON.stringify(agent)).changes;
+}