Add method to notify for incoming funds to the SDK

Author: altafan

pkg/client-sdk/ark_sdk.go

@@ -39,6 +39,7 @@ type ArkClient interface {
 	RedeemNotes(ctx context.Context, notes []string, opts ...Option) (string, error)
 	SetNostrNotificationRecipient(ctx context.Context, nostrRecipient string) error
 	SignTransaction(ctx context.Context, tx string) (string, error)
+	NotifyIncomingFunds(ctx context.Context, address string) ([]types.Vtxo, error)
 	Stop() error
 }
 

pkg/client-sdk/client.go

@@ -5,6 +5,7 @@ import (
 	"encoding/hex"
 	"fmt"
 	"strings"
+	"sync"
 	"time"
 
 	"github.com/ark-network/ark/common"
@@ -142,6 +143,35 @@ func (a *arkClient) ListVtxos(
 	return
 }
 
+func (a *arkClient) NotifyIncomingFunds(
+	ctx context.Context, addr string,
+) ([]types.Vtxo, error) {
+	eventCh, closeFn, err := a.client.SubscribeForAddress(ctx, addr)
+	if err != nil {
+		return nil, err
+	}
+
+	wg := &sync.WaitGroup{}
+	wg.Add(1)
+	incomingVtxos := make([]types.Vtxo, 0)
+	go func() {
+		for event := range eventCh {
+			if event.Err != nil {
+				err = event.Err
+			} else {
+				for _, vtxo := range event.NewVtxos {
+					incomingVtxos = append(incomingVtxos, toTypesVtxo(vtxo))
+				}
+			}
+			closeFn()
+			return
+		}
+	}()
+	wg.Done()
+
+	return incomingVtxos, nil
+}
+
 func (a *arkClient) initWithWallet(
 	ctx context.Context, args InitWithWalletArgs,
 ) error {

pkg/client-sdk/wasm/browser/exports.go

@@ -43,6 +43,7 @@ func init() {
 	js.Global().Set("setNostrNotificationRecipient", SetNostrNotificationRecipientWrapper())
 	js.Global().Set("listVtxos", ListVtxosWrapper())
 	js.Global().Set("signTransaction", SignTransactionWrapper())
+	js.Global().Set("notifyIncomingFunds", NotifyIncomingFundsWrapper())
 
 	js.Global().Set("getServerUrl", GetServerUrlWrapper())
 	js.Global().Set("getServerPubkeyHex", GetServerPubkeyWrapper())

pkg/client-sdk/wasm/browser/wrappers.go

@@ -553,6 +553,31 @@ func SignTransactionWrapper() js.Func {
 	})
 }
 
+func NotifyIncomingFundsWrapper() js.Func {
+	return JSPromise(func(args []js.Value) (interface{}, error) {
+		if len(args) != 1 {
+			return nil, fmt.Errorf("invalid number of args")
+		}
+
+		address := args[0].String()
+
+		incomingVtxos, err := arkSdkClient.NotifyIncomingFunds(context.Background(), address)
+		if err != nil {
+			return nil, err
+		}
+
+		rawList := map[string]interface{}{
+			"incomingVtxos": incomingVtxos,
+		}
+		result, err := json.Marshal(rawList)
+		if err != nil {
+			return nil, err
+		}
+
+		return result, nil
+	})
+}
+
 type promise func(args []js.Value) (interface{}, error)
 
 func JSPromise(fn promise) js.Func {