/*
 * 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.main;

import java.io.File;
import java.io.StringWriter;

import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;

import org.apache.tuscany.sca.diagram.generator.DiagramGenerator;
import org.apache.tuscany.sca.diagram.html.HTMLWrapper;
import org.apache.tuscany.sca.diagram.io.XMLReader;
import org.apache.tuscany.sca.diagram.io.XMLWriter;
import org.apache.tuscany.sca.diagram.layout.CompositeEntity;
import org.apache.tuscany.sca.diagram.layout.EntityBuilder;
import org.w3c.dom.Document;

public class Main {
    private static String outFileDir; //= System.getProperty("user.dir")+"/output/";
    private static boolean isHtml = false;
    private static boolean isSvg = false;

    /**
     * @param args
     * @throws Exception 
     */
    public static void main(String[] args) throws Exception {

        if (args.length == 0) {

            String help =
                "" + "----------Apache Tuscany Composite Diagram Generator----------\n"
                    + "\n"
                    + "Error: You have not specified an input \"Composite XML\" file.\n"
                    + "\n"
                    + "-----------------------------Help-----------------------------\n"
                    + "You have 3 options:\n"
                    + "---1) Default: Specify \"Composite XML\" file paths separated \n"
                    + "---            by a space, with no arguments. This will"
                    + "---            generate SVG and HTML output.\n"
                    + "---     Eg: java -jar tuscany-composite-diagram-0.0.1.jar \n"
                    + "---         //a//aa//a.composite //b//bb//b.composite \n"
                    + "---\n"
                    + "---2) SVG Only: Specify \"Composite XML\" file paths separated\n"
                    + "---            by a space, with '-svg' argument.\n"
                    + "---     Eg: java -jar tuscany-composite-diagram-0.0.1.jar -svg\n"
                    + "---          //a//aa//a.composite //b//bb//b.composite \n"
                    + "---\n"
                    + "---3) HTML Only: Specify \"Composite XML\" file paths separated\n"
                    + "---            by a space, with '-html' argument.\n"
                    + "---     Eg: java -jar tuscany-composite-diagram-0.0.1.jar -html\n"
                    + "---          //a//aa//a.composite //b//bb//b.composite \n"
                    + "---\n";
            System.out.println(help);
        } else {

            if (args[0].equalsIgnoreCase("-svg")) {
                isSvg = true;
                args[0] = null;
            } else if (args[0].equalsIgnoreCase("-html")) {
                isHtml = true;
                args[0] = null;
            } else {
                isSvg = true;
                isHtml = true;
            }

            for (String str : args) {

                if (str != null) {

                    File in = new File(str);
                    outFileDir = in.getParent() + "//";
                    XMLReader reader = new XMLReader();
                    Document doc = reader.parseXMLFile(str);//composite1.xml composite_with_compositeService.xml");
                    EntityBuilder eb = new EntityBuilder(doc);
                    CompositeEntity comp = eb.buildCompositeEntity();

                    DiagramGenerator dg = new DiagramGenerator(comp, isHtml);
                    Document svg = dg.buildSVGDocument();

                    String svgContent = extractSvg(svg);

                    if (isSvg) {
                        writeSVG(svgContent, comp);
                    }

                    if (isHtml) {
                        writeHTML(svgContent, comp);
                    }

                }
            }
        }
    }

    private static void writeHTML(String svg, CompositeEntity comp) throws Exception {

        File htmlFile = new File(outFileDir + comp.getName() + comp.getFileNameSuffix() + ".html");
        HTMLWrapper html = new HTMLWrapper(svg, comp.getName(), htmlFile);
        String content = html.buildHTML();

        System.err.println("--------------HTML Output for " + comp.getName() + "--------------\n");
        System.out.println(content);
        System.err.println("--------------------------------------------------------------\n");

    }

    private static void writeSVG(String svg, CompositeEntity comp) throws Exception {

        String svgFileName = comp.getName() + comp.getFileNameSuffix() + ".svg";
        File svgFile = new File(outFileDir + svgFileName);
        XMLWriter writer = new XMLWriter();
        writer.fileWriter(svg, svgFile);

        System.err.println("--------------SVG Output for " + comp.getName() + "--------------\n");
        System.out.println(svg);
        System.err.println("-------------------------------------------------------------\n");
    }

    private static String extractSvg(Document svg) throws Exception {

        // Set up the output transformer
        TransformerFactory transfac = TransformerFactory.newInstance();
        Transformer trans = transfac.newTransformer();

        // Print the DOM node

        StringWriter sw = new StringWriter();
        StreamResult result = new StreamResult(sw);
        DOMSource source = new DOMSource(svg);
        trans.transform(source, result);
        String svgString = sw.toString();

        return svgString;
    }

}