mirror of
https://github.com/lightningnetwork/lnd.git
synced 2024-11-19 09:53:54 +01:00
65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
//go:build dev
|
|
// +build dev
|
|
|
|
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
|
|
"github.com/lightninglabs/protobuf-hex-display/jsonpb"
|
|
"github.com/lightningnetwork/lnd/lncfg"
|
|
"github.com/lightningnetwork/lnd/lnrpc"
|
|
"github.com/lightningnetwork/lnd/lnrpc/devrpc"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
// devCommands will return the set of commands to enable for devrpc builds.
|
|
func devCommands() []cli.Command {
|
|
return []cli.Command{
|
|
{
|
|
Name: "importgraph",
|
|
Category: "Development",
|
|
Description: "Imports graph from describegraph JSON",
|
|
Usage: "Import the network graph.",
|
|
ArgsUsage: "graph-json-file",
|
|
Action: actionDecorator(importGraph),
|
|
},
|
|
}
|
|
}
|
|
|
|
func getDevClient(ctx *cli.Context) (devrpc.DevClient, func()) {
|
|
conn := getClientConn(ctx, false)
|
|
cleanUp := func() {
|
|
conn.Close()
|
|
}
|
|
return devrpc.NewDevClient(conn), cleanUp
|
|
}
|
|
|
|
func importGraph(ctx *cli.Context) error {
|
|
ctxc := getContext()
|
|
client, cleanUp := getDevClient(ctx)
|
|
defer cleanUp()
|
|
|
|
jsonFile := lncfg.CleanAndExpandPath(ctx.Args().First())
|
|
jsonBytes, err := ioutil.ReadFile(jsonFile)
|
|
if err != nil {
|
|
return fmt.Errorf("error reading JSON from file %v: %v",
|
|
jsonFile, err)
|
|
}
|
|
|
|
jsonGraph := &lnrpc.ChannelGraph{}
|
|
err = jsonpb.Unmarshal(bytes.NewReader(jsonBytes), jsonGraph)
|
|
if err != nil {
|
|
return fmt.Errorf("error parsing JSON: %v", err)
|
|
}
|
|
res, err := client.ImportGraph(ctxc, jsonGraph)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
printRespJSON(res)
|
|
return nil
|
|
}
|