Add command 'gp ports list' to gitpod-cli

Author: felladrin

components/gitpod-cli/cmd/ports.go

@@ -0,0 +1,33 @@
+// Copyright (c) 2022 Gitpod GmbH. All rights reserved.
+// Licensed under the GNU Affero General Public License (AGPL).
+// See License-AGPL.txt in the project root for license information.
+
+package cmd
+
+import (
+	"github.com/gitpod-io/gitpod/gitpod-cli/cmd/ports"
+	"github.com/spf13/cobra"
+)
+
+// portsCmd represents the ports command
+var portsCmd = &cobra.Command{
+	Use:   "ports",
+	Short: "Interact with workspace ports",
+	Run: func(cmd *cobra.Command, args []string) {
+		if len(args) == 0 {
+			_ = cmd.Help()
+		}
+	},
+}
+
+// listPortsCmd represents the ports list command
+var listPortsCmd = &cobra.Command{
+	Use:   "list",
+	Short: "Lists the workspace ports and their state",
+	Run:   ports.ListPortsCmd,
+}
+
+func init() {
+	rootCmd.AddCommand(portsCmd)
+	portsCmd.AddCommand(listPortsCmd)
+}

components/gitpod-cli/cmd/ports/ports-list.go

@@ -0,0 +1,75 @@
+// Copyright (c) 2022 Gitpod GmbH. All rights reserved.
+// Licensed under the GNU Affero General Public License (AGPL).
+// See License-AGPL.txt in the project root for license information.
+
+package ports
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"time"
+
+	"github.com/gitpod-io/gitpod/gitpod-cli/pkg/supervisor"
+	"github.com/gitpod-io/gitpod/supervisor/api"
+	"github.com/spf13/cobra"
+
+	"github.com/olekukonko/tablewriter"
+)
+
+func ListPortsCmd(cmd *cobra.Command, args []string) {
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	conn := supervisor.Dial()
+	client := api.NewStatusServiceClient(conn)
+
+	ports := supervisor.GetPortsList(ctx, client)
+
+	if len(ports) == 0 {
+		fmt.Println("No ports detected")
+		return
+	}
+
+	table := tablewriter.NewWriter(os.Stdout)
+	table.SetHeader([]string{"Port", "Status", "URL", "Name & Description"})
+	table.SetBorders(tablewriter.Border{Left: true, Top: false, Right: true, Bottom: false})
+	table.SetCenterSeparator("|")
+
+	for _, port := range ports {
+		status := "not served"
+		statusColor := tablewriter.FgHiBlackColor
+		if port.Exposed == nil && port.Tunneled == nil {
+			if port.AutoExposure == api.PortAutoExposure_failed {
+				status = "failed to expose"
+				statusColor = tablewriter.FgRedColor
+			} else {
+				status = "detecting..."
+				statusColor = tablewriter.FgYellowColor
+			}
+		} else if port.Served {
+			status = "open (" + port.Exposed.Visibility.String() + ")"
+			if port.Exposed.Visibility == api.PortVisibility_public {
+				statusColor = tablewriter.FgHiCyanColor
+			} else {
+				statusColor = tablewriter.FgHiWhiteColor
+			}
+		}
+
+		nameAndDescription := port.Name
+		if len(port.Description) > 0 {
+			if len(nameAndDescription) > 0 {
+				nameAndDescription = fmt.Sprint(nameAndDescription, ": ", port.Description)
+			} else {
+				nameAndDescription = port.Description
+			}
+		}
+
+		table.Rich(
+			[]string{fmt.Sprint(port.LocalPort), status, port.Exposed.Url, nameAndDescription},
+			[]tablewriter.Colors{{}, {statusColor}, {}, {}},
+		)
+	}
+
+	table.Render()
+}

components/gitpod-cli/pkg/supervisor/ports.go

@@ -0,0 +1,42 @@
+// Copyright (c) 2022 Gitpod GmbH. All rights reserved.
+// Licensed under the GNU Affero General Public License (AGPL).
+// See License-AGPL.txt in the project root for license information.
+
+package supervisor
+
+import (
+	"context"
+	"io"
+
+	"github.com/gitpod-io/gitpod/supervisor/api"
+	log "github.com/sirupsen/logrus"
+)
+
+func GetPortsList(ctx context.Context, client api.StatusServiceClient) []*api.PortsStatus {
+	listen, err := client.PortsStatus(ctx, &api.PortsStatusRequest{Observe: false})
+	if err != nil {
+		log.WithError(err).Error("Cannot list ports")
+	}
+
+	portschan := make(chan []*api.PortsStatus)
+
+	go func() {
+		for {
+			resp, err := listen.Recv()
+			if err != nil && err != io.EOF {
+				panic(err)
+			}
+
+			chunk := resp.GetPorts()
+
+			if chunk == nil {
+				close(portschan)
+				break
+			} else {
+				portschan <- chunk
+			}
+		}
+	}()
+
+	return <-portschan
+}