/*
 * 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.cli.result;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import org.apache.geode.management.internal.cli.json.GfJsonArray;
import org.apache.geode.management.internal.cli.json.GfJsonException;
import org.apache.geode.management.internal.cli.json.GfJsonObject;

/**
 * 
 * 
 * @since GemFire 7.0
 */
public class TabularResultData extends AbstractResultData {
  /* package */ TabularResultData() {
    super();
  }

  /* package */ TabularResultData(GfJsonObject gfJsonObject) {
    super(gfJsonObject);
  }

  public TabularResultData accumulate(String accumulateFor, Object value) {
    try {
      contentObject.accumulate(accumulateFor, value);
    } catch (GfJsonException e) {
      throw new ResultDataException(e.getMessage());
    }
    return this;
  }

  public GfJsonArray getHeaders() {
    try {
      return this.contentObject.names();
    } catch (GfJsonException e) {
      e.printStackTrace();
    }
    return null;
  }

  /**
   * @return the gfJsonObject
   */
  public GfJsonObject getGfJsonObject() {
    return gfJsonObject;
  }

  @Override
  public String getType() {
    return TYPE_TABULAR;
  }

  /**
   * 
   * @param headerText
   * @return this TabularResultData
   * @throws ResultDataException If the value is non-finite number or if the key is null.
   */
  public TabularResultData setHeader(String headerText) {
    return (TabularResultData) super.setHeader(headerText);
  }

  /**
   * 
   * @param footerText
   * @return this TabularResultData
   * @throws ResultDataException If the value is non-finite number or if the key is null.
   */
  public TabularResultData setFooter(String footerText) {
    return (TabularResultData) super.setFooter(footerText);
  }

  @Override
  public String getHeader() {
    return gfJsonObject.getString(RESULT_HEADER);
  }

  @Override
  public String getFooter() {
    return gfJsonObject.getString(RESULT_FOOTER);
  }

  public Map<String, String> retrieveDataByValueInColumn(String columnName, String valueToSearch) {
    Map<String, String> foundValues = Collections.emptyMap();
    try {
      GfJsonArray jsonArray = contentObject.getJSONArray(columnName);
      int size = jsonArray.size();
      int foundIndex = -1;
      for (int i = 0; i < size; i++) {
        Object object = jsonArray.get(i);
        if (object != null && object.equals(valueToSearch)) {
          foundIndex = i;
          break;
        }
      }

      if (foundIndex != -1) {
        foundValues = new LinkedHashMap<String, String>();
        for (Iterator<String> iterator = contentObject.keys(); iterator.hasNext();) {
          String storedColumnNames = (String) iterator.next();
          GfJsonArray storedColumnValues = contentObject.getJSONArray(storedColumnNames);
          foundValues.put(storedColumnNames, String.valueOf(storedColumnValues.get(foundIndex)));
        }
      }
    } catch (GfJsonException e) {
      throw new ResultDataException(e.getMessage());
    }
    return foundValues;
  }

  public List<Map<String, String>> retrieveAllDataByValueInColumn(String columnName,
      String valueToSearch) {
    List<Map<String, String>> foundValuesList = new ArrayList<Map<String, String>>();
    try {
      GfJsonArray jsonArray = contentObject.getJSONArray(columnName);
      int size = jsonArray.size();
      for (int i = 0; i < size; i++) {
        Object object = jsonArray.get(i);
        if (object != null && object.equals(valueToSearch)) {
          Map<String, String> foundValues = new LinkedHashMap<String, String>();

          for (Iterator<String> iterator = contentObject.keys(); iterator.hasNext();) {
            String storedColumnNames = (String) iterator.next();
            GfJsonArray storedColumnValues = contentObject.getJSONArray(storedColumnNames);
            foundValues.put(storedColumnNames, String.valueOf(storedColumnValues.get(i)));
          }

          foundValuesList.add(foundValues);
        }
      }
    } catch (GfJsonException e) {
      throw new ResultDataException(e.getMessage());
    }
    return foundValuesList;
  }

  public List<String> retrieveAllValues(String columnName) {
    List<String> values = new ArrayList<String>();

    try {
      GfJsonArray jsonArray = contentObject.getJSONArray(columnName);
      int size = jsonArray.size();
      for (int i = 0; i < size; i++) {
        values.add(String.valueOf(jsonArray.get(i)));
      }
    } catch (GfJsonException e) {
      throw new ResultDataException(e.getMessage());
    }
    return values;
  }
}