/*
 * 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.demonwav.mcdev.asset.MCDevBundle
import com.intellij.ide.wizard.AbstractNewProjectWizardStep
import com.intellij.ide.wizard.NewProjectWizardStep
import com.intellij.openapi.progress.ProgressIndicator
import com.intellij.openapi.progress.ProgressManager
import com.intellij.openapi.progress.Task
import com.intellij.openapi.project.Project
import com.intellij.openapi.util.Key
import com.intellij.openapi.util.UserDataHolderEx
import java.util.concurrent.ConcurrentLinkedQueue
 
private typealias TaskQueue = ConcurrentLinkedQueue<AbstractLongRunningStep>
 
/**
 * Creator steps that either take a long time to complete, or need to be run after other steps that take a long time to
 * complete.
 *
 * These steps show an indeterminate progress bar to the user while they are running.
 */
abstract class AbstractLongRunningStep(parent: NewProjectWizardStep) : AbstractNewProjectWizardStep(parent) {
 
    /**
     * The text to display on the progress bar
     */
    abstract val description: String
 
    abstract fun perform(project: Project)
 
    final override fun setupProject(project: Project) {
        val newQueue = TaskQueue()
        val queue = (data as UserDataHolderEx).putUserDataIfAbsent(TASK_QUEUE_KEY, newQueue)
        queue += this
        if (queue === newQueue) {
            startTaskQueue(project, queue)
        }
    }
 
    private fun startTaskQueue(project: Project, queue: TaskQueue) {
        val task = object : Task.Backgroundable(
            project,
            MCDevBundle("creator.step.generic.project_created.message")
        ) {
            override fun run(indicator: ProgressIndicator) {
                if (project.isDisposed) {
                    return
                }
 
                indicator.text = MCDevBundle("creator.step.generic.project_created.message")
                var currentQueue = queue
                while (true) {
                    while (true) {
                        val task = currentQueue.poll() ?: break
                        indicator.text2 = task.description
                        if (project.isDisposed) {
                            return
                        }
                        task.perform(project)
                        if (project.isDisposed) {
                            return
                        }
                    }
                    if ((data as UserDataHolderEx).replace(TASK_QUEUE_KEY, currentQueue, null)) {
                        break
                    }
                    currentQueue = data.getUserData(TASK_QUEUE_KEY) ?: break
                }
                indicator.text2 = null
            }
        }
 
        ProgressManager.getInstance().run(task)
    }
 
    companion object {
        private val TASK_QUEUE_KEY = Key.create<TaskQueue>("${AbstractLongRunningStep::class.java.name}.queue")
    }
}

      

        
/*
 * Minecraft Development for IntelliJ
 *
 * https://mcdev.io/
 *
 * Copyright (C) 2025 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.demonwav.mcdev.asset.MCDevBundle
import com.intellij.ide.wizard.AbstractNewProjectWizardStep
import com.intellij.ide.wizard.NewProjectWizardStep
import com.intellij.openapi.progress.ProgressIndicator
import com.intellij.openapi.progress.ProgressManager
import com.intellij.openapi.progress.Task
import com.intellij.openapi.project.Project
import com.intellij.openapi.util.Key
import com.intellij.openapi.util.UserDataHolderEx
import java.util.concurrent.ConcurrentLinkedQueue
 
private typealias TaskQueue = ConcurrentLinkedQueue<AbstractLongRunningStep>
 
/**
 * Creator steps that either take a long time to complete, or need to be run after other steps that take a long time to
 * complete.
 *
 * These steps show an indeterminate progress bar to the user while they are running.
 */
abstract class AbstractLongRunningStep(parent: NewProjectWizardStep) : AbstractNewProjectWizardStep(parent) {
 
    /**
     * The text to display on the progress bar
     */
    abstract val description: String
 
    abstract fun perform(project: Project)
 
    final override fun setupProject(project: Project) {
        val newQueue = TaskQueue()
        val queue = (data as UserDataHolderEx).putUserDataIfAbsent(TASK_QUEUE_KEY, newQueue)
        queue += this
        if (queue === newQueue) {
            startTaskQueue(project, queue)
        }
    }
 
    private fun startTaskQueue(project: Project, queue: TaskQueue) {
        val task = object : Task.Backgroundable(
            project,
            MCDevBundle("creator.step.generic.project_created.message")
        ) {
            override fun run(indicator: ProgressIndicator) {
                if (project.isDisposed) {
                    return
                }
 
                indicator.text = MCDevBundle("creator.step.generic.project_created.message")
                var currentQueue = queue
                while (true) {
                    while (true) {
                        val task = currentQueue.poll() ?: break
                        indicator.text2 = task.description
                        if (project.isDisposed) {
                            return
                        }
                        task.perform(project)
                        if (project.isDisposed) {
                            return
                        }
                    }
                    if ((data as UserDataHolderEx).replace(TASK_QUEUE_KEY, currentQueue, null)) {
                        break
                    }
                    currentQueue = data.getUserData(TASK_QUEUE_KEY) ?: break
                }
                indicator.text2 = null
            }
        }
 
        ProgressManager.getInstance().run(task)
    }
 
    companion object {
        private val TASK_QUEUE_KEY = Key.create<TaskQueue>("${AbstractLongRunningStep::class.java.name}.queue")
    }
}