1 files changed, 98 insertions, 0 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
new file mode 100644
index 0000000000..7adde00b29
--- /dev/null
+++ b/collaboration/GSoC-2011-Nirmal/CompositeDiagramShellPlugin/src/main/java/org/apache/tuscany/sca/diagram/shell/DrawShellCommand.java
@@ -0,0 +1,98 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ * 
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.    
+ */
+
+package org.apache.tuscany.sca.diagram.shell;
+
+import java.io.ByteArrayInputStream;
+import java.io.File;
+
+import javax.xml.parsers.DocumentBuilder;
+import javax.xml.parsers.DocumentBuilderFactory;
+
+import jline.Completor;
+import jline.NullCompletor;
+
+import org.apache.tuscany.sca.assembly.Composite;
+import org.apache.tuscany.sca.assembly.xml.Utils;
+import org.apache.tuscany.sca.impl.NodeImpl;
+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.shell.Command;
+import org.apache.tuscany.sca.shell.Shell;
+import org.apache.tuscany.sca.shell.jline.CompositeURICompletor;
+import org.apache.tuscany.sca.shell.jline.ICURICompletor;
+import org.w3c.dom.Document;
+
+public class DrawShellCommand implements Command {
+
+    private Shell shell;
+    
+    public DrawShellCommand(Shell shell) {
+        this.shell = shell;
+    }
+
+    public String getName() {
+        return "draw";
+    }
+
+    public String getShortHelp() {
+        return "draw <contributionURI> <compositeURI>";
+    }
+
+    public String getHelp() {
+        StringBuilder helpText = new StringBuilder();
+        helpText.append("Draws a picture of the composite.\n");
+        helpText.append("\n");
+        helpText.append("Arguments:\n");
+        helpText.append("   contributionURI - (required) the URI of the installed contribution\n");
+        helpText.append("   compositeURL    - (required) the URI of a started composite\n");
+        return helpText.toString();
+    }
+
+    public boolean invoke(String[] args) throws Exception {
+
+        NodeImpl node = (NodeImpl)shell.getNode();
+        Composite composite = node.getContribution(args[0]).getArtifactModel(args[1]);
+        
+        //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);
+        CompositeEntity comp = eb.buildCompositeEntity();
+        DiagramGenerator dg = new DiagramGenerator(comp);
+        Document svg = dg.buildSVGDocument();
+        File outFile = File.createTempFile("tmp", "svg");
+        XMLWriter writer = new XMLWriter();
+        writer.fileWriter(svg, outFile);
+
+        new SVGViewer(outFile.toURI().toURL().toString());
+
+        return true;
+    }
+
+    public Completor[] getCompletors() {
+        return new Completor[]{new ICURICompletor(shell), new CompositeURICompletor(shell), new NullCompletor()};
+    }
+
+}