// Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

package software.aws.toolkits.jetbrains.services.lambda.nodejs

import com.intellij.javascript.nodejs.interpreter.NodeJsInterpreterField
import com.intellij.javascript.nodejs.interpreter.NodeJsInterpreterManager
import com.intellij.javascript.nodejs.interpreter.NodeJsInterpreterRef
import com.intellij.lang.javascript.dialects.JSLanguageLevel
import com.intellij.lang.javascript.settings.JSRootConfiguration
import com.intellij.openapi.project.ProjectManager
import com.intellij.openapi.roots.ModifiableRootModel
import com.intellij.openapi.ui.TextFieldWithBrowseButton
import com.intellij.openapi.ui.ValidationInfo
import software.aws.toolkits.core.lambda.LambdaRuntime
import software.aws.toolkits.jetbrains.services.lambda.wizard.SamAppTemplateBased
import software.aws.toolkits.jetbrains.services.lambda.wizard.SamProjectTemplate
import software.aws.toolkits.jetbrains.services.lambda.wizard.SamProjectWizard
import software.aws.toolkits.jetbrains.services.lambda.wizard.SdkSelector
import software.aws.toolkits.jetbrains.utils.ui.validationInfo
import software.aws.toolkits.resources.message
import javax.swing.JComponent
import javax.swing.JLabel

class NodeJsSamProjectWizard : SamProjectWizard {
    override fun createSdkSelectionPanel(projectLocation: TextFieldWithBrowseButton?): SdkSelector? = NodeJsSdkSelectionPanel()

    override fun listTemplates(): Collection<SamProjectTemplate> = listOf(
        SamHelloWorldNodeJs(),
        SamHelloWorldNodeJsTypeScript()
    )
}

class NodeJsSdkSelectionPanel : SdkSelector {
    private val interpreterPanel = createInterpreterField()

    private fun createInterpreterField(): NodeJsInterpreterField {
        val project = ProjectManager.getInstance().defaultProject
        return object : NodeJsInterpreterField(project, false) {
            override fun isDefaultProjectInterpreterField(): Boolean = true
        }
    }

    override fun sdkSelectionLabel() = JLabel(message("sam.init.node_interpreter.label"))

    override fun sdkSelectionPanel(): JComponent = interpreterPanel

    override fun applySdkSettings(model: ModifiableRootModel) {
        NodeJsInterpreterManager.getInstance(model.project).setInterpreterRef(NodeJsInterpreterRef.create(interpreterPanel.interpreter))
        JSRootConfiguration.getInstance(model.project).storeLanguageLevelAndUpdateCaches(JSLanguageLevel.ES6)
    }

    override fun validateSelection(): ValidationInfo? = interpreterPanel.interpreter?.validate(null)?.let {
        interpreterPanel.validationInfo(it)
    }
}

class SamHelloWorldNodeJs : SamAppTemplateBased() {
    override fun displayName() = message("sam.init.template.hello_world.name")

    override fun description() = message("sam.init.template.hello_world.description")

    override fun supportedZipRuntimes(): Set<LambdaRuntime> = setOf(
        LambdaRuntime.NODEJS14_X,
        LambdaRuntime.NODEJS16_X,
        LambdaRuntime.NODEJS18_X
    )

    override fun supportedImageRuntimes(): Set<LambdaRuntime> = setOf(
        LambdaRuntime.NODEJS14_X,
        LambdaRuntime.NODEJS16_X,
        LambdaRuntime.NODEJS18_X
    )

    override val appTemplateName: String = "hello-world"

    override val dependencyManager: String = "npm"
}

class SamHelloWorldNodeJsTypeScript : SamAppTemplateBased() {
    override fun displayName() = message("sam.init.template.hello_world_typescript.name")

    override fun description() = message("sam.init.template.hello_world_typescript.description")

    override fun supportedZipRuntimes(): Set<LambdaRuntime> = setOf(
        LambdaRuntime.NODEJS14_X,
        LambdaRuntime.NODEJS16_X,
        LambdaRuntime.NODEJS18_X
    )

    override fun supportedImageRuntimes(): Set<LambdaRuntime> = setOf(
        LambdaRuntime.NODEJS14_X,
        LambdaRuntime.NODEJS16_X,
        LambdaRuntime.NODEJS18_X
    )

    override val appTemplateName: String = "hello-world-typescript"

    override val dependencyManager: String = "npm"
}