Add the file command with read subcommand

cmd/speedrun/cli/file.go

@@ -0,0 +1,92 @@
+package cli
+
+import (
+	"context"
+	"crypto/tls"
+	"fmt"
+	"strings"
+	"time"
+
+	"github.com/alitto/pond"
+	"github.com/apex/log"
+	"github.com/dpogorzelski/speedrun/pkg/speedrun/cloud"
+	portalpb "github.com/dpogorzelski/speedrun/proto/portal"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+	"storj.io/drpc/drpcconn"
+)
+
+var fileCmd = &cobra.Command{
+	Use:              "file",
+	Short:            "Manage files",
+	TraverseChildren: true,
+}
+
+var readCmd = &cobra.Command{
+	Use:     "read <path>",
+	Short:   "Read a file",
+	Example: "  speedrun file read /etc/resolv.conf",
+	Args:    cobra.MinimumNArgs(1),
+	RunE:    read,
+}
+
+func init() {
+	fileCmd.SetUsageTemplate(usage)
+	fileCmd.AddCommand(readCmd)
+}
+
+func read(cmd *cobra.Command, args []string) error {
+	usePrivateIP := viper.GetBool("portal.use-private-ip")
+
+	tlsConfig, err := cloud.SetupTLS()
+	if err != nil {
+		return err
+	}
+
+	target, err := cmd.Flags().GetString("target")
+	if err != nil {
+		return err
+	}
+
+	portals, err := cloud.GetInstances(target)
+	if err != nil {
+		return err
+	}
+
+	pool := pond.New(1000, 10000)
+	for _, p := range portals {
+		portal := p
+		pool.Submit(func() {
+			fields := log.Fields{
+				"host":    portal.Name,
+				"address": portal.GetAddress(usePrivateIP),
+			}
+			log := log.WithFields(fields)
+
+			addr := fmt.Sprintf("%s:%d", portal.GetAddress(usePrivateIP), 1337)
+			rawconn, err := tls.Dial("tcp", addr, tlsConfig)
+			if err != nil {
+				log.Error(err.Error())
+				return
+			}
+
+			conn := drpcconn.New(rawconn)
+			defer conn.Close()
+
+			c := portalpb.NewDRPCPortalClient(conn)
+			ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
+			defer cancel()
+
+			path := strings.Join(args, " ")
+			r, err := c.FileRead(ctx, &portalpb.FileReadRequest{Path: path})
+			if err != nil {
+				log.Error(err.Error())
+				return
+			}
+			log.WithField("state", r.GetState()).Infof("Contents of %s:\n%s", path, r.GetContent())
+
+		})
+	}
+	pool.StopAndWait()
+	return nil
+}

cmd/speedrun/cli/root.go

@@ -36,7 +36,7 @@ func Execute() {
 
 	cobra.OnInitialize(initConfig)
 	rootCmd.SetUsageTemplate(rootUsage)
-	rootCmd.AddCommand(runCmd, serviceCmd)
+	rootCmd.AddCommand(runCmd, serviceCmd, fileCmd)
 
 	home, err := homedir.Dir()
 	if err != nil {

cmd/speedrun/cli/templates/root.tmpl

@@ -5,7 +5,7 @@ Usage:{{if .Runnable}}
 Core Commands:{{range .Commands}}{{if (or (eq .Name "help") (eq .Name "completion"))}}
   {{rpad .Name .NamePadding }} {{.Short}}{{end}}{{end}}
 
-Action Commands:{{range .Commands}}{{if (or (eq .Name "run") (eq .Name "exec") (eq .Name "service"))}}
+Action Commands:{{range .Commands}}{{if (or (eq .Name "run") (eq .Name "exec") (eq .Name "service") (eq .Name "file"))}}
   {{rpad .Name .NamePadding }} {{.Short}}{{end}}{{end}}
 {{if .HasAvailableLocalFlags}}
 Flags: