generator-standard: Use Project Mutator instead of modifying map dire…

…ctly
speedment · Feb 13, 2018 · 2c36090 · 2c36090
1 parent 2c565ee
commit 2c36090
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/...src/main/java/com/speedment/generator/standard/lifecycle/GeneratedMetadataTranslator.java b/...src/main/java/com/speedment/generator/standard/lifecycle/GeneratedMetadataTranslator.java
@@ -24,6 +24,7 @@
 import com.speedment.common.json.Json;
 import com.speedment.generator.translator.AbstractJavaClassTranslator;
 import com.speedment.runtime.config.Project;
+import com.speedment.runtime.config.mutator.ProjectMutator;
 import com.speedment.runtime.config.util.DocumentTranscoder;
 import com.speedment.runtime.core.ApplicationMetadata;
 import com.speedment.runtime.core.component.InfoComponent;
@@ -74,9 +75,13 @@ protected Class makeCodeGenModel(File file) {
 
         final Method initializer = Method.of("init", String.class).static_().private_();
 
-        final Project project = getSupport().projectOrThrow();
+        final ProjectMutator<? extends Project> project = getSupport().projectOrThrow().mutator();
-        project.getData().put(Project.SPEEDMENT_VERSION, infoComponent.getEditionAndVersionString());
+        project.setSpeedmentVersion(infoComponent.getEditionAndVersionString());
-        final List<String> lines = Stream.of(DocumentTranscoder.save(project, Json::toJson).split("\\R")).collect(toList());
+
+        final List<String> lines = Stream.of(
+            DocumentTranscoder.save(project.document(), Json::toJson)
+                .split("\\R")).collect(toList());
+
         final List<List<String>> segments = new ArrayList<>();
         List<String> segment = new ArrayList<>();
         segments.add(segment);