/*
 * 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.webbeans.test.managed;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Set;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicReference;

import jakarta.enterprise.context.ApplicationScoped;
import jakarta.enterprise.context.spi.CreationalContext;
import jakarta.enterprise.inject.Produces;
import jakarta.enterprise.inject.spi.Bean;
import jakarta.enterprise.inject.spi.InterceptionFactory;
import jakarta.enterprise.util.AnnotationLiteral;
import jakarta.inject.Named;
import jakarta.interceptor.Interceptor;
import jakarta.interceptor.InterceptorBinding;
import jakarta.interceptor.InvocationContext;

import org.junit.Assert;

import org.apache.webbeans.config.WebBeansContext;
import org.apache.webbeans.proxy.NormalScopeProxyFactory;
import org.apache.webbeans.test.AbstractUnitTest;
import org.apache.webbeans.test.managed.multipleinterfaces.MyEntityServiceImpl;
import org.junit.Test;

import static org.junit.Assert.assertEquals;

public class ProxyFactoryTest extends AbstractUnitTest
{
    @InterceptorBinding
    @Retention(RetentionPolicy.RUNTIME)
    @Target({ElementType.METHOD, ElementType.TYPE})
    public @interface Foo
    {
    }

    @Interceptor
    @Foo
    public static class FooInterceptor
    {
        public Object foo(InvocationContext ctx) throws Exception
        {
            return ctx.proceed();
        }
    }

    public static class ABean
    {
        public String foo()
        {
            return "foo";
        }

        public String bar()
        {
            return "bar";
        }
    }

    @ApplicationScoped
    public static class ABeanFactories
    {
        @Produces
        @Named("bean1")
        @ApplicationScoped
        public ABean bean1()
        {
            return new ABean();
        }

        @Foo
        @Named("bean2")
        @Produces
        public ABean bean2(final InterceptionFactory<ABean> factory)
        {
            factory.configure()
                    .methods()
                    .iterator().next().add(new AnnotationLiteral<Foo>()
            {
            });
            return factory.createInterceptedInstance(new ABean());
        }
    }

    @Test
    public void stableNameMultipleTypes()
    {
        addConfiguration("org.apache.webbeans.proxy.useStaticNames", "true");
        addConfiguration("org.apache.webbeans.proxy.staticNames.useXxHash64", "true");
        startContainer(ABeanFactories.class, FooInterceptor.class);
        final ABean bean1 = getInstance("bean1");
        final ABean bean2 = getInstance("bean2");
        assertEquals(
                "org.apache.webbeans.test.managed.ProxyFactoryTest$ABean$$OwbNormalScopeProxy5751833139562769786",
                bean1.getClass().getName());
        assertEquals(
                "org.apache.webbeans.test.managed.ProxyFactoryTest$ABean$$OwbInterceptProxy5751833139562769786",
                bean2.getClass().getName());
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testProxyFactoryWithMultipleInterfaces()
    {
        Collection<String> beanXmls = new ArrayList<String>();

        Collection<Class<?>> beanClasses = new ArrayList<Class<?>>();
        beanClasses.add(MyEntityServiceImpl.class);

        startContainer(beanClasses, beanXmls);

        Set<Bean<?>> beans = getBeanManager().getBeans(MyEntityServiceImpl.class);
        Assert.assertNotNull(beans);

        Bean<MyEntityServiceImpl> bean = (Bean<MyEntityServiceImpl>) beans.iterator().next();
        Assert.assertNotNull(bean);
        
        CreationalContext<MyEntityServiceImpl> ctx = getBeanManager().createCreationalContext(bean);
        
        Object reference = getBeanManager().getReference(bean, MyEntityServiceImpl.class, ctx);
        Assert.assertNotNull(reference);
        
        shutDownContainer();
    }

    @Test
    public void threadSafe() throws Throwable
    {
        Collection<String> beanXmls = new ArrayList<>();

        Collection<Class<?>> beanClasses = new ArrayList<>();
        beanClasses.add(MyEntityServiceImpl.class);

        startContainer(beanClasses, beanXmls);

        final AtomicReference<Throwable> failed = new AtomicReference<>();
        final ClassLoader loader = Thread.currentThread().getContextClassLoader();
        final NormalScopeProxyFactory normalScopeProxyFactory = WebBeansContext.currentInstance().getNormalScopeProxyFactory();
        final Thread[] concurrentRequests = new Thread[16];
        final CountDownLatch starter = new CountDownLatch(1);
        for (int i = 0; i < concurrentRequests.length; i++)
        {
            final int idx = i;
            concurrentRequests[i] = new Thread()
            {
                {
                    setName(getClass().getName() + ".threadSafe-#" + (idx + 1));
                }

                @Override
                public void run()
                {
                    try
                    {
                        starter.await();
                        normalScopeProxyFactory.createProxyClass(loader, MyEntityServiceImpl.class);
                    }
                    catch (final Throwable t)
                    {
                        failed.compareAndSet(null, t);
                    }
                }
            };
        }

        for (final Thread t : concurrentRequests)
        {
            t.start();
        }
        starter.countDown();
        for (final Thread t : concurrentRequests)
        {
            t.join(TimeUnit.MINUTES.toMillis(1));
        }
        if (failed.get() != null) {
            throw failed.get();
        }

        shutDownContainer();
    }
}
