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

import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

import org.apache.geode.management.cli.Result;
import org.apache.geode.management.internal.cli.AbstractCliAroundInterceptor;
import org.apache.geode.management.internal.cli.GfshParseResult;
import org.apache.geode.management.internal.cli.commands.ShowMetricsCommand.Category;
import org.apache.geode.management.internal.cli.i18n.CliStrings;
import org.apache.geode.management.internal.cli.result.CommandResult;
import org.apache.geode.management.internal.cli.result.ResultBuilder;

public class ShowMetricsInterceptor extends AbstractCliAroundInterceptor {
  @Override
  public Result preExecution(GfshParseResult parseResult) {
    String export_to_report_to = parseResult.getParamValueAsString(CliStrings.SHOW_METRICS__FILE);
    if (export_to_report_to != null && !export_to_report_to.endsWith(".csv")) {
      return ResultBuilder
          .createUserErrorResult(CliStrings.format(CliStrings.INVALID_FILE_EXTENSION, ".csv"));
    }

    String regionName = parseResult.getParamValueAsString(CliStrings.SHOW_METRICS__REGION);
    String port = parseResult.getParamValueAsString(CliStrings.SHOW_METRICS__CACHESERVER__PORT);
    String member = parseResult.getParamValueAsString(CliStrings.MEMBER);
    String[] categoryArgs = (String[]) parseResult.getParamValue(CliStrings.SHOW_METRICS__CATEGORY);

    if (regionName != null && port != null) {
      return ResultBuilder.createUserErrorResult(
          CliStrings.SHOW_METRICS__CANNOT__USE__REGION__WITH__CACHESERVERPORT);
    }

    if (port != null && member == null) {
      return ResultBuilder
          .createUserErrorResult(CliStrings.SHOW_METRICS__CANNOT__USE__CACHESERVERPORT);
    }

    if (categoryArgs != null) {
      boolean regionProvided = regionName != null;
      boolean portProvided = port != null;
      boolean memberProvided = member != null;
      List<String> validCategories =
          getValidCategoriesAsStrings(regionProvided, memberProvided, portProvided);
      Set<String> userCategories = new HashSet<>(Arrays.asList(categoryArgs));
      userCategories.removeAll(validCategories);
      if (!userCategories.isEmpty()) {
        return getInvalidCategoryResult(userCategories);
      }
    }

    return ResultBuilder.createInfoResult("OK");
  }

  static List<Category> getValidCategories(boolean regionProvided, boolean memberProvided,
      boolean portProvided) {
    if (regionProvided && memberProvided) {
      return ShowMetricsCommand.REGION_METRIC_CATEGORIES;
    }
    if (regionProvided) {
      return ShowMetricsCommand.SYSTEM_REGION_METRIC_CATEGORIES;
    }
    if (memberProvided && portProvided) {
      return ShowMetricsCommand.MEMBER_WITH_PORT_METRIC_CATEGORIES;
    }
    if (memberProvided) {
      return ShowMetricsCommand.MEMBER_METRIC_CATEGORIES;
    }
    return ShowMetricsCommand.SYSTEM_METRIC_CATEGORIES;
  }

  static List<String> getValidCategoriesAsStrings(boolean regionProvided, boolean memberProvided,
      boolean portProvided) {

    return getValidCategories(regionProvided, memberProvided, portProvided).stream().map(Enum::name)
        .collect(Collectors.toList());
  }


  private CommandResult getInvalidCategoryResult(Set<String> invalidCategories) {
    StringBuilder sb = new StringBuilder();
    sb.append("Invalid Categories\n");
    for (String category : invalidCategories) {
      sb.append(category);
      sb.append('\n');
    }
    return ResultBuilder.createUserErrorResult(sb.toString());
  }
}
