From 114ac731a4dc3a0112891fac4cbf3f34bab4963f Mon Sep 17 00:00:00 2001
From: Calvin Kim <calvin@kcalvinalvin.info>
Date: Fri, 1 Nov 2024 14:09:18 +0900
Subject: [PATCH] btcutil: add utreexotx

UtreexoTx is a wrapper around MsgUtreexoTx and adds helpful method just
like Tx adds helpful methods for MsgTx.
---
 btcutil/utreexotx.go | 61 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 btcutil/utreexotx.go

diff --git a/btcutil/utreexotx.go b/btcutil/utreexotx.go
new file mode 100644
index 000000000..f36cd16df
--- /dev/null
+++ b/btcutil/utreexotx.go
@@ -0,0 +1,61 @@
+// Copyright (c) 2024 The utreexo developers
+// Use of this source code is governed by an ISC
+// license that can be found in the LICENSE file.
+
+package btcutil
+
+import (
+	"bytes"
+	"io"
+
+	"github.com/utreexo/utreexod/wire"
+)
+
+// UtreexoTx defines a utreexo bitcoin transaction that provides easier and more
+// efficient manipulation of raw transactions. It also memoizes the hash for the
+// transaction on its first access so subsequent accesses don't have to repeat
+// the relatively expensive hashing operations.
+type UtreexoTx struct {
+	Tx
+	msgUtreexoTx *wire.MsgUtreexoTx
+}
+
+// MsgUtreexoTx returns the underlying wire.MsgUtreexoTx for the utreexo transaction.
+func (t *UtreexoTx) MsgUtreexoTx() *wire.MsgUtreexoTx {
+	return t.msgUtreexoTx
+}
+
+// NewUtreexoTx returns a new instance of a bitcoin transaction given an underlying
+// wire.MsgUtreexoTx. See UtreexoTx.
+func NewUtreexoTx(msgUtreexoTx *wire.MsgUtreexoTx) *UtreexoTx {
+	return &UtreexoTx{
+		Tx:           *NewTx(&msgUtreexoTx.MsgTx),
+		msgUtreexoTx: msgUtreexoTx,
+	}
+}
+
+// NewUtreexoTxFromBytes returns a new instance of a bitcoin utreexo transaction given the
+// serialized bytes. See UtreexoTx.
+func NewUtreexoTxFromBytes(serializedUtreexoTx []byte) (*UtreexoTx, error) {
+	br := bytes.NewReader(serializedUtreexoTx)
+	return NewUtreexoTxFromReader(br)
+}
+
+// NewUtreexoTxFromReader returns a new instance of a bitcoin transaction given a
+// Reader to deserialize the transaction. See UtreexoTx.
+func NewUtreexoTxFromReader(r io.Reader) (*UtreexoTx, error) {
+	// Deserialize the bytes into a MsgUtreexoTx.
+	var msgUtreexoTx wire.MsgUtreexoTx
+	err := msgUtreexoTx.Deserialize(r)
+	if err != nil {
+		return nil, err
+	}
+
+	tx := NewTx(&msgUtreexoTx.MsgTx)
+
+	t := UtreexoTx{
+		Tx:           *tx,
+		msgUtreexoTx: &msgUtreexoTx,
+	}
+	return &t, nil
+}