129 lines
2.7 KiB
Go
129 lines
2.7 KiB
Go
package cli
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
logging "github.com/ipfs/go-log"
|
|
manet "github.com/multiformats/go-multiaddr-net"
|
|
"golang.org/x/xerrors"
|
|
"gopkg.in/urfave/cli.v2"
|
|
|
|
"github.com/filecoin-project/lotus/api"
|
|
"github.com/filecoin-project/lotus/api/client"
|
|
"github.com/filecoin-project/lotus/lib/jsonrpc"
|
|
"github.com/filecoin-project/lotus/node/repo"
|
|
)
|
|
|
|
var log = logging.Logger("cli")
|
|
|
|
const (
|
|
metadataTraceConetxt = "traceContext"
|
|
)
|
|
|
|
// ApiConnector returns API instance
|
|
type ApiConnector func() api.FullNode
|
|
|
|
func getAPI(ctx *cli.Context, repoFlag string) (string, http.Header, error) {
|
|
r, err := repo.NewFS(ctx.String(repoFlag))
|
|
if err != nil {
|
|
return "", nil, err
|
|
}
|
|
|
|
ma, err := r.APIEndpoint()
|
|
if err != nil {
|
|
return "", nil, xerrors.Errorf("failed to get api endpoint: %w", err)
|
|
}
|
|
_, addr, err := manet.DialArgs(ma)
|
|
if err != nil {
|
|
return "", nil, err
|
|
}
|
|
var headers http.Header
|
|
token, err := r.APIToken()
|
|
if err != nil {
|
|
log.Warnf("Couldn't load CLI token, capabilities may be limited: %w", err)
|
|
} else {
|
|
headers = http.Header{}
|
|
headers.Add("Authorization", "Bearer "+string(token))
|
|
}
|
|
|
|
return "ws://" + addr + "/rpc/v0", headers, nil
|
|
}
|
|
|
|
func GetAPI(ctx *cli.Context) (api.Common, jsonrpc.ClientCloser, error) {
|
|
f := "repo"
|
|
if ctx.String("storagerepo") != "" {
|
|
f = "storagerepo"
|
|
}
|
|
|
|
addr, headers, err := getAPI(ctx, f)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
return client.NewCommonRPC(addr, headers)
|
|
}
|
|
|
|
func GetFullNodeAPI(ctx *cli.Context) (api.FullNode, jsonrpc.ClientCloser, error) {
|
|
addr, headers, err := getAPI(ctx, "repo")
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
return client.NewFullNodeRPC(addr, headers)
|
|
}
|
|
|
|
func GetStorageMinerAPI(ctx *cli.Context) (api.StorageMiner, jsonrpc.ClientCloser, error) {
|
|
addr, headers, err := getAPI(ctx, "storagerepo")
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
return client.NewStorageMinerRPC(addr, headers)
|
|
}
|
|
|
|
func DaemonContext(cctx *cli.Context) context.Context {
|
|
if mtCtx, ok := cctx.App.Metadata[metadataTraceConetxt]; ok {
|
|
return mtCtx.(context.Context)
|
|
}
|
|
|
|
return context.Background()
|
|
}
|
|
|
|
// ReqContext returns context for cli execution. Calling it for the first time
|
|
// installs SIGTERM handler that will close returned context.
|
|
// Not safe for concurrent execution.
|
|
func ReqContext(cctx *cli.Context) context.Context {
|
|
tCtx := DaemonContext(cctx)
|
|
|
|
ctx, done := context.WithCancel(tCtx)
|
|
sigChan := make(chan os.Signal, 2)
|
|
go func() {
|
|
<-sigChan
|
|
done()
|
|
}()
|
|
signal.Notify(sigChan, syscall.SIGTERM, syscall.SIGINT)
|
|
|
|
return ctx
|
|
}
|
|
|
|
var Commands = []*cli.Command{
|
|
authCmd,
|
|
chainCmd,
|
|
clientCmd,
|
|
createMinerCmd,
|
|
fetchParamCmd,
|
|
mpoolCmd,
|
|
netCmd,
|
|
paychCmd,
|
|
sendCmd,
|
|
stateCmd,
|
|
syncCmd,
|
|
unregisterMinerCmd,
|
|
versionCmd,
|
|
walletCmd,
|
|
}
|