From 5f705be63b930e566d3431f0ec213bd22cf591fb Mon Sep 17 00:00:00 2001
From: Olaoluwa Osuntokun <laolu32@gmail.com>
Date: Wed, 6 Jan 2016 15:22:12 -0800
Subject: [PATCH] chainntfs: introduce ChainConnection interface to avoid
 import cycle and for future tests

* Looks rather hack atm. Put in place so progress can be had with
lnwallet before notes is finished.
---
 chainntfs/btcdnotify/btcd.go   | 31 ++++++-------------------------
 chainntfs/btcdnotify/source.go | 14 ++++++++++++++
 2 files changed, 20 insertions(+), 25 deletions(-)
 create mode 100644 chainntfs/btcdnotify/source.go

diff --git a/chainntfs/btcdnotify/btcd.go b/chainntfs/btcdnotify/btcd.go
index 1e6dc33f5..3d02a2190 100644
--- a/chainntfs/btcdnotify/btcd.go
+++ b/chainntfs/btcdnotify/btcd.go
@@ -7,17 +7,15 @@ import (
 	"sync/atomic"
 
 	"github.com/btcsuite/btcd/wire"
-	"github.com/btcsuite/btcutil"
 	"github.com/btcsuite/btcwallet/chain"
 	"github.com/btcsuite/btcwallet/wtxmgr"
 	"li.lan/labs/plasma/chainntfs"
-	"li.lan/labs/plasma/lnwallet"
 )
 
 // BtcdNotifier...
 type BtcdNotifier struct {
 	// TODO(roasbeef): refactor to use the new NotificationServer
-	wallet *lnwallet.LightningWallet
+	conn ChainConnection
 
 	notificationRegistry chan interface{}
 
@@ -28,9 +26,6 @@ type BtcdNotifier struct {
 	connectedBlocks    <-chan wtxmgr.BlockMeta
 	disconnectedBlocks <-chan wtxmgr.BlockMeta
 	relevantTxs        <-chan chain.RelevantTx
-	managerLocked      <-chan bool
-	confirmedBalance   <-chan btcutil.Amount
-	unconfirmedBalance <-chan btcutil.Amount
 
 	rpcConnected chan struct{}
 
@@ -43,8 +38,9 @@ type BtcdNotifier struct {
 var _ chainntnfs.ChainNotifier = (*BtcdNotifier)(nil)
 
 // NewBtcdNotifier...
-func NewBtcdNotifier() (*BtcdNotifier, error) {
+func NewBtcdNotifier(c ChainConnection) (*BtcdNotifier, error) {
 	return &BtcdNotifier{
+		conn:                 c,
 		notificationRegistry: make(chan interface{}),
 
 		spendNotifications: make(map[wire.OutPoint]*spendNotification),
@@ -54,9 +50,6 @@ func NewBtcdNotifier() (*BtcdNotifier, error) {
 		connectedBlocks:    make(chan wtxmgr.BlockMeta),
 		disconnectedBlocks: make(chan wtxmgr.BlockMeta),
 		relevantTxs:        make(chan chain.RelevantTx),
-		managerLocked:      make(chan bool),
-		confirmedBalance:   make(chan btcutil.Amount),
-		unconfirmedBalance: make(chan btcutil.Amount),
 
 		rpcConnected: make(chan struct{}, 1),
 
@@ -186,27 +179,15 @@ out:
 func (b *BtcdNotifier) initAllNotifications() error {
 	var err error
 
-	b.connectedBlocks, err = b.wallet.ListenConnectedBlocks()
+	b.connectedBlocks, err = b.conn.ListenConnectedBlocks()
 	if err != nil {
 		return err
 	}
-	b.disconnectedBlocks, err = b.wallet.ListenDisconnectedBlocks()
+	b.disconnectedBlocks, err = b.conn.ListenDisconnectedBlocks()
 	if err != nil {
 		return err
 	}
-	b.relevantTxs, err = b.wallet.ListenRelevantTxs()
-	if err != nil {
-		return err
-	}
-	b.managerLocked, err = b.wallet.ListenLockStatus()
-	if err != nil {
-		return err
-	}
-	b.confirmedBalance, err = b.wallet.ListenConfirmedBalance()
-	if err != nil {
-		return err
-	}
-	b.unconfirmedBalance, err = b.wallet.ListenUnconfirmedBalance()
+	b.relevantTxs, err = b.conn.ListenRelevantTxs()
 	if err != nil {
 		return err
 	}
diff --git a/chainntfs/btcdnotify/source.go b/chainntfs/btcdnotify/source.go
new file mode 100644
index 000000000..dc7e5a2b0
--- /dev/null
+++ b/chainntfs/btcdnotify/source.go
@@ -0,0 +1,14 @@
+package btcdnotify
+
+import (
+	"github.com/btcsuite/btcwallet/chain"
+	"github.com/btcsuite/btcwallet/wtxmgr"
+)
+
+// ChainConnection...
+// Required in order to avoid an import cycle, and do aide in testing.
+type ChainConnection interface {
+	ListenConnectedBlocks() (<-chan wtxmgr.BlockMeta, error)
+	ListenDisconnectedBlocks() (<-chan wtxmgr.BlockMeta, error)
+	ListenRelevantTxs() (<-chan chain.RelevantTx, error)
+}