1 files changed, 24 insertions, 0 deletions
diff --git a/nodejs/database/contracts.js b/nodejs/database/contracts.js
new file mode 100644
index 0000000..8790701
--- /dev/null
+++ b/nodejs/database/contracts.js
@@ -0,0 +1,24 @@
+import db from './db.js';
+
+const addContractStatement = db.prepare(`INSERT INTO contracts(data) VALUES (json(?));`);
+const getContractStatement = db.prepare(`SELECT data FROM contracts WHERE data->>'id' = ?;`);
+const updateContractStatement = db.prepare(`UPDATE contracts SET data = json(:data) WHERE data->>'id' = :id;`);
+
+export function getContract(id) {
+	const data = getContractStatement.get(id);
+	if (data === undefined) {
+		return null;
+	}
+	return data.data;
+}
+
+export function setContract(data) {
+	if (getContract(data.id) === null) {
+		return addContractStatement.run(JSON.stringify(data)).lastInsertRowid;
+	} else {
+		return updateContractStatement.run({
+			data: JSON.stringify(data),
+			id: data.id,
+		}).changes;
+	}
+}