mirror of
https://github.com/lightningnetwork/lnd.git
synced 2025-01-19 05:45:21 +01:00
22a3988222
We want to export some of our CLI code to re-use in other projects. But in Golang you cannot import code from a `main` package. So we need to move the actual code into its own package and only have the `func main()` in the `main` package.
84 lines
1.4 KiB
Go
84 lines
1.4 KiB
Go
package commands
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"fmt"
|
|
|
|
"github.com/lightningnetwork/lnd/lnrpc"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
var sendCustomCommand = cli.Command{
|
|
Name: "sendcustom",
|
|
Flags: []cli.Flag{
|
|
cli.StringFlag{
|
|
Name: "peer",
|
|
},
|
|
cli.Uint64Flag{
|
|
Name: "type",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "data",
|
|
},
|
|
},
|
|
Action: actionDecorator(sendCustom),
|
|
}
|
|
|
|
func sendCustom(ctx *cli.Context) error {
|
|
ctxc := getContext()
|
|
client, cleanUp := getClient(ctx)
|
|
defer cleanUp()
|
|
|
|
peer, err := hex.DecodeString(ctx.String("peer"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
msgType := ctx.Uint64("type")
|
|
|
|
data, err := hex.DecodeString(ctx.String("data"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = client.SendCustomMessage(
|
|
ctxc,
|
|
&lnrpc.SendCustomMessageRequest{
|
|
Peer: peer,
|
|
Type: uint32(msgType),
|
|
Data: data,
|
|
},
|
|
)
|
|
|
|
return err
|
|
}
|
|
|
|
var subscribeCustomCommand = cli.Command{
|
|
Name: "subscribecustom",
|
|
Action: actionDecorator(subscribeCustom),
|
|
}
|
|
|
|
func subscribeCustom(ctx *cli.Context) error {
|
|
ctxc := getContext()
|
|
client, cleanUp := getClient(ctx)
|
|
defer cleanUp()
|
|
|
|
stream, err := client.SubscribeCustomMessages(
|
|
ctxc,
|
|
&lnrpc.SubscribeCustomMessagesRequest{},
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for {
|
|
msg, err := stream.Recv()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("Received from peer %x: type=%d, data=%x\n",
|
|
msg.Peer, msg.Type, msg.Data)
|
|
}
|
|
}
|