fix: align container and extension definitions to the spec

Signed-off-by: Armin Graf <arminhammer@gmail.com>

Author: arminhammer

core/src/lib.rs

@@ -373,4 +373,74 @@ mod unit_tests {
         assert!(script.environment.is_some());
         assert_eq!(script.environment.as_ref().unwrap().get("FOO"), Some(&"bar".to_string()));
     }
+
+    #[test]
+    fn test_extension_definition_serialization() {
+        let extension_json = json!({
+            "document": {
+                "dsl": "1.0.2",
+                "namespace": "test",
+                "name": "sample-workflow",
+                "version": "0.1.0"
+            },
+            "use": {
+                "extensions": [
+                    {
+                    "mockService": {
+                        "extend": "call",
+                        "when": "($task.with.endpoint != null and ($task.with.endpoint | startswith(\"https://mocked.service.com\"))) or ($task.with.endpoint.uri != null and ($task.with.endpoint.uri | startswith(\"https://mocked.service.com\")))",
+                        "before": [
+                            {
+                                "mockResponse": {
+                                    "set": {
+                                        "statusCode": 200,
+                                        "headers": {
+                                            "Content-Type": "application/json"
+                                        },
+                                        "content": {
+                                            "foo": {
+                                                "bar": "baz"
+                                            }
+                                        }
+                                    },
+                                    "then": "exit"
+                                }
+                            }
+                        ]
+                    }
+                }
+                ]
+            },
+            "do": [
+                {
+                    "callHttp": {
+                        "call": "http",
+                        "with": {
+                            "method": "get",
+                            "endpoint": {
+                                "uri": "https://fake.com/sample"
+                            }
+                        }
+                    }
+                }
+            ]
+        });
+        let result: Result<WorkflowDefinition, _> = serde_json::from_value(extension_json);
+        match result {
+            Ok(workflow) => {
+                // Verify the workflow was deserialized correctly
+                assert_eq!(workflow.document.namespace, "test");
+                assert_eq!(workflow.document.name, "sample-workflow");
+                assert_eq!(workflow.document.version, "0.1.0");
+                // Verify the use section exists with extensions
+                assert!(workflow.use_.is_some());
+                if let Some(use_def) = workflow.use_ {
+                    assert!(use_def.extensions.is_some());
+                }
+            }
+            Err(e) => {
+                panic!("Failed to deserialize workflow with extension: {}", e);
+            }
+        }
+    }
 }

core/src/models/extension.rs

@@ -16,10 +16,9 @@ pub struct ExtensionDefinition{
 
     /// Gets/sets a name/definition map, if any, of the tasks to execute before the extended task
     #[serde(rename = "before", skip_serializing_if = "Option::is_none")]
-    pub before: Option<HashMap<String, TaskDefinition>>,
+    pub before: Option<Vec<HashMap<String, TaskDefinition>>>,
 
     /// Gets/sets a name/definition map, if any, of the tasks to execute after the extended task
     #[serde(rename = "after", skip_serializing_if = "Option::is_none")]
-    pub after: Option<HashMap<String, TaskDefinition>>
-
+    pub after: Option<Vec<HashMap<String, TaskDefinition>>>
 }

core/src/models/task.rs

@@ -715,16 +715,23 @@ pub struct ContainerProcessDefinition{
     #[serde(rename = "environment", skip_serializing_if = "Option::is_none")]
     pub environment: Option<HashMap<String, String>>,
 
+    #[serde(rename = "stdin", skip_serializing_if = "Option::is_none")]
+    pub stdin: Option<String>,
+
+    #[serde(rename = "arguments", skip_serializing_if = "Option::is_none")]
+    pub arguments: Option<Vec<String>>,
 }
 impl ContainerProcessDefinition {
-    pub fn new(image: &str, name: Option<String>, command: Option<String>, ports: Option<HashMap<u16, u16>>, volumes: Option<HashMap<String, String>>, environment: Option<HashMap<String, String>>) -> Self{
+    pub fn new(image: &str, name: Option<String>, command: Option<String>, ports: Option<HashMap<u16, u16>>, volumes: Option<HashMap<String, String>>, environment: Option<HashMap<String, String>>, stdin: Option<String>, arguments: Option<Vec<String>>) -> Self{
         Self { 
             image: image.to_string(), 
             name,
             command, 
             ports, 
             volumes, 
-            environment
+            environment,
+            stdin,
+            arguments,
         }
     }
 }

core/src/models/workflow.rs

@@ -207,7 +207,7 @@ pub struct ComponentDefinitionCollection{
 
     /// Gets/sets a name/value mapping of the workflow's extensions, if any
     #[serde(rename = "extensions", skip_serializing_if = "Option::is_none")]
-    pub extensions: Option<HashMap<String, ExtensionDefinition>>,
+    pub extensions: Option<Vec<HashMap<String, ExtensionDefinition>>>,
 
     /// Gets/sets a name/value mapping of the workflow's reusable functions
     #[serde(rename = "functions", skip_serializing_if = "Option::is_none")]