/*
 * 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.configuration.messages;

import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.TransformerException;
import javax.xml.transform.TransformerFactoryConfigurationError;

import org.xml.sax.SAXException;

import org.apache.geode.DataSerializer;
import org.apache.geode.InternalGemFireError;
import org.apache.geode.internal.DataSerializableFixedID;
import org.apache.geode.internal.Version;
import org.apache.geode.internal.lang.StringUtils;
import org.apache.geode.management.internal.configuration.domain.Configuration;
import org.apache.geode.management.internal.configuration.utils.XmlUtils;

/***
 * Response containing the configuration requested by the {@link ConfigurationRequest}
 */
public class ConfigurationResponse implements DataSerializableFixedID {

  private Map<String, Configuration> requestedConfiguration = new HashMap<String, Configuration>();
  private byte[][] jarBytes;
  private String[] jarNames;
  private boolean failedToGetSharedConfig = false;

  public ConfigurationResponse() {

  }

  public ConfigurationResponse(Map<String, Configuration> requestedConfiguration) {
    this.requestedConfiguration.putAll(requestedConfiguration);
  }

  @Override
  public int getDSFID() {
    return DataSerializableFixedID.CONFIGURATION_RESPONSE;
  }

  @Override
  public void toData(DataOutput out) throws IOException {
    DataSerializer.writeHashMap((HashMap<?, ?>) requestedConfiguration, out);
    DataSerializer.writeStringArray(jarNames, out);
    DataSerializer.writeArrayOfByteArrays(jarBytes, out);
    DataSerializer.writeBoolean(Boolean.valueOf(failedToGetSharedConfig), out);
  }

  @Override
  public void fromData(DataInput in) throws IOException, ClassNotFoundException {
    this.requestedConfiguration = DataSerializer.readHashMap(in);
    this.jarNames = DataSerializer.readStringArray(in);
    this.jarBytes = DataSerializer.readArrayOfByteArrays(in);
    this.failedToGetSharedConfig = DataSerializer.readBoolean(in);
  }

  public Map<String, Configuration> getRequestedConfiguration() {
    return this.requestedConfiguration;
  }

  public void setRequestedConfiguration(Map<String, Configuration> requestedConfiguration) {
    this.requestedConfiguration = requestedConfiguration;
  }

  public void addConfiguration(Configuration configuration) {
    if (configuration != null) {
      this.requestedConfiguration.put(configuration.getConfigName(), configuration);
    }
  }

  public String toString() {
    StringBuffer sb = new StringBuffer();
    Set<String> configNames = requestedConfiguration.keySet();
    for (String configName : configNames) {
      sb.append("\n" + requestedConfiguration.get(configName));
    }
    return sb.toString();
  }

  public String describeConfig() {
    StringBuffer sb = new StringBuffer();
    if (requestedConfiguration.isEmpty()) {
      sb.append("Received an empty shared configuration");
    } else {
      Set<Entry<String, Configuration>> entries = requestedConfiguration.entrySet();
      Iterator<Entry<String, Configuration>> iter = entries.iterator();

      while (iter.hasNext()) {
        Entry<String, Configuration> entry = iter.next();
        String configType = entry.getKey();
        Configuration config = entry.getValue();

        if (config != null) {
          sb.append("\n***************************************************************");
          sb.append("\nConfiguration for  '" + configType + "'");
          sb.append("\n\nJar files to deployed");

          Set<String> jarNames = config.getJarNames();
          Iterator<String> jarIter = jarNames.iterator();
          int jarCounter = 0;

          while (jarIter.hasNext()) {
            sb.append("\n" + ++jarCounter + "." + jarIter.next());
          }

          try {
            String cacheXmlContent = config.getCacheXmlContent();
            if (!StringUtils.isBlank(cacheXmlContent)) {
              sb.append("\n" + XmlUtils.prettyXml(cacheXmlContent));
            }
          } catch (IOException | TransformerFactoryConfigurationError | TransformerException
              | SAXException | ParserConfigurationException e) {
            throw new InternalGemFireError(e);
          }
        }

      }
    }
    return sb.toString();
  }


  public String[] getJarNames() {
    return this.jarNames;
  }

  public byte[][] getJars() {
    return this.jarBytes;
  }

  public void addJarsToBeDeployed(String[] jarNames, byte[][] jarBytes) {
    this.jarNames = jarNames;
    this.jarBytes = jarBytes;
  }

  // TODO Sourabh, please review for correctness
  public Version[] getSerializationVersions() {
    return new Version[] {Version.CURRENT};
  }

  public boolean failedToGetSharedConfig() {
    return failedToGetSharedConfig;
  }

  public void setFailedToGetSharedConfig(boolean failedToGetSharedConfig) {
    this.failedToGetSharedConfig = failedToGetSharedConfig;
  }
}