/*
 * Copyright (c) 2017, 2020 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.jdbc.jdbcjmsauth.ejb;

import jakarta.ejb.SessionBean;
import jakarta.ejb.SessionContext;
import jakarta.ejb.CreateException;
import jakarta.jms.QueueConnectionFactory;
import jakarta.jms.QueueConnection;
import jakarta.jms.QueueSession;
import jakarta.jms.Session;
import javax.naming.InitialContext;

public class JmsAuthBean implements SessionBean {

    private SessionContext ctxt_;
    private QueueConnectionFactory qcf_;
    private QueueConnectionFactory qcf1_;

    public void ejbCreate() throws CreateException {
        try {
            InitialContext ic = new InitialContext();
            qcf_ = (QueueConnectionFactory)
                ic.lookup("java:comp/env/jms/MyQueueConnectionFactory");
            qcf1_ = (QueueConnectionFactory)
                ic.lookup("java:comp/env/jms/MyQueueConnectionFactory_CM");
        } catch( Exception e ) {
            e.printStackTrace();
            CreateException ce = new CreateException( e.getMessage() );
            ce.initCause( e );
            throw ce;
        }

    }

    public void ejbRemove() {
    }

    public void ejbActivate() {
    }

    public void ejbPassivate() {
    }

    public void setSessionContext( SessionContext ctxt ) {
        ctxt_ = ctxt;
    }

    public boolean test1() {
        //application auth - getConnection w/ user/pass - must pass
        boolean passed = true;
        QueueConnection con = null;
        QueueSession session = null;

        try {
            con = qcf_.createQueueConnection("guest", "guest");
            session = con.createQueueSession(true, Session.AUTO_ACKNOWLEDGE);
        } catch( Exception e ) {
            System.out.println("------------jms test 1--------------");
            e.printStackTrace();
            System.out.println("------------------------------------");
            passed = false;
        } finally {
            try {
                session.close();
                con.close();

            } catch( Exception e ) {}
        }

        return passed;
    }

    public boolean test2() {
        //application auth - getConnection w/o user/pass - must fail
        boolean passed = true;
        QueueConnection con = null;
        QueueSession session = null;

        try {
            con = qcf_.createQueueConnection();
            session = con.createQueueSession(true, Session.AUTO_ACKNOWLEDGE);
        } catch( Exception e ) {
            System.out.println("------------jms test 2--------------");
            e.printStackTrace();
            System.out.println("------------------------------------");
            passed = false;
        } finally {
            try {
                session.close();
                con.close();
            } catch( Exception e ) {}
        }

        return passed;
    }

    public boolean test3() {
        //application auth - getConnection w/ wrong user/pass - must fail
        boolean passed = false;
        QueueConnection con = null;
        QueueSession session = null;

        try {
            con = qcf_.createQueueConnection("xyz", "xyz");
            session = con.createQueueSession(true, Session.AUTO_ACKNOWLEDGE);
        } catch( Exception e ) {
            System.out.println("------------jms test 3--------------");
            e.printStackTrace();
            System.out.println("------------------------------------");
            passed = true;
        } finally {
            try {
                session.close();
                con.close();
            } catch( Exception e ) {}
        }

        return passed;
    }

    public boolean test4() {
        //application auth - getConnection w/ correct user/pass
        //and then wrong - must pass
        boolean passed = false;
        QueueConnection con = null;
        QueueSession session = null;

        try {
            con = qcf_.createQueueConnection("guest", "guest");
            session = con.createQueueSession(true, Session.AUTO_ACKNOWLEDGE);
        } catch( Exception e ) {
            System.out.println("------------jms test 4--------------");
            e.printStackTrace();
            System.out.println("------------------------------------");
            return passed;
        } finally {
            try {
                session.close();
                con.close();
            } catch( Exception e ) {}
        }

        try {
            con = qcf_.createQueueConnection("xyz", "xyz" );
            session = con.createQueueSession( true, Session.AUTO_ACKNOWLEDGE );
        } catch( Exception e ) {
            System.out.println("------------jms test 4--------------");
            e.printStackTrace();
            System.out.println("------------------------------------");
            passed = true;
        } finally {
            try {
                session.close();
                con.close();
            } catch( Exception e ) {}
        }

        return passed;
    }

    public boolean test5() {
        //container auth - getConnection w/o user/pass - must pass
        boolean passed = true;
        QueueConnection con = null;
        QueueSession session = null;

        try {
            con = qcf1_.createQueueConnection();
            session = con.createQueueSession(true, Session.AUTO_ACKNOWLEDGE);
        } catch( Exception e ) {
            System.out.println("------------jms test 5--------------");
            e.printStackTrace();
            System.out.println("------------------------------------");
            passed = false;
        } finally {
            try {
                session.close();
                con.close();

            } catch( Exception e ) {}
        }

        return passed;
    }

    public boolean test6() {
        //container auth - getConnection w/ user/pass - must pass
        boolean passed = true;
        QueueConnection con = null;
        QueueSession session = null;

        try {
            con = qcf1_.createQueueConnection("guest","guest");
            session = con.createQueueSession(true, Session.AUTO_ACKNOWLEDGE);
        } catch( Exception e ) {
            System.out.println("------------jms test 6--------------");
            e.printStackTrace();
            System.out.println("------------------------------------");
            passed = false;
        } finally {
            try {
                session.close();
                con.close();

            } catch( Exception e ) {}
        }

        return passed;
    }

    public boolean test7() {
        //container auth - getConnection w/ wrong user/pass - must fail
        boolean passed = false;
        QueueConnection con = null;
        QueueSession session = null;

        try {
            con = qcf1_.createQueueConnection("xyz", "xyz");
            session = con.createQueueSession(true, Session.AUTO_ACKNOWLEDGE);
        } catch( Exception e ) {
            System.out.println("------------jms test 7--------------");
            e.printStackTrace();
            System.out.println("------------------------------------");
            passed = true;
        } finally {
            try {
                session.close();
                con.close();

            } catch( Exception e ) {}
        }

        return passed;
    }
}