all info stuff

Author: hieblmi

loopd/daemon.go

@@ -604,6 +604,7 @@ func (d *Daemon) initialize(withMacaroonService bool) error {
 	depositManager = deposit.NewManager(depoCfg)
 
 	// Static address deposit withdrawal manager setup.
+	withdrawalStore := withdraw.NewSqlStore(baseDb)
 	withdrawalCfg := &withdraw.ManagerConfig{
 		StaticAddressServerClient: staticAddressClient,
 		AddressManager:            staticAddressManager,
@@ -612,6 +613,7 @@ func (d *Daemon) initialize(withMacaroonService bool) error {
 		ChainParams:               d.lnd.ChainParams,
 		ChainNotifier:             d.lnd.ChainNotifier,
 		Signer:                    d.lnd.Signer,
+		Store:                     withdrawalStore,
 	}
 	withdrawalManager = withdraw.NewManager(withdrawalCfg, blockHeight)
 

loopdb/sqlc/migrations/000015_static_address_withdrawals.down.sql

@@ -0,0 +1 @@
+DROP TABLE IF EXISTS withdrawals;

loopdb/sqlc/migrations/000015_static_address_withdrawals.up.sql

@@ -0,0 +1,16 @@
+CREATE TABLE IF NOT EXISTS withdrawals (
+    -- id is the auto-incrementing primary key for a withdrawal.
+    id INTEGER PRIMARY KEY,
+
+    withdrawal_tx_id TEXT NOT NULL UNIQUE,
+
+    deposit_outpoints TEXT NOT NULL,
+
+    total_deposit_amount BIGINT NOT NULL,
+
+    withdrawn_amount BIGINT NOT NULL,
+
+    change_amount BIGINT NOT NULL,
+
+    confirmation_height BIGINT NOT NULL
+);

loopdb/sqlc/models.go

@@ -0,0 +1,24 @@
+-- name: CreateWithdrawal :exec
+INSERT INTO withdrawals (
+    withdrawal_tx_id,
+    deposit_outpoints,
+    total_deposit_amount,
+    withdrawn_amount,
+    change_amount,
+    confirmation_height
+) VALUES (
+             $1,
+             $2,
+             $3,
+             $4,
+             $5,
+             $6
+         );
+
+-- name: AllWithdrawals :many
+SELECT
+    *
+FROM
+    withdrawals
+ORDER BY
+    id ASC;

loopdb/sqlc/querier.go

@@ -82,6 +82,8 @@ type ManagerConfig struct {
 
 	// Signer is the signer client that is used to sign transactions.
 	Signer lndclient.SignerClient
+
+	Store *SqlStore
 }
 
 // newWithdrawalRequest is used to send withdrawal request to the manager main
@@ -621,7 +623,7 @@ func (m *Manager) handleWithdrawal(ctx context.Context,
 					int32(m.initiationHeight.Load()),
 				)
 			select {
-			case <-confChan:
+			case tx := <-confChan:
 				err = m.cfg.DepositManager.TransitionDeposits(
 					ctx, deposits, deposit.OnWithdrawn,
 					deposit.Withdrawn,
@@ -637,6 +639,11 @@ func (m *Manager) handleWithdrawal(ctx context.Context,
 				delete(m.finalizedWithdrawalTxns, txHash)
 				m.mu.Unlock()
 
+				m.cfg.Store.CreateWithdrawal(
+					ctx, tx.Tx, tx.BlockHeight, deposits,
+					addrParams.PkScript,
+				)
+
 			case err := <-errChan:
 				log.Errorf("Error waiting for confirmation: %v",
 					err)

loopdb/sqlc/queries/static_address_withdrawals.sql

@@ -0,0 +1,118 @@
+package withdraw
+
+import (
+	"bytes"
+	"context"
+	"strings"
+
+	"github.com/btcsuite/btcd/btcutil"
+	"github.com/btcsuite/btcd/chaincfg/chainhash"
+	"github.com/btcsuite/btcd/wire"
+	"github.com/lightninglabs/loop/loopdb"
+	"github.com/lightninglabs/loop/loopdb/sqlc"
+	"github.com/lightninglabs/loop/staticaddr/deposit"
+	"github.com/lightningnetwork/lnd/clock"
+)
+
+// SqlStore is the backing store for static address withdrawals.
+type SqlStore struct {
+	baseDB *loopdb.BaseDB
+
+	clock clock.Clock
+}
+
+// NewSqlStore constructs a new SQLStore from a BaseDB. The BaseDB is agnostic
+// to the underlying driver which can be postgres or sqlite.
+func NewSqlStore(db *loopdb.BaseDB) *SqlStore {
+	return &SqlStore{
+		baseDB: db,
+
+		clock: clock.NewDefaultClock(),
+	}
+}
+
+// CreateWithdrawal creates a static address withdrawal record in the database.
+func (s *SqlStore) CreateWithdrawal(ctx context.Context, tx *wire.MsgTx,
+	confirmationHeight uint32, deposits []*deposit.Deposit,
+	changePkScript []byte) error {
+
+	strOutpoints := make([]string, len(deposits))
+	totalAmount := int64(0)
+	for i, deposit := range deposits {
+		strOutpoints[i] = deposit.OutPoint.String()
+		totalAmount += int64(deposit.Value)
+	}
+
+	withdrawnAmount, changeAmount := int64(0), int64(0)
+	if len(tx.TxOut) == 1 {
+		withdrawnAmount = tx.TxOut[0].Value
+	} else if len(tx.TxOut) == 2 {
+		withdrawnAmount, changeAmount = tx.TxOut[0].Value, tx.TxOut[1].Value
+		if bytes.Equal(changePkScript, tx.TxOut[0].PkScript) {
+			changeAmount = tx.TxOut[0].Value
+			withdrawnAmount = tx.TxOut[1].Value
+		}
+	}
+
+	createArgs := sqlc.CreateWithdrawalParams{
+		WithdrawalTxID:     tx.TxHash().String(),
+		DepositOutpoints:   strings.Join(strOutpoints, ","),
+		TotalDepositAmount: totalAmount,
+		WithdrawnAmount:    withdrawnAmount,
+		ChangeAmount:       changeAmount,
+		ConfirmationHeight: int64(confirmationHeight),
+	}
+
+	return s.baseDB.ExecTx(ctx, &loopdb.SqliteTxOptions{},
+		func(q *sqlc.Queries) error {
+			return q.CreateWithdrawal(ctx, createArgs)
+		})
+}
+
+// AllWithdrawals retrieves all known withdrawals.
+func (s *SqlStore) AllWithdrawals(ctx context.Context) ([]*Withdrawal, error) {
+	var allWithdrawals []*Withdrawal
+
+	err := s.baseDB.ExecTx(ctx, loopdb.NewSqlReadOpts(),
+		func(q *sqlc.Queries) error {
+			var err error
+
+			withdrawals, err := q.AllWithdrawals(ctx)
+			if err != nil {
+				return err
+			}
+
+			for _, withdrawal := range withdrawals {
+				w, err := s.toWithdrawal(withdrawal)
+				if err != nil {
+					return err
+				}
+
+				allWithdrawals = append(allWithdrawals, w)
+			}
+
+			return nil
+		})
+	if err != nil {
+		return nil, err
+	}
+
+	return allWithdrawals, nil
+}
+
+// toDeposit converts an sql deposit to a deposit.
+func (s *SqlStore) toWithdrawal(row sqlc.Withdrawal) (*Withdrawal, error) {
+	txHash, err := chainhash.NewHashFromStr(row.WithdrawalTxID)
+	if err != nil {
+		return nil, err
+	}
+
+	return &Withdrawal{
+		TxID:               *txHash,
+		DepositOutpoints:   strings.Split(row.DepositOutpoints, ","),
+		TotalDepositAmount: btcutil.Amount(row.TotalDepositAmount),
+		WithdrawnAmount:    btcutil.Amount(row.WithdrawnAmount),
+		ChangeAmount:       btcutil.Amount(row.ChangeAmount),
+		ConfirmationHeight: uint32(row.ConfirmationHeight),
+	}, nil
+}

loopdb/sqlc/static_address_withdrawals.sql.go

@@ -0,0 +1,15 @@
+package withdraw
+
+import (
+	"github.com/btcsuite/btcd/btcutil"
+	"github.com/btcsuite/btcd/chaincfg/chainhash"
+)
+
+type Withdrawal struct {
+	TxID               chainhash.Hash
+	DepositOutpoints   []string
+	TotalDepositAmount btcutil.Amount
+	WithdrawnAmount    btcutil.Amount
+	ChangeAmount       btcutil.Amount
+	ConfirmationHeight uint32
+}