Port updated

Author: mgiamberardino

controllers/plantuml.js

@@ -7,30 +7,51 @@ var fs = require('fs'),
     request = require('request');
 
 exports.generatePng = function(req, res) {
-    console.log("Trying to generate an UML image.")
+    generate(res,req,false)
+    //request('http://www.plantuml.com/plantuml/img/'+encoded).pipe(res);
+};
 
-    plantuml.useNailgun();
-    entities = new Entities();
-    res.set('Content-Type', 'image/png');
-    var text = req.body.text;
-    console.log("body " + req.body.text);
-    console.log("params " + req.params.text);
-    splited = text.split("```");
-    if(splited.length > 1){
-      code=splited[1];
-      if (code.includes("@startuml")){
-        console.log(entities.decode(code));
-        var encoded = plantumlEncoder.encode(entities.decode(code));
-        console.log('http://www.plantuml.com/plantuml/img/'+encoded);
-        res.status(200).jsonp({text: 'http://www.plantuml.com/plantuml/img/'+encoded})
+exports.generatePng = function(req, res) {
+    generate(res,req,true)
+    //request('http://www.plantuml.com/plantuml/img/'+encoded).pipe(res);
+};
+
+function generate(res, req, command){
+  console.log("Trying to generate an UML image.")
+
+  plantuml.useNailgun();
+  entities = new Entities();
+  res.set('Content-Type', 'image/png');
+  var text = req.body.text;
+  console.log("body " + req.body.text);
+  console.log("params " + req.params.text);
+  splited = text.split("```");
+  if(splited.length > 1){
+    code=splited[1];
+    if (code.includes("@startuml")){
+      console.log(entities.decode(code));
+      var encoded = plantumlEncoder.encode(entities.decode(code));
+      console.log('http://www.plantuml.com/plantuml/img/'+encoded);
+      if (command){
+        res.status(200).jsonp({   text:
+                                  attachments: [
+                                      {
+                                          text: "Optional text that appears within the attachment",
+                                          image_url: "http://www.plantuml.com/plantuml/img/"+encoded,
+                                          footer: "Generated UML"
+                                      }
+                                  ]
+                              });
       } else {
-        res.status(200).jsonp({});
+        res.status(200).jsonp({text: 'http://www.plantuml.com/plantuml/img/'+encoded})
       }
     } else {
       res.status(200).jsonp({});
     }
-    //request('http://www.plantuml.com/plantuml/img/'+encoded).pipe(res);
-};
+  } else {
+    res.status(200).jsonp({});
+  }
+}
 
 exports.generateSvg = function(req, res) {
     plantuml.useNailgun();

routes/all.js

@@ -8,6 +8,9 @@ exports.getRoutes = function(){
   routes.route('/png')
     .post(PlantCtrl.generatePng);
 
+  routes.route('/commands/png')
+    .post(PlantCtrl.generateCommandPng);
+
   routes.route('/svg')
     .post(PlantCtrl.generateSvg);