updated mapping and schema to support icons (#37)

* updated mapping and schema to support icons * typo * update entity schema * update entity schema * updated test * controller test * removed typo * white spaces
port-labs · Feb 1, 2024 · b24e223 · b24e223
1 parent 58b7a20
commit b24e223
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 4 deletions.
diff --git a/pkg/defaults/defaults_test.go b/pkg/defaults/defaults_test.go
@@ -205,11 +205,12 @@ func Test_InitIntegration_DeprecatedResourcesConfiguration(t *testing.T) {
 				Entity: port.EntityMappings{
 					Mappings: []port.EntityMapping{
 						{
-							Identifier: "workload",
-							Title:      "Workload",
-							Blueprint:  "workload",
+							Identifier: "\"workload\"",
+							Title:      "\"Workload\"",
+							Blueprint:  "\"workload\"",
+							Icon:       "\"Microservice\"",
 							Properties: map[string]string{
-								"namespace": "default",
+								"namespace": "\"default\"",
 							},
 						},
 					},

diff --git a/pkg/k8s/controller_test.go b/pkg/k8s/controller_test.go
@@ -164,6 +164,7 @@ func TestCreateDeployment(t *testing.T) {
 		{
 			Identifier: ".metadata.name",
 			Blueprint:  "\"k8s-export-test-bp\"",
+			Icon:       "\"Microservice\"",
 			Team:       "\"Test\"",
 			Properties: map[string]string{
 				"text": "\"pod\"",
@@ -190,6 +191,7 @@ func TestUpdateDeployment(t *testing.T) {
 		{
 			Identifier: ".metadata.name",
 			Blueprint:  "\"k8s-export-test-bp\"",
+			Icon:       "\"Microservice\"",
 			Team:       "[\"Test\", \"Test2\"]",
 			Properties: map[string]string{
 				"text": "\"pod\"",

diff --git a/pkg/port/mapping/entity.go b/pkg/port/mapping/entity.go
@@ -28,6 +28,12 @@ func NewEntity(obj interface{}, mapping port.EntityMapping) (*port.Entity, error
 			return &port.Entity{}, err
 		}
 	}
+	if mapping.Icon != "" {
+		entity.Icon, err = jq.ParseString(mapping.Icon, obj)
+		if err != nil {
+			return &port.Entity{}, err
+		}
+	}
 	entity.Properties, err = jq.ParseMapInterface(mapping.Properties, obj)
 	if err != nil {
 		return &port.Entity{}, err

diff --git a/pkg/port/models.go b/pkg/port/models.go
@@ -22,6 +22,7 @@ type (
 		Identifier string                 `json:"identifier,omitempty"`
 		Title      string                 `json:"title"`
 		Blueprint  string                 `json:"blueprint"`
+		Icon       string                 `json:"icon,omitempty"`
 		Team       interface{}            `json:"team,omitempty"`
 		Properties map[string]interface{} `json:"properties"`
 		Relations  map[string]interface{} `json:"relations"`
@@ -181,6 +182,7 @@ type EntityMapping struct {
 	Identifier string            `json:"identifier"`
 	Title      string            `json:"title"`
 	Blueprint  string            `json:"blueprint"`
+	Icon	   string            `json:"icon,omitempty"`
 	Team       string            `json:"team,omitempty"`
 	Properties map[string]string `json:"properties,omitempty"`
 	Relations  map[string]string `json:"relations,omitempty"`