org.jaxws.wsdl2bytecodes.service.Wsdl2ByteCodes Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of wsdl2html Show documentation
A java tool to convert wsdl xml to readable doc
There is a newer version: 4.2.2
package org.jaxws.wsdl2bytecodes.service;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Date;
import java.util.List;

import javax.tools.Diagnostic;
import javax.tools.DiagnosticCollector;
import javax.tools.JavaCompiler;
import javax.tools.JavaFileObject;
import javax.tools.StandardJavaFileManager;
import javax.tools.ToolProvider;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.StringUtils;
import org.apache.commons.lang.math.RandomUtils;
import org.jaxws.util.lang.RandomStringUtils;
import org.jaxws.util.os.SystemProcessException;
import org.jaxws.util.os.SystemProcessUtils;
import org.jaxws.wsdl2bytecodes.model.ByteCodePackage;

/**
 * 
 * @author chenjianjx
 * 
 */
public class Wsdl2ByteCodes {

	public static ByteCodePackage generate(String byteCodesDirParent, String wsdlUrl) throws WsdlImportException {
		String currentTime = formatDate(new Date(), "yyyyMMddHHmmssSSS");
		File byteCodeDir = createByteCodesDir(byteCodesDirParent, currentTime);
		String packageName = generatePakcageName(currentTime);
		byteCodeDir.mkdirs();
		doWsImport(byteCodeDir.getAbsolutePath(), wsdlUrl, packageName);
		doCompile(byteCodeDir);
		System.out.println("Java files generated at: " + byteCodeDir);
		return new ByteCodePackage(byteCodeDir, packageName);
	}

	private static void doCompile(File sourceDir) throws WsdlImportException {
		System.out.println("Compiling stubs");
		JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
		DiagnosticCollector diagnostics = new DiagnosticCollector();
		StandardJavaFileManager fileManager = compiler.getStandardFileManager(diagnostics, null, null);
		Collection files = FileUtils.listFiles(sourceDir, new String[] { "java" }, true);
		// there is one file that you may encounter compliation errors
		removeWebServiceClientFile(files);
		Iterable compilationUnits = fileManager.getJavaFileObjectsFromFiles(files);
		List compilerOptions = Arrays.asList("-source", "1.6", "-target", "1.6");
		JavaCompiler.CompilationTask task = compiler.getTask(null, fileManager, diagnostics, compilerOptions, null, compilationUnits);
		boolean success = task.call();
		try {
			fileManager.close();
		} catch (IOException e) {
			// ignore it
		}
		if (!success) {
			StringBuffer errors = new StringBuffer();
			for (Diagnostic diagnostic : diagnostics.getDiagnostics()) {
				errors.append(String.format("Compilation Error on line %d in %s", diagnostic.getLineNumber(), diagnostic)).append("\n");
			}
			throw new WsdlImportException(errors.toString());
		}
	}

	private static void removeWebServiceClientFile(Collection files) {
		File theFile = null;
		for (File file : files) {
			String content = null;
			try {
				content = FileUtils.readFileToString(file);
			} catch (IOException e) {
				throw new IllegalStateException(e);
			}
			if (content.contains("@WebServiceClient")) {
				theFile = file;
				break;
			}
		}
		if (theFile != null) {
			files.remove(theFile);
		}
	}

	private static File createByteCodesDir(String byteCodeDirParent, String currentTime) {
		String nextDir = RandomStringUtils.getRandomLetters(10);
		return new File(byteCodeDirParent + "/" + currentTime + "/" + nextDir);
	}

	static String generatePakcageName(String currentTime) {

		List fragments = new ArrayList();
		fragments.add("wsdl2bytecodes" + currentTime); // first level
		for (int i = 0; i < 2; i++) { // the next 9 levels
			// each level may have 3 to 13 words
			fragments.add(RandomStringUtils.getRandomLetters(3 + RandomUtils.nextInt(10)));
		}
		return StringUtils.join(fragments, ".");
	}

	private static void doWsImport(String outputDir, String wsdlUrl, String packageName) throws WsdlImportException {

		File jaxbFile = copyDefaultJaxbFile(outputDir);

		List cmdList = new ArrayList();
		if (isWindows()) {
			cmdList.add("cmd.exe");
			cmdList.add("/c");
		}

		cmdList.add("wsimport");
		cmdList.add("-B-XautoNameResolution");
		cmdList.add("-b");
		cmdList.add(jaxbFile.getAbsolutePath());
		cmdList.add("-s");
		cmdList.add(outputDir);
		cmdList.add("-d");
		cmdList.add(outputDir);
		// We will do the compilation manually, otherwise there may be problems
		// if there are multiple JDKS installed. wsimport doesn't allow you to
		// pass jdk-version option. JAX-WS sucks.
		cmdList.add("-Xnocompile");
		if (packageName != null) {
			cmdList.add("-p");
			cmdList.add(packageName);
		}
		cmdList.add("-verbose");
		cmdList.add("-extension");
		cmdList.add(wsdlUrl);

		String[] cmdArray = new String[cmdList.size()];
		cmdList.toArray(cmdArray);

		System.out.println(StringUtils.join(cmdArray, " "));

		try {
			String consoleOutput = SystemProcessUtils.exec(cmdArray);
			if (consoleOutput.contains("Two declarations cause a collision in the ObjectFactory class")) {
				throw new DeclarationCollisionException(consoleOutput);
			}

		} catch (SystemProcessException e) {
			throw new WsdlImportException(e.getConsoleOutput());
		}

	}

	private static File copyDefaultJaxbFile(String outputDir) {

		File generatedJaxbFile = new File(outputDir, "jaxb.xml");
		InputStream inputStream = null;
		FileOutputStream outputStream = null;
		try {
			String defaultJaxbClasspath = "/default-jaxb.xml";
			inputStream = Wsdl2ByteCodes.class.getResourceAsStream(defaultJaxbClasspath);
			outputStream = new FileOutputStream(generatedJaxbFile);
			IOUtils.copy(inputStream, outputStream);
		} catch (IOException e) {
			throw new IllegalStateException(e);
		} finally {
			IOUtils.closeQuietly(inputStream);
			IOUtils.closeQuietly(outputStream);
		}

		return generatedJaxbFile;
	}

	private static boolean isWindows() {
		return System.getProperty("os.name").toLowerCase().contains("win");
	}

	static String formatDate(Date date, String pattern) {
		SimpleDateFormat format = new SimpleDateFormat(pattern);
		return format.format(date);
	}

	public static final class DeclarationCollisionException extends WsdlImportException {
		public DeclarationCollisionException(String readableReason) {
			super(readableReason);
		}

		private static final long serialVersionUID = 7625306326132684237L;
	}
}