/*

  * Copyright 2007 The Kuali Foundation

  *

  * Licensed under the Educational Community 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.opensource.org/licenses/ecl2.php

  *

  * 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.kuali.rice.ksb.messaging.serviceconnectors;

 import org.apache.cxf.endpoint.Client;

 import org.apache.cxf.frontend.ClientProxy;

 import org.apache.cxf.transport.http.HTTPConduit;

 import org.apache.cxf.transports.http.configuration.HTTPClientPolicy;

 import org.kuali.rice.core.security.credentials.CredentialsSource;

 import org.kuali.rice.ksb.messaging.BusClientFailureProxy;

 import org.kuali.rice.ksb.messaging.ServiceInfo;

 import org.kuali.rice.ksb.messaging.bam.BAMClientProxy;

 import org.kuali.rice.ksb.service.KSBServiceLocator;

 import org.springframework.util.Assert;

 /**

  * Abstract implementation of the ServiceConnector that holds the ServiceInfo

  * and the CredentialsSource as well as providing convenience proxy methods.

  * 

  * @author Kuali Rice Team (rice.collab@kuali.org)

  * @since 0.9

  * 

  */

 public abstract class AbstractServiceConnector implements ServiceConnector {

         private static final String HTTP_CLIENT_POLICY_BEAN = "httpClientPolicy";

         /**

          * Maintains the information about the service.  This should never be null.

          */

         private ServiceInfo serviceInfo;

         /**

          * Maintains the credentials needed by the service.  This may be null.

          */

         private CredentialsSource credentialsSource;

         public AbstractServiceConnector(final ServiceInfo serviceInfo) {

                 Assert.notNull(serviceInfo, "serviceInfo cannot be null");

                 this.serviceInfo = serviceInfo;

         }

         public ServiceInfo getServiceInfo() {

                 return this.serviceInfo;

         }

         public void setServiceInfo(final ServiceInfo serviceInfo) {

                 this.serviceInfo = serviceInfo;

         }

         public void setCredentialsSource(final CredentialsSource credentialsSource) {

                 this.credentialsSource = credentialsSource;

         }

         protected CredentialsSource getCredentialsSource() {

                 return this.credentialsSource;

         }

         protected Object getServiceProxyWithFailureMode(final Object service,

                         final ServiceInfo serviceInfo) {

                 // HACK: TODO: rip out or clean up

                 // set http client policy

                 HTTPClientPolicy policy = 

                         (HTTPClientPolicy)KSBServiceLocator.getService(HTTP_CLIENT_POLICY_BEAN);

                 if (policy != null) {

                         try {

                                 Client cl = ClientProxy.getClient(service);

                                 ((HTTPConduit)cl.getConduit()).setClient(policy);

                         } catch (ClassCastException e) {

                                 // ignore for now

                         }

                 }

                 Object bamWrappedClientProxy = BAMClientProxy

                                 .wrap(service, serviceInfo);

                 return BusClientFailureProxy.wrap(bamWrappedClientProxy, serviceInfo);

         }

 }