/*
    * Copyright 2009 the original author or authors.
    * 
    * Licensed 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.openehealth.ipf.commons.ihe.ws;
  
  import org.apache.cxf.binding.soap.SoapBindingConfiguration;
  import org.apache.cxf.frontend.ServerFactoryBean;
  import org.apache.cxf.interceptor.InterceptorProvider;
  import org.apache.cxf.jaxws.JaxWsServerFactoryBean;
  import org.apache.cxf.ws.addressing.WSAddressingFeature;
  import org.openehealth.ipf.commons.audit.AuditContext;
  import org.openehealth.ipf.commons.ihe.core.atna.AuditStrategy;
  import org.openehealth.ipf.commons.ihe.ws.cxf.Cxf3791WorkaroundInterceptor;
  import org.openehealth.ipf.commons.ihe.ws.cxf.RejectionHandlerInterceptor;
  import org.openehealth.ipf.commons.ihe.ws.cxf.WsRejectionHandlingStrategy;
  import org.openehealth.ipf.commons.ihe.ws.cxf.audit.WsAuditDataset;
  import org.openehealth.ipf.commons.ihe.ws.cxf.payload.InPayloadExtractorInterceptor;
  
  import java.util.Collections;
  
  import static org.openehealth.ipf.commons.ihe.ws.cxf.payload.StringPayloadHolder.PayloadType.HTTP;
  
  /**
   * Factory for Web Services
   * @author Jens Riemschneider
   */
  public class JaxWsServiceFactory<AuditDatasetType extends WsAuditDataset> {
  
      /**
       * Transaction configuration.
       */
      protected final WsTransactionConfiguration<AuditDatasetType> wsTransactionConfiguration;
      /**
       * Service endpoint address.
       */
      protected final String serviceAddress;
      /**
       * User-defined custom CXF interceptors.
       */
      protected final InterceptorProvider customInterceptors;
      /**
       * User-defined strategy for handling rejected messages.
       */
      protected final WsRejectionHandlingStrategy rejectionHandlingStrategy;
      /**
       * Server-side ATNA audit strategy.
       */
      protected final AuditStrategy<AuditDatasetType> auditStrategy;
  
      protected final AuditContext auditContext;
  
      /**
       * Constructs the factory.
       * @param wsTransactionConfiguration
       *          the info about the service to produce.
       * @param serviceAddress
       *          the address of the service that it should be published with.
       * @param auditStrategy
       *          server-side ATNA audit strategy.
       * @param customInterceptors
       *          user-defined custom CXF interceptors.
       * @param rejectionHandlingStrategy
       *          user-defined rejection handling strategy.
       *
       */
      public JaxWsServiceFactory(
              WsTransactionConfiguration<AuditDatasetType> wsTransactionConfiguration,
              String serviceAddress,
              AuditStrategy<AuditDatasetType> auditStrategy,
              AuditContext auditContext,
              InterceptorProvider customInterceptors,
              WsRejectionHandlingStrategy rejectionHandlingStrategy) {
          this.wsTransactionConfiguration = wsTransactionConfiguration;
          this.serviceAddress = serviceAddress;
          this.auditStrategy = auditStrategy;
          this.auditContext = auditContext;
          this.customInterceptors = customInterceptors;
          this.rejectionHandlingStrategy = rejectionHandlingStrategy;
      }
      
      /**
       * Creates and configures a server factory.
       * Use the server factory to create a server instance that can be used
       * to start and stop the service.
       * @param serviceImplClass
       *          the type of the service implementation.
       * @return the server factory.
      */
     public ServerFactoryBean createServerFactory(Class<?> serviceImplClass) {
         try {
             return createServerFactory(serviceImplClass.newInstance());
         } catch (InstantiationException | IllegalAccessException e) {
             throw new IllegalStateException(e);
         }
     }
 
 
     public ServerFactoryBean createServerFactory(Object serviceImpl) {
         ServerFactoryBean svrFactory = new JaxWsServerFactoryBean();
         configureService(svrFactory, serviceImpl);
         configureBinding(svrFactory);
         configureInterceptors(svrFactory);
         svrFactory.setStart(false);
         return svrFactory;
     }
 
 
     private void configureService(ServerFactoryBean svrFactory, Object service) {
         svrFactory.setServiceClass(wsTransactionConfiguration.getSei());
         svrFactory.setServiceName(wsTransactionConfiguration.getServiceName());
         svrFactory.setWsdlLocation(wsTransactionConfiguration.getWsdlLocation());
         svrFactory.setAddress(serviceAddress);
         svrFactory.setServiceBean(service);
         svrFactory.getFeatures().add(new WSAddressingFeature());
         if (wsTransactionConfiguration.isMtom()) {
             svrFactory.setProperties(Collections.singletonMap("mtom-enabled", "true"));
         }
     }
 
     private void configureBinding(ServerFactoryBean svrFactory) {
         SoapBindingConfiguration bindingConfig = new SoapBindingConfiguration();
         bindingConfig.setBindingName(wsTransactionConfiguration.getBindingName());
         svrFactory.setBindingConfig(bindingConfig);
     }
 
     /**
      * Called to configure any interceptors of the service.
      * @param svrFactory
      *          the server factory.
      */
     protected void configureInterceptors(ServerFactoryBean svrFactory) {
         svrFactory.getInInterceptors().add(new Cxf3791WorkaroundInterceptor());
         InterceptorUtils.copyInterceptorsFromProvider(customInterceptors, svrFactory);
 
         if (rejectionHandlingStrategy != null) {
             svrFactory.getInInterceptors().add(new InPayloadExtractorInterceptor(HTTP));
 
             RejectionHandlerInterceptor rejectionHandlerInterceptor =
                     new RejectionHandlerInterceptor(rejectionHandlingStrategy);
 
             svrFactory.getOutInterceptors().add(rejectionHandlerInterceptor);
             svrFactory.getOutFaultInterceptors().add(rejectionHandlerInterceptor);
         }
 
     }
 
 }