cli: Start a trivial readline-style interface

This commit is contained in:
John Brooks 2016-08-16 17:58:39 -07:00
parent 0b44f08f32
commit 9fb3aa33b5
1 changed files with 76 additions and 26 deletions

View File

@ -1,40 +1,31 @@
package main
import (
"fmt"
rpc "github.com/special/notricochet/rpc"
"golang.org/x/net/context"
"google.golang.org/grpc"
"gopkg.in/readline.v1"
"io"
"log"
"strings"
)
const (
defaultAddress = "127.0.0.1:58281"
)
func main() {
conn, err := grpc.Dial(defaultAddress, grpc.WithInsecure())
if err != nil {
log.Fatalf("connection failed: %v", err)
}
defer conn.Close()
type Client struct {
Backend rpc.RicochetCoreClient
Input *readline.Instance
}
c := rpc.NewRicochetCoreClient(conn)
r, err := c.GetServerStatus(context.Background(), &rpc.ServerStatusRequest{
RpcVersion: 1,
})
func (c *Client) Initialize() error {
stream, err := c.Backend.MonitorNetwork(context.Background(), &rpc.MonitorNetworkRequest{})
if err != nil {
log.Fatalf("could not get status: %v", err)
}
log.Printf("get status result: %v", r)
stream, err := c.MonitorNetwork(context.Background(), &rpc.MonitorNetworkRequest{})
if err != nil {
log.Fatalf("could not start stream: %v", err)
return err
}
wait := make(chan struct{})
go func() {
for {
resp, err := stream.Recv()
@ -43,19 +34,78 @@ func main() {
break
}
if err != nil {
log.Fatalf("stream error: %v", err)
log.Printf("stream error: %v", err)
break
}
log.Printf("stream says: %v", resp)
}
close(wait)
}()
status, err := c.StartNetwork(context.Background(), &rpc.StartNetworkRequest{})
return nil
}
func main() {
conn, err := grpc.Dial(defaultAddress, grpc.WithInsecure())
if err != nil {
log.Fatalf("connection failed: %v", err)
}
defer conn.Close()
input, err := readline.NewEx(&readline.Config{})
if err != nil {
log.Fatal(err)
}
defer input.Close()
log.SetOutput(input.Stdout())
c := &Client{
Backend: rpc.NewRicochetCoreClient(conn),
Input: input,
}
if err := c.Initialize(); err != nil {
log.Fatal(err)
}
input.SetPrompt("> ")
for {
line := input.Line()
if line.CanContinue() {
continue
} else if line.CanBreak() {
break
}
words := strings.SplitN(line.Line, " ", 1)
switch words[0] {
case "clear":
readline.ClearScreen(readline.Stdout)
case "status":
r, err := c.Backend.GetServerStatus(context.Background(), &rpc.ServerStatusRequest{
RpcVersion: 1,
})
if err != nil {
log.Fatalf("could not get status: %v", err)
}
log.Printf("get status result: %v", r)
case "connect":
status, err := c.Backend.StartNetwork(context.Background(), &rpc.StartNetworkRequest{})
if err != nil {
log.Printf("start network error: %v", err)
} else {
log.Printf("network started: %v", status)
}
<-wait
case "disconnect":
status, err := c.Backend.StopNetwork(context.Background(), &rpc.StopNetworkRequest{})
if err != nil {
log.Printf("stop network error: %v", err)
} else {
log.Printf("network stopped: %v", status)
}
case "help":
fallthrough
default:
fmt.Println("Commands: clear, status, connect, disconnect, help")
}
}
}