org.eclipse.jkube.kit.enricher.api.DefaultEnricherManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jkube-kit-enricher-api Show documentation
The newest version!
/*
 * Copyright (c) 2019 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.jkube.kit.enricher.api;

import io.fabric8.kubernetes.api.model.KubernetesListBuilder;
import org.eclipse.jkube.kit.common.KitLogger;
import org.eclipse.jkube.kit.common.util.ClassUtil;
import org.eclipse.jkube.kit.common.util.PluginServiceFactory;
import org.eclipse.jkube.kit.config.resource.EnricherManager;
import org.eclipse.jkube.kit.config.resource.PlatformMode;
import org.eclipse.jkube.kit.config.resource.ProcessorConfig;

import java.util.Collections;
import java.util.List;
import java.util.Optional;
import java.util.function.Function;

import static org.eclipse.jkube.kit.enricher.api.util.Misc.filterEnrichers;

public class DefaultEnricherManager implements EnricherManager {

  private static final String[] SERVICE_PATHS = new String[] {
      "META-INF/jkube/enricher-default",
      "META-INF/jkube-enricher-default",
      "META-INF/jkube-enricher",
      "META-INF/jkube/enricher"
  };

  // List of enrichers used for customizing the generated deployment descriptors
  private final List enrichers;

  // context used by enrichers
  private final ProcessorConfig defaultEnricherConfig;

  private final KitLogger log;

  public DefaultEnricherManager(EnricherContext enricherContext) {
    this(enricherContext, Collections.emptyList());
  }

  public DefaultEnricherManager(EnricherContext enricherContext, List extraClasspathElements) {
    this.defaultEnricherConfig = Optional.ofNullable(enricherContext.getConfiguration().getProcessorConfig())
        .orElse(ProcessorConfig.EMPTY);
    this.log = enricherContext.getLog();
    final PluginServiceFactory pluginFactory = new PluginServiceFactory<>(enricherContext);
    if (!extraClasspathElements.isEmpty()) {
      pluginFactory.addAdditionalClassLoader(
          ClassUtil.createProjectClassLoader(extraClasspathElements, enricherContext.getLog()));
    }
    this.enrichers = pluginFactory.createServiceObjects(SERVICE_PATHS);

    logEnrichers(filterEnrichers(defaultEnricherConfig, enrichers));
  }

  @Override
  public void createDefaultResources(PlatformMode platformMode, final KubernetesListBuilder builder) {
    createDefaultResources(platformMode, defaultEnricherConfig, builder);
  }

  @Override
  public void createDefaultResources(PlatformMode platformMode, ProcessorConfig enricherConfig,
      final KubernetesListBuilder builder) {
    // Add default resources
    loop(enricherConfig, enricher -> {
      enricher.create(platformMode, builder);
      return null;
    });
  }

  @Override
  public void enrich(PlatformMode platformMode, KubernetesListBuilder builder) {
    enrich(platformMode, defaultEnricherConfig, builder);
  }

  /**
   * Allow enricher to add Metadata to the resources.
   *
   * @param builder builder to customize
   */
  @Override
  public void enrich(PlatformMode platformMode, final ProcessorConfig enricherConfig, final KubernetesListBuilder builder) {
    loop(enricherConfig, enricher -> {
      enricher.enrich(platformMode, builder);
      return null;
    });
  }

  // =============================================================================================
  private void logEnrichers(List enrichers) {
    log.verbose("Enrichers:");
    for (Enricher enricher : enrichers) {
      log.verbose("- %s", enricher.getName());
    }
  }

  private void loop(ProcessorConfig config, Function function) {
    for (Enricher enricher : filterEnrichers(config, enrichers)) {
      function.apply(enricher);
    }
  }
}