Notify wallets when mempool txs pay to a wallet address.

mempool.go

@@ -876,7 +876,10 @@ func (mp *txMemPool) maybeAcceptTransaction(tx *btcutil.Tx, isOrphan *bool) erro
 	txmpLog.Debugf("Accepted transaction %v (pool size: %v)", txHash,
 		len(mp.pool))
 
-	// TODO(davec): Notifications
+	// Notify wallets of mempool transactions to wallet addresses.
+	if mp.server.rpcServer != nil {
+		mp.server.rpcServer.NotifyForTxOuts(tx, nil)
+	}
 
 	return nil
 }

rpcserver.go

@@ -1728,7 +1728,7 @@ func (s *rpcServer) NotifyBlockDisconnected(block *btcutil.Block) {
 func (s *rpcServer) NotifyBlockTXs(db btcdb.Db, block *btcutil.Block) {
 	for _, tx := range block.Transactions() {
 		s.newBlockNotifyCheckTxIn(tx)
-		s.newBlockNotifyCheckTxOut(block, tx)
+		s.NotifyForTxOuts(tx, block)
 	}
 }
 
@@ -1786,12 +1786,10 @@ func (s *rpcServer) newBlockNotifyCheckTxIn(tx *btcutil.Tx) {
 	}
 }
 
-// newBlockNotifyCheckTxOut is a helper function to iterate through
+// NotifyForTxOuts iterates through all outputs of a tx, performing any
-// each transaction output of a new block and perform any checks and
+// necessary notifications for wallets.  If a non-nil block is passed,
-// notify listening frontends when necessary.
+// additional block information is passed with the notifications.
-func (s *rpcServer) newBlockNotifyCheckTxOut(block *btcutil.Block,
+func (s *rpcServer) NotifyForTxOuts(tx *btcutil.Tx, block *btcutil.Block) {
-	tx *btcutil.Tx) {
-
 	for i, txout := range tx.MsgTx().TxOut {
 		stype, txaddrhash, err := btcscript.ScriptToAddrHash(txout.PkScript)
 		if stype != btcscript.ScriptAddr || err != nil {
@@ -1802,18 +1800,14 @@ func (s *rpcServer) newBlockNotifyCheckTxOut(block *btcutil.Block,
 			for e := idlist.Front(); e != nil; e = e.Next() {
 				ctx := e.Value.(*notificationCtx)
 
-				blkhash, err := block.Sha()
-				if err != nil {
-					rpcsLog.Error("Error getting block sha; dropping Tx notification.")
-					break
-				}
 				txaddr, err := btcutil.EncodeAddress(txaddrhash, s.server.btcnet)
 				if err != nil {
 					rpcsLog.Error("Error encoding address; dropping Tx notification.")
 					break
 				}
-				reply := &btcjson.Reply{
+
-					Result: struct {
+				// TODO(jrick): shove this in btcws
+				result := struct {
 					Receiver   string `json:"receiver"`
 					Height     int64  `json:"height"`
 					BlockHash  string `json:"blockhash"`
@@ -1825,24 +1819,38 @@ func (s *rpcServer) newBlockNotifyCheckTxOut(block *btcutil.Block,
 					PkScript   string `json:"pkscript"`
 				}{
 					Receiver:   txaddr,
-						Height:     block.Height(),
-						BlockHash:  blkhash.String(),
-						BlockIndex: tx.Index(),
-						BlockTime:  block.MsgBlock().Header.Timestamp.Unix(),
 					TxID:       tx.Sha().String(),
 					TxOutIndex: uint32(i),
 					Amount:     txout.Value,
 					PkScript:   btcutil.Base58Encode(txout.PkScript),
-					},
+				}
+
+				if block != nil {
+					blkhash, err := block.Sha()
+					if err != nil {
+						rpcsLog.Error("Error getting block sha; dropping Tx notification.")
+						break
+					}
+					result.Height = block.Height()
+					result.BlockHash = blkhash.String()
+					result.BlockIndex = tx.Index()
+					result.BlockTime = block.MsgBlock().Header.Timestamp.Unix()
+				} else {
+					result.Height = -1
+					result.BlockIndex = -1
+				}
+
+				reply := &btcjson.Reply{
+					Result: result,
 					Error:  nil,
 					Id:     &ctx.id,
 				}
-				replyBytes, err := json.Marshal(reply)
+				mreply, err := json.Marshal(reply)
 				if err != nil {
 					rpcsLog.Errorf("Unable to marshal tx notification: %v", err)
 					continue
 				}
-				ctx.connection <- replyBytes
+				ctx.connection <- mreply
 			}
 		}
 	}