Apply user template to included IDL files (#472) (#474)

* Refs #23150: Apply user template to included IDL files

* Refs #23150: Apply Review

---------

Signed-off-by: cferreiragonz <[email protected]>
(cherry picked from commit fbb0672)

# Conflicts:
#	src/main/java/com/eprosima/fastdds/fastddsgen.java

Author: cferreiragonz

src/main/java/com/eprosima/fastdds/fastddsgen.java

@@ -580,9 +580,50 @@ private boolean isIncludePathDuplicated(String pathToCheck)
         {
             // path operations failed, just returning false
         }
+<<<<<<< HEAD
 
         return false;
      }
+=======
+        return false;
+    }
+
+    /*
+     * ----------------------------------------------------------------------------------------
+     * Arguments
+     */
+    private static final String case_sensitive_arg = "-cs";
+    private static final String output_path_arg = "-d";
+    private static final String default_container_prealloc_size = "-default-container-prealloc-size";
+    private static final String default_extensibility_arg = "-default_extensibility";
+    private static final String default_extensibility_short_arg = "-de";
+    private static final String specific_platform_arg = "-example";
+    private static final String extra_template_arg = "-extrastg";
+    private static final String flat_output_directory_arg = "-flat-output-dir";
+    private static final String fusion_arg = "-fusion";
+    private static final String help_arg = "-help";
+    private static final String include_path_arg = "-I";
+    private static final String language_arg = "-language";
+    private static final String no_typesupport_arg = "-no-typesupport";
+    private static final String no_typeobjectsupport_arg = "-no-typeobjectsupport";
+    private static final String no_dependencies_arg = "-no-dependencies";
+    private static final String package_arg = "-package";
+    private static final String disable_preprocessor_arg = "-ppDisable";
+    private static final String preprocessor_path_arg = "-ppPath";
+    private static final String python_bindings_arg = "-python";
+    private static final String replace_arg = "-replace";
+    private static final String temp_dir_arg = "-t";
+    private static final String ros2_names_arg = "-typeros2";
+    private static final String cnames_arg = "-typesc";
+    private static final String version_arg = "-version";
+
+    /*
+     * ----------------------------------------------------------------------------------------
+     * Developer Arguments
+     */
+    private static final String generate_api_arg = "-genapi";
+    private static final String execute_test_arg = "-test";
+>>>>>>> fbb0672 (Apply user template to included IDL files (#472) (#474))
 
     public static void printHelp()
     {
@@ -783,9 +824,26 @@ private Project parseIDL(
                 for (Map.Entry<String, String> entry : m_customStgOutput.entrySet())
                 {
                     System.out.println("Loading custom template " + entry.getKey() + "...");
+<<<<<<< HEAD
                     Path path = Paths.get(entry.getKey());
                     String templateName = path.getFileName().toString().substring(0, path.getFileName().toString().lastIndexOf('.'));
                     tmanager.addGroup(templateName);
+=======
+                    loadAndAddTemplate(entry.getKey(), tmanager);
+                }
+            }
+            else
+            {
+                // Check if there is a '@' in the output_file_name
+                for (Map.Entry<String, String> entry : m_customStgOutput.entrySet())
+                {
+                    if (entry.getValue().contains("@"))
+                    {
+                        System.out.println("Loading custom template " +
+                                entry.getKey() + " for included IDL file " + idlFilename + "...");
+                        loadAndAddTemplate(entry.getKey(), tmanager);
+                    }
+>>>>>>> fbb0672 (Apply user template to included IDL files (#472) (#474))
                 }
             }
 
@@ -828,6 +886,7 @@ private Project parseIDL(
                 {
                     for (Map.Entry<String, String> entry : m_customStgOutput.entrySet())
                     {
+<<<<<<< HEAD
                         Path path = Paths.get(entry.getKey());
                         String templateName = path.getFileName().toString().substring(0, path.getFileName().toString().lastIndexOf('.'));
                         System.out.println("Generating from custom " + templateName + " to " + entry.getValue());
@@ -845,11 +904,32 @@ private Project parseIDL(
                             }
                         }
                         else
+=======
+                        if (! (returnedValue = createOutputCustomTemplate(
+                                    entry, idlFilename, output_dir, relative_dir, ctx.getFilename(),
+                                    maintemplates, m_replace, project)))
+>>>>>>> fbb0672 (Apply user template to included IDL files (#472) (#474))
                         {
                             break;
                         }
                     }
                 }
+                else
+                {
+                    // Check if there is a '$' in the output_file_name
+                    for (Map.Entry<String, String> entry : m_customStgOutput.entrySet())
+                    {
+                        if (entry.getValue().contains("@"))
+                        {
+                            if (! (returnedValue = createOutputCustomTemplate(
+                                        entry, idlFilename, output_dir, relative_dir, ctx.getFilename(),
+                                        maintemplates, m_replace, project)))
+                            {
+                                break;
+                            }
+                        }
+                    }
+                }
 
                 System.out.println("Generating Type definition files...");
                 if ((returnedValue) && (returnedValue =
@@ -1048,6 +1128,55 @@ private Project parseIDL(
         return returnedValue ? project : null;
     }
 
+    private void loadAndAddTemplate(
+            String templatePath,
+            TemplateManager tmanager)
+    {
+        try
+        {
+            Path path = Paths.get(templatePath);
+            String templateName = path.getFileName().toString();
+            templateName = templateName.substring(0, templateName.lastIndexOf('.'));
+            String content = new String(Files.readAllBytes(path));
+            tmanager.addGroupFromString(templateName, content);
+        }
+        catch (IOException e)
+        {
+            System.out.println(ColorMessage.error("IOException") + "Cannot read content from " + templatePath);
+        }
+    }
+
+    private boolean createOutputCustomTemplate(
+            Map.Entry<String, String> entry,
+            String idlFilename,
+            String outputDir,
+            String relativeDir,
+            String contextFilename,
+            TemplateGroup maintemplates,
+            boolean replace,
+            Project project)
+    {
+        Path path = Paths.get(entry.getKey());
+        String templateName = path.getFileName().toString();
+        templateName = templateName.substring(0, templateName.lastIndexOf('.'));
+        String outputName = entry.getValue().replace("@", idlFilename.substring(0, idlFilename.lastIndexOf('.')));
+        System.out.println("Generating from custom " + templateName + " to " + outputName);
+
+        boolean ret_val = Utils.writeFile(outputDir + outputName, maintemplates.getTemplate(templateName), replace);
+        if (ret_val)
+        {
+            if (outputName.contains(".hpp") || outputName.contains(".h"))
+            {
+                project.addCommonIncludeFile(relativeDir + outputName);
+            }
+            else
+            {
+                project.addCommonSrcFile(relativeDir + contextFilename + outputName);
+            }
+        }
+        return ret_val;
+    }
+
     private boolean genSolution(
             Solution solution)
     {