From 958916e2ed6fe50e9f06adba345003e77641c52e Mon Sep 17 00:00:00 2001
From: Oliver Gugger <gugger@gmail.com>
Date: Thu, 14 Mar 2024 19:16:08 +0100
Subject: [PATCH] cmd/tapcli: add events send/receive subcommands

---
 cmd/tapcli/events.go | 114 +++++++++++++++++++++++++++++++++++++++++++
 cmd/tapcli/main.go   |   1 +
 2 files changed, 115 insertions(+)
 create mode 100644 cmd/tapcli/events.go

diff --git a/cmd/tapcli/events.go b/cmd/tapcli/events.go
new file mode 100644
index 000000000..805c9cbd1
--- /dev/null
+++ b/cmd/tapcli/events.go
@@ -0,0 +1,114 @@
+package main
+
+import (
+	"encoding/hex"
+	"fmt"
+
+	"github.com/lightninglabs/taproot-assets/taprpc"
+	"github.com/urfave/cli"
+)
+
+var eventCommands = []cli.Command{
+	{
+		Name:      "events",
+		ShortName: "e",
+		Usage: "Subscribe to live events from the Taproot Asset " +
+			"daemon.",
+		Category: "Events",
+		Subcommands: []cli.Command{
+			receiveEventsCommand,
+			sendEventsCommand,
+		},
+	},
+}
+
+var receiveEventsCommand = cli.Command{
+	Name:      "receive",
+	ShortName: "r",
+	Usage:     "Subscribe to events around receiving inbound assets",
+	Description: "Get live updates on the status of inbound asset " +
+		"transfers to the local node. This command will block " +
+		"until aborted manually by hitting Ctrl+C.",
+	Flags: []cli.Flag{
+		cli.StringFlag{
+			Name: addrName,
+			Usage: "(optional) the address to receive events " +
+				"for; if not set, events for all addresses " +
+				"will be shown",
+		},
+	},
+	Action: receiveEvents,
+}
+
+func receiveEvents(ctx *cli.Context) error {
+	ctxc := getContext()
+	client, cleanUp := getClient(ctx)
+	defer cleanUp()
+
+	addr, err := client.SubscribeReceiveEvents(
+		ctxc, &taprpc.SubscribeReceiveEventsRequest{
+			FilterAddr: ctx.String(addrName),
+		},
+	)
+	if err != nil {
+		return fmt.Errorf("unable to subscribe to receive events: %w",
+			err)
+	}
+
+	for {
+		event, err := addr.Recv()
+		if err != nil {
+			return fmt.Errorf("unable to receive event: %w", err)
+		}
+
+		printRespJSON(event)
+	}
+}
+
+var sendEventsCommand = cli.Command{
+	Name:      "send",
+	ShortName: "s",
+	Usage:     "Subscribe to events around sending outbound assets",
+	Description: "Get live updates on the status of outbound asset " +
+		"transfers to the local node. This command will block " +
+		"until aborted manually by hitting Ctrl+C.",
+	Flags: []cli.Flag{
+		cli.StringFlag{
+			Name: scriptKeyName,
+			Usage: "(optional) the script key to receive events " +
+				"for; if not set, events for all transfers " +
+				"will be shown",
+		},
+	},
+	Action: sendEvents,
+}
+
+func sendEvents(ctx *cli.Context) error {
+	ctxc := getContext()
+	client, cleanUp := getClient(ctx)
+	defer cleanUp()
+
+	scriptKeyBytes, err := hex.DecodeString(ctx.String(scriptKeyName))
+	if err != nil {
+		return fmt.Errorf("unable to hex decode script key: %w", err)
+	}
+
+	send, err := client.SubscribeSendEvents(
+		ctxc, &taprpc.SubscribeSendEventsRequest{
+			FilterScriptKey: scriptKeyBytes,
+		},
+	)
+	if err != nil {
+		return fmt.Errorf("unable to subscribe to send events: %w",
+			err)
+	}
+
+	for {
+		event, err := send.Recv()
+		if err != nil {
+			return fmt.Errorf("unable to receive event: %w", err)
+		}
+
+		printRespJSON(event)
+	}
+}
diff --git a/cmd/tapcli/main.go b/cmd/tapcli/main.go
index d3134ab36..c27b7c414 100644
--- a/cmd/tapcli/main.go
+++ b/cmd/tapcli/main.go
@@ -332,6 +332,7 @@ func main() {
 	}
 	app.Commands = append(app.Commands, assetsCommands...)
 	app.Commands = append(app.Commands, addrCommands...)
+	app.Commands = append(app.Commands, eventCommands...)
 	app.Commands = append(app.Commands, proofCommands...)
 	app.Commands = append(app.Commands, universeCommands...)
 	app.Commands = append(app.Commands, devCommands...)