160 lines
3.0 KiB
Go
160 lines
3.0 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
"text/scanner"
|
|
|
|
"github.com/chzyer/readline"
|
|
"github.com/urfave/cli/v2"
|
|
"golang.org/x/xerrors"
|
|
|
|
lcli "github.com/filecoin-project/lotus/cli"
|
|
"github.com/filecoin-project/lotus/node/repo"
|
|
)
|
|
|
|
var rpcCmd = &cli.Command{
|
|
Name: "rpc",
|
|
Usage: "Interactive JsonPRC shell",
|
|
// TODO: flag for miner/worker
|
|
Action: func(cctx *cli.Context) error {
|
|
addr, headers, err := lcli.GetRawAPI(cctx, repo.FullNode)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
u, err := url.Parse(addr)
|
|
if err != nil {
|
|
return xerrors.Errorf("parsing api URL: %w", err)
|
|
}
|
|
|
|
switch u.Scheme {
|
|
case "ws":
|
|
u.Scheme = "http"
|
|
case "wss":
|
|
u.Scheme = "https"
|
|
}
|
|
|
|
addr = u.String()
|
|
|
|
ctx := lcli.ReqContext(cctx)
|
|
ctx, cancel := context.WithCancel(ctx)
|
|
defer cancel()
|
|
afmt := lcli.NewAppFmt(cctx.App)
|
|
|
|
cs := readline.NewCancelableStdin(afmt.Stdin)
|
|
go func() {
|
|
<-ctx.Done()
|
|
cs.Close() // nolint:errcheck
|
|
}()
|
|
|
|
send := func(method, params string) error {
|
|
jreq, err := json.Marshal(struct {
|
|
Jsonrpc string `json:"jsonrpc"`
|
|
ID int `json:"id"`
|
|
Method string `json:"method"`
|
|
Params json.RawMessage `json:"params"`
|
|
}{
|
|
Jsonrpc: "2.0",
|
|
Method: "Filecoin." + method,
|
|
Params: json.RawMessage(params),
|
|
ID: 0,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
req, err := http.NewRequest("POST", addr, bytes.NewReader(jreq))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
req.Header = headers
|
|
resp, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
rb, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Println(string(rb))
|
|
|
|
if err := resp.Body.Close(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
if cctx.Args().Present() {
|
|
if cctx.Args().Len() > 2 {
|
|
return xerrors.Errorf("expected 1 or 2 arguments: method [params]")
|
|
}
|
|
|
|
params := cctx.Args().Get(1)
|
|
if params == "" {
|
|
// TODO: try to be smart and use zero-values for method
|
|
params = "[]"
|
|
}
|
|
|
|
return send(cctx.Args().Get(0), params)
|
|
}
|
|
|
|
cctx.App.Metadata["repoType"] = repo.FullNode
|
|
if err := lcli.VersionCmd.Action(cctx); err != nil {
|
|
return err
|
|
}
|
|
fmt.Println("Usage: > Method [Param1, Param2, ...]")
|
|
|
|
rl, err := readline.NewEx(&readline.Config{
|
|
Stdin: cs,
|
|
HistoryFile: "/tmp/lotusrpc.tmp",
|
|
Prompt: "> ",
|
|
EOFPrompt: "exit",
|
|
HistorySearchFold: true,
|
|
|
|
// TODO: Some basic auto completion
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for {
|
|
line, err := rl.Readline()
|
|
if err == readline.ErrInterrupt {
|
|
if len(line) == 0 {
|
|
break
|
|
} else {
|
|
continue
|
|
}
|
|
} else if err == io.EOF {
|
|
break
|
|
}
|
|
|
|
var s scanner.Scanner
|
|
s.Init(strings.NewReader(line))
|
|
s.Scan()
|
|
method := s.TokenText()
|
|
|
|
s.Scan()
|
|
params := line[s.Position.Offset:]
|
|
|
|
if err := send(method, params); err != nil {
|
|
_, _ = fmt.Fprintf(os.Stderr, "%v", err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
},
|
|
}
|