Sicherung

Kalk/Kalk.Core/KalkModuleWithFunctions.cs

@@ -0,0 +1,126 @@
+using System;
+using Scriban.Runtime;
+
+namespace Kalk.Core
+{
+    public abstract class KalkModuleWithFunctions : KalkModule
+    {
+        protected KalkModuleWithFunctions() : this(null)
+        {
+        }
+
+        protected KalkModuleWithFunctions(string name) : base(name)
+        {
+            Content = new ScriptObject();
+        }
+
+        public ScriptObject Content { get; }
+        
+        protected override void Import()
+        {
+            base.Import();
+
+            // Feed the engine with our new builtins
+            Engine.Builtins.Import(Content);
+
+            DisplayImport();
+        }
+
+        protected virtual void DisplayImport()
+        {
+            if (!IsBuiltin && Content.Count > 0)
+            {
+                Engine.WriteHighlightLine($"# {Content.Count} functions successfully imported from module `{Name}`.");
+            }
+        }
+
+        protected virtual void RegisterFunctionsAuto()
+        {
+        }
+        
+        protected void RegisterConstant(string name, object value, string category = null)
+        {
+            RegisterVariable(name, value, category);
+        }
+        protected void RegisterAction(string name, Action action, string category = null)
+        {
+            RegisterCustomFunction(name, DelegateCustomFunction.Create(action), category);
+        }
+
+        protected void RegisterAction<T1>(string name, Action<T1> action, string category = null)
+        {
+            RegisterCustomFunction(name, DelegateCustomFunction.Create(action), category);
+        }
+
+        protected void RegisterAction<T1, T2>(string name, Action<T1, T2> action, string category = null)
+        {
+            RegisterCustomFunction(name, DelegateCustomFunction.Create(action), category);
+        }
+
+        protected void RegisterAction<T1, T2, T3>(string name, Action<T1, T2, T3> action, string category = null)
+        {
+            RegisterCustomFunction(name, DelegateCustomFunction.Create(action), category);
+        }
+
+        protected void RegisterAction<T1, T2, T3, T4>(string name, Action<T1, T2, T3, T4> action, string category = null)
+        {
+            RegisterCustomFunction(name, DelegateCustomFunction.Create(action), category);
+        }
+
+        protected void RegisterFunction<T1>(string name, Func<T1> func, string category = null)
+        {
+            RegisterCustomFunction(name, DelegateCustomFunction.CreateFunc(func), category);
+        }
+
+        protected void RegisterFunction<T1, T2>(string name, Func<T1, T2> func, string category = null)
+        {
+            RegisterCustomFunction(name, DelegateCustomFunction.CreateFunc(func), category);
+        }
+
+        protected void RegisterFunction<T1, T2, T3>(string name, Func<T1, T2, T3> func, string category = null)
+        {
+            RegisterCustomFunction(name, DelegateCustomFunction.CreateFunc(func), category);
+        }
+
+        protected void RegisterFunction<T1, T2, T3, T4>(string name, Func<T1, T2, T3, T4> func, string category = null)
+        {
+            RegisterCustomFunction(name, DelegateCustomFunction.CreateFunc(func), category);
+        }
+
+        protected void RegisterFunction<T1, T2, T3, T4, T5>(string name, Func<T1, T2, T3, T4, T5> func, string category = null)
+        {
+            RegisterCustomFunction(name, DelegateCustomFunction.CreateFunc(func), category);
+        }
+
+        protected void RegisterFunction<T1, T2, T3, T4, T5, T6>(string name, Func<T1, T2, T3, T4, T5, T6> func, string category = null)
+        {
+            RegisterCustomFunction(name, DelegateCustomFunction.CreateFunc(func), category);
+        }
+
+        protected void RegisterCustomFunction(string name, IScriptCustomFunction func, string category = null)
+        {
+            RegisterVariable(name, func, category);
+        }
+
+        protected void RegisterVariable(string name, object value, string category = null)
+        {
+            if (name == null) throw new ArgumentNullException(nameof(name));
+            if (value == null) throw new ArgumentNullException(nameof(value));
+            var names = name.Split(',');
+
+            KalkDescriptor descriptor = null;
+
+            foreach (var subName in names)
+            {
+                Content.SetValue(subName, value, true);
+
+                if (descriptor == null || !Descriptors.TryGetValue(names[0], out descriptor))
+                {
+                    descriptor = new KalkDescriptor();
+                }
+                Descriptors.Add(subName, descriptor);
+                descriptor.Names.Add(subName);
+            }
+        }
+    }
+}