| /* |
| * 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.owsm; |
| |
| import java.util.Collection; |
| import java.util.Collections; |
| import java.util.HashMap; |
| import java.util.HashSet; |
| import java.util.LinkedList; |
| import java.util.Map; |
| import java.util.Map.Entry; |
| import java.util.Set; |
| |
| import javax.lang.model.element.TypeElement; |
| |
| import org.netbeans.api.java.source.Comment; |
| import org.netbeans.api.java.source.TreeMaker; |
| import org.netbeans.api.java.source.WorkingCopy; |
| import org.netbeans.modules.j2ee.deployment.devmodules.api.J2eePlatform; |
| import org.netbeans.modules.websvc.api.jaxws.project.config.Client; |
| import org.netbeans.modules.websvc.core.jaxws.policies.JaxWsPoliciesCodeGenerator; |
| import org.openide.filesystems.FileObject; |
| import org.openide.util.Lookup; |
| import org.openide.util.lookup.ServiceProvider; |
| |
| import com.sun.source.tree.ExpressionTree; |
| import com.sun.source.tree.NewClassTree; |
| import com.sun.source.tree.Tree; |
| import com.sun.source.tree.VariableTree; |
| import org.netbeans.modules.javaee.specs.support.api.JaxWsPoliciesSupport; |
| |
| |
| /** |
| * @author ads |
| * |
| */ |
| @ServiceProvider( service=JaxWsPoliciesCodeGenerator.class ) |
| public class OWSMPoliciesCodeGenerator implements JaxWsPoliciesCodeGenerator { |
| |
| private static final String ORACLE = "oracle/"; // NOI18N |
| |
| private static final String SECURITY_POLICY_FEATURE = |
| "weblogic.wsee.jws.jaxws.owsm.SecurityPolicyFeature"; // NOI18N |
| |
| private static final String SECURITY_FEATURE_INIT = "private static final " + |
| "SecurityPolicyFeature[] securityFeature = new SecurityPolicyFeature[]" + |
| " { new SecurityPolicyFeature(\""; // NOI18N |
| |
| @Override |
| public Collection<String> getRequiredClasses( String id ) { |
| Collection<String> result = new LinkedList<String>(); |
| result.add("weblogic.wsee.jws.jaxws.owsm.SecurityPolicyFeature"); // NOI18N |
| result.add("com.sun.xml.ws.developer.WSBindingProvider"); // NOI18N |
| result.add("javax.xml.ws.BindingProvider"); // NOI18N |
| result.add("javax.xml.ws.WebServiceFeature"); // NOI18N |
| result.add(Map.class.getCanonicalName()); |
| |
| OWSMPolicyCodeGenerator generator = GENERATORS.get( id ); |
| if ( generator != null ){ |
| generator.addRequiredClassesFqns( result ); |
| } |
| return result; |
| } |
| |
| |
| /* (non-Javadoc) |
| * @see org.netbeans.modules.websvc.core.jaxws.policies.JaxWsPoliciesCodeGenerator#generatePolicyAccessCode(java.util.Set, org.netbeans.modules.websvc.api.jaxws.project.config.Client, java.lang.StringBuilder) |
| */ |
| @Override |
| public String generatePolicyAccessCode( Set<String> policyIds, |
| Client client, StringBuilder code ) |
| { |
| if ( policyIds.isEmpty() ){ |
| return ""; |
| } |
| Map<String,OWSMPolicyCodeGenerator> map = new HashMap<String, |
| OWSMPolicyCodeGenerator>( GENERATORS); |
| Set<String> keySet = map.keySet(); |
| keySet.retainAll( policyIds ); |
| if ( keySet.isEmpty() ){ |
| String id = policyIds.iterator().next(); |
| generateDefaultCode( id , client , code ); |
| return id; |
| } |
| else { |
| String mainId = null; |
| for (Entry<String, OWSMPolicyCodeGenerator> entry : map.entrySet()) { |
| String id = entry.getKey(); |
| OWSMPolicyCodeGenerator generator = entry.getValue(); |
| if ( mainId == null ){ |
| mainId = id; |
| generator.generatePolicyAccessCode(code, client); |
| } |
| else { |
| StringBuilder builder = new StringBuilder(); |
| generator.generatePolicyAccessCode(builder, client); |
| code.append("/* "); // NOI18N |
| code.append( builder.toString() ); |
| code.append(" */"); // NOI18N |
| } |
| } |
| return mainId; |
| } |
| } |
| |
| @Override |
| public boolean isApplicable( Set<String> policyIds, J2eePlatform platform, |
| FileObject wsdl ) |
| { |
| JaxWsPoliciesSupport support = JaxWsPoliciesSupport.getInstance( platform ); |
| if ( support == null ){ |
| return false; |
| } |
| String id = support.getId(); |
| boolean result = id!=null && ORACLE.startsWith( id ) && id.length() == ORACLE.length()-1; |
| |
| if ( result ){ |
| policyIds.retainAll( GENERATORS.keySet() ); |
| } |
| |
| return result; |
| } |
| |
| /* (non-Javadoc) |
| * @see org.netbeans.modules.websvc.core.jaxws.policies.JaxWsPoliciesCodeGenerator#createSecurityFeatureType(org.netbeans.api.java.source.WorkingCopy, org.netbeans.api.java.source.TreeMaker) |
| */ |
| @Override |
| public Tree createSecurityFeatureType( WorkingCopy workingCopy, |
| TreeMaker make ) |
| { |
| return make.ArrayType( createSecurityFeatureComponentType(workingCopy, |
| make) ); |
| } |
| |
| |
| private Tree createSecurityFeatureComponentType( WorkingCopy workingCopy, |
| TreeMaker make ) |
| { |
| TypeElement securityType = workingCopy.getElements().getTypeElement( |
| SECURITY_POLICY_FEATURE); |
| Tree securityTreeType = securityType != null ? make.Type(securityType.asType()) : |
| make.Identifier(SECURITY_POLICY_FEATURE); |
| return securityTreeType; |
| } |
| |
| |
| |
| private void generateDefaultCode( String id, Client client, |
| StringBuilder code ) |
| { |
| code.append("WSBindingProvider wsbp = (WSBindingProvider)port;\n"); // NOI18N |
| code.append("Map<String, Object> requestContext = wsbp.getRequestContext();\n");// NOI18N |
| code.append("// Override the endpoint - useful when switching target environments without regenerating the jax-ws client\n");// NOI18N |
| code.append("requestContext.put(WSBindingProvider.ENDPOINT_ADDRESS_PROPERTY, \"");// NOI18N |
| code.append(client.getWsdlUrl()); |
| code.append("\");\n"); // NOI18N |
| code.append("// Use request context to initialize poliy access \n" );// NOI18N |
| code.append("//requestContext.put( some_property_key, some_property_value );\n"); // NOI18N |
| } |
| |
| /* (non-Javadoc) |
| * @see org.netbeans.modules.websvc.core.jaxws.policies.JaxWsPoliciesCodeGenerator#createSecurityFeatureInitializer(org.netbeans.api.java.source.WorkingCopy, org.netbeans.api.java.source.TreeMaker, java.lang.String) |
| */ |
| @Override |
| public ExpressionTree createSecurityFeatureInitializer( |
| WorkingCopy workingCopy, TreeMaker make, String id ) |
| { |
| String clientId = id; |
| OWSMPolicyCodeGenerator generator = GENERATORS.get(id); |
| if ( generator != null){ |
| clientId = generator.getClientId(); |
| } |
| if ( !clientId.startsWith( ORACLE )){ |
| clientId = ORACLE+clientId; |
| } |
| TypeElement securityType = workingCopy.getElements().getTypeElement( |
| SECURITY_POLICY_FEATURE); |
| NewClassTree initClassTree = make.NewClass(null, |
| Collections.<ExpressionTree>emptyList(), |
| securityType != null ? make.Identifier( securityType): |
| make.Identifier( SECURITY_POLICY_FEATURE), |
| Collections.singletonList( |
| make.Literal(clientId)), null); |
| return make.NewArray( |
| createSecurityFeatureComponentType(workingCopy, make ), |
| Collections.<ExpressionTree>emptyList(), |
| Collections.singletonList(initClassTree) ); |
| } |
| |
| /* (non-Javadoc) |
| * @see org.netbeans.modules.websvc.core.jaxws.policies.JaxWsPoliciesCodeGenerator#modifySecurityFeatureAttribute(com.sun.source.tree.VariableTree, org.netbeans.api.java.source.WorkingCopy, org.netbeans.api.java.source.TreeMaker, java.lang.String, java.util.Set) |
| */ |
| @Override |
| public void modifySecurityFeatureAttribute( VariableTree var, |
| WorkingCopy workingCopy, TreeMaker make, String id, |
| Set<String> ids ) |
| { |
| Collection<String> clientIds = new LinkedList<String>(); |
| Set<String> otherIds = new HashSet<String>(ids ); |
| otherIds.remove( id ); |
| |
| Map<String,OWSMPolicyCodeGenerator> map = new HashMap<String, |
| OWSMPolicyCodeGenerator>( GENERATORS); |
| Set<String> keySet = map.keySet(); |
| keySet.retainAll( otherIds ); |
| for (OWSMPolicyCodeGenerator generator :map.values()) { |
| clientIds.add(generator.getClientId()); |
| } |
| |
| OWSMPolicyCodeGenerator generator = GENERATORS.get( id ); |
| if ( generator != null ){ |
| for( String relatedId : generator.getRelatedPolicyIds() ){ |
| clientIds.add( relatedId ); |
| } |
| } |
| |
| addComments(var, make, clientIds ); |
| } |
| |
| |
| private void addComments( VariableTree var , TreeMaker make , |
| Collection<String> ids) |
| { |
| for (String relatedId : ids) { |
| if ( !relatedId.startsWith( ORACLE )){ |
| relatedId = ORACLE +relatedId; |
| } |
| StringBuilder builder = new StringBuilder(SECURITY_FEATURE_INIT); |
| builder.append(relatedId); |
| builder.append("\")};\n"); // NOI18N |
| make.addComment(var,Comment.create(Comment.Style.LINE, |
| builder.toString()),true); |
| } |
| } |
| |
| private static void initGenerators() { |
| Collection<? extends OWSMPolicyCodeGenerator> generators = |
| Lookup.getDefault().lookupAll( OWSMPolicyCodeGenerator.class); |
| for (OWSMPolicyCodeGenerator generator : generators) { |
| GENERATORS.put( generator.getId() , generator ); |
| } |
| } |
| |
| private static Map<String,OWSMPolicyCodeGenerator> GENERATORS = |
| new HashMap<String, OWSMPolicyCodeGenerator>(); |
| |
| static { |
| initGenerators(); |
| } |
| } |
