/*
 * Minecraft Development for IntelliJ
 *
 * https://mcdev.io/
 *
 * Copyright (C) 2023 minecraft-dev
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, version 3.0 only.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
package com.demonwav.mcdev.creator.step
 
import com.intellij.codeInsight.actions.ReformatCodeProcessor
import com.intellij.ide.projectView.ProjectView
import com.intellij.ide.projectWizard.generators.AssetsNewProjectWizardStep
import com.intellij.ide.starters.local.GeneratorAsset
import com.intellij.ide.starters.local.GeneratorEmptyDirectory
import com.intellij.ide.starters.local.GeneratorResourceFile
import com.intellij.ide.starters.local.GeneratorTemplateFile
import com.intellij.ide.wizard.AbstractNewProjectWizardStep
import com.intellij.ide.wizard.NewProjectWizardStep
import com.intellij.openapi.application.ApplicationManager
import com.intellij.openapi.application.WriteAction
import com.intellij.openapi.fileEditor.FileEditorManager
import com.intellij.openapi.project.Project
import com.intellij.openapi.startup.StartupManager
import com.intellij.openapi.util.io.NioFiles
import com.intellij.openapi.vfs.VfsUtil
import com.intellij.openapi.vfs.VirtualFile
import com.intellij.psi.PsiManager
import java.io.IOException
import java.nio.file.Files
import java.nio.file.Path
 
/**
 * Fixed version of [AssetsNewProjectWizardStep], to be removed in 2022.3 when
 * [IDEA-297489 is fixed](https://github.com/JetBrains/intellij-community/commit/fefae70bf621f3181ee9f2d0815c43d0325cd6c4).
 *
 * Written to be drop-in replaced with [AssetsNewProjectWizardStep] when it's ready.
 */
abstract class FixedAssetsNewProjectWizardStep(parent: NewProjectWizardStep) : AbstractNewProjectWizardStep(parent) {
    lateinit var outputDirectory: String
    private val assets = arrayListOf<FixedGeneratorAsset>()
    val templateProperties = hashMapOf<String, Any?>()
    private val filesToOpen = hashSetOf<String>()
 
    fun addAssets(vararg assets: Any) = addAssets(assets.toList())
 
    fun addAssets(assets: Iterable<Any>) {
        assets.mapTo(this.assets) { asset ->
            when (asset) {
                is GeneratorAsset -> GeneratorAssetDelegate(asset)
                is FixedGeneratorAsset -> asset
                else -> throw IllegalArgumentException("$asset is not a valid asset")
            }
        }
    }
 
    fun addTemplateProperties(vararg properties: Pair<String, Any?>) = addTemplateProperties(properties.toMap())
 
    fun addTemplateProperties(properties: Map<String, Any?>) = templateProperties.putAll(properties)
 
    fun addFilesToOpen(vararg relativeCanonicalPaths: String) = addFilesToOpen(relativeCanonicalPaths.toList())
 
    fun addFilesToOpen(relativeCanonicalPaths: Iterable<String>) {
        relativeCanonicalPaths.mapTo(filesToOpen) { "$outputDirectory/$it" }
    }
 
    abstract fun setupAssets(project: Project)
 
    override fun setupProject(project: Project) {
        setupAssets(project)
 
        WriteAction.runAndWait<Throwable> {
            if (project.isDisposed) {
                return@runAndWait
            }
 
            val generatedFiles = mutableSetOf<VirtualFile>()
            for (asset in assets) {
                generateFile(asset)?.let { generatedFiles += it }
            }
 
            runWhenCreated(project) {
                fixupFiles(project, generatedFiles)
            }
        }
    }
 
    fun runWhenCreated(project: Project, action: () -> Unit) {
        if (ApplicationManager.getApplication().isUnitTestMode) {
            action()
        } else if (!project.isDisposed) {
            StartupManager.getInstance(project).runAfterOpened {
                WriteAction.runAndWait<Throwable>(action)
            }
        }
    }
 
    private fun generateFile(asset: FixedGeneratorAsset): VirtualFile? {
        return when (asset) {
            is GeneratorAssetDelegate -> when (val delegate = asset.delegate) {
                is GeneratorTemplateFile -> generateFile(delegate)
                is GeneratorResourceFile -> generateFile(delegate)
                is GeneratorEmptyDirectory -> generateFile(delegate)
            }
            is GeneratorFile -> generateFile(asset)
        }
    }
 
    private fun generateFile(asset: GeneratorTemplateFile): VirtualFile? {
        val code = try {
            asset.template.getText(templateProperties)
        } catch (e: Exception) {
            throw IOException("Unable to process template", e)
        }
 
        val pathStr = "$outputDirectory/${asset.relativePath}"
        val path = Path.of(pathStr)
        path.parent?.let(NioFiles::createDirectories)
        Files.writeString(path, code)
 
        return VfsUtil.findFile(path, true)
    }
 
    private fun generateFile(asset: GeneratorResourceFile): VirtualFile? {
        val content = asset.resource.openStream().use { it.readAllBytes() }
 
        val pathStr = "$outputDirectory/${asset.relativePath}"
        val path = Path.of(pathStr)
        path.parent?.let(NioFiles::createDirectories)
        Files.write(path, content)
 
        return VfsUtil.findFile(path, true)
    }
 
    private fun generateFile(asset: GeneratorEmptyDirectory): VirtualFile? {
        val pathStr = "$outputDirectory/${asset.relativePath}"
        val path = Path.of(pathStr)
        NioFiles.createDirectories(path)
        return VfsUtil.findFile(path, true)
    }
 
    private fun generateFile(asset: GeneratorFile): VirtualFile? {
        val pathStr = "$outputDirectory/${asset.targetFileName}"
        val path = Path.of(pathStr)
        path.parent?.let(NioFiles::createDirectories)
        Files.write(path, asset.content)
 
        return VfsUtil.findFile(path, true)
    }
 
    private fun fixupFiles(project: Project, generatedFiles: Iterable<VirtualFile>) {
        val psiManager = PsiManager.getInstance(project)
        val psiFiles = generatedFiles.mapNotNull { psiManager.findFile(it) }
 
        ReformatCodeProcessor(project, psiFiles.toTypedArray(), null, false).run()
 
        val fileEditorManager = FileEditorManager.getInstance(project)
        val projectView = ProjectView.getInstance(project)
        for (file in generatedFiles) {
            if (file.path in filesToOpen) {
                fileEditorManager.openFile(file, true)
                projectView.select(null, file, false)
            }
        }
    }
}
 
// This can be removed when https://github.com/JetBrains/intellij-community/pull/2304 is merged
sealed class FixedGeneratorAsset {
    abstract val targetFileName: String
}
 
data class GeneratorAssetDelegate(val delegate: GeneratorAsset) : FixedGeneratorAsset() {
    override val targetFileName get() = delegate.relativePath
}
 
class GeneratorFile(
    override val targetFileName: String,
    val content: ByteArray,
) : FixedGeneratorAsset() {
    constructor(targetFileName: String, contents: String) : this(targetFileName, contents.encodeToByteArray())
}

      

        
/*
 * Minecraft Development for IntelliJ
 *
 * https://mcdev.io/
 *
 * Copyright (C) 2024 minecraft-dev
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, version 3.0 only.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
package com.demonwav.mcdev.creator.step
 
import com.intellij.codeInsight.actions.ReformatCodeProcessor
import com.intellij.ide.projectView.ProjectView
import com.intellij.ide.projectWizard.generators.AssetsNewProjectWizardStep
import com.intellij.ide.starters.local.GeneratorAsset
import com.intellij.ide.starters.local.GeneratorEmptyDirectory
import com.intellij.ide.starters.local.GeneratorResourceFile
import com.intellij.ide.starters.local.GeneratorTemplateFile
import com.intellij.ide.wizard.AbstractNewProjectWizardStep
import com.intellij.ide.wizard.NewProjectWizardStep
import com.intellij.openapi.application.ApplicationManager
import com.intellij.openapi.application.WriteAction
import com.intellij.openapi.fileEditor.FileEditorManager
import com.intellij.openapi.project.Project
import com.intellij.openapi.startup.StartupManager
import com.intellij.openapi.util.io.NioFiles
import com.intellij.openapi.vfs.VfsUtil
import com.intellij.openapi.vfs.VirtualFile
import com.intellij.psi.PsiManager
import java.io.IOException
import java.nio.file.Files
import java.nio.file.Path
 
/**
 * Fixed version of [AssetsNewProjectWizardStep], to be removed in 2022.3 when
 * [IDEA-297489 is fixed](https://github.com/JetBrains/intellij-community/commit/fefae70bf621f3181ee9f2d0815c43d0325cd6c4).
 *
 * Written to be drop-in replaced with [AssetsNewProjectWizardStep] when it's ready.
 */
abstract class FixedAssetsNewProjectWizardStep(parent: NewProjectWizardStep) : AbstractNewProjectWizardStep(parent) {
    lateinit var outputDirectory: String
    private val assets = arrayListOf<FixedGeneratorAsset>()
    val templateProperties = hashMapOf<String, Any?>()
    private val filesToOpen = hashSetOf<String>()
 
    fun addAssets(vararg assets: Any) = addAssets(assets.toList())
 
    fun addAssets(assets: Iterable<Any>) {
        assets.mapTo(this.assets) { asset ->
            when (asset) {
                is GeneratorAsset -> GeneratorAssetDelegate(asset)
                is FixedGeneratorAsset -> asset
                else -> throw IllegalArgumentException("$asset is not a valid asset")
            }
        }
    }
 
    fun addTemplateProperties(vararg properties: Pair<String, Any?>) = addTemplateProperties(properties.toMap())
 
    fun addTemplateProperties(properties: Map<String, Any?>) = templateProperties.putAll(properties)
 
    fun addFilesToOpen(vararg relativeCanonicalPaths: String) = addFilesToOpen(relativeCanonicalPaths.toList())
 
    fun addFilesToOpen(relativeCanonicalPaths: Iterable<String>) {
        relativeCanonicalPaths.mapTo(filesToOpen) { "$outputDirectory/$it" }
    }
 
    abstract fun setupAssets(project: Project)
 
    override fun setupProject(project: Project) {
        setupAssets(project)
 
        WriteAction.runAndWait<Throwable> {
            if (project.isDisposed) {
                return@runAndWait
            }
 
            val generatedFiles = mutableSetOf<VirtualFile>()
            for (asset in assets) {
                generateFile(asset)?.let { generatedFiles += it }
            }
 
            runWhenCreated(project) {
                fixupFiles(project, generatedFiles)
            }
        }
    }
 
    fun runWhenCreated(project: Project, action: () -> Unit) {
        if (ApplicationManager.getApplication().isUnitTestMode) {
            action()
        } else if (!project.isDisposed) {
            StartupManager.getInstance(project).runAfterOpened {
                WriteAction.runAndWait<Throwable>(action)
            }
        }
    }
 
    private fun generateFile(asset: FixedGeneratorAsset): VirtualFile? {
        return when (asset) {
            is GeneratorAssetDelegate -> when (val delegate = asset.delegate) {
                is GeneratorTemplateFile -> generateFile(delegate)
                is GeneratorResourceFile -> generateFile(delegate)
                is GeneratorEmptyDirectory -> generateFile(delegate)
            }
            is GeneratorFile -> generateFile(asset)
        }
    }
 
    private fun generateFile(asset: GeneratorTemplateFile): VirtualFile? {
        val code = try {
            asset.template.getText(templateProperties)
        } catch (e: Exception) {
            throw IOException("Unable to process template", e)
        }
 
        val pathStr = "$outputDirectory/${asset.relativePath}"
        val path = Path.of(pathStr)
        path.parent?.let(NioFiles::createDirectories)
        Files.writeString(path, code)
 
        return VfsUtil.findFile(path, true)
    }
 
    private fun generateFile(asset: GeneratorResourceFile): VirtualFile? {
        val content = asset.resource.openStream().use { it.readAllBytes() }
 
        val pathStr = "$outputDirectory/${asset.relativePath}"
        val path = Path.of(pathStr)
        path.parent?.let(NioFiles::createDirectories)
        Files.write(path, content)
 
        return VfsUtil.findFile(path, true)
    }
 
    private fun generateFile(asset: GeneratorEmptyDirectory): VirtualFile? {
        val pathStr = "$outputDirectory/${asset.relativePath}"
        val path = Path.of(pathStr)
        NioFiles.createDirectories(path)
        return VfsUtil.findFile(path, true)
    }
 
    private fun generateFile(asset: GeneratorFile): VirtualFile? {
        val pathStr = "$outputDirectory/${asset.targetFileName}"
        val path = Path.of(pathStr)
        path.parent?.let(NioFiles::createDirectories)
        Files.write(path, asset.content)
 
        return VfsUtil.findFile(path, true)
    }
 
    private fun fixupFiles(project: Project, generatedFiles: Iterable<VirtualFile>) {
        val psiManager = PsiManager.getInstance(project)
        val psiFiles = generatedFiles.mapNotNull { psiManager.findFile(it) }
 
        ReformatCodeProcessor(project, psiFiles.toTypedArray(), null, false).run()
 
        val fileEditorManager = FileEditorManager.getInstance(project)
        val projectView = ProjectView.getInstance(project)
        for (file in generatedFiles) {
            if (file.path in filesToOpen) {
                fileEditorManager.openFile(file, true)
                projectView.select(null, file, false)
            }
        }
    }
}
 
// This can be removed when https://github.com/JetBrains/intellij-community/pull/2304 is merged
sealed class FixedGeneratorAsset {
    abstract val targetFileName: String
}
 
data class GeneratorAssetDelegate(val delegate: GeneratorAsset) : FixedGeneratorAsset() {
    override val targetFileName get() = delegate.relativePath
}
 
class GeneratorFile(
    override val targetFileName: String,
    val content: ByteArray,
) : FixedGeneratorAsset() {
    constructor(targetFileName: String, contents: String) : this(targetFileName, contents.encodeToByteArray())
}