Add package versions in STUtils

PlayPen · Apr 23, 2017 · b889480 · b889480
1 parent 4d7a266
commit b889480
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/main/java/io/playpen/core/utils/STUtils.java b/src/main/java/io/playpen/core/utils/STUtils.java
@@ -4,8 +4,10 @@
 import io.playpen.core.p3.P3Package;
 import io.playpen.core.p3.PackageContext;
 import org.stringtemplate.v4.ST;
+import org.stringtemplate.v4.misc.Aggregate;
 
 import java.nio.file.Paths;
+import java.util.HashMap;
 import java.util.Map;
 
 public class STUtils {
@@ -20,6 +22,14 @@ public static void buildSTProperties(P3Package p3, PackageContext ctx, ST templa
         for(Map.Entry<String, String> entry : ctx.getProperties().entrySet()) {
             template.add(entry.getKey(), entry.getValue());
         }
+
+        Map<String, String> versions = new HashMap<>();
+        for (P3Package p3Package : ctx.getDependencyChain()) {
+            versions.put(p3Package.getId(), p3Package.getVersion());
+        }
+        versions.put(p3.getId(), p3.getVersion());
+
+        template.add("package_versions", versions);
     }
 
     private STUtils() {}