| /* |
| * 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.netbeans.modules.websvc.saas.codegen.j2ee; |
| |
| import java.io.IOException; |
| import java.util.List; |
| import javax.swing.text.Document; |
| import org.netbeans.modules.editor.NbEditorUtilities; |
| import org.netbeans.modules.websvc.saas.codegen.Constants; |
| import org.netbeans.modules.websvc.saas.codegen.Constants.SaasAuthenticationType; |
| import org.netbeans.modules.websvc.saas.codegen.java.CustomClientPojoCodeGenerator; |
| import org.netbeans.modules.websvc.saas.codegen.model.CustomClientSaasBean; |
| import org.netbeans.modules.websvc.saas.codegen.model.ParameterInfo; |
| import org.netbeans.modules.websvc.saas.codegen.model.SaasBean; |
| import org.netbeans.modules.websvc.saas.codegen.j2ee.support.J2eeUtil; |
| import org.netbeans.modules.websvc.saas.codegen.util.Util; |
| import org.netbeans.modules.websvc.saas.model.CustomSaasMethod; |
| import org.netbeans.modules.websvc.saas.model.SaasMethod; |
| |
| /** |
| * Code generator for Accessing Saas services. |
| * |
| * @author nam |
| */ |
| @org.openide.util.lookup.ServiceProvider(service=org.netbeans.modules.websvc.saas.codegen.spi.SaasClientCodeGenerationProvider.class) |
| public class CustomClientServletCodeGenerator extends CustomClientPojoCodeGenerator { |
| private SaasClientJ2eeAuthenticationGenerator j2eeAuthGen; |
| |
| public CustomClientServletCodeGenerator() { |
| setDropFileType(Constants.DropFileType.SERVLET); |
| setPrecedence(1); |
| } |
| |
| @Override |
| public boolean canAccept(SaasMethod method, Document doc) { |
| if (SaasBean.canAccept(method, CustomSaasMethod.class, getDropFileType()) && |
| J2eeUtil.isServlet(NbEditorUtilities.getDataObject(doc))) { |
| return true; |
| } |
| return false; |
| } |
| |
| @Override |
| public void init(SaasMethod m, Document doc) throws IOException { |
| super.init(m, new CustomClientSaasBean((CustomSaasMethod) m, true), doc); |
| |
| this.j2eeAuthGen = new SaasClientJ2eeAuthenticationGenerator(getBean(), getProject()); |
| this.j2eeAuthGen.setLoginArguments(getLoginArguments()); |
| this.j2eeAuthGen.setAuthenticatorMethodParameters(getAuthenticatorMethodParameters()); |
| this.j2eeAuthGen.setSaasServiceFolder(getSaasServiceFolder()); |
| this.j2eeAuthGen.setAuthenticationProfile(getBean().getProfile(m, getDropFileType())); |
| this.j2eeAuthGen.setDropFileType(getDropFileType()); |
| } |
| |
| @Override |
| public CustomClientSaasBean getBean() { |
| return super.getBean(); |
| } |
| |
| @Override |
| public SaasClientJ2eeAuthenticationGenerator getAuthenticationGenerator() { |
| return j2eeAuthGen; |
| } |
| |
| @Override |
| protected List<ParameterInfo> getServiceMethodParameters() { |
| if(getBean().getAuthenticationType() == SaasAuthenticationType.SESSION_KEY || |
| getBean().getAuthenticationType() == SaasAuthenticationType.HTTP_BASIC) |
| return getServiceMethodParametersForWeb(getBean()); |
| else |
| return super.getServiceMethodParameters(); |
| } |
| |
| @Override |
| protected void addImportsToTargetFile() { |
| } |
| |
| protected String getCustomMethodBody(String paramDecl, String paramUse, |
| String resultVarName, String indent2) { |
| String indent = " "; |
| String methodBody = "\n"; |
| methodBody += indent + "try {\n"; |
| methodBody += paramDecl + "\n"; |
| methodBody += indent2 + REST_RESPONSE + " "+resultVarName+" = " + getBean().getSaasServiceName() + |
| "." + getBean().getSaasServiceMethodName() + "(" + paramUse + ");\n"; |
| methodBody += Util.createPrintStatement( |
| getBean().getOutputWrapperPackageNames(), |
| getBean().getOutputWrapperNames(), |
| getDropFileType(), |
| getBean().getHttpMethod(), |
| getBean().canGenerateJAXBUnmarshaller(), resultVarName, indent2); |
| methodBody += indent+"} catch (Exception ex) {\n"; |
| methodBody += indent2+"ex.printStackTrace();\n"; |
| methodBody += indent+"}\n"; |
| return methodBody; |
| } |
| |
| @Override |
| protected List<ParameterInfo> getAuthenticatorMethodParameters() { |
| if(getBean().getAuthenticationType() == SaasAuthenticationType.SESSION_KEY || |
| getBean().getAuthenticationType() == SaasAuthenticationType.HTTP_BASIC) |
| return Util.getAuthenticatorMethodParametersForWeb(); |
| else |
| return super.getAuthenticatorMethodParameters(); |
| } |
| |
| @Override |
| protected String getLoginArguments() { |
| return Util.getLoginArgumentsForWeb(); |
| } |
| } |