1 files changed, 8 insertions, 5 deletions

diff --git a/collaboration/GSoC-2011-Nirmal/CompositeDiagramShellPlugin/src/main/java/org/apache/tuscany/sca/diagram/shell/DrawShellCommand.java b/collaboration/GSoC-2011-Nirmal/CompositeDiagramShellPlugin/src/main/java/org/apache/tuscany/sca/diagram/shell/DrawShellCommand.java
index 7adde00b29..6fb8d06fd6 100644
--- a/collaboration/GSoC-2011-Nirmal/CompositeDiagramShellPlugin/src/main/java/org/apache/tuscany/sca/diagram/shell/DrawShellCommand.java
+++ b/collaboration/GSoC-2011-Nirmal/CompositeDiagramShellPlugin/src/main/java/org/apache/tuscany/sca/diagram/shell/DrawShellCommand.java
@@ -35,6 +35,7 @@ import org.apache.tuscany.sca.impl.diagram.DiagramGenerator;
 import org.apache.tuscany.sca.impl.io.XMLWriter;

 import org.apache.tuscany.sca.impl.layout.CompositeEntity;

 import org.apache.tuscany.sca.impl.layout.EntityBuilder;

+import org.apache.tuscany.sca.impl.layout.TuscanyCompositeEntityBuilder;

 import org.apache.tuscany.sca.shell.Command;

 import org.apache.tuscany.sca.shell.Shell;

 import org.apache.tuscany.sca.shell.jline.CompositeURICompletor;

@@ -75,14 +76,16 @@ public class DrawShellCommand implements Command {
         //TODO: really all we want is to pass the Tuscany Composite object to the SVG drawing code,

         //      but for now as that requires an XML file to the Composite to XML:

         

-        String compositeXML = Utils.modelToXML(composite, false, node.getExtensionPointRegistry());

-        DocumentBuilder db = DocumentBuilderFactory.newInstance().newDocumentBuilder();

-        Document doc = db.parse(new ByteArrayInputStream(compositeXML.getBytes()));

-        EntityBuilder eb = new EntityBuilder(doc);

+//        String compositeXML = Utils.modelToXML(composite, false, node.getExtensionPointRegistry());

+//        DocumentBuilder db = DocumentBuilderFactory.newInstance().newDocumentBuilder();

+//        Document doc = db.parse(new ByteArrayInputStream(compositeXML.getBytes()));

+//        EntityBuilder eb = new EntityBuilder(doc);

+        

+        TuscanyCompositeEntityBuilder eb = new TuscanyCompositeEntityBuilder(composite);

         CompositeEntity comp = eb.buildCompositeEntity();

         DiagramGenerator dg = new DiagramGenerator(comp);

         Document svg = dg.buildSVGDocument();

-        File outFile = File.createTempFile("tmp", "svg");

+        File outFile = File.createTempFile("tmp", ".svg");

         XMLWriter writer = new XMLWriter();

         writer.fileWriter(svg, outFile);