/*
 * Copyright (c) 2002, 2018 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

package com.sun.s1asdev.ejb.ejb30.hello.session2;

import jakarta.ejb.Stateful;
import jakarta.ejb.Remote;
import jakarta.ejb.EJB;
import jakarta.annotation.PostConstruct;
import jakarta.interceptor.Interceptors;
import jakarta.ejb.EJBs;
import jakarta.ejb.Remove;
import jakarta.annotation.PreDestroy;
import jakarta.ejb.EJBException;
import jakarta.ejb.SessionContext;
import jakarta.ejb.TransactionManagement;
import jakarta.ejb.TransactionManagementType;
import javax.naming.InitialContext;

import jakarta.annotation.Resource;
import jakarta.transaction.UserTransaction;

import java.util.Collection;
import java.util.HashSet;

@Stateful
@TransactionManagement(TransactionManagementType.BEAN)

@EJBs(
 { @EJB(name="ejb/TypeLevelSless1", beanName="SlessEJB",
       beanInterface=Sless.class),
   @EJB(name="ejb/TypeLevelSless2", beanName="SlessEJB2",
       beanInterface=Sless.class) ,
   @EJB(name="ejb/SfulEJB2", beanInterface=Sful2.class)
 })

@Interceptors(MyCallbackHandler.class)
@EJB(name="ejb/TypeLevelSless3", beanInterface=SlessSub.class)
@Remote({Sful.class})
// ejb is not required to put business interface in implements clause
public class SfulEJB
{

    private @Resource SessionContext sc;

    private @EJB(beanName="SlessEJB") Sless sless;
    private @EJB(beanName="SlessEJB2") Sless sless2;
    private @EJB(beanName="SlessEJB3") SlessSub sless3;

    private Sless sless4;
    private @EJB(beanName="SlessEJB") void setSless4(Sless sless) {
        sless4 = sless;
    }

    private Sless sless5;
    @EJB(beanName="SlessEJB2") void setSless5(Sless sless) {
        sless5 = sless;
    }

    private SlessSub sless6;
    public @EJB void setSless6(SlessSub sless) {
        sless6 = sless;
    }

    @Resource
    private void setSessionContext(SessionContext context) {

        try {
            context.getUserTransaction();
            throw new RuntimeException("Should have gotten IllegalStateEx");
        } catch(IllegalStateException ise) {
            System.out.println("Successfully got exception when accessing " +
                               "context.getUserTransaction() in " +
                               "setContext method of BMT SFSB");
        }

        try {
            context.getTimerService();
            throw new RuntimeException("Should have gotten IllegalStateEx");
        } catch(IllegalStateException ise) {
            System.out.println("Successfully got exception when accessing " +
                               "context.getTimerService() in " +
                               "setContext method");
        }

        try {
            context.getBusinessObject(Sful.class);
            throw new RuntimeException("Should have gotten IllegalStateEx");
        } catch(IllegalStateException ise) {
            System.out.println("Successfully got exception when accessing " +
                               "getBusinessObject() in " +
                               "setContext method");
        }

    }

    private String state;

    public void set(String value) {
        state = value;
    }

    public String get() {
        return state;
    }

    public String hello() {
        System.out.println("In SfulEJB:hello()");

        try {

            testSlessRefs(sless, sless2, sless3);

            testSlessRefs(sless4, sless5, sless6);


            InitialContext ic = new InitialContext();

            Sless sless7 = (Sless)
                ic.lookup("java:comp/env/ejb/TypeLevelSless1");
            Sless sless8 = (Sless)
                ic.lookup("java:comp/env/ejb/TypeLevelSless2");
            SlessSub sless9 = (SlessSub)
                ic.lookup("java:comp/env/ejb/TypeLevelSless3");

            testSlessRefs(sless7, sless8, sless9);

            Sless sless10 = (Sless)
                sc.lookup("ejb/TypeLevelSless1");
            Sless sless11 = (Sless)
                sc.lookup("ejb/TypeLevelSless2");
            SlessSub sless12 = (SlessSub)
                sc.lookup("ejb/TypeLevelSless3");
            testSlessRefs(sless10, sless11, sless12);

            // Make sure we can getUserTransaction via SessionContext.
            // This must work if bean has bean-managed transactions.
            UserTransaction ut = sc.getUserTransaction();

            int status = ut.getStatus();
            System.out.println("context ut status = " + status);

            // Do bogus lookup to ensure exception is thrown
            try {
                sc.lookup("abcdef");
                throw new Exception("lookup should have thrown exception");
            } catch(IllegalArgumentException iae) {
                System.out.println("Got expected exception for bogus lookup");
            }

        } catch(Exception e) {
            e.printStackTrace();
            throw new EJBException("hello failure", e);
        }

        return "hello";
    }

    public void foo() {}
    public void foo(int a, String b) {
    }

    private void testSlessRefs(Sless s1, Sless s2, SlessSub s3)
        throws Exception {

        String sless1Id = s1.getId();
        String sless2Id = s2.getId();
        String sless3Id = s3.getId();

        System.out.println("sless1Id = " + sless1Id);
        System.out.println("sless2Id = " + sless2Id);
        System.out.println("sless3Id = " + sless3Id);

        if( sless1Id.equals(sless2Id) ||
            sless1Id.equals(sless3Id) ||
            sless2Id.equals(sless3Id) ) {
            throw new EJBException("sless bean ejb linking error");
        }

        Sless r1 = s1.roundTrip(s1);
        s1.roundTrip(s2);
        s1.roundTrip(s3);

        s2.roundTrip(s1);
        Sless r2 = s2.roundTrip(s2);
        s2.roundTrip(s3);

        s3.roundTrip(s1);
        s3.roundTrip(s2);
        Sless r3 = s3.roundTrip(s3);

        sless1Id = r1.getId();
        sless2Id = r2.getId();
        sless3Id = r3.getId();

        System.out.println("sless1Id = " + sless1Id);
        System.out.println("sless2Id = " + sless2Id);
        System.out.println("sless3Id = " + sless3Id);

        if( sless1Id.equals(sless2Id) ||
            sless1Id.equals(sless3Id) ||
            sless2Id.equals(sless3Id) ) {
            throw new EJBException("remote 3.0 param passing error");
        }

        Collection c = new HashSet();
        c.add(s2);
        s1.roundTrip2(c);
        s2.roundTrip2(c);
        s3.roundTrip2(c);

        s3.hello3();

    }

    @PostConstruct
    public void afterCreate() {
        System.out.println("In SfulEJB::afterCreate() marked as PostConstruct");

        // jndi lookup should work within postconstruct
        Sless sless = (Sless) sc.lookup("ejb/TypeLevelSless1");

        // Access to getBusinessObject is allowed here
        sc.getBusinessObject(Sful.class);

    }

    @Remove
    public void remove() {
        System.out.println("In SfulEJB " + state + " @Remove method");
        // Access to getBusinessObject is allowed here
        sc.getBusinessObject(Sful.class);

    }

    @Remove(retainIfException=true)
    public void removeRetainIfException(boolean throwException)
        throws Exception {

        System.out.println("In SfulEJB " + state + " removeRetainIfException");
        System.out.println("throwException = " + throwException);
        if( throwException ) {
            throw new Exception("throwing app exception from @Remove method");
        }
    }

    @Remove
    public void removeNotRetainIfException(boolean throwException)
        throws Exception {

        System.out.println("In SfulEJB " + state +
                           "removeNotRetainIfException");
        System.out.println("throwException = " + throwException);
        if( throwException ) {
            throw new Exception("throwing app exception from @Remove method");
        }
    }

    @Remove
    public void removeMethodThrowSysException(boolean throwException) {

        System.out.println("In SfulEJB " + state +
                           "removeMethodThrowSysException");
        System.out.println("throwException = " + throwException);
        if( throwException ) {
            throw new EJBException
                ("throwing system exception from @Remove method");
        }
    }

    public void doRemoveMethodSessionSyncTests() {

        Sful2 sful2_1 = (Sful2) sc.lookup("ejb/SfulEJB2");
        Sful2 sful2_2 = (Sful2) sc.lookup("ejb/SfulEJB2");
        Sful2 sful2_3 = (Sful2) sc.lookup("ejb/SfulEJB2");

        sful2_1.hello();
        try {
            sful2_1.removeRetainIfException(true);
        } catch(Exception e) {
            e.printStackTrace();
        }

        // all SessionSynch callbacks should have happened
        if( SfulEJB2.afterBeginCalled &&
            SfulEJB2.beforeCompletionCalled &&
            SfulEJB2.afterCompletionCalled ) {
            System.out.println("Got expected SessionSynch behavior for " +
                               "removeRetainIfException(true)");
        } else {
            throw new EJBException("SessionSynch failure for " +
                                   "removeRetainIfException(true)");
        }

        // invocation should still work since bean wasn't removed
        sful2_1.hello();

        try {
            sful2_1.removeRetainIfException(false);
        } catch(Exception e) {
            throw new EJBException(e);
        }

        // bean should have been removed and completion-time SessionSynch
        // callbacks should not have been called.
        if( SfulEJB2.afterBeginCalled &&
            !SfulEJB2.beforeCompletionCalled &&
            !SfulEJB2.afterCompletionCalled ) {
            System.out.println("Got expected SessionSynch behavior for " +
                               "removeRetainIfException(false)");
        } else {
            throw new EJBException("SessionSynch failure for " +
                                   "removeRetainIfException(true)");
        }


    }


}