/*
 * Copyright (c) 2017, 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 client;

import com.sun.xml.ws.transport.tcp.io.DataInOutUtils;
import com.sun.xml.ws.transport.tcp.util.TCPConstants;
import com.sun.xml.ws.transport.tcp.util.Version;
import com.sun.xml.ws.transport.tcp.util.VersionController;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.net.Socket;
import java.net.SocketException;
import java.net.UnknownHostException;
import jakarta.annotation.Resource;

import servlet_endpoint.ServletHelloService;
import servlet_endpoint.ServletHello;

import ejb_endpoint.WSHelloEJBService;
import ejb_endpoint.WSHelloEJB;

import com.sun.ejte.ccl.reporter.SimpleReporterAdapter;

public class Client {

    @Resource
    static jakarta.transaction.UserTransaction ut;

    public static final int TIME_OUT = 1000 * 60;

    private static SimpleReporterAdapter stat =
            new SimpleReporterAdapter("appserv-tests");

    public static void main(String[] args) {
        stat.addDescription("soaptcp-test");
        String host = "localhost";
        int port = 8080;
        if (args.length == 1) { // run with appclient?
            args = args[0].split(" ");
        }

        if (args.length >= 2) {
            host = args[0];
            port = Integer.parseInt(args[1]);
        }

        System.out.println("Connecting to host: " + host + " port: " + port);
        Client client = new Client();
        if (client.doPortUnificationTest(host, port)) {
            client.doServletTest();
            client.doEjbTest();
        } else {
            stat.addStatus("SOAP/TCP-Servlet-Endpoint", stat.DID_NOT_RUN);
            stat.addStatus("SOAP/TCP-EJB-Endpoint", stat.DID_NOT_RUN);
        }

        stat.printSummary("soaptcp-test");
    }

    public void doServletTest() {
        try {
            ServletHelloService svc = new ServletHelloService();
            ServletHello port = svc.getServletHelloPort();

            for (int i=0;i<10;i++) {
                String ret = port.sayServletHello("Appserver Tester !");
                if(ret.indexOf("WebSvcTest-Servlet-Hello") == -1) {
                    System.out.println("Unexpected greeting " + ret);
                    stat.addStatus("SOAP/TCP-Servlet-Endpoint", stat.FAIL);
                    return;
                }
                System.out.println(ret);
            }
            stat.addStatus("SOAP/TCP-Servlet-Endpoint", stat.PASS);
        } catch(Exception e) {
            e.printStackTrace();
            stat.addStatus("SOAP/TCP-Servlet-Endpoint", stat.FAIL);
        }
    }

    public void doEjbTest() {
        try {
            WSHelloEJBService svc = new  WSHelloEJBService();
            WSHelloEJB port = svc.getWSHelloEJBPort();
            ut.getStatus();
            for (int i=0;i<10;i++) {
                String ret = port.sayEjbHello("Appserver Tester !");
                if(ret.indexOf("WebSvcTest-EJB-Hello") == -1) {
                    System.out.println("Unexpected greeting " + ret);
                    stat.addStatus("SOAP/TCP-EJB-Endpoint", stat.FAIL);
                    return;
                }
                System.out.println(ret);
            }
            stat.addStatus("SOAP/TCP-EJB-Endpoint", stat.PASS);
        } catch(Exception e) {
            e.printStackTrace();
            stat.addStatus("SOAP/TCP-EJB-Endpoint", stat.FAIL);
        }
    }

    public boolean doPortUnificationTest(String host, int port) {
        try {
            Socket s = new Socket("localhost", 8080);
            s.setSoTimeout(TIME_OUT);
            final VersionController versionController = VersionController.getInstance();
            final Version framingVersion = versionController.getFramingVersion();
            final Version connectionManagementVersion = versionController.getConnectionManagementVersion();

            final OutputStream outputStream = s.getOutputStream();
            outputStream.write(TCPConstants.PROTOCOL_SCHEMA.getBytes("US-ASCII"));

            DataInOutUtils.writeInts4(outputStream, framingVersion.getMajor(),
                    framingVersion.getMinor(),
                    connectionManagementVersion.getMajor(),
                    connectionManagementVersion.getMinor());
            outputStream.flush();

            final InputStream inputStream = s.getInputStream();
            final int[] versionInfo = new int[4];

            DataInOutUtils.readInts4(inputStream, versionInfo, 4);

            final Version serverFramingVersion = new Version(versionInfo[0], versionInfo[1]);
            final Version serverConnectionManagementVersion = new Version(versionInfo[2], versionInfo[3]);

            final boolean success = versionController.isVersionSupported(serverFramingVersion, serverConnectionManagementVersion);
            if (success) {
                stat.addStatus("SOAP/TCP-PortUnification", stat.PASS);
                return true;
            } else {
                System.out.println("SOAP/TCP version mismatch. Sent{fv: " + framingVersion + "; cmv: " + connectionManagementVersion + "}" +
                        " Received{fv: " + serverFramingVersion + "; cmv: " + serverConnectionManagementVersion + "}");
                stat.addStatus("SOAP/TCP-PortUnification", stat.FAIL);
            }
        } catch (Exception e) {
            e.printStackTrace();
            stat.addStatus("SOAP/TCP-PortUnification", stat.FAIL);
        }

        return false;
    }
}