TUN-3669: Teamnet commands to add/show Teamnet routes.

teamnet/api.go

@@ -0,0 +1,88 @@
+package teamnet
+
+import (
+	"encoding/json"
+	"fmt"
+	"net"
+	"time"
+
+	"github.com/google/uuid"
+)
+
+// Route is a mapping from customer's IP space to a tunnel.
+// Each route allows the customer to route eyeballs in their corporate network
+// to certain private IP ranges. Each Route represents an IP range in their
+// network, and says that eyeballs can reach that route using the corresponding
+// tunnel.
+type Route struct {
+	Network   net.IPNet
+	TunnelID  uuid.UUID
+	Comment   string
+	CreatedAt time.Time
+	DeletedAt time.Time
+}
+
+// TableString outputs a table row summarizing the route, to be used
+// when showing the user their routing table.
+func (r Route) TableString() string {
+	deletedColumn := "-"
+	if !r.DeletedAt.IsZero() {
+		deletedColumn = r.DeletedAt.Format(time.RFC3339)
+	}
+	return fmt.Sprintf(
+		"%s\t%s\t%s\t%s\t%s\t",
+		r.Network.String(),
+		r.Comment,
+		r.TunnelID,
+		r.CreatedAt.Format(time.RFC3339),
+		deletedColumn,
+	)
+}
+
+// UnmarshalJSON handles fields with non-JSON types (e.g. net.IPNet).
+func (r *Route) UnmarshalJSON(data []byte) error {
+
+	// This is the raw JSON format that cloudflared receives from tunnelstore.
+	// Note it does not understand types like IPNet.
+	var resp struct {
+		Network   string    `json:"network"`
+		TunnelID  uuid.UUID `json:"tunnel_id"`
+		Comment   string    `json:"comment"`
+		CreatedAt time.Time `json:"created_at"`
+		DeletedAt time.Time `json:"deleted_at"`
+	}
+	if err := json.Unmarshal(data, &resp); err != nil {
+		return err
+	}
+
+	// Parse the raw JSON into a properly-typed response.
+	_, network, err := net.ParseCIDR(resp.Network)
+	if err != nil || network == nil {
+		return fmt.Errorf("backend returned invalid network %s", resp.Network)
+	}
+	r.Network = *network
+	r.TunnelID = resp.TunnelID
+	r.Comment = resp.Comment
+	r.CreatedAt = resp.CreatedAt
+	r.DeletedAt = resp.DeletedAt
+	return nil
+}
+
+// NewRoute has all the parameters necessary to add a new route to the table.
+type NewRoute struct {
+	Network  net.IPNet
+	TunnelID uuid.UUID
+	Comment  string
+}
+
+// MarshalJSON handles fields with non-JSON types (e.g. net.IPNet).
+func (r NewRoute) MarshalJSON() ([]byte, error) {
+	return json.Marshal(&struct {
+		Network  string    `json:"network"`
+		TunnelID uuid.UUID `json:"tunnel_id"`
+		Comment  string    `json:"comment"`
+	}{
+		TunnelID: r.TunnelID,
+		Comment:  r.Comment,
+	})
+}