Add import logic

Author: sandrofigo

src/Assets/SimpleCodeGenerator/Editor/IntermediateTemplate.cs

@@ -18,6 +18,18 @@ public IntermediateTemplate(IEnumerable<string> template, object data)
             content = new List<string>(template);
         }
 
+        public IntermediateTemplate RenderImports()
+        {
+            var result = new IntermediateTemplate(content, data);
+
+            for (int i = 0; i < result.content.Count; i++)
+            {
+                result.content[i] = InsertImportedTemplates(result.content[i]);
+            }
+
+            return result;
+        }
+        
         public IntermediateTemplate RenderStandaloneValues()
         {
             var result = new IntermediateTemplate(content, data);
@@ -82,6 +94,30 @@ public IntermediateTemplate RenderForLoops()
             return result;
         }
 
+        private static string InsertImportedTemplates(string input)
+        {
+            while (true)
+            {
+                Match importMatch = Regex.Match(input, @"\{\{\s*import\s+(?<templateName>\S+)\s*\}\}");
+
+                if (!importMatch.Success)
+                    break;
+
+                string templateName = importMatch.Groups["templateName"].Value;
+
+                if (Template.TryFindTemplateInAssets(templateName, out Template template))
+                {
+                    input = input.Replace(importMatch.Value, template.Render(new { }));
+                }
+                else
+                {
+                    input = input.Replace(importMatch.Value, "");
+                }
+            }
+
+            return input;
+        }
+        
         private static string InsertValuesFromObject(string input, object objectToInsert, string propertyRootPath = "")
         {
             while (true)

src/Assets/SimpleCodeGenerator/Editor/Template.cs

@@ -19,6 +19,16 @@ public static Template ParseFromFile(string path)
             return new Template(File.ReadAllLines(path));
         }
 
+        public static Template ParseFromLines(string[] lines)
+        {
+            return new Template(lines);
+        }
+
+        public static Template Parse(string contentWithLineBreaks)
+        {
+            return ParseFromLines(contentWithLineBreaks.Split(new[] { "\r\n", "\r", "\n" }, StringSplitOptions.None));
+        }
+
         public string Render(object data)
         {
             return new IntermediateTemplate(content, data)
@@ -59,5 +69,10 @@ internal static AbsolutePath GetAbsolutePathToBuiltInTemplate(string templateNam
         {
             return GetPathToBuiltInTemplates() / $"{templateName}.txt";
         }
+
+        public Template Import(string key, Template template)
+        {
+            return this;
+        }
     }
 }

tests/TemplateTests.cs

@@ -85,19 +85,21 @@ public Task Render_RenderValidTemplate2_TemplateIsRenderedCorrectly()
         settings.UseDirectory("Verify");
         return Verify(result, settings);
     }
-    
+
     [Fact]
     public Task Render_TemplateWithImportedTemplate_TemplateIsRenderedCorrectly()
     {
-        Template rootTemplate = Template.ParseFromFile(TemplateFileImport);
+        Template template = Template.ParseFromFile(TemplateFileImport);
+        template.Import("Template_To_Import", Template.ParseFromFile(TemplateFileImportPartial));
+        template.Import("This_Can_Be_Any_Key", Template.Parse("1234567890"));
 
-        var rootData = new
+        var data = new
         {
             Template = "Import",
             SomeNumbers = new[] { 1, 2, 3, 4, 5 }
         };
 
-        string result = rootTemplate.Render(rootData);
+        string result = template.Render(data);
 
         var settings = new VerifySettings();
         settings.UseDirectory("Verify");

tests/Templates/Test_Template_Import.txt

@@ -1,8 +1,8 @@
-{{ import "Test_Template_1" }}
+{{ import "Template_To_Import" }}
 
 // Test Template {{ Template }} with imported template
 
-{{ import "Test_Template_1" }}
+{{ import "Template_To_Import" }}
 {{ import "This_Can_Be_Any_Key" }}
 
 {{ for number in SomeNumbers }}