com.github.dkschlos.wsdlget.internal.WsdlDownloader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of wsdlget-maven-plugin Show documentation

Maven plugin which downloads a wsdl and all referenced schema-filed to a specified target folder (default: src/main/resources/wsdl) and rewrites schema-locations in all files.

The newest version!

package com.github.dkschlos.wsdlget.internal;

import com.github.dkschlos.wsdlget.WsdlDefinition;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.*;
import javax.wsdl.Definition;
import javax.wsdl.Import;
import javax.wsdl.Service;
import javax.wsdl.WSDLException;
import javax.wsdl.factory.WSDLFactory;
import javax.wsdl.xml.WSDLReader;
import javax.wsdl.xml.WSDLWriter;

public class WsdlDownloader {

	private Path baseFolder;
	private int wsdlCount = 0;
	private WsdlDefinition wsdl;

	private Map uriToWsdl = new HashMap<>();

	public WsdlDownloader(Path baseFolder, WsdlDefinition wsdl) {
		this.baseFolder = baseFolder;
		this.wsdl = wsdl;
	}


	public void download() {
		try {
			Definition wsdlDefinition = readWsdl(wsdl.getUrl());

			String serviceName = wsdl.getServiceName() == null
					? extractServiceName(wsdlDefinition)
					: wsdl.getServiceName();

			uriToWsdl.put(wsdlDefinition.getDocumentBaseURI(), serviceName + ".wsdl");
			write(wsdlDefinition, serviceName, serviceName + ".wsdl");
		} catch (Exception e) {
			throw new RuntimeException("WSDL writing failed!", e);
		}
	}

	private String extractServiceName(Definition wsdlDefinition) {
		String serviceName = (String) wsdlDefinition.getServices().values().stream()
				.map(o -> ((Service) o).getQName().getLocalPart())
				.findFirst()
				.orElse("UnknownService");
		return serviceName;
	}

	private Definition readWsdl(String url) throws IllegalArgumentException, WSDLException {
		WSDLReader reader = WSDLFactory.newInstance().newWSDLReader();
		reader.setFeature("javax.wsdl.importDocuments", true);
		Definition wsdlDefinition = reader.readWSDL(url);
		return wsdlDefinition;
	}

	@SuppressWarnings("unchecked")
	private void write(Definition definition, String serviceName, String fileName) throws Exception {
		Map> importMap = definition.getImports();
		List imports = new ArrayList<>();
		for (List value : importMap.values()) {
			imports.addAll(value);
		}

		if (!imports.isEmpty()) {
			for (Import wsdlImport : imports) {
				String wsdlLocation = wsdlImport.getDefinition().getDocumentBaseURI();
				String extension = wsdlLocation.contains("xsd") ? ".xsd" : ".wsdl";
				String fileNamePart = serviceName + wsdlCount++;
				String wsdlName = fileNamePart + extension;
				if (!uriToWsdl.containsKey(wsdlLocation)) {
					uriToWsdl.put(wsdlLocation, wsdlName);
					Definition innerDefinition = wsdlImport.getDefinition();
					write(innerDefinition, serviceName, wsdlName);
				}

				wsdlImport.setLocationURI((String) uriToWsdl.get(wsdlLocation));
			}

		}
		Path serviceFolder = getOrCreateServiceFolder(serviceName);

		SchemaDownloader schemaDownloader = new SchemaDownloader(serviceFolder, definition, serviceName);
		schemaDownloader.download();

		WSDLWriter wsdlWriter = WSDLFactory.newInstance().newWSDLWriter();

		try (OutputStreamWriter out = new OutputStreamWriter(new FileOutputStream(serviceFolder.resolve(fileName).toFile()), StandardCharsets.UTF_8)) {
			wsdlWriter.writeWSDL(definition, out);
		}
	}

	private Path getOrCreateServiceFolder(String serviceName) throws IOException {
		Path serviceFolder = baseFolder.resolve(serviceName);
		Files.createDirectories(serviceFolder);
		return serviceFolder;
	}


}