/*
 * Copyright (c) 2000, 2017 Oracle and/or its affiliates. All rights reserved.
 * Copyright (c) 2021 Contributors to the Eclipse Foundation
 *
 * 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
 */

/*
 * @(#)GenericPortMapperClient.java	1.8 06/27/07
 */

package com.sun.messaging.jmq.jmsclient;

import java.io.InputStream;
import java.io.OutputStream;
import java.io.IOException;
import java.net.Socket;
import java.util.Iterator;
import java.util.Map;

import com.sun.messaging.jmq.io.PortMapperTable;
import com.sun.messaging.jmq.io.PortMapperEntry;

/**
 *
 * This is a generic class (ie does not use any JMX/JMX specific classes to read the portmapper.
 *
 * It is assumed that the user of this class will perform the JMS or JMX specific checks e.g. version checking and act
 * accordingly.
 *
 */

public class GenericPortMapperClient {

    protected PortMapperTable portMapperTable = null;
    protected String host;
    protected int port;
    private boolean debug = Debug.debug;

    public GenericPortMapperClient(String host, int port) {
        this.host = host;
        this.port = port;
        init();
    }

    public int getPortForProtocol(String protocol, String type) {
        return getPort(protocol, type, null);
    }

    public int getPortForService(String protocol, String service, String type) {
        return getPort(protocol, type, service);
    }

    /*
     * Get key property value of portmapper entry with the given protocol, type, and serviename. If any of
     * protocol/type/servicename is null, that means it is a "don't care" e.g. if protocol is null, any protocol value is OK
     * - the protocol won't be used to find a matching entry.
     *
     * The portmapper entry can look like the following:
     *
     * jmxrmi rmi JMX 0 [url=theURL]
     *
     * From the above entry, the only property or key is "url" and it's value is "theURL".
     */
    public String getProperty(String key, String protocol, String type, String servicename) {
        String propVal = null;
        Map table = portMapperTable.getServices();
        PortMapperEntry pme = null;

        Iterator pmeIterator = table.values().iterator();
        while (pmeIterator.hasNext()) {
            pme = (PortMapperEntry) pmeIterator.next();

            if ((protocol != null) && !pme.getProtocol().equals(protocol)) {
                continue;
            }

            if ((type != null) && !pme.getType().equals(type)) {
                continue;
            }

            if ((servicename != null) && !pme.getName().equals(servicename)) {
                continue;
            }

            propVal = pme.getProperty(key);
            if (propVal != null) {
                break;
            }

            /*
             * OLD logic - did not handle nulls for protocol if (pme.getProtocol().equals(protocol)){ if
             * (pme.getType().equals(type)){ if (servicename == null){ propVal = pme.getProperty(key); break; } else { if
             * (pme.getName().equals(servicename)){ propVal = pme.getProperty(key); break; } } } }
             */
        }
        return propVal;
    }

    private int getPort(String protocol, String type, String servicename) {
        int port = 25374;
        Map table = portMapperTable.getServices();
        PortMapperEntry pme = null;

        Iterator pmeIterator = table.values().iterator();
        while (pmeIterator.hasNext()) {
            pme = (PortMapperEntry) pmeIterator.next();
            if (pme.getProtocol().equals(protocol)) {
                if (pme.getType().equals(type)) {
                    if (servicename == null) {
                        port = pme.getPort();
                        break;
                    } else {
                        if (pme.getName().equals(servicename)) {
                            port = pme.getPort();
                            break;
                        }
                    }
                }
            }
        }
        return port;
    }

    protected void init() {
        readBrokerPorts();
    }

    public String getPaketVersion() {
        return portMapperTable.getPacketVersion();
    }

    protected void readBrokerPorts() {
        if (debug) {
            Debug.println("PortMapper connecting to host: " + host + "  port: " + port);
        }

        try {
            String version = String.valueOf(PortMapperTable.PORTMAPPER_VERSION) + "\n";

            Socket socket = new Socket(host, port);
            InputStream is = socket.getInputStream();
            OutputStream os = socket.getOutputStream();

            // Write version of portmapper we support to broker
            try {
                os.write(version.getBytes());
                os.flush();
            } catch (IOException e) {
                // This can sometimes fail if the server already wrote
                // the port table and closed the connection
            }

            portMapperTable = new PortMapperTable();
            portMapperTable.read(is);

            is.close();
            socket.close();

        } catch (Exception e) {
            throw new RuntimeException("Exception caught when reading portmapper.", e);
        }
    }

    public static void main(String args[]) {
        try {
            GenericPortMapperClient pmc = new GenericPortMapperClient("localhost", 7676);

            String url = pmc.getProperty("url", "rmi", "JMX", null);

            System.out.println("url = " + url);

        } catch (Exception e) {
            Debug.printStackTrace(e);
        }
    }
}
