cli: Start a trivial readline-style interface

cli/cli.go

@@ -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")
+		}
+	}
 }