add: new actoin api logic

odigos-io · Oct 22, 2024 · de8ac43 · de8ac43
1 parent f10f9a0
commit de8ac43
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 7 deletions.
diff --git a/frontend/graph/schema.resolvers.go b/frontend/graph/schema.resolvers.go
diff --git a/frontend/services/actions/piimasking.go b/frontend/services/actions/piimasking.go
@@ -51,14 +51,19 @@ func CreatePiiMasking(ctx context.Context, action model.ActionInput) (model.Acti
 		return nil, fmt.Errorf("failed to create PiiMasking: %v", err)
 	}
 
+	piiCategories := make([]string, len(details.PiiCategories))
+	for i, category := range details.PiiCategories {
+		piiCategories[i] = string(category)
+	}
+
 	response := &model.PiiMaskingAction{
 		ID:      generatedAction.Name,
 		Type:    ActionTypePiiMasking,
 		Name:    action.Name,
 		Notes:   action.Notes,
 		Disable: action.Disable,
 		Signals: action.Signals,
-		Details: details.PiiCategories,
+		Details: piiCategories,
 	}
 
 	return response, nil
@@ -95,14 +100,19 @@ func UpdatePiiMasking(ctx context.Context, id string, action model.ActionInput)
 		return nil, fmt.Errorf("failed to update PiiMasking: %v", err)
 	}
 
+	piiCategories := make([]string, len(details.PiiCategories))
+	for i, category := range details.PiiCategories {
+		piiCategories[i] = string(category)
+	}
+
 	response := &model.PiiMaskingAction{
 		ID:      updatedAction.Name,
 		Type:    ActionTypePiiMasking,
 		Name:    action.Name,
 		Notes:   action.Notes,
 		Disable: action.Disable,
 		Signals: action.Signals,
-		Details: details.PiiCategories,
+		Details: piiCategories,
 	}
 
 	return response, nil