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,23 @@ 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;
+
+                // TODO
+            }
+
+            return input;
+        }
+
         private static string InsertValuesFromObject(string input, object objectToInsert, string propertyRootPath = "")
         {
             while (true)

src/Assets/SimpleCodeGenerator/Editor/Template.cs

@@ -1,4 +1,7 @@
-using System.IO;
+using System;
+using System.Collections.Generic;
+using System.IO;
+using JetBrains.Annotations;
 using SimpleCodeGenerator.Core;
 using UnityEditor;
 using UnityEngine;
@@ -8,20 +11,36 @@ namespace SimpleCodeGenerator.Editor
     public class Template
     {
         private readonly string[] content;
+        private readonly Dictionary<string, Template> importedTemplates = new();
 
         private Template(string[] content)
         {
             this.content = content;
         }
 
+        [PublicAPI]
         public static Template ParseFromFile(string path)
         {
             return new Template(File.ReadAllLines(path));
         }
 
+        [PublicAPI]
+        public static Template ParseFromLines(string[] lines)
+        {
+            return new Template(lines);
+        }
+
+        [PublicAPI]
+        public static Template Parse(string contentWithLineBreaks)
+        {
+            return ParseFromLines(contentWithLineBreaks.Split(new[] { "\r\n", "\r", "\n" }, StringSplitOptions.None));
+        }
+
+        [PublicAPI]
         public string Render(object data)
         {
             return new IntermediateTemplate(content, data)
+                .RenderImports()
                 .RenderStandaloneValues()
                 .RenderForLoops()
                 .Build();
@@ -32,6 +51,7 @@ internal static Template FindBuiltInTemplate(string templateName)
             return ParseFromFile(GetAbsolutePathToBuiltInTemplate(templateName));
         }
 
+        [PublicAPI]
         public static bool TryFindTemplateInAssets(string templateAssetPath, out Template template)
         {
             if (!templateAssetPath.StartsWith("Assets/"))
@@ -59,5 +79,12 @@ internal static AbsolutePath GetAbsolutePathToBuiltInTemplate(string templateNam
         {
             return GetPathToBuiltInTemplates() / $"{templateName}.txt";
         }
+
+        [PublicAPI]
+        public Template Import(string key, Template template)
+        {
+            importedTemplates.Add(key, 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 }}