/*
 * 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.streampipes.sdk.builder.adapter;

import org.apache.streampipes.extensions.api.assets.AssetResolver;
import org.apache.streampipes.extensions.api.assets.DefaultAssetResolver;
import org.apache.streampipes.extensions.api.connect.IParser;
import org.apache.streampipes.extensions.api.connect.StreamPipesAdapter;
import org.apache.streampipes.model.AdapterType;
import org.apache.streampipes.model.connect.adapter.AdapterDescription;
import org.apache.streampipes.sdk.builder.AbstractConfigurablePipelineElementBuilder;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.function.Supplier;
import java.util.stream.Collectors;

public class AdapterConfigurationBuilder extends
    AbstractConfigurablePipelineElementBuilder<AdapterConfigurationBuilder, AdapterDescription> {

  private final List<IParser> supportedParsers;
  private final Supplier<StreamPipesAdapter> supplier;

  private AssetResolver assetResolver;

  protected AdapterConfigurationBuilder(String appId,
                                        int version,
                                        Supplier<StreamPipesAdapter> supplier) {
    super(appId, new AdapterDescription(version));
    supportedParsers = new ArrayList<>();
    assetResolver = new DefaultAssetResolver(appId);
    this.supplier = supplier;
  }

  /**
   * Creates a new adapter configuration using the builder pattern.
   * @deprecated
   * This method is no longer recommend since we rely on a version for migration purposes.
   * <p> Please adopt {@link #create(String, int, Supplier)} instead.
   * @param appId     A unique identifier of the new adapter, e.g., com.mycompany.processor.mynewdataprocessor
   * @param supplier  instance of the adapter to be described
   */
  @Deprecated(since = "0.93.0", forRemoval = true)
  public static AdapterConfigurationBuilder create(String appId,
                                                   Supplier<StreamPipesAdapter> supplier) {
    return new AdapterConfigurationBuilder(appId, 0, supplier);
  }

  /**
   * Creates a new adapter configuration using the builder pattern.
   * @param appId     A unique identifier of the new adapter, e.g., com.mycompany.processor.mynewdataprocessor
   * @param supplier  instance of the adapter to be described
   * @param version   version of the processing element for migration purposes. Should be 0 in standard cases.
   *                  Only in case there exist migrations for the specific element the version needs to be aligned.
   */
  public static AdapterConfigurationBuilder create(String appId,
                                                   int version,
                                                   Supplier<StreamPipesAdapter> supplier) {
    return new AdapterConfigurationBuilder(appId, version, supplier);
  }

  @Override
  protected AdapterConfigurationBuilder me() {
    return this;
  }

  @Override
  protected void prepareBuild() {
  }

  @Override
  public AdapterDescription build() {
    return this.elementDescription;
  }

  public AdapterConfiguration buildConfiguration() {
    this.elementDescription.setConfig(getStaticProperties());
    return new AdapterConfiguration(
        this.elementDescription,
        this.supportedParsers,
        assetResolver,
        supplier
    );
  }

  public AdapterConfigurationBuilder withSupportedParsers(IParser... parsers) {
    Collections.addAll(supportedParsers, parsers);
    return this;
  }

  public AdapterConfigurationBuilder withSupportedParsers(List<IParser> parsers) {
    this.supportedParsers.addAll(parsers);
    return this;
  }

  public AdapterConfigurationBuilder withCategory(AdapterType... categories) {
    this.elementDescription
        .setCategory(Arrays
            .stream(categories)
            .map(Enum::name)
            .collect(Collectors.toList()));
    return me();
  }

  public AdapterConfigurationBuilder withVersion(int version) {
    this.elementDescription.setVersion(version);
    return this;
  }

  public AdapterConfigurationBuilder withAssetResolver(AssetResolver assetResolver) {
    this.assetResolver = assetResolver;
    return this;
  }
}
