@izara_project/izara-core-generate-service-code 1.0.54 → 1.0.56

package/src/codeGenerators/app/sls_yaml/FunctionYamlGenerator.js

@@ -3,228 +3,330 @@ import path from 'path';
 import ejs from 'ejs';
 import { fileURLToPath } from 'url';
 import { getFlowsForGeneration } from '../src/generatedCode/Flow/_shared/shared/flowSelection.js';
+import { isRbacFlow } from '../src/generatedCode/Flow/_shared/shared/flowClassifier.js';
 
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
 
 export async function generateFunctionYaml(allSchemas, options) {
-    console.log('  [FunctionYamlGenerator] Generating Function YAMLs...');
-    
-    const slsYamlDir = path.join(options.outputPath, 'app', 'sls_yaml', 'generatedCode', 'source');
-    await fs.mkdir(slsYamlDir, { recursive: true });
-
-    const templates = {
-        sqs: await fs.readFile(path.join(__dirname, 'templates', 'FunctionYaml_Sqs.ejs'), 'utf-8'),
-        inv: await fs.readFile(path.join(__dirname, 'templates', 'FunctionYaml_Inv.ejs'), 'utf-8'),
-        api: await fs.readFile(path.join(__dirname, 'templates', 'FunctionYaml_Api.ejs'), 'utf-8'),
-        wbs: await fs.readFile(path.join(__dirname, 'templates', 'FunctionYaml_Wbs.ejs'), 'utf-8'),
-        eventBridge: await fs.readFile(path.join(__dirname, 'templates', 'FunctionYaml_EventBridge.ejs'), 'utf-8')
-    };
-    const flows = getFlowsForGeneration(allSchemas);
-
-    const CRUD_FLOW_TAGS = ['Create', 'Update', 'Delete', 'Get'];
-    const REL_FLOW_TAGS = [
-        'createRelationship', 'updateRelationship', 'deleteRelationship',
-        'getRelationship', 'changeRelationship', 'moveRelationship'
-    ];
-
-    let perActionYaml = '';
-    let relActionYaml = '';
-    let flowSchemaYaml = '';
-
-    const upperCase = (str) => str.charAt(0).toUpperCase() + str.slice(1);
-
-    // 1. Find all generated Handlers
-    const generatedCodeDir = path.join(options.outputPath, 'app', 'src', 'generatedCode');
-    const handlerFiles = [];
-    async function getHandlerFiles(dir) {
-        try {
-            const dirents = await fs.readdir(dir, { withFileTypes: true });
-            for (const dirent of dirents) {
-                const res = path.resolve(dir, dirent.name);
-                if (dirent.isDirectory()) {
-                    await getHandlerFiles(res);
-                } else if (dirent.name.includes('_Hdr') && dirent.name.endsWith('.js')) {
-                    // Extract relative path from generatedCodeDir
-                    const relativePath = path.relative(generatedCodeDir, res);
-                    // normalize to forward slashes
-                    handlerFiles.push(relativePath.replace(/\\/g, '/'));
-                }
-            }
-        } catch (e) {
-            // directory might not exist if no flows generated
+  console.log('  [FunctionYamlGenerator] Generating Function YAMLs...');
+
+  const slsYamlDir = path.join(
+    options.outputPath,
+    'app',
+    'sls_yaml',
+    'generatedCode',
+    'source'
+  );
+  await fs.mkdir(slsYamlDir, { recursive: true });
+
+  const templates = {
+    sqs: await fs.readFile(
+      path.join(__dirname, 'templates', 'FunctionYaml_Sqs.ejs'),
+      'utf-8'
+    ),
+    inv: await fs.readFile(
+      path.join(__dirname, 'templates', 'FunctionYaml_Inv.ejs'),
+      'utf-8'
+    ),
+    api: await fs.readFile(
+      path.join(__dirname, 'templates', 'FunctionYaml_Api.ejs'),
+      'utf-8'
+    ),
+    wbs: await fs.readFile(
+      path.join(__dirname, 'templates', 'FunctionYaml_Wbs.ejs'),
+      'utf-8'
+    ),
+    eventBridge: await fs.readFile(
+      path.join(__dirname, 'templates', 'FunctionYaml_EventBridge.ejs'),
+      'utf-8'
+    )
+  };
+  const flows = getFlowsForGeneration(allSchemas);
+
+  const CRUD_FLOW_TAGS = ['Create', 'Update', 'Delete', 'Get'];
+  const REL_FLOW_TAGS = [
+    'createRelationship',
+    'updateRelationship',
+    'deleteRelationship',
+    'getRelationship',
+    'changeRelationship',
+    'moveRelationship'
+  ];
+
+  let perActionYaml = '';
+  let relActionYaml = '';
+  let rbacYaml = '';
+  let flowSchemaYaml = '';
+  const customYmls = {};
+
+  const upperCase = str => str.charAt(0).toUpperCase() + str.slice(1);
+
+  // 1. Find all generated Handlers
+  const generatedCodeDir = path.join(
+    options.outputPath,
+    'app',
+    'src',
+    'generatedCode'
+  );
+  const handlerFiles = [];
+  async function getHandlerFiles(dir) {
+    try {
+      const dirents = await fs.readdir(dir, { withFileTypes: true });
+      for (const dirent of dirents) {
+        const res = path.resolve(dir, dirent.name);
+        if (dirent.isDirectory()) {
+          await getHandlerFiles(res);
+        } else if (
+          dirent.name.includes('_Hdr') &&
+          dirent.name.endsWith('.js')
+        ) {
+          // Extract relative path from generatedCodeDir
+          const relativePath = path.relative(generatedCodeDir, res);
+          // normalize to forward slashes
+          handlerFiles.push(relativePath.replace(/\\/g, '/'));
         }
+      }
+    } catch (e) {
+      // directory might not exist if no flows generated
     }
-    await getHandlerFiles(generatedCodeDir);
-
-    // Prepare API routes
-    const crudApiRoutes = allSchemas.objects.map(obj => ({
-        objectName: obj.objectType,
-        upperName: upperCase(obj.objectType)
-    }));
-    const relApiRoutes = allSchemas.relationships.map(rel => {
-        const relName = Object.keys(rel)[0];
-        return {
-            objectName: relName,
-            upperName: upperCase(relName)
-        };
-    });
-
-    for (const handlerFile of handlerFiles) {
-        // e.g. "Endpoints/Create/source/ProcessCreate_HdrSqs.js"
-        const parts = handlerFile.split('/');
-        if (parts.length < 4) continue;
-
-        const subDir = parts[0]; // e.g. FlowEndpoints, FlowSchemas, SystemFlowSchemas
-        if (subDir === 'SystemFlowSchemas') continue;
-
-        const flowTag = parts[1];
-        const fileName = parts[3]; // e.g. ProcessCreate_HdrSqs.js
-        
-        const fileNameWithoutExt = fileName.replace('.js', '');
-        // e.g. ProcessCreate_HdrSqs
-        const splitFileName = fileNameWithoutExt.split('_');
-        const handlerBaseName = splitFileName[0]; // e.g. ProcessCreate or CreateComplete
-        const handlerSuffix = splitFileName[1]; // e.g. HdrSqs, HdrApi, HdrInv
-
-        const isCrud = CRUD_FLOW_TAGS.map(t => t.toLowerCase()).includes(flowTag.toLowerCase());
-        const isRel = REL_FLOW_TAGS.map(t => t.toLowerCase()).includes(flowTag.toLowerCase());
-        const isCustom = !isCrud && !isRel;
-
-        // Role mapping
-        let roleName = '';
-        if (isCrud) {
-            roleName = 'PerActionEndpointRole';
-        } else if (isRel) {
-            roleName = 'RelationshipRole';
-        } else {
-            roleName = `${upperCase(flowTag)}Role`;
-        }
+  }
+  await getHandlerFiles(generatedCodeDir);
 
-        // Function Name Config
-        // e.g. ProcessCreateSqs or CreateCompleteSqs or GetApi
-        let suffixType = handlerSuffix.replace('Hdr', ''); // Sqs, Api, Inv
-        let functionNameConfig = `${handlerBaseName}${suffixType}`;
-        if (isCrud && handlerBaseName === `Process${upperCase(flowTag)}`) {
-            // Legacy matches: CreateSqs, GetApi instead of ProcessCreateSqs
-            functionNameConfig = `${upperCase(flowTag)}${suffixType}`;
-        }
-        if (isRel && handlerBaseName === `Process${upperCase(flowTag)}`) {
-            // Legacy matches: CreateRelSqs
-            functionNameConfig = `${upperCase(flowTag).replace('Relationship', 'Rel')}${suffixType}`;
-        }
-        if (handlerBaseName.endsWith('Complete')) {
-            // e.g. CreateObjectComp or CreateCompleteSqs
-            if (isCrud) {
-                // Legacy: DeleteNodeComp, UpdateNodeComp, CreateObjectComp
-                const crudType = upperCase(flowTag);
-                if (crudType === 'Create') functionNameConfig = `CreateObjectComp`;
-                else functionNameConfig = `${crudType}NodeComp`;
-            } else if (isRel) {
-                // Legacy: UpdateRelComp
-                functionNameConfig = `${upperCase(flowTag).replace('Relationship', 'Rel')}Comp`;
-            } else {
-                functionNameConfig = `${handlerBaseName}${suffixType}`;
-            }
-        }
+  // Prepare API routes
+  const crudApiRoutes = allSchemas.objects.map(obj => ({
+    objectName: obj.objectType,
+    upperName: upperCase(obj.objectType)
+  }));
+  const relApiRoutes = allSchemas.relationships.map(rel => {
+    const relName = Object.keys(rel)[0];
+    return {
+      objectName: relName,
+      upperName: upperCase(relName)
+    };
+  });
 
-        let queueName = '';
-        const flowDefForQueue = flows.find(f => f.flowTag === flowTag);
-        
-        if (isCrud || isRel) {
-            queueName = handlerBaseName.replace('Process', ''); 
-            if (handlerSuffix === 'HdrSqs' || handlerSuffix === 'HdrDsq') {
-                queueName += handlerSuffix.replace('Hdr', '');
-            }
-        } else {
-            // For Custom Flows (e.g. OwnTopicFlowProcess, ProcessExtTopicFlow)
-            // The queue name exactly matches the handler name without the underscore.
-            if (handlerSuffix === 'HdrSqs' || handlerSuffix === 'HdrDsq') {
-                if (handlerBaseName === `Process${upperCase(flowTag)}` && flowDefForQueue && flowDefForQueue.event && flowDefForQueue.event.includes('ownTopic')) {
-                    // ownTopic flows in FlowResourceYamlGenerator omit 'Process' for the queue
-                    queueName = flowTag + handlerSuffix.replace('Hdr', '');
-                } else {
-                    queueName = handlerBaseName + handlerSuffix.replace('Hdr', '');
-                }
-            } else {
-                queueName = handlerBaseName;
-            }
-        }
+  for (const handlerFile of handlerFiles) {
+    // e.g. "Endpoints/Create/source/ProcessCreate_HdrSqs.js"
+    const parts = handlerFile.split('/');
+    if (parts.length < 4) continue;
 
-        // Api routes
-        let apiRoutes = [];
-        if (suffixType === 'Api') {
-            const routesList = isCrud ? crudApiRoutes : (isRel ? relApiRoutes : []);
-            const actionPath = flowTag.toLowerCase().replace('relationship', '');
-            apiRoutes = routesList.map(r => ({
-                path: `/TST/${r.objectName}/${actionPath}`,
-                upperName: r.upperName
-            }));
-        }
+    const subDir = parts[0]; // e.g. FlowObjects, FlowSchemas, SystemFlowSchemas
+    if (subDir === 'SystemFlowSchemas') continue;
 
-        // Handler Path
-        const handlerPath = `src/generatedCode/${subDir}/${flowTag}/source/${fileNameWithoutExt}.main`;
+    const flowTag = parts[1];
+    const fileName = parts[3]; // e.g. ProcessCreate_HdrSqs.js
 
-        let filterPatterns = '';
-        if (handlerBaseName.endsWith('Complete') && (suffixType === 'Sqs' || suffixType === 'Dsq')) {
-            filterPatterns = `        filterPatterns: #**** need to update serverless framework upper v.2.69.1
-          - body: {"MessageAttributes":{"callingFlow":{"Value":["\${self:custom.iz_resourcePrefix}${functionNameConfig}"]}}} # functionName of callingFlow
-          - body: {"MessageAttributes":{"callingFlow":{"Value":[{"exists":false}]}}}`;
-        }
+    const fileNameWithoutExt = fileName.replace('.js', '');
+    // e.g. ProcessCreate_HdrSqs
+    const splitFileName = fileNameWithoutExt.split('_');
+    const handlerBaseName = splitFileName[0]; // e.g. ProcessCreate or CreateComplete
+    const handlerSuffix = splitFileName[1]; // e.g. HdrSqs, HdrApi, HdrInv
 
-        const flowDef = flows.find(f => f.flowTag === flowTag);
-        let schedules = [];
-        if (flowDef && flowDef.schedules) {
-            schedules = flowDef.schedules;
-        }
+    const isCrud = CRUD_FLOW_TAGS.map(t => t.toLowerCase()).includes(
+      flowTag.toLowerCase()
+    );
+    const isRel = REL_FLOW_TAGS.map(t => t.toLowerCase()).includes(
+      flowTag.toLowerCase()
+    );
+    const isRbac =
+      isRbacFlow(flowTag) ||
+      String(flowTag || '')
+        .toLowerCase()
+        .endsWith('rbacflow');
+    const isCustom = !isCrud && !isRel && !isRbac;
 
-        const renderVars = {
-            functionNameConfig,
-            handlerPath,
-            roleName,
-            queueName,
-            apiRoutes,
-            filterPatterns,
-            schedules,
-            flowTag
-        };
-
-        let yamlBlock = '';
-        if (schedules.length > 0 && suffixType === 'Inv') {
-            yamlBlock = ejs.render(templates.eventBridge, renderVars);
-        } else if (suffixType === 'Sqs' || suffixType === 'Dsq') {
-            yamlBlock = ejs.render(templates.sqs, renderVars);
-        } else if (suffixType === 'Inv') {
-            yamlBlock = ejs.render(templates.inv, renderVars);
-        } else if (suffixType === 'Api') {
-            yamlBlock = ejs.render(templates.api, renderVars);
-        } else if (suffixType === 'Wbs') {
-            renderVars.routeName = flowTag;
-            yamlBlock = ejs.render(templates.wbs, renderVars);
-        } else {
-            // Fallback for others
-            yamlBlock = ejs.render(templates.inv, renderVars); // Simple no-events template
-        }
+    // Role mapping
+    let roleName = '';
+  if (isCrud) {
+    roleName = 'FlowObjectsRole';
+} else if (isRbac) {
+  roleName = 'FlowRbacRole';
+} else if (isRel) {
+  roleName = 'FlowRelationshipRole';
+} else {
+      roleName = `${upperCase(flowTag)}Role`;
+    }
+
+    // Function Name Config
+    // e.g. ProcessCreateSqs or CreateCompleteSqs or GetApi
+    let suffixType = handlerSuffix.startsWith('Hdr')
+      ? handlerSuffix
+      : handlerSuffix.replace('Hdr', '');
+    let functionNameConfig = `${handlerBaseName}${suffixType}`;
+    if (isCrud && handlerBaseName === `Process${upperCase(flowTag)}`) {
+      // Legacy matches: CreateSqs, GetApi instead of ProcessCreateSqs
+      functionNameConfig = `${upperCase(flowTag)}${suffixType}`;
+    }
+    if (isRel && handlerBaseName === `Process${upperCase(flowTag)}`) {
+      // Legacy matches: CreateRelSqs
+      functionNameConfig = `${upperCase(flowTag).replace('Relationship', 'Rel')}${suffixType}`;
+    }
+    if (handlerBaseName.endsWith('Complete')) {
+      // e.g. CreateObjectComp or CreateCompleteSqs
+      if (isCrud) {
+        // Legacy: DeleteNodeComp, UpdateNodeComp, CreateObjectComp
+        const crudType = upperCase(flowTag);
+        if (crudType === 'Create') functionNameConfig = `CreateObjectComp`;
+        else functionNameConfig = `${crudType}NodeComp`;
+      } else if (isRel) {
+        // Legacy: UpdateRelComp
+        functionNameConfig = `${upperCase(flowTag).replace('Relationship', 'Rel')}Comp`;
+      } else {
+        functionNameConfig = `${handlerBaseName}${suffixType}`;
+      }
+    }
 
-        if (isCrud) {
-            perActionYaml += yamlBlock + '\n\n';
-        } else if (isRel) {
-            relActionYaml += yamlBlock + '\n\n';
+    let queueName = '';
+    const flowDefForQueue = flows.find(f => f.flowTag === flowTag);
+
+    if (isCrud || isRel) {
+      queueName = handlerBaseName.replace('Process', '');
+      if (handlerSuffix === 'HdrSqs' || handlerSuffix === 'HdrDsq') {
+        queueName += handlerSuffix;
+      }
+    } else {
+      // For Custom Flows (e.g. OwnTopicFlowProcess, ProcessExtTopicFlow)
+      // The queue name exactly matches the handler name without the underscore.
+      if (handlerSuffix === 'HdrSqs' || handlerSuffix === 'HdrDsq') {
+        if (
+          handlerBaseName === `Process${upperCase(flowTag)}` &&
+          flowDefForQueue &&
+          flowDefForQueue.event &&
+          flowDefForQueue.event.includes('ownTopic')
+        ) {
+          // ownTopic flows in FlowResourceYamlGenerator omit 'Process' for the queue
+          queueName = flowTag + handlerSuffix;
         } else {
-            flowSchemaYaml += yamlBlock + '\n\n';
+          queueName = handlerBaseName + handlerSuffix;
         }
+      } else {
+        queueName = handlerBaseName;
+      }
     }
 
-    if (perActionYaml) {
-        await fs.writeFile(path.join(slsYamlDir, 'per-action-function.yml'), perActionYaml, 'utf-8');
+    // Api routes
+    let apiRoutes = [];
+    if (suffixType === 'HdrApi' || suffixType === 'Api') {
+      const routesList = isCrud ? crudApiRoutes : isRel ? relApiRoutes : [];
+      const actionPath = flowTag.toLowerCase().replace('relationship', '');
+      apiRoutes = routesList.map(r => ({
+        path: `/TST/${r.objectName}/${actionPath}`,
+        upperName: r.upperName
+      }));
     }
-    if (relActionYaml) {
-        await fs.writeFile(path.join(slsYamlDir, 'relationship-per-action.yml'), relActionYaml, 'utf-8');
+
+    // Handler Path
+    const handlerPath = `src/generatedCode/${subDir}/${flowTag}/source/${fileNameWithoutExt}.main`;
+
+    let filterPatterns = '';
+    if (
+      handlerBaseName.endsWith('Complete') &&
+      (suffixType === 'HdrSqs' ||
+        suffixType === 'HdrDsq' ||
+        suffixType === 'Sqs' ||
+        suffixType === 'Dsq')
+    ) {
+      filterPatterns = `        filterPatterns: #**** need to update serverless framework upper v.2.69.1
+          - body: {"MessageAttributes":{"callingFlow":{"Value":["\${self:custom.iz_resourcePrefix}${functionNameConfig}"]}}} # functionName of callingFlow
+          - body: {"MessageAttributes":{"callingFlow":{"Value":[{"exists":false}]}}}`;
     }
-    if (flowSchemaYaml) {
-        // Append to flow-schema.yml just in case WebSocket/TriggerCache already created it, or create new.
-        await fs.appendFile(path.join(slsYamlDir, 'flow-schema.yml'), flowSchemaYaml, 'utf-8');
+
+    const flowDef = flows.find(f => f.flowTag === flowTag);
+    let schedules = [];
+    if (flowDef && flowDef.schedules) {
+      schedules = flowDef.schedules;
+    }
+
+    const renderVars = {
+      functionNameConfig,
+      handlerPath,
+      roleName,
+      queueName,
+      apiRoutes,
+      filterPatterns,
+      schedules,
+      flowTag
+    };
+
+    let yamlBlock = '';
+    if (
+      schedules.length > 0 &&
+      (suffixType === 'HdrInv' || suffixType === 'Inv')
+    ) {
+      yamlBlock = ejs.render(templates.eventBridge, renderVars);
+    } else if (
+      suffixType === 'HdrSqs' ||
+      suffixType === 'HdrDsq' ||
+      suffixType === 'Sqs' ||
+      suffixType === 'Dsq'
+    ) {
+      yamlBlock = ejs.render(templates.sqs, renderVars);
+    } else if (suffixType === 'HdrInv' || suffixType === 'Inv') {
+      yamlBlock = ejs.render(templates.inv, renderVars);
+    } else if (suffixType === 'HdrApi' || suffixType === 'Api') {
+      yamlBlock = ejs.render(templates.api, renderVars);
+    } else if (suffixType === 'HdrWbs' || suffixType === 'Wbs') {
+      renderVars.routeName = flowTag;
+      yamlBlock = ejs.render(templates.wbs, renderVars);
+    } else {
+      // Fallback for others
+      yamlBlock = ejs.render(templates.inv, renderVars); // Simple no-events template
     }
 
-    console.log(`  [FunctionYamlGenerator] Wrote YAML functions for CRUD, Relationships, and Custom Flows.`);
+    const customYamlFile =
+      flowDef && flowDef.fileName ? flowDef.fileName : null;
+
+    if (customYamlFile) {
+      if (!customYmls[customYamlFile]) customYmls[customYamlFile] = '';
+      customYmls[customYamlFile] += yamlBlock + '\n\n';
+    } else if (isCrud) {
+      perActionYaml += yamlBlock + '\n\n';
+    } else if (isRel) {
+      relActionYaml += yamlBlock + '\n\n';
+    } else if (isRbac) {
+      rbacYaml += yamlBlock + '\n\n';
+    } else {
+      flowSchemaYaml += yamlBlock + '\n\n';
+    }
+  }
+
+  if (perActionYaml) {
+    await fs.writeFile(
+      path.join(slsYamlDir, 'flow-objects.yml'),
+      perActionYaml,
+      'utf-8'
+    );
+  }
+  if (relActionYaml) {
+    await fs.writeFile(
+      path.join(slsYamlDir, 'flow-relationships.yml'),
+      relActionYaml,
+      'utf-8'
+    );
+  }
+  if (rbacYaml) {
+    await fs.writeFile(
+      path.join(slsYamlDir, 'flow-rbac.yml'),
+      rbacYaml,
+      'utf-8'
+    );
+  }
+  if (flowSchemaYaml) {
+    // Append to flow-schema.yml just in case WebSocket/TriggerCache already created it, or create new.
+    await fs.appendFile(
+      path.join(slsYamlDir, 'flow-schema.yml'),
+      flowSchemaYaml,
+      'utf-8'
+    );
+  }
+
+  for (const [fName, content] of Object.entries(customYmls)) {
+    const fullFileName = fName.endsWith('.yml') ? fName : `${fName}.yml`;
+    await fs.writeFile(path.join(slsYamlDir, fullFileName), content, 'utf-8');
+  }
+
+  console.log(
+    `  [FunctionYamlGenerator] Wrote YAML functions for CRUD, Relationships, and Custom Flows.`
+  );
 }