/*
 * 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.geode.management.internal;

import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;
import java.net.InetAddress;
import java.util.Map;
import java.util.Properties;

import org.apache.geode.distributed.internal.DistributionConfigImpl;
import org.apache.geode.distributed.internal.tcpserver.TcpClient;
import org.apache.geode.internal.DataSerializableFixedID;
import org.apache.geode.internal.Version;
import org.apache.geode.internal.net.*;

/**
 * Sent to a locator to request it to find (and possibly start) a jmx manager for us. It returns a
 * JmxManagerLocatorResponse.
 * 
 * @since GemFire 7.0
 *
 */
public class JmxManagerLocatorRequest implements DataSerializableFixedID {

  public JmxManagerLocatorRequest() {
    super();
  }

  public void fromData(DataInput in) throws IOException, ClassNotFoundException {}

  public void toData(DataOutput out) throws IOException {}

  public int getDSFID() {
    return DataSerializableFixedID.JMX_MANAGER_LOCATOR_REQUEST;
  }

  @Override
  public String toString() {
    return "JmxManagerLocatorRequest";
  }

  private static final JmxManagerLocatorRequest SINGLETON = new JmxManagerLocatorRequest();

  /**
   * Send a request to the specified locator asking it to find (and start if needed) a jmx manager.
   * A jmx manager will only be started
   * 
   * @param locatorHost the name of the host the locator is on
   * @param locatorPort the port the locator is listening on
   * @param msTimeout how long in milliseconds to wait for a response from the locator
   * @param sslConfigProps Map carrying SSL configuration that can be used by SocketCreator
   * @return a response object that describes the jmx manager.
   * @throws IOException if we can not connect to the locator, timeout waiting for a response, or
   *         have trouble communicating with it.
   */
  public static JmxManagerLocatorResponse send(String locatorHost, int locatorPort, int msTimeout,
      Map<String, String> sslConfigProps) throws IOException {
    Properties distributionConfigProps = new Properties();
    InetAddress networkAddress = InetAddress.getByName(locatorHost);

    try {
      if (sslConfigProps != null) {
        distributionConfigProps.putAll(sslConfigProps);
      }

      TcpClient client = new TcpClient(new DistributionConfigImpl(distributionConfigProps));
      Object responseFromServer =
          client.requestToServer(networkAddress, locatorPort, SINGLETON, msTimeout);

      return (JmxManagerLocatorResponse) responseFromServer;
    } catch (ClassNotFoundException unexpected) {
      throw new IllegalStateException(unexpected);
    } catch (ClassCastException unexpected) {
      // FIXME - Abhishek: object read is type "int" instead of
      // JmxManagerLocatorResponse when the Locator is using SSL & the request
      // didn't use SSL -> this causes ClassCastException. Not sure how to make
      // locator meaningful message
      throw new IllegalStateException(unexpected);
    } finally {
      distributionConfigProps.clear();
    }
  }

  public static JmxManagerLocatorResponse send(String locatorHost, int locatorPort, int msTimeout)
      throws IOException {
    return send(locatorHost, locatorPort, msTimeout, null);
  }

  @Override
  public Version[] getSerializationVersions() {
    // TODO Auto-generated method stub
    return null;
  }
}