src/dochelper/ant/AntMain.java

Go to the documentation of this file.

/*
 * Main.java
 *
 * Created on February 3, 2009, 3:33 PM
 *
 * To change this template, choose Tools | Template Manager
 * and open the template in the editor.
 */

package dochelper.ant;

import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.lang.reflect.Method;
import java.net.URL;
import java.net.URLClassLoader;

import javax.xml.transform.Source;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.stream.StreamResult;
import javax.xml.transform.stream.StreamSource;

import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.Task;

import dochelper.Main;
import dochelper.exceptions.InitializationException;

public class AntMain extends Task {
    
    public AntMain() {
    }
    
    String _docdef;

    String _dochelperjar;       
    
    String _temppath = null;    
    
    
    
    public void setDocdef(String path) {
        _docdef = path;
    }     

    public void setTemppath(String path) {
        _temppath = path;
    }        
    
    public void setDochelperjar(String path) {
        _dochelperjar = path;
    }        

    public void init() throws BuildException {
                super.init();
                
                _dochelperjar = System.getProperty("ant.home") + "/lib/dochelper_ant.jar";
                
        try {
            //TODO: proper file checking
            File t = new File(_dochelperjar);
            
            if (!t.exists()) {
                throw new BuildException("The dochelper_ant.jar not installed in $ANT_HOME/lib.");
            }
        }
        catch(Exception e) {
            throw new BuildException("The dochelper_ant.jar not installed in $ANT_HOME/lib.");
        }
        }

    public void execute() {

        try {
            
            //STEP 0: Make sure that the parameters that were passed in are valid
            try {
                //TODO: proper file checking
                new File(_docdef);                
            }
            catch(Exception e) {
                throw new InitializationException(e.getMessage());
            }
            
            if(_temppath == null)
                _temppath = System.getProperty("user.dir");
            
            //STEP 1: get the StyleSheet and the DocDef
            String styleSheet = "xslt/CreateMain.xsl";        
            InputStream in = Main.class.getClassLoader().getResourceAsStream(styleSheet);
            Source xsltSource = new StreamSource(in);

            //Source, load Document Definition.
            File xmlFile = new File(_docdef);
            Source xmlSource = new StreamSource(xmlFile);

            // the factory pattern supports different XSLT processors
            TransformerFactory transFact = TransformerFactory.newInstance();
            Transformer trans = transFact.newTransformer(xsltSource);

            //STEP 2: Create the Temp Java File and compile it : both java ans class should go in temp, then temp deleted
            FileOutputStream fos = new FileOutputStream(_temppath + "/DocDef.java");
            trans.transform(xmlSource, new StreamResult(fos));                
            fos.flush();
            fos.close();
            
            //Compile the generated class            
            int errorCode = com.sun.tools.javac.Main.compile(new String[] {                                                                
                "-cp", _dochelperjar,
                "-d", _temppath,
                _temppath + "/DocDef.java" });

            //STEP 3: Load the class and Execute the Main method
            File classesDir = new File(_temppath);
            
            // Load class "sample.PostmanImpl" with our own classloader.
            URLClassLoader loader1 = new URLClassLoader( new URL[] { classesDir.toURI().toURL() }, Main.class.getClassLoader());
            Class<?> docdef = loader1.loadClass("DocDef");
            
            Class<?> param[] = { String[].class };
            Method docdefMain = docdef.getDeclaredMethod("main", param);
            String params[] = { _docdef, _temppath};
            Object args[] = { params };
            docdefMain.invoke(null, args);
            
            //STEP 4: Delete the java files ?
            
        }
        catch(Exception e) {
            e.printStackTrace();
        }
        
        
    }
    
}

---