Fixed heroku#129 according to the @friism's comment

jgkim · May 10, 2016 · 99bb365 · 99bb365
1 parent add0357
commit 99bb365
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/commands/init.js b/commands/init.js
@@ -140,9 +140,21 @@ function createDockerCompose(procfile, addons, mountDir) {
   }
 
   function addonToService(addon) {
-    return {
-      image: ADDONS[addon].image
-    };
+    var service = {}
+
+    // fallback for backward compatibility
+    if ((typeof ADDONS[addon].image) === 'string') {
+      service.image = ADDONS[addon].image;
+    } else {
+      service.image = ADDONS[addon].image.name;
+
+      // All the other properties except `image` are optional.
+      if ('env' in ADDONS[addon].image) {
+        service.environment = ADDONS[addon].image.env;
+      }
+    }
+
+    return service;
   }
 }