Add missing ports field to workspace model

Signed-off-by: Yehudit Kerido <[email protected]>

Author: Yehudit Kerido

workspaces/backend/internal/models/workspaces/funcs.go

@@ -128,6 +128,39 @@ func NewWorkspaceModelFromWorkspace(ws *kubefloworgv1beta1.Workspace, wsk *kubef
 	return workspaceModel
 }
 
+func buildPortsList(ws *kubefloworgv1beta1.Workspace, wsk *kubefloworgv1beta1.WorkspaceKind) []ImagePort {
+	var ports []ImagePort
+
+	// Return an empty list if wsk is nil.
+	if !wskExists(wsk) {
+		return ports
+	}
+
+	// Get the image configuration from the WorkspaceKind's PodTemplate options.
+	imageConfig := wsk.Spec.PodTemplate.Options.ImageConfig
+
+	for _, val := range imageConfig.Values {
+		if len(val.Spec.Ports) == 0 {
+			continue
+		}
+		firstPort := val.Spec.Ports[0]
+		portStr := fmt.Sprintf("%d", firstPort.Port)
+		id := firstPort.Id
+		displayName := firstPort.DisplayName
+		if displayName == "" {
+			displayName = val.Id
+		}
+		imagePort := ImagePort{
+			ID:          id,
+			DisplayName: displayName,
+			Port:        portStr,
+		}
+		ports = append(ports, imagePort)
+	}
+
+	return ports
+}
+
 func wskExists(wsk *kubefloworgv1beta1.WorkspaceKind) bool {
 	return wsk != nil && wsk.UID != ""
 }
@@ -218,6 +251,7 @@ func buildImageConfig(ws *kubefloworgv1beta1.Workspace, wsk *kubefloworgv1beta1.
 		Current:       currentImageConfig,
 		Desired:       desiredImageConfig,
 		RedirectChain: redirectChain,
+		Ports:         buildPortsList(ws, wsk),
 	}
 }
 

workspaces/backend/internal/models/workspaces/types.go

@@ -84,6 +84,13 @@ type ImageConfig struct {
 	Current       OptionInfo     `json:"current"`
 	Desired       *OptionInfo    `json:"desired,omitempty"`
 	RedirectChain []RedirectStep `json:"redirect_chain,omitempty"`
+	Ports         []ImagePort    `json:"ports,omitempty"`
+}
+
+type ImagePort struct {
+	ID          string `json:"id"`
+	DisplayName string `json:"display_name"`
+	Port        string `json:"port"`
 }
 
 type PodConfig struct {