From d8763a2ce57075d8bce3a490b3921ff67bd6c58d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Farias?= <jpfarias9@gmail.com>
Date: Sat, 11 Mar 2023 00:10:54 +0000
Subject: [PATCH] Implemented Set Local Variable block

---
 Assets/Content/XML UI/Blocks.xml              | 12 +++++
 Assets/Content/XML UI/Blocks.xml.meta         |  7 +++
 .../Content/XML UI/UIResourceDatabase.asset   |  2 +
 Assets/Scripts/Blocks.meta                    |  8 ++++
 .../Blocks/SetLocalVariableInstruction.cs     | 32 ++++++++++++++
 .../SetLocalVariableInstruction.cs.meta       | 11 +++++
 .../Patches/ProgramSerializerPatches.cs       | 30 ++++++++++++-
 .../Patches/VariableElementScriptPatches.cs   | 44 +++++++++++++++++++
 .../VariableElementScriptPatches.cs.meta      | 11 +++++
 Assets/Scripts/Patches/VizzyToolboxPatches.cs | 41 +++++++++++++++++
 .../Patches/VizzyToolboxPatches.cs.meta       | 11 +++++
 11 files changed, 208 insertions(+), 1 deletion(-)
 create mode 100644 Assets/Content/XML UI/Blocks.xml
 create mode 100644 Assets/Content/XML UI/Blocks.xml.meta
 create mode 100644 Assets/Scripts/Blocks.meta
 create mode 100644 Assets/Scripts/Blocks/SetLocalVariableInstruction.cs
 create mode 100644 Assets/Scripts/Blocks/SetLocalVariableInstruction.cs.meta
 create mode 100644 Assets/Scripts/Patches/VariableElementScriptPatches.cs
 create mode 100644 Assets/Scripts/Patches/VariableElementScriptPatches.cs.meta
 create mode 100644 Assets/Scripts/Patches/VizzyToolboxPatches.cs
 create mode 100644 Assets/Scripts/Patches/VizzyToolboxPatches.cs.meta

diff --git a/Assets/Content/XML UI/Blocks.xml b/Assets/Content/XML UI/Blocks.xml
new file mode 100644
index 0000000..cebb1f5
--- /dev/null
+++ b/Assets/Content/XML UI/Blocks.xml	
@@ -0,0 +1,12 @@
+<XmlLayout>
+  <Styles>
+    <Style id="set-local-variable" color="Instruction" format="set local variable |var| to (0)" tooltip="Creates a local variable with the given value." />
+  </Styles>
+  <Categories>
+    <Category name="Variables">
+      <SetLocalVariable style="set-local-variable">
+        <Constant number="0" />
+      </SetLocalVariable>
+    </Category>
+  </Categories>
+</XmlLayout>
\ No newline at end of file
diff --git a/Assets/Content/XML UI/Blocks.xml.meta b/Assets/Content/XML UI/Blocks.xml.meta
new file mode 100644
index 0000000..833e820
--- /dev/null
+++ b/Assets/Content/XML UI/Blocks.xml.meta	
@@ -0,0 +1,7 @@
+fileFormatVersion: 2
+guid: 9f14c504674ecc9459d72bd76a860351
+TextScriptImporter:
+  externalObjects: {}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 
diff --git a/Assets/Content/XML UI/UIResourceDatabase.asset b/Assets/Content/XML UI/UIResourceDatabase.asset
index 0619339..044b22e 100644
--- a/Assets/Content/XML UI/UIResourceDatabase.asset	
+++ b/Assets/Content/XML UI/UIResourceDatabase.asset	
@@ -18,6 +18,8 @@ MonoBehaviour:
   - Content/XML UI
   AutomaticallyRemoveEntries: 1
   entries:
+  - path: Vizzy Studio/Blocks
+    resource: {fileID: 4900000, guid: 9f14c504674ecc9459d72bd76a860351, type: 3}
   - path: Vizzy Studio/UIResourceDatabase
     resource: {fileID: 11400000}
   - path: Vizzy Studio/VizzyFileExplorer
diff --git a/Assets/Scripts/Blocks.meta b/Assets/Scripts/Blocks.meta
new file mode 100644
index 0000000..c16cd25
--- /dev/null
+++ b/Assets/Scripts/Blocks.meta
@@ -0,0 +1,8 @@
+fileFormatVersion: 2
+guid: 092aa03563e46fa4e85d2e9ed99abdf7
+folderAsset: yes
+DefaultImporter:
+  externalObjects: {}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 
diff --git a/Assets/Scripts/Blocks/SetLocalVariableInstruction.cs b/Assets/Scripts/Blocks/SetLocalVariableInstruction.cs
new file mode 100644
index 0000000..7d80ee3
--- /dev/null
+++ b/Assets/Scripts/Blocks/SetLocalVariableInstruction.cs
@@ -0,0 +1,32 @@
+using HarmonyLib;
+using ModApi.Craft.Program;
+using ModApi.Craft.Program.Instructions;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Assets.Scripts.Blocks
+{
+    [Serializable]
+    public class SetLocalVariableInstruction : ProgramInstruction
+    {
+        [ProgramNodeProperty]
+        private string _variable = "var";
+
+        public string VariableName
+        {
+            get => _variable;
+            set => _variable = value;
+        }
+
+        public override ProgramInstruction Execute(IThreadContext context)
+        {
+            ExpressionResult variableValue = GetExpression(0).Evaluate(context);
+            context.CreateLocalVariable(VariableName).Value.Set(variableValue);
+
+            return base.Execute(context);
+        }
+    }
+}
diff --git a/Assets/Scripts/Blocks/SetLocalVariableInstruction.cs.meta b/Assets/Scripts/Blocks/SetLocalVariableInstruction.cs.meta
new file mode 100644
index 0000000..5cc843b
--- /dev/null
+++ b/Assets/Scripts/Blocks/SetLocalVariableInstruction.cs.meta
@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: e0a12dc37dbeb19488b03b190dee20a2
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 
diff --git a/Assets/Scripts/Patches/ProgramSerializerPatches.cs b/Assets/Scripts/Patches/ProgramSerializerPatches.cs
index 568194d..06b9512 100644
--- a/Assets/Scripts/Patches/ProgramSerializerPatches.cs
+++ b/Assets/Scripts/Patches/ProgramSerializerPatches.cs
@@ -1,6 +1,9 @@
-using HarmonyLib;
+using Assets.Scripts.Blocks;
+using HarmonyLib;
 using ModApi.Craft.Program;
 using ModApi.Craft.Program.Instructions;
+using System;
+using System.Reflection;
 using System.Xml.Linq;
 using UnityEngine;
 
@@ -92,5 +95,30 @@ private static void OnSerializedFlightProgram(ProgramNode node, XElement parentE
                 __result.SetAttributeValue("module", module.Name);
             }
         }
+
+        private static bool _initializedBlocks = false;
+        [HarmonyPrefix]
+        [HarmonyPatch(nameof(ProgramSerializer.CreateProgramNode))]
+        private static void OnCreateProgramNode()
+        {
+            if (_initializedBlocks)
+                return;
+
+            _initializedBlocks = true;
+
+            ConstructorInfo programNodeCreatorConstructor = AccessTools.FirstConstructor(AccessTools.TypeByName("ProgramSerializer.ProgramNodeCreator"), c => true);
+            ConstructorInfo programNodeCreatorConstructor2 = AccessTools.FirstConstructor(AccessTools.TypeByName("ProgramNodeCreator"), c => true);
+
+            object typeNameLookup = AccessTools.Field(typeof(ProgramSerializer), "_typeNameLookup").GetValue(null);
+            object xmlNameLookup = AccessTools.Field(typeof(ProgramSerializer), "_xmlNameLookup").GetValue(null);
+
+            MethodInfo typeNameLookupAddMethod = AccessTools.Method(typeNameLookup.GetType(), "Add");
+            MethodInfo xmlNameLookupAddMethod = AccessTools.Method(xmlNameLookup.GetType(), "Add");
+
+            object setLocalVariableCreator = programNodeCreatorConstructor2.Invoke(new object[] { "SetLocalVariable", typeof(SetLocalVariableInstruction), (Func<ProgramNode>)(() => new SetLocalVariableInstruction()) });
+
+            typeNameLookupAddMethod.Invoke(typeNameLookup, new object[] { typeof(SetLocalVariableInstruction).Name, setLocalVariableCreator });
+            xmlNameLookupAddMethod.Invoke(xmlNameLookup, new object[] { "SetLocalVariable", setLocalVariableCreator });
+        }
     }
 }
diff --git a/Assets/Scripts/Patches/VariableElementScriptPatches.cs b/Assets/Scripts/Patches/VariableElementScriptPatches.cs
new file mode 100644
index 0000000..e7d47fe
--- /dev/null
+++ b/Assets/Scripts/Patches/VariableElementScriptPatches.cs
@@ -0,0 +1,44 @@
+using Assets.Scripts.Blocks;
+using Assets.Scripts.Vizzy.UI;
+using Assets.Scripts.Vizzy.UI.Elements;
+using HarmonyLib;
+using ModApi.Craft.Program;
+using ModApi.Ui;
+using UnityEngine;
+
+namespace Assets.Scripts.Patches
+{
+    [HarmonyPatch(typeof(VariableElementScript))]
+    internal static class VariableElementScriptPatches
+    {
+        [HarmonyPostfix]
+        [HarmonyPatch("OnPointerClick")]
+        private static void OnPointerClick(VariableElementScript __instance)
+        {
+            if (__instance.Parent?.Node is SetLocalVariableInstruction setLocalVarInstruction)
+            {
+                VariableExpression variableExpression = AccessTools.Field(typeof(VariableElementScript), "_expression").GetValue(__instance) as VariableExpression;
+                InputDialogScript variableNameInputDialog = VizzyUIController.CreateVariableNameInputDialog(variableExpression.VariableName);
+                variableNameInputDialog.MessageText = "RENAME LOCAL VARIABLE";
+                variableNameInputDialog.OkayClicked += dialog =>
+                {
+                    if (string.IsNullOrWhiteSpace(dialog.InputText))
+                        return;
+
+                    string oldName = variableExpression.VariableName;
+                    variableExpression.VariableName = dialog.InputText;
+                    setLocalVarInstruction.VariableName = dialog.InputText;
+
+                    AccessTools.Method(typeof(VariableElementScript), "RenameLocalVariableInScope").Invoke(__instance, new object[] { setLocalVarInstruction, oldName, setLocalVarInstruction.VariableName });
+
+                    foreach (VariableElementScript varScript in __instance.VizzyUI.ProgramTransform.GetComponentsInChildren<VariableElementScript>())
+                    {
+                        varScript.Parent?.LayoutElement();
+                    }
+
+                    dialog.Close();
+                };
+            }
+        }
+    }
+}
diff --git a/Assets/Scripts/Patches/VariableElementScriptPatches.cs.meta b/Assets/Scripts/Patches/VariableElementScriptPatches.cs.meta
new file mode 100644
index 0000000..4f412c9
--- /dev/null
+++ b/Assets/Scripts/Patches/VariableElementScriptPatches.cs.meta
@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 5dcdf07fd4829dc4fab8f740c78159f8
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 
diff --git a/Assets/Scripts/Patches/VizzyToolboxPatches.cs b/Assets/Scripts/Patches/VizzyToolboxPatches.cs
new file mode 100644
index 0000000..7c7060a
--- /dev/null
+++ b/Assets/Scripts/Patches/VizzyToolboxPatches.cs
@@ -0,0 +1,41 @@
+using Assets.Scripts.Vizzy.UI;
+using HarmonyLib;
+using ModApi.Common.Extensions;
+using System;
+using System.Linq;
+using System.Xml.Linq;
+using UnityEngine;
+
+namespace Assets.Scripts.Patches
+{
+    [HarmonyPatch(typeof(VizzyToolbox))]
+    internal static class VizzyToolboxPatches
+    {
+        [HarmonyPrefix]
+        [HarmonyPatch(MethodType.Constructor)]
+        [HarmonyPatch(new Type[] { typeof(XElement), typeof(bool) })]
+        private static void OnConstructor(XElement xml)
+        {
+            XElement blocksXml = XElement.Parse(Game.Instance.UserInterface.ResourceDatabase.GetResource<TextAsset>("Vizzy Studio/Blocks").text);
+            XElement toolboxStyles = xml.Element("Styles");
+            foreach (XElement style in blocksXml.Element("Styles").Elements())
+            {
+                toolboxStyles.Add(style);
+            }
+
+            XElement toolboxCategories = xml.Element("Categories");
+            foreach (XElement category in blocksXml.Element("Categories").Elements())
+            {
+                XElement toolboxCategory = toolboxCategories.Elements().FirstOrDefault(e => e.GetStringAttribute("name") == category.GetStringAttribute("name"));
+                if (toolboxCategory != null)
+                {
+                    foreach (XElement block in category.Elements())
+                    {
+                        //#TODO: Setting as first is specific to "SetLocalVariable". Add an attribute to specify position
+                        toolboxCategory.AddFirst(block);
+                    }
+                }
+            }
+        }
+    }
+}
diff --git a/Assets/Scripts/Patches/VizzyToolboxPatches.cs.meta b/Assets/Scripts/Patches/VizzyToolboxPatches.cs.meta
new file mode 100644
index 0000000..a7997f9
--- /dev/null
+++ b/Assets/Scripts/Patches/VizzyToolboxPatches.cs.meta
@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 5f701ffd279a7584cbe4ee01922716b0
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: