/**
 * 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.apache.openejb.server.cxf;

import org.apache.cxf.endpoint.Client;
import org.apache.cxf.frontend.ClientProxy;
import org.apache.cxf.interceptor.Interceptor;
import org.apache.cxf.interceptor.LoggingOutInterceptor;
import org.apache.cxf.message.Message;
import org.apache.cxf.ws.addressing.impl.MAPAggregatorImpl;
import org.apache.cxf.ws.addressing.soap.MAPCodec;
import org.apache.cxf.ws.security.wss4j.WSS4JInInterceptor;
import org.apache.cxf.ws.security.wss4j.WSS4JOutInterceptor;
import org.apache.openejb.config.sys.MapFactory;
import org.apache.openejb.jee.WebApp;
import org.apache.openejb.junit.ApplicationComposer;
import org.apache.openejb.server.cxf.config.WSS4JInInterceptorFactory;
import org.apache.openejb.testing.ApplicationConfiguration;
import org.apache.openejb.testing.Classes;
import org.apache.openejb.testing.EnableServices;
import org.apache.openejb.testing.Module;
import org.apache.openejb.testng.PropertiesBuilder;
import org.junit.Test;
import org.junit.runner.RunWith;

import javax.ejb.Singleton;
import javax.jws.WebService;
import javax.xml.namespace.QName;
import javax.xml.ws.Service;
import javax.xml.ws.WebEndpoint;
import javax.xml.ws.WebServiceClient;
import javax.xml.ws.WebServiceException;
import javax.xml.ws.WebServiceFeature;
import javax.xml.ws.WebServiceRef;
import javax.xml.ws.soap.AddressingFeature;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Iterator;
import java.util.Properties;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

@EnableServices("jax-ws")
@RunWith(ApplicationComposer.class)
public class WebServiceInjectionTest {
    @Module
    @Classes(innerClassesAsBean = true)
    public WebApp module() {
        return new WebApp();
    }

    @WebServiceRef
    private MyWsApi api;

    @WebServiceRef
    private MyWebservice_Service service;

    @Test
    public void checkConfiguration() {
        // assertEquals("ok", api.test()); // local call so skip it but check config which is actually the only interesting thing
        final Client client = ClientProxy.getClient(api);
        testPort(client);

        testPort(ClientProxy.getClient(service.getMyWsApi()));
        testPortWithFeature(ClientProxy.getClient(service.getMyWsApi(new AddressingFeature())));
    }

    private void testPort(final Client client) {
        assertNotNull(client);
        assertEquals(2, client.getOutInterceptors().size());
        assertEquals(1, client.getInInterceptors().size());
        final Iterator<Interceptor<? extends Message>> iterator = client.getOutInterceptors().iterator();
        assertTrue(LoggingOutInterceptor.class.isInstance(iterator.next()));
        final Interceptor<? extends Message> wss4jout = iterator.next();
        assertTrue(WSS4JOutInterceptor.class.isInstance(wss4jout));
        assertEquals("d", WSS4JOutInterceptor.class.cast(wss4jout).getProperties().get("c"));
        final Interceptor<? extends Message> wss4jin = client.getInInterceptors().iterator().next();
        assertTrue(WSS4JInInterceptor.class.isInstance(wss4jin));
        assertEquals("b", WSS4JInInterceptor.class.cast(wss4jin).getProperties().get("a"));
    }

    private void testPortWithFeature(final Client client) {
        assertNotNull(client);
        assertEquals(4, client.getOutInterceptors().size());
        assertEquals(3, client.getInInterceptors().size());
        final Iterator<Interceptor<? extends Message>> Out = client.getOutInterceptors().iterator();
        assertTrue(MAPAggregatorImpl.class.isInstance(Out.next()));
        assertTrue(MAPCodec.class.isInstance(Out.next()));
        assertTrue(LoggingOutInterceptor.class.isInstance(Out.next()));
        final Interceptor<? extends Message> wss4jout = Out.next();
        assertTrue(WSS4JOutInterceptor.class.isInstance(wss4jout));

        final Iterator<Interceptor<? extends Message>> iteratorIn = client.getInInterceptors().iterator();
        assertTrue(MAPAggregatorImpl.class.isInstance(iteratorIn.next()));
        assertTrue(MAPCodec.class.isInstance(iteratorIn.next()));
        assertTrue(WSS4JInInterceptor.class.isInstance(iteratorIn.next()));
    }

    @ApplicationConfiguration
    public Properties props() {
        // return new PropertiesBuilder().p("cxf.jaxws.client.out-interceptors", LoggingOutInterceptor.class.getName()).build();
        // return new PropertiesBuilder().p("cxf.jaxws.client.{http://cxf.server.openejb.apache.org/}MyWebservicePort.out-interceptors", LoggingOutInterceptor.class.getName()).build();
        return new PropertiesBuilder()
                .p("cxf.jaxws.client.{http://cxf.server.openejb.apache.org/}MyWebservicePort.in-interceptors", "wss4jin")
                .p("cxf.jaxws.client.{http://cxf.server.openejb.apache.org/}MyWebservicePort.out-interceptors", "loo,wss4jout")

                .p("cxf.jaxws.client.{http://cxf.server.openejb.apache.org/}myWebservice.in-interceptors", "wss4jin")
                .p("cxf.jaxws.client.{http://cxf.server.openejb.apache.org/}myWebservice.out-interceptors", "loo,wss4jout")

                .p("loo", "new://Service?class-name=" + LoggingOutInterceptor.class.getName())

                .p("wss4jin", "new://Service?class-name=" + WSS4JInInterceptorFactory.class.getName() + "&factory-name=create")
                .p("wss4jin.a", "b")

                .p("wss4jout", "new://Service?class-name=" + WSS4JOutInterceptor.class.getName() + "&constructor=properties")
                .p("wss4jout.properties", "$properties")

                .p("properties", "new://Service?class-name=" + MapFactory.class.getName())
                .p("properties.c", "d")

                .build();
    }

    @WebService
    public static interface MyWsApi {
        String test();
    }

    @WebService
    @Singleton
    public static class MyWebservice implements MyWsApi {
        @Override
        public String test() {
            return "ok";
        }
    }

    @WebServiceClient(name = "MyWebservice")
    public static class MyWebservice_Service extends Service {
        public MyWebservice_Service() {
            super(getWsdl(), new QName("http://cxf.server.openejb.apache.org/", "MyWebservice_Service"));
        }

        public MyWebservice_Service(final URL wsdlDocumentLocation, final QName serviceName) {
            super(wsdlDocumentLocation, serviceName);
        }

        static URL getWsdl() throws WebServiceException {
            try {
                return new URL("http://wsdl");
            } catch (final MalformedURLException ex) {
                throw new WebServiceException(ex);
            }
        }

        @WebEndpoint(name = "myWebserviceWS")
        public MyWsApi getMyWsApi(final WebServiceFeature... features) {
            return super.getPort(new QName("http://cxf.server.openejb.apache.org/", "myWebservice"), MyWsApi.class, features);
        }


    }
}
