1
2
3
4
5
6
7
8
9
10
11
12
13 package com.eviware.soapui.impl.wsdl.actions.iface.tools.jbossws;
14
15 import java.io.File;
16 import java.io.IOException;
17
18 import org.jboss.jbosswsTools.ConfigurationDocument;
19 import org.jboss.jbosswsTools.ConfigurationType;
20 import org.jboss.jbosswsTools.JavaToWsdlType;
21 import org.jboss.jbosswsTools.MappingType;
22 import org.jboss.jbosswsTools.NamespacesType;
23 import org.jboss.jbosswsTools.ServiceType;
24 import org.jboss.jbosswsTools.WsxmlType;
25 import org.jboss.jbosswsTools.ServiceType.ParameterStyle;
26 import org.jboss.jbosswsTools.ServiceType.Style;
27
28 import com.eviware.soapui.SoapUI;
29 import com.eviware.soapui.impl.WsdlInterfaceFactory;
30 import com.eviware.soapui.impl.wsdl.WsdlProject;
31 import com.eviware.soapui.impl.wsdl.actions.iface.tools.support.AbstractToolsAction;
32 import com.eviware.soapui.impl.wsdl.actions.iface.tools.support.ArgumentBuilder;
33 import com.eviware.soapui.impl.wsdl.actions.iface.tools.support.ProcessToolRunner;
34 import com.eviware.soapui.impl.wsdl.actions.iface.tools.support.RunnerContext;
35 import com.eviware.soapui.impl.wsdl.actions.iface.tools.support.ShowConfigFileAction;
36 import com.eviware.soapui.impl.wsdl.actions.iface.tools.support.ToolHost;
37 import com.eviware.soapui.impl.wsdl.support.HelpUrls;
38 import com.eviware.soapui.model.iface.Interface;
39 import com.eviware.soapui.settings.ToolsSettings;
40 import com.eviware.soapui.support.SoapUIException;
41 import com.eviware.soapui.support.Tools;
42 import com.eviware.soapui.support.UISupport;
43 import com.eviware.soapui.support.action.swing.ActionList;
44 import com.eviware.soapui.support.types.StringToStringMap;
45 import com.eviware.x.form.XForm;
46 import com.eviware.x.form.XFormDialog;
47 import com.eviware.x.form.XFormDialogBuilder;
48 import com.eviware.x.form.XFormFactory;
49
50 /***
51 * Invokes jbossws java2wsdl tools
52 *
53 * @author Ole.Matzura
54 */
55
56 public class WSToolsJava2WsdlAction extends AbstractToolsAction<WsdlProject>
57 {
58 public static final String SOAPUI_ACTION_ID = "WSToolsJava2WsdlAction";
59
60 private static final String CLASSPATH = "Classpath";
61 private static final String OUTPUT = "Output Directory";
62 private static final String ENDPOINT = "Endpoint";
63 private static final String MAPPING = "Mapping file";
64 private static final String SERVICE_NAME = "Service Name";
65 private static final String STYLE = "Style";
66 private static final String PARAMETER_STYLE = "Parameter Style";
67 private static final String TARGET_NAMESPACE = "Target Namespace";
68 private static final String TYPES_NAMESPACE = "Types Namespace";
69 private static final String EJB_LINK = "ejb-link";
70 private static final String SERVLET_LINK = "servlet-link";
71
72 public WSToolsJava2WsdlAction()
73 {
74 super( "Generate WSDL with JBossWS", "Generates WSDL with the jbossws wstools utility" );
75 }
76
77 protected XFormDialog buildDialog( WsdlProject project )
78 {
79 XFormDialogBuilder builder = XFormFactory.createDialogBuilder( "Generate JBossWS WSDL Artifacts" );
80
81 XForm mainForm = builder.createForm( "Basic" );
82
83 mainForm.addTextField( ENDPOINT, "Serice Endpoint Interface", XForm.FieldType.JAVA_CLASS );
84 mainForm.addTextField( SERVICE_NAME, "The name of the generated Service", XForm.FieldType.TEXT );
85 mainForm
86 .addComboBox( STYLE, new String[] { Style.DOCUMENT.toString(), Style.RPC.toString() }, "The style to use" );
87 mainForm.addComboBox( PARAMETER_STYLE, new String[] { ParameterStyle.BARE.toString(),
88 ParameterStyle.WRAPPED.toString() }, "The style to use" );
89 mainForm.addTextField( CLASSPATH, "Classpath to use", XForm.FieldType.PROJECT_FOLDER );
90 mainForm.addTextField( OUTPUT, "The root directory for all emitted files.", XForm.FieldType.PROJECT_FOLDER );
91 mainForm.addTextField( MAPPING, "mapping file to generate", XForm.FieldType.PROJECT_FILE );
92 mainForm.addTextField( TARGET_NAMESPACE, "The target namespace for the generated WSDL", XForm.FieldType.TEXT );
93 mainForm.addTextField( TYPES_NAMESPACE, "The namespace for the generated types", XForm.FieldType.TEXT );
94 mainForm.addTextField( EJB_LINK, "The name of the source EJB to link to", XForm.FieldType.TEXT );
95 mainForm.addTextField( SERVLET_LINK, "The name of the source Servlet to link to", XForm.FieldType.TEXT );
96
97 buildArgsForm( builder, false, "wstools" );
98
99 ActionList actions = buildDefaultActions( HelpUrls.WSTOOLS_HELP_URL, project );
100 actions.addAction( new ShowConfigFileAction( "JBossWS Java2Wsdl", "Contents of generated wsconfig.xml file" )
101 {
102 protected String getConfigFile()
103 {
104 ConfigurationDocument configDocument = createConfigFile( getDialog().getValues() );
105 return configDocument.toString();
106 }
107 } );
108
109 return builder.buildDialog( actions, "Specify arguments for JBossWS wstools java2wsdl functionality",
110 UISupport.TOOL_ICON );
111 }
112
113 protected void generate( StringToStringMap values, ToolHost toolHost, WsdlProject project ) throws Exception
114 {
115 String wstoolsDir = SoapUI.getSettings().getString( ToolsSettings.JBOSSWS_WSTOOLS_LOCATION, null );
116 if( Tools.isEmpty( wstoolsDir ) )
117 {
118 UISupport.showErrorMessage( "wstools directory must be set in global preferences" );
119 return;
120 }
121
122 String wsToolsExtension = UISupport.isWindows() ? ".bat" : ".sh";
123
124 File wstoolsFile = new File( wstoolsDir + File.separatorChar + "wstools" + wsToolsExtension );
125 if( !wstoolsFile.exists() )
126 {
127 UISupport.showErrorMessage( "Could not find wstools script at [" + wstoolsFile + "]" );
128 return;
129 }
130
131 ProcessBuilder builder = new ProcessBuilder();
132 ArgumentBuilder args = buildArgs( values, UISupport.isWindows() );
133 builder.command( args.getArgs() );
134 builder.directory( new File( wstoolsDir ) );
135
136 toolHost.run( new ToolRunner( builder, new File( values.get( OUTPUT ) ), values.get( SERVICE_NAME ), project ) );
137 }
138
139 private ArgumentBuilder buildArgs( StringToStringMap values, boolean isWindows ) throws IOException
140 {
141 values.put( OUTPUT, Tools.ensureDir( values.get( OUTPUT ), "" ) );
142
143 ArgumentBuilder builder = new ArgumentBuilder( values );
144 builder.startScript( "wstools" );
145
146 builder.addString( CLASSPATH, "-cp" );
147 builder.addArgs( "-config", buildConfigFile( values ) );
148 builder.addString( OUTPUT, "-dest" );
149 addToolArgs( values, builder );
150 return builder;
151 }
152
153 private String buildConfigFile( StringToStringMap values ) throws IOException
154 {
155 File file = File.createTempFile( "wstools-config", ".xml" );
156 ConfigurationDocument configDocument = createConfigFile( values );
157
158 configDocument.save( file );
159
160 return file.getAbsolutePath();
161 }
162
163 private ConfigurationDocument createConfigFile( StringToStringMap values )
164 {
165 ConfigurationDocument configDocument = ConfigurationDocument.Factory.newInstance();
166 ConfigurationType config = configDocument.addNewConfiguration();
167
168 JavaToWsdlType java2Wsdl = config.addNewJavaWsdl();
169 ServiceType service = java2Wsdl.addNewService();
170 service.setEndpoint( values.get( ENDPOINT ) );
171 service.setStyle( Style.Enum.forString( values.get( STYLE ) ) );
172 service.setParameterStyle( ParameterStyle.Enum.forString( values.get( PARAMETER_STYLE ) ) );
173 service.setName( values.get( SERVICE_NAME ) );
174
175 MappingType mapping = java2Wsdl.addNewMapping();
176 mapping.setFile( values.get( MAPPING ) );
177
178 NamespacesType namespaces = java2Wsdl.addNewNamespaces();
179 namespaces.setTargetNamespace( values.get( TARGET_NAMESPACE ) );
180 namespaces.setTypeNamespace( values.get( TYPES_NAMESPACE ) );
181
182 WsxmlType webservices = java2Wsdl.addNewWebservices();
183 if( values.get( EJB_LINK ) != null && values.get( EJB_LINK ).length() > 0 )
184 webservices.setEjbLink( values.get( EJB_LINK ) );
185 if( values.get( SERVLET_LINK ) != null && values.get( SERVLET_LINK ).length() > 0 )
186 webservices.setServletLink( values.get( SERVLET_LINK ) );
187 return configDocument;
188 }
189
190 private class ToolRunner extends ProcessToolRunner
191 {
192 private final File outDir;
193 private final String serviceName;
194 private final WsdlProject project;
195
196 public ToolRunner( ProcessBuilder builder, File outDir, String serviceName, WsdlProject modelItem )
197 {
198 super( builder, "JBossWS wstools", modelItem );
199 this.outDir = outDir;
200 this.serviceName = serviceName;
201 this.project = modelItem;
202 }
203
204 protected void afterRun( RunnerContext context )
205 {
206 if( context.getStatus() != RunnerContext.RunnerStatus.FINISHED )
207 return;
208
209 try
210 {
211 String wsdlUrl = "file:" + outDir.getAbsolutePath() + File.separatorChar + "wsdl" + File.separatorChar
212 + serviceName + ".wsdl";
213 Interface[] ifaces = WsdlInterfaceFactory.importWsdl( project, wsdlUrl, true );
214
215 if( ifaces.length > 0 )
216 {
217 context.log( "Added Interface [" + ifaces[0].getName() + "] to project" );
218 ifaces[0].getSettings().setString( WSToolsRegenerateJava2WsdlAction.class.getName() + "@values",
219 getDialog().getValues().toXml() );
220 UISupport.select( ifaces[0] );
221 }
222 }
223 catch( SoapUIException e )
224 {
225 SoapUI.logError( e );
226 }
227 }
228 }
229 }