Set variables to initial values on start

Author: desplesda

Program.cs

@@ -139,6 +139,7 @@ public class JSCompilation
         public Dictionary<string, string> StringTable { get; set; } = new Dictionary<string, string>();
         public List<Diagnostic> Diagnostics { get; set; } = new List<Diagnostic>();
         public byte[] ProgramData { get; set; } = Array.Empty<byte>();
+        public Dictionary<string, object> InitialValues { get; set; } = new Dictionary<string, object>();
     }
 
     /// <summary>
@@ -257,6 +258,20 @@ public Task<JSCompilation> SetProgramSource(string source)
             StringTable = result.StringTable?.ToDictionary(kv => kv.Key, kv => kv.Value.text ?? "") ?? new Dictionary<string, string>(),
             Diagnostics = result.Diagnostics.ToList(),
             ProgramData = result.Program?.ToByteArray() ?? Array.Empty<byte>(),
+            InitialValues = result.Program?.InitialValues.ToDictionary(kv => kv.Key, kv =>
+            {
+                switch (kv.Value.ValueCase)
+                {
+                    case Operand.ValueOneofCase.StringValue:
+                        return (object)kv.Value.StringValue;
+                    case Operand.ValueOneofCase.BoolValue:
+                        return (object)kv.Value.BoolValue;
+                    case Operand.ValueOneofCase.FloatValue:
+                        return (object)kv.Value.FloatValue;
+                    default:
+                        throw new Exception($"Variable {kv.Key} has a value of invalid type {kv.Value.ValueCase}");
+                }
+            }) ?? new()
         });
     }
 

src/playground.ts

@@ -346,6 +346,12 @@ export async function load(script: string) {
       var source = editor.getModel().getValue();
       var compilation = await dialogue.compileSource(source);
 
+      for (const [varName, value] of Object.entries(
+        compilation.initialValues,
+      )) {
+        variableStorage.setValue(varName, value);
+      }
+
       // Display any diagnostics we have
       for (let diagnostic of compilation.diagnostics) {
         let displayPosition = `Line ${diagnostic.range.start.line + 1}`;

src/yarnspinner.ts

@@ -21,6 +21,7 @@ interface CompileResult {
   stringTable: StringTable;
   diagnostics: Diagnostic[];
   programData?: Uint8Array;
+  initialValues: Record<string, number | string | boolean>;
 }
 
 export interface Line {
@@ -220,6 +221,7 @@ class Dialogue implements IDialogue {
           nodes: [],
           diagnostics: [],
           stringTable: {},
+          initialValues: {},
         };
         this.onError(error);
       }