lotus/cli/cmd.go

128 lines
2.7 KiB
Go
Raw Normal View History

package cli
import (
"context"
2019-07-23 18:49:09 +00:00
"net/http"
"os"
"os/signal"
"syscall"
2019-07-23 18:49:09 +00:00
logging "github.com/ipfs/go-log"
2019-07-10 17:28:49 +00:00
manet "github.com/multiformats/go-multiaddr-net"
2019-07-26 11:45:25 +00:00
"golang.org/x/xerrors"
"gopkg.in/urfave/cli.v2"
2019-07-10 17:28:49 +00:00
"github.com/filecoin-project/go-lotus/api"
"github.com/filecoin-project/go-lotus/api/client"
"github.com/filecoin-project/go-lotus/node/repo"
)
2019-07-23 18:49:09 +00:00
var log = logging.Logger("cli")
const (
metadataTraceConetxt = "traceContext"
metadataContext = "context"
)
// ApiConnector returns API instance
2019-07-24 00:09:34 +00:00
type ApiConnector func() api.FullNode
2019-07-08 19:07:16 +00:00
func getAPI(ctx *cli.Context, repoFlag string) (string, http.Header, error) {
r, err := repo.NewFS(ctx.String(repoFlag))
2019-07-10 17:28:49 +00:00
if err != nil {
return "", nil, err
2019-07-10 17:28:49 +00:00
}
ma, err := r.APIEndpoint()
if err != nil {
return "", nil, xerrors.Errorf("failed to get api endpoint: %w", err)
2019-07-10 17:28:49 +00:00
}
_, addr, err := manet.DialArgs(ma)
2019-07-11 11:52:07 +00:00
if err != nil {
return "", nil, err
2019-07-11 11:52:07 +00:00
}
2019-07-23 18:49:09 +00:00
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 {
2019-07-23 20:05:44 +00:00
headers = http.Header{}
2019-07-23 18:49:09 +00:00
headers.Add("Authorization", "Bearer "+string(token))
}
return "ws://" + addr + "/rpc/v0", headers, nil
}
2019-08-02 16:18:44 +00:00
func GetAPI(ctx *cli.Context) (api.Common, error) {
f := "repo"
if ctx.String("storagerepo") != "" {
f = "storagerepo"
}
addr, headers, err := getAPI(ctx, f)
if err != nil {
return nil, err
}
return client.NewCommonRPC(addr, headers)
}
func GetFullNodeAPI(ctx *cli.Context) (api.FullNode, error) {
addr, headers, err := getAPI(ctx, "repo")
if err != nil {
return nil, err
}
return client.NewFullNodeRPC(addr, headers)
}
func GetStorageMinerAPI(ctx *cli.Context) (api.StorageMiner, error) {
addr, headers, err := getAPI(ctx, "storagerepo")
if err != nil {
return nil, err
}
return client.NewStorageMinerRPC(addr, headers)
}
2019-07-18 23:16:23 +00:00
// 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.
2019-07-18 23:16:23 +00:00
func ReqContext(cctx *cli.Context) context.Context {
if uctx, ok := cctx.App.Metadata[metadataContext]; ok {
// unchecked cast as if something else is in there
// it is crash worthy either way
return uctx.(context.Context)
}
var tCtx context.Context
if mtCtx, ok := cctx.App.Metadata[metadataTraceConetxt]; ok {
tCtx = mtCtx.(context.Context)
} else {
tCtx = context.Background()
}
ctx, done := context.WithCancel(tCtx)
sigChan := make(chan os.Signal, 2)
go func() {
<-sigChan
done()
}()
signal.Notify(sigChan, syscall.SIGTERM, syscall.SIGINT)
return ctx
2019-07-08 19:07:16 +00:00
}
var Commands = []*cli.Command{
2019-07-09 15:19:27 +00:00
chainCmd,
2019-07-20 21:55:51 +00:00
clientCmd,
minerCmd,
mpoolCmd,
2019-07-08 19:07:16 +00:00
netCmd,
versionCmd,
2019-07-13 00:41:32 +00:00
walletCmd,
createMinerCmd,
stateCmd,
sendCmd,
}