feat: add support for EventFlow in TargetEndpoint and ProxyEndpoint

Fixes #37

Author: micovery

pkg/apigee/v1/apiproxymodel_test.go

@@ -53,6 +53,9 @@ func TestNewAPIProxyModel(t *testing.T) {
 		{
 			"google-id-token-auth",
 		},
+		{
+			"event-flow",
+		},
 	}
 	for _, tt := range tests {
 		ttDir := filepath.Join("testdata", "yaml-first", tt.name)

pkg/apigee/v1/eventflow.go

@@ -0,0 +1,42 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package v1
+
+import "fmt"
+
+type EventFlow struct {
+	Name        string    `xml:"name,attr,omitempty"`
+	ContentType string    `xml:"content-type,attr"`
+	Description string    `xml:"Description,omitempty"`
+	Response    *Response `xml:"Response,omitempty"`
+
+	UnknownNode AnyList `xml:",any"`
+}
+
+func ValidateEventFlow(v *EventFlow, path string) []error {
+	if v == nil {
+		return nil
+	}
+
+	subPath := fmt.Sprintf("%s.EventFlow", path)
+	if len(v.UnknownNode) > 0 {
+		return []error{NewUnknownNodeError(subPath, v.UnknownNode[0])}
+	}
+
+	var subErrors []error
+	subErrors = append(subErrors, ValidateResponse(v.Response, subPath)...)
+
+	return subErrors
+}

pkg/apigee/v1/proxyendpoint.go

@@ -26,6 +26,7 @@ type ProxyEndpoint struct {
 	FaultRules          *FaultRules          `xml:"FaultRules"`
 	DefaultFaultRule    *DefaultFaultRule    `xml:"DefaultFaultRule,omitempty"`
 	PreFlow             *PreFlow             `xml:"PreFlow,omitempty"`
+	EventFlow           *EventFlow           `xml:"EventFlow,omitempty"`
 	Flows               *Flows               `xml:"Flows,omitempty"`
 	PostFlow            *PostFlow            `xml:"PostFlow,omitempty"`
 	PostClientFlow      *PostClientFlow      `xml:"PostClientFlow,omitempty"`
@@ -64,6 +65,7 @@ func ValidateProxyEndpoint(v *ProxyEndpoint, path string) []error {
 	var subErrors []error
 	subErrors = append(subErrors, ValidateHTTPProxyConnection(v.HTTPProxyConnection, subPath)...)
 	subErrors = append(subErrors, ValidatePreFlow(v.PreFlow, subPath)...)
+	subErrors = append(subErrors, ValidateEventFlow(v.EventFlow, subPath)...)
 	subErrors = append(subErrors, ValidateFlows(v.Flows, subPath)...)
 	subErrors = append(subErrors, ValidatePostFlow(v.PostFlow, subPath)...)
 	subErrors = append(subErrors, ValidateRouteRules(v.RouteRules, subPath)...)

pkg/apigee/v1/targetendpoint.go

@@ -26,6 +26,7 @@ type TargetEndpoint struct {
 	FaultRules            *FaultRules            `xml:"FaultRules,omitempty"`
 	DefaultFaultRule      *DefaultFaultRule      `xml:"DefaultFaultRule,omitempty"`
 	PreFlow               PreFlow                `xml:"PreFlow"`
+	EventFlow             *EventFlow             `xml:"EventFlow,omitempty"`
 	Flows                 Flows                  `xml:"Flows"`
 	PostFlow              PostFlow               `xml:"PostFlow"`
 	HTTPTargetConnection  *HTTPTargetConnection  `xml:"HTTPTargetConnection,omitempty"`
@@ -63,6 +64,7 @@ func ValidateTargetEndpoint(v *TargetEndpoint, path string) []error {
 	var subErrors []error
 	subErrors = append(subErrors, ValidateHTTPTargetConnection(v.HTTPTargetConnection, subPath)...)
 	subErrors = append(subErrors, ValidatePreFlow(&v.PreFlow, subPath)...)
+	subErrors = append(subErrors, ValidateEventFlow(v.EventFlow, subPath)...)
 	subErrors = append(subErrors, ValidateFlows(&v.Flows, subPath)...)
 	subErrors = append(subErrors, ValidatePostFlow(&v.PostFlow, subPath)...)
 	subErrors = append(subErrors, ValidateFaultRules(v.FaultRules, subPath)...)

pkg/apigee/v1/testdata/yaml-first/event-flow/apiproxy.yaml

@@ -0,0 +1,80 @@
+#  Copyright 2025 Google LLC
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#       http:#www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+APIProxy:
+  .revision: 1
+  .name: httpbin
+Policies:
+  - RaiseFault:
+      .name: RF-InvalidCred
+      DisplayName: RF-InvalidCred
+      FaultResponse:
+        Set:
+          - StatusCode: 401
+          - ReasonPhrase: Unauthorized
+          - Payload:
+              .contentType: application/json
+              -Data: '{"error": "Unauthorized", "message": "Invalid access token"}'
+          - Headers:
+              Header:
+                .name: Content-Type
+                -Data: application/json
+  - Javascript:
+      .name: JS-combine-resp
+      .timeLimit: 200
+      .languageVersion: VERSION_ES6
+      DisplayName: JS-combine-resp
+      Source: |
+        // Access a variable from the flow
+        var myVariable = context.getVariable("myFlowVariable");
+
+        // Set a new variable
+        context.setVariable("javascript.policy.executed", true);
+        
+        print("JS-combine-resp policy executed!");
+  - SanitizeModelResponse:
+      .name: SMR-ModelResp
+      DisplayName: SMR-ModelResp
+      ModelArmor:
+        TemplateName: projects/{organization.name}/locations/{system.region.name}/templates/default-template
+      UserPromptSource: request.content
+      LLMResponseSource: response.content
+ProxyEndpoints:
+  - ProxyEndpoint:
+      .name: default
+      EventFlow:
+        .name: EF-1
+        .content-type: text/event-stream
+        Response:
+          - Step:
+              Name: RF-InvalidCred
+              Condition: fault.name equals "invalid_access_token"
+      HTTPProxyConnection:
+        BasePath: /httpbin
+      RouteRule:
+        .name: default
+        TargetEndpoint: default
+TargetEndpoints:
+  - TargetEndpoint:
+      .name: default
+      EventFlow:
+        .name: EventFlow
+        .content-type: text/event-stream
+        Response:
+          - Step:
+              Name: JS-combine-resp
+          - Step:
+              Condition: buff_ready = true
+              Name: SMR-ModelResp
+      HTTPTargetConnection:
+        URL: https://httpbin.org/