add cp command

cmd/speedrun/cli/file.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"crypto/tls"
 	"net"
+	"os"
 	"strings"
 	"time"
 
@@ -30,9 +31,18 @@ var readCmd = &cobra.Command{
 	RunE:    read,
 }
 
+var cpCmd = &cobra.Command{
+	Use:     "cp <src> <dst>",
+	Short:   "Copy a file",
+	Example: "  speedrun file cp myfile /tmp/myfile",
+	Args:    cobra.MinimumNArgs(1),
+	RunE:    cp,
+}
+
 func init() {
 	fileCmd.SetUsageTemplate(usage)
 	fileCmd.AddCommand(readCmd)
+	fileCmd.AddCommand(cpCmd)
 }
 
 func read(cmd *cobra.Command, args []string) error {
@@ -83,10 +93,70 @@ func read(cmd *cobra.Command, args []string) error {
 				log.Error(err.Error())
 				return
 			}
-			log.WithField("state", r.GetState()).Infof("Contents of %s:\n%s", path, r.GetContent())
+			log.WithField("state", r.GetState()).Infof("Contents of %s:", path)
+			println(r.GetContent())
 
 		})
 	}
 	pool.StopAndWait()
 	return nil
 }
+
+func cp(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
+	}
+
+	content, err := os.ReadFile(args[0])
+	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 := net.JoinHostPort(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()
+
+			r, err := c.FileCp(ctx, &portalpb.FileCpRequest{Path: args[1], Content: content})
+			if err != nil {
+				log.Error(err.Error())
+				return
+			}
+			log.WithField("state", r.GetState()).Infof("Done")
+		})
+	}
+	pool.StopAndWait()
+	return nil
+}