/*
 * Copyright (c) 2012-2021 Red Hat, Inc.
 * This program and the accompanying materials are made
 * available under the terms of the Eclipse Public License 2.0
 * which is available at https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *   Red Hat, Inc. - initial API and implementation
 */
package org.eclipse.che.api.system.server;

import com.google.common.annotations.VisibleForTesting;
import com.google.common.collect.ImmutableSortedSet;
import java.util.Comparator;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;
import javax.inject.Inject;
import org.eclipse.che.api.core.notification.EventService;
import org.eclipse.che.api.system.shared.event.service.StoppingSystemServiceEvent;
import org.eclipse.che.api.system.shared.event.service.SuspendingSystemServiceEvent;
import org.eclipse.che.api.system.shared.event.service.SystemServiceStoppedEvent;
import org.eclipse.che.api.system.shared.event.service.SystemServiceSuspendedEvent;
import org.eclipse.che.inject.ConfigurationException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Terminates or suspends system services.
 *
 * @author Yevhenii Voevodin
 */
class ServiceTerminator {

  private static final Logger LOG = LoggerFactory.getLogger(ServiceTerminator.class);

  private final EventService eventService;
  private final Set<ServiceTermination> terminations;

  @Inject
  ServiceTerminator(EventService eventService, Set<ServiceTermination> terminations) {
    this.eventService = eventService;
    checkNamesAndDependencies(terminations);
    this.terminations =
        ImmutableSortedSet.copyOf(new ServiceTerminationComparator(terminations), terminations);
  }

  /**
   * Terminates system services in a order satisfying termination dependencies.
   *
   * @throws InterruptedException when termination is interrupted
   */
  void terminateAll() throws InterruptedException {
    for (ServiceTermination termination : terminations) {
      LOG.info("Shutting down '{}' service", termination.getServiceName());
      doTerminate(termination);
    }
  }

  /**
   * Suspends system services in a order satisfying termination dependencies.
   *
   * @throws InterruptedException when suspending is interrupted
   */
  void suspendAll() throws InterruptedException {
    for (ServiceTermination termination : terminations) {
      LOG.info("Suspending down '{}' service", termination.getServiceName());
      eventService.publish(new SuspendingSystemServiceEvent(termination.getServiceName()));
      try {
        termination.suspend();
        eventService.publish(new SystemServiceSuspendedEvent(termination.getServiceName()));
        LOG.info("Service '{}' is suspended", termination.getServiceName());
      } catch (UnsupportedOperationException e) {
        LOG.info(
            "Suspending down '{}' service isn't supported, terminating it",
            termination.getServiceName());
        doTerminate(termination);
      } catch (InterruptedException x) {
        LOG.error(
            "Interrupted while waiting for '{}' service to suspend", termination.getServiceName());
        throw x;
      }
    }
  }

  @VisibleForTesting
  void doTerminate(ServiceTermination termination) throws InterruptedException {
    eventService.publish(new StoppingSystemServiceEvent(termination.getServiceName()));
    try {
      termination.terminate();
    } catch (InterruptedException x) {
      LOG.error(
          "Interrupted while waiting for '{}' service to shutdown", termination.getServiceName());
      throw x;
    }
    LOG.info("Service '{}' is shut down", termination.getServiceName());
    eventService.publish(new SystemServiceStoppedEvent(termination.getServiceName()));
  }

  private void checkNamesAndDependencies(Set<ServiceTermination> terminationSet) {
    Set<String> uniqueNamesSet = new HashSet<>();
    terminationSet.forEach(
        t -> {
          if (!uniqueNamesSet.add(t.getServiceName())) {
            throw new ConfigurationException(
                String.format(
                    "Duplicate termination found with service name %s", t.getServiceName()));
          }
        });

    terminationSet.forEach(
        t -> {
          if (!uniqueNamesSet.containsAll(t.getDependencies())) {
            throw new RuntimeException(
                String.format("Unknown dependency found in termination %s", t.getServiceName()));
          }
        });
  }

  public static class ServiceTerminationComparator implements Comparator<ServiceTermination> {

    private final Map<String, Set<String>> dependencies;

    public ServiceTerminationComparator(Set<ServiceTermination> terminations) {
      this.dependencies =
          terminations.stream()
              .collect(
                  Collectors.toMap(
                      ServiceTermination::getServiceName, ServiceTermination::getDependencies));
    }

    @Override
    public int compare(ServiceTermination o1, ServiceTermination o2) {
      return checkTransitiveDependency(o1.getServiceName(), o2.getServiceName(), new HashSet<>());
    }

    // Recursively dig into dependencies and sort them out
    private int checkTransitiveDependency(String o1, String o2, Set<String> loopList) {
      if (loopList.contains(o1)) {
        throw new RuntimeException("Circular dependency found between terminations " + loopList);
      }
      Set<String> directDependencies = dependencies.get(o1);
      if (directDependencies.isEmpty()) {
        return -1;
      } else {
        if (directDependencies.contains(o2)) {
          return 1;
        } else {
          loopList.add(o1);
          for (String dependency : directDependencies) {
            if (checkTransitiveDependency(dependency, o2, loopList) > 0) {
              return 1;
            }
          }
          return -1;
        }
      }
    }
  }
}
