From 633ffeec6f514770f7e03b3b539678df70291c8f Mon Sep 17 00:00:00 2001
From: Christian Decker <decker.christian@gmail.com>
Date: Sat, 7 Apr 2018 14:39:11 +0200
Subject: [PATCH] wallet: Add methods to add transactions to the DB

Signed-off-by: Christian Decker <decker.christian@gmail.com>
---
 wallet/wallet.c | 43 +++++++++++++++++++++++++++++++++++++++++++
 wallet/wallet.h |  4 ++++
 2 files changed, 47 insertions(+)

diff --git a/wallet/wallet.c b/wallet/wallet.c
index 7bdc16fe9..f3deecabd 100644
--- a/wallet/wallet.c
+++ b/wallet/wallet.c
@@ -2158,3 +2158,46 @@ struct outpoint *wallet_outpoint_for_scid(struct wallet *w, tal_t *ctx,
 
 	return op;
 }
+
+void wallet_transaction_add(struct wallet *w, const struct bitcoin_tx *tx,
+			    const u32 blockheight, const u32 txindex)
+{
+	struct bitcoin_txid txid;
+	sqlite3_stmt *stmt = db_prepare(w->db, "SELECT blockheight FROM transactions WHERE id=?");
+	bool known;
+
+	bitcoin_txid(tx, &txid);
+	sqlite3_bind_sha256(stmt, 1, &txid.shad.sha);
+	known = sqlite3_step(stmt) == SQLITE_ROW;
+	sqlite3_finalize(stmt);
+
+	if (!known) {
+		/* This transaction is still unknown, insert */
+		stmt = db_prepare(w->db,
+				  "INSERT INTO transactions ("
+				  "  id"
+				  ", blockheight"
+				  ", txindex"
+				  ", rawtx) VALUES (?, ?, ?, ?);");
+		sqlite3_bind_sha256(stmt, 1, &txid.shad.sha);
+		if (blockheight) {
+			sqlite3_bind_int(stmt, 2, blockheight);
+			sqlite3_bind_int(stmt, 3, txindex);
+		} else {
+			sqlite3_bind_null(stmt, 2);
+			sqlite3_bind_null(stmt, 3);
+		}
+		sqlite3_bind_tx(stmt, 4, tx);
+		db_exec_prepared(w->db, stmt);
+	} else if (blockheight){
+		/* We know about the transaction, update */
+		stmt = db_prepare(w->db,
+				  "UPDATE transactions "
+				  "SET blockheight = ?, txindex = ? "
+				  "WHERE id = ?");
+		sqlite3_bind_int(stmt, 1, blockheight);
+		sqlite3_bind_int(stmt, 2, txindex);
+		sqlite3_bind_sha256(stmt, 3, &txid.shad.sha);
+		db_exec_prepared(w->db, stmt);
+	}
+}
diff --git a/wallet/wallet.h b/wallet/wallet.h
index 4e70e143a..2dc3f764a 100644
--- a/wallet/wallet.h
+++ b/wallet/wallet.h
@@ -810,4 +810,8 @@ void wallet_utxoset_add(struct wallet *w, const struct bitcoin_tx *tx,
 			const u32 outnum, const u32 blockheight,
 			const u32 txindex, const u8 *scriptpubkey,
 			const u64 satoshis);
+
+void wallet_transaction_add(struct wallet *w, const struct bitcoin_tx *tx,
+			    const u32 blockheight, const u32 txindex);
+
 #endif /* LIGHTNING_WALLET_WALLET_H */