Add compiletime natives for map name and build date

See MagicFunctions package in the standard library. wurstscript/WurstStdlib2@80ea0d0

Author: Frotty

de.peeeq.wurstscript/src/main/java/de/peeeq/wurstio/CompilationProcess.java

@@ -84,8 +84,7 @@ public CompilationProcess(WurstGui gui, RunArgs runArgs) {
                     () -> runTests(compiler.getImTranslator(), compiler, runArgs.getTestTimeout()));
         }
 
-        timeTaker.measure("Run compiletime functions",
-            compiler::runCompiletime);
+        timeTaker.measure("Run compiletime functions", () ->compiler.runCompiletime(null));
 
         JassProg jassProg = timeTaker.measure("Transform program to Jass",
             compiler::transformProgToJass);

de.peeeq.wurstscript/src/main/java/de/peeeq/wurstio/CompiletimeFunctionRunner.java

@@ -4,6 +4,7 @@
 import com.google.common.collect.ArrayListMultimap;
 import com.google.common.collect.Lists;
 import com.google.common.collect.Maps;
+import config.WurstProjectConfigData;
 import de.peeeq.wurstio.intermediateLang.interpreter.CompiletimeNatives;
 import de.peeeq.wurstio.intermediateLang.interpreter.ProgramStateIO;
 import de.peeeq.wurstio.jassinterpreter.InterpreterException;
@@ -79,15 +80,15 @@ public boolean matches(ImFunction f) {
 
 
     public CompiletimeFunctionRunner(
-            ImTranslator tr, ImProg imProg, Optional<File> mapFile, MpqEditor mpqEditor, WurstGui gui,
-            FunctionFlagToRun flag) {
+        ImTranslator tr, ImProg imProg, Optional<File> mapFile, MpqEditor mpqEditor, WurstGui gui,
+        FunctionFlagToRun flag, WurstProjectConfigData projectConfigData) {
         Preconditions.checkNotNull(imProg);
         this.translator = tr;
         this.imProg = imProg;
         globalState = new ProgramStateIO(mapFile, mpqEditor, gui, imProg, true);
         this.interpreter = new ILInterpreter(imProg, gui, mapFile, globalState);
 
-        interpreter.addNativeProvider(new CompiletimeNatives(globalState));
+        interpreter.addNativeProvider(new CompiletimeNatives(globalState, projectConfigData));
         interpreter.addNativeProvider(new ReflectionNativeProvider(interpreter));
         this.gui = gui;
         this.functionFlag = flag;

de.peeeq.wurstscript/src/main/java/de/peeeq/wurstio/WurstCompilerJassImpl.java

@@ -4,6 +4,7 @@
 import com.google.common.base.Preconditions;
 import com.google.common.collect.*;
 import com.google.common.io.Files;
+import config.WurstProjectConfigData;
 import de.peeeq.wurstio.languageserver.requests.RequestFailedException;
 import de.peeeq.wurstio.map.importer.ImportFile;
 import de.peeeq.wurstio.mpq.MpqEditor;
@@ -99,13 +100,13 @@ public void loadFiles(File... files) {
     }
 
     @Override
-    public void runCompiletime() {
+    public void runCompiletime(WurstProjectConfigData projectConfigData) {
         if (runArgs.runCompiletimeFunctions()) {
             // compile & inject object-editor data
             // TODO run optimizations later?
             gui.sendProgress("Running compiletime functions");
             CompiletimeFunctionRunner ctr = new CompiletimeFunctionRunner(imTranslator, getImProg(), getMapFile(), getMapfileMpqEditor(), gui,
-                    CompiletimeFunctions);
+                    CompiletimeFunctions, projectConfigData);
             ctr.setInjectObjects(runArgs.isInjectObjects());
             ctr.setOutputStream(new PrintStream(System.err));
             ctr.run();

de.peeeq.wurstscript/src/main/java/de/peeeq/wurstio/intermediateLang/interpreter/CompiletimeNatives.java

@@ -1,6 +1,7 @@
 package de.peeeq.wurstio.intermediateLang.interpreter;
 
 
+import config.WurstProjectConfigData;
 import de.peeeq.wurstio.jassinterpreter.InterpreterException;
 import de.peeeq.wurstio.jassinterpreter.ReflectionBasedNativeProvider;
 import de.peeeq.wurstio.objectreader.*;
@@ -12,12 +13,19 @@
 import de.peeeq.wurstscript.intermediatelang.interpreter.ProgramState;
 import de.peeeq.wurstscript.intermediatelang.interpreter.VariableType;
 
+import java.text.SimpleDateFormat;
+import java.time.LocalDateTime;
+import java.time.temporal.ChronoUnit;
+import java.time.temporal.TemporalUnit;
+
 @SuppressWarnings("ucd") // ignore unused code detector warnings, because this class uses reflection
 public class CompiletimeNatives extends ReflectionBasedNativeProvider implements NativesProvider {
     private ProgramStateIO globalState;
+    private WurstProjectConfigData projectConfigData;
 
-    public CompiletimeNatives(ProgramStateIO globalState) {
+    public CompiletimeNatives(ProgramStateIO globalState, WurstProjectConfigData projectConfigData) {
         this.globalState = globalState;
+        this.projectConfigData = projectConfigData;
     }
 
 
@@ -133,4 +141,11 @@ public void compileError(ILconstString msg) {
         throw new InterpreterException(msg.getVal());
     }
 
+    public ILconstString getMapName() {
+        return new ILconstString(projectConfigData.getBuildMapData().getName());
+    }
+
+    public ILconstString getBuildDate() {
+        return new ILconstString(LocalDateTime.now().truncatedTo(ChronoUnit.MINUTES).toString());
+    }
 }

de.peeeq.wurstscript/src/main/java/de/peeeq/wurstio/languageserver/requests/BuildMap.java

@@ -61,7 +61,7 @@ public Object execute(ModelManager modelManager) throws IOException {
             String fileName = projectConfig.getBuildMapData().getFileName();
             Optional<File> targetMap = Optional.of(
                 new File(buildDir, fileName.isEmpty() ? projectConfig.getProjectName() + ".w3x" : fileName + ".w3x"));
-            File compiledScript = compileScript(modelManager, gui, targetMap);
+            File compiledScript = compileScript(modelManager, gui, targetMap, projectConfig);
 
             gui.sendProgress("Applying Map Config...");
             ProjectConfigBuilder.apply(projectConfig, targetMap.get(), compiledScript, buildDir, runArgs, w3data);

de.peeeq.wurstscript/src/main/java/de/peeeq/wurstio/languageserver/requests/MapRequest.java

@@ -2,6 +2,7 @@
 
 import com.google.common.base.Charsets;
 import com.google.common.io.Files;
+import config.WurstProjectConfigData;
 import de.peeeq.wurstio.Pjass;
 import de.peeeq.wurstio.UtilsIO;
 import de.peeeq.wurstio.WurstCompilerJassImpl;
@@ -140,7 +141,8 @@ protected void processMapScript(
         modelManager.syncCompilationUnit(WFile.create(existingScript));
     }
 
-    protected File compileMap(File projectFolder, WurstGui gui, Optional<File> mapCopy, RunArgs runArgs, WurstModel model) {
+    protected File compileMap(File projectFolder, WurstGui gui, Optional<File> mapCopy, RunArgs runArgs, WurstModel model,
+                              WurstProjectConfigData projectConfigData) {
         try (@Nullable MpqEditor mpqEditor = MpqEditorFactory.getEditor(mapCopy)) {
             if (mpqEditor != null && !mpqEditor.canWrite()) {
                 WLogger.severe("The supplied map is invalid/corrupted/protected and Wurst cannot write to it.\n" +
@@ -166,7 +168,7 @@ protected File compileMap(File projectFolder, WurstGui gui, Optional<File> mapCo
             }
 
 
-            compiler.runCompiletime();
+            compiler.runCompiletime(projectConfigData);
 
             if (runArgs.isLua()) {
                 print("translating program to Lua ... ");
@@ -314,7 +316,8 @@ protected void println(String s) {
     }
 
 
-    protected File compileScript(WurstGui gui, ModelManager modelManager, List<String> compileArgs, Optional<File> mapCopy) throws Exception {
+    protected File compileScript(WurstGui gui, ModelManager modelManager, List<String> compileArgs, Optional<File> mapCopy,
+                                 WurstProjectConfigData projectConfigData) throws Exception {
         RunArgs runArgs = new RunArgs(compileArgs);
         print("Compile Script : ");
         for (File dep : modelManager.getDependencyWurstFiles()) {
@@ -345,10 +348,10 @@ protected File compileScript(WurstGui gui, ModelManager modelManager, List<Strin
             model = ModelManager.copy(model);
         }
 
-        return compileMap(modelManager.getProjectPath(), gui, mapCopy, runArgs, model);
+        return compileMap(modelManager.getProjectPath(), gui, mapCopy, runArgs, model, projectConfigData);
     }
 
-    protected File compileScript(ModelManager modelManager, WurstGui gui, Optional<File> testMap) throws Exception {
+    protected File compileScript(ModelManager modelManager, WurstGui gui, Optional<File> testMap, WurstProjectConfigData projectConfigData) throws Exception {
         if (testMap.isPresent() && testMap.get().exists()) {
             boolean deleteOk = testMap.get().delete();
             if (!deleteOk) {
@@ -360,7 +363,7 @@ protected File compileScript(ModelManager modelManager, WurstGui gui, Optional<F
         }
 
         // first compile the script:
-        File compiledScript = compileScript(gui, modelManager, compileArgs, testMap);
+        File compiledScript = compileScript(gui, modelManager, compileArgs, testMap, projectConfigData);
 
         Optional<WurstModel> model = Optional.ofNullable(modelManager.getModel());
         if (!model.isPresent()

de.peeeq.wurstscript/src/main/java/de/peeeq/wurstio/languageserver/requests/RunMap.java

@@ -84,7 +84,7 @@ public Object execute(ModelManager modelManager) throws IOException {
             // first we copy in same location to ensure validity
             File buildDir = getBuildDir();
             Optional<File> testMap = map.map($ -> new File(buildDir, "WurstRunMap.w3x"));
-            File compiledScript = compileScript(modelManager, gui, testMap);
+            File compiledScript = compileScript(modelManager, gui, testMap, projectConfig);
 
             if (runArgs.isHotReload()) {
                 // call jhcr update

de.peeeq.wurstscript/src/main/java/de/peeeq/wurstio/languageserver/requests/RunTests.java

@@ -147,7 +147,7 @@ public TestResult runTests(ImTranslator translator, ImProg imProg, Optional<Func
         WurstGui gui = new TestGui();
 
         CompiletimeFunctionRunner cfr = new CompiletimeFunctionRunner(translator, imProg, Optional.empty(), null, gui,
-            CompiletimeFunctions);
+            CompiletimeFunctions, null);
         ILInterpreter interpreter = cfr.getInterpreter();
         ProgramState globalState = cfr.getGlobalState();
         if (globalState == null) {

de.peeeq.wurstscript/src/main/java/de/peeeq/wurstscript/WurstCompiler.java

@@ -1,5 +1,6 @@
 package de.peeeq.wurstscript;
 
+import config.WurstProjectConfigData;
 import de.peeeq.wurstscript.ast.WurstModel;
 import org.eclipse.jdt.annotation.Nullable;
 
@@ -14,5 +15,5 @@ public interface WurstCompiler {
 
     void loadFiles(File... files);
 
-    void runCompiletime();
+    void runCompiletime(WurstProjectConfigData projectConfigData);
 }