/*
 * Copyright 2019 Netflix, Inc.
 *
 * Licensed 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 com.netflix.spinnaker.kork.plugins

import com.netflix.spinnaker.kork.plugins.config.ConfigFactory
import com.netflix.spinnaker.kork.plugins.sdk.SdkFactory
import org.pf4j.ExtensionFactory

/**
 * Creates extensions, injecting in dependencies as needed.
 */
class SpinnakerExtensionFactory(
  private val pluginManager: SpinnakerPluginManager,
  private val configFactory: ConfigFactory,
  private val pluginSdkFactories: List<SdkFactory>
) : ExtensionFactory {

  override fun <T> create(extensionClass: Class<T>): T {
    val pluginWrapper = pluginManager.whichPlugin(extensionClass)

    @Suppress("UNCHECKED_CAST")
    return extensionClass.createWithConstructor(
      ClassKind.EXTENSION,
      pluginSdkFactories,
      configFactory,
      pluginWrapper
    ) as T
  }
}
