fix(1428): add support for multiValue

KaotoIO · Sep 17, 2024 · 10c31e5 · 10c31e5
1 parent 9b7cd80
commit 10c31e5
Show file tree

Hide file tree

Showing 5 changed files with 110 additions and 1 deletion.
diff --git a/packages/ui/src/models/camel-properties-common.ts b/packages/ui/src/models/camel-properties-common.ts
@@ -5,6 +5,8 @@ export interface CamelPropertyCommon {
   displayName: string;
   label?: string;
   required: boolean;
+  multiValue?: boolean;
+  prefix?: string;
   javaType: string;
   enum?: string[];
   autowired: boolean;

diff --git a/packages/ui/src/models/visualization/flows/abstract-camel-visual-entity.ts b/packages/ui/src/models/visualization/flows/abstract-camel-visual-entity.ts
@@ -72,7 +72,8 @@ export abstract class AbstractCamelVisualEntity<T extends object> implements Bas
 
   updateModel(path: string | undefined, value: unknown): void {
     if (!path) return;
-    const updatedValue = CamelComponentSchemaService.getUriSerializedDefinition(path, value);
+    let updatedValue = CamelComponentSchemaService.getUriSerializedDefinition(path, value);
+    updatedValue = CamelComponentSchemaService.getMultiValueSerializedDefinition(path, updatedValue);
 
     setValue(this.route, path, updatedValue);
   }

diff --git a/...els/visualization/flows/support/__snapshots__/camel-component-schema.service.test.ts.snap b/...els/visualization/flows/support/__snapshots__/camel-component-schema.service.test.ts.snap
@@ -820,6 +820,7 @@ exports[`CamelComponentSchemaService getVisualComponentSchema should transform a
     "parameters": {
       "beanName": "myBean",
       "method": "hello",
+      "parameters": {},
     },
     "uri": "bean",
   },
@@ -935,6 +936,7 @@ exports[`CamelComponentSchemaService getVisualComponentSchema should transform a
     "parameters": {
       "beanName": "myBean",
       "method": "hello",
+      "parameters": {},
     },
     "uri": "bean",
   },

diff --git a/packages/ui/src/models/visualization/flows/support/camel-component-schema.service.test.ts b/packages/ui/src/models/visualization/flows/support/camel-component-schema.service.test.ts
@@ -559,6 +559,43 @@ describe('CamelComponentSchemaService', () => {
     });
   });
 
+  describe('getMultiValueSerializedDefinition', () => {
+    it('should return the same parameters if the definition is not a component', () => {
+      const definition = { log: { message: 'Hello World' } };
+      const result = CamelComponentSchemaService.getMultiValueSerializedDefinition('from', definition);
+
+      expect(result).toEqual(definition);
+    });
+
+    it('should return the same parameters if the component is not found', () => {
+      const definition = {
+        uri: 'unknown-component',
+        parameters: { jobParameters: { test: 'test' }, triggerParameters: { test: 'test' } },
+      };
+      const result = CamelComponentSchemaService.getMultiValueSerializedDefinition('from', definition);
+
+      expect(result).toEqual(definition);
+    });
+
+    it('should query the catalog service', () => {
+      const definition = { uri: 'log', parameters: { message: 'Hello World' } };
+      const catalogServiceSpy = jest.spyOn(CamelCatalogService, 'getCatalogLookup');
+
+      CamelComponentSchemaService.getMultiValueSerializedDefinition('from', definition);
+      expect(catalogServiceSpy).toHaveBeenCalledWith('log');
+    });
+
+    it('should return the serialized definition', () => {
+      const definition = {
+        uri: 'quartz',
+        parameters: { jobParameters: { test: 'test' }, triggerParameters: { test: 'test' } },
+      };
+      const result = CamelComponentSchemaService.getMultiValueSerializedDefinition('from', definition);
+
+      expect(result).toEqual({ uri: 'quartz', parameters: { 'job.test': 'test', 'trigger.test': 'test' } });
+    });
+  });
+
   describe('getUriSerializedDefinition', () => {
     it('should return the same parameters if the definition is not a component', () => {
       const definition = { log: { message: 'Hello World' } };

diff --git a/packages/ui/src/models/visualization/flows/support/camel-component-schema.service.ts b/packages/ui/src/models/visualization/flows/support/camel-component-schema.service.ts
@@ -257,6 +257,42 @@ export class CamelComponentSchemaService {
     return definition;
   }
 
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  static getMultiValueSerializedDefinition(path: string, definition: any): ParsedParameters | undefined {
+    const camelElementLookup = this.getCamelComponentLookup(path, definition);
+    if (camelElementLookup.componentName === undefined) {
+      return definition;
+    }
+
+    const catalogLookup = CamelCatalogService.getCatalogLookup(camelElementLookup.componentName);
+    if (catalogLookup.catalogKind === CatalogKind.Component) {
+      const multiValueParameters: Map<string, string> = new Map<string, string>();
+      if (catalogLookup.definition?.properties !== undefined) {
+        Object.entries(catalogLookup.definition.properties).forEach(([key, value]) => {
+          if (value.multiValue) multiValueParameters.set(key, value.prefix!);
+        });
+      }
+      const defaultMultiValues: ParsedParameters = {};
+      const filteredParameters = definition.parameters;
+
+      if (definition.parameters !== undefined) {
+        Object.keys(definition.parameters).forEach((key) => {
+          if (multiValueParameters.has(key)) {
+            if (definition.parameters[key] === undefined) {
+              return;
+            }
+            Object.keys(definition.parameters[key]).forEach((subKey) => {
+              defaultMultiValues[multiValueParameters.get(key) + subKey] = definition.parameters[key][subKey];
+            });
+            delete filteredParameters[key];
+          }
+        });
+      }
+      return Object.assign({}, definition, { parameters: { ...filteredParameters, ...defaultMultiValues } });
+    }
+    return definition;
+  }
+
   /**
    * If the processor is a `from` or `to` processor, we need to extract the component name from the uri property
    * and return both the processor name and the underlying component name to build the combined schema
@@ -385,11 +421,42 @@ export class CamelComponentSchemaService {
     if (camelElementLookup.componentName !== undefined) {
       updatedDefinition.parameters = updatedDefinition.parameters ?? {};
       this.applyParametersFromSyntax(camelElementLookup.componentName, updatedDefinition);
+      this.readMultiValue(camelElementLookup.componentName, updatedDefinition);
     }
 
     return updatedDefinition;
   }
 
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  private static readMultiValue(componentName: string, definition: any) {
+    const catalogLookup = CamelCatalogService.getCatalogLookup(componentName);
+
+    const multiValueParameters: Map<string, string> = new Map<string, string>();
+    if (catalogLookup !== undefined && catalogLookup.definition?.properties !== undefined) {
+      Object.entries(catalogLookup.definition.properties).forEach(([key, value]) => {
+        if (value.multiValue) multiValueParameters.set(key, value.prefix!);
+      });
+    }
+    if (multiValueParameters.size > 0) {
+      // eslint-disable-next-line @typescript-eslint/no-explicit-any
+      const parameters: any = {};
+      const filteredParameters = definition.parameters;
+
+      for (const [key, value] of multiValueParameters) {
+        const nestParameters: ParsedParameters = {};
+
+        Object.entries(definition.parameters).forEach(([paramKey, paramValue]) => {
+          if (paramKey.startsWith(value)) {
+            nestParameters[paramKey.replace(value, '')] = paramValue as string;
+            delete filteredParameters[paramKey];
+          }
+          parameters[key] = { ...nestParameters };
+        });
+      }
+      Object.assign(definition, { parameters: { ...filteredParameters, ...parameters } });
+    }
+  }
+
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
   private static applyParametersFromSyntax(componentName: string, definition: any) {
     const catalogLookup = CamelCatalogService.getCatalogLookup(componentName);