/*
 * 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.translations.inspections
 
import com.demonwav.mcdev.translations.identification.LiteralTranslationIdentifier
import com.demonwav.mcdev.translations.reference.TranslationGotoModel
import com.demonwav.mcdev.util.runWriteAction
import com.intellij.codeInspection.LocalQuickFix
import com.intellij.codeInspection.ProblemDescriptor
import com.intellij.ide.util.gotoByName.ChooseByNamePopup
import com.intellij.ide.util.gotoByName.ChooseByNamePopupComponent
import com.intellij.openapi.application.ModalityState
import com.intellij.openapi.project.Project
import com.intellij.psi.JavaPsiFacade
import com.intellij.psi.PsiLiteralExpression
import com.intellij.psi.PsiNamedElement
import com.intellij.util.IncorrectOperationException
 
class ChangeTranslationQuickFix(private val name: String) : LocalQuickFix {
    override fun getName() = name
 
    override fun applyFix(project: Project, descriptor: ProblemDescriptor) {
        try {
            val literal = descriptor.psiElement as PsiLiteralExpression
            val key = LiteralTranslationIdentifier().identify(literal)?.key ?: return
            val popup = ChooseByNamePopup.createPopup(
                project,
                TranslationGotoModel(project, key.prefix, key.suffix),
                null,
            )
            popup.invoke(
                object : ChooseByNamePopupComponent.Callback() {
                    override fun elementChosen(element: Any) {
                        val selectedKey = (element as PsiNamedElement).name ?: return
                        literal.containingFile.runWriteAction {
                            val insertion = selectedKey.substring(
                                key.prefix.length,
                                selectedKey.length - key.suffix.length,
                            )
                            literal.replace(
                                JavaPsiFacade.getInstance(project).elementFactory.createExpressionFromText(
                                    "\"$insertion\"",
                                    literal.context,
                                ),
                            )
                        }
                    }
                },
                ModalityState.current(),
                false,
            )
        } catch (ignored: IncorrectOperationException) {
        }
    }
 
    override fun startInWriteAction() = false
 
    override fun getFamilyName() = name
}

      

        
/*
 * 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.translations.inspections
 
import com.demonwav.mcdev.translations.identification.LiteralTranslationIdentifier
import com.demonwav.mcdev.translations.reference.TranslationGotoModel
import com.demonwav.mcdev.util.runWriteAction
import com.intellij.codeInspection.LocalQuickFix
import com.intellij.codeInspection.ProblemDescriptor
import com.intellij.ide.util.gotoByName.ChooseByNamePopup
import com.intellij.ide.util.gotoByName.ChooseByNamePopupComponent
import com.intellij.openapi.application.ModalityState
import com.intellij.openapi.project.Project
import com.intellij.psi.PsiNamedElement
import com.intellij.util.IncorrectOperationException
import org.jetbrains.uast.ULiteralExpression
import org.jetbrains.uast.generate.getUastElementFactory
import org.jetbrains.uast.generate.replace
import org.jetbrains.uast.toUElementOfType
 
class ChangeTranslationQuickFix(private val name: String) : LocalQuickFix {
    override fun getName() = name
 
    override fun applyFix(project: Project, descriptor: ProblemDescriptor) {
        try {
            val literal = descriptor.psiElement.toUElementOfType<ULiteralExpression>() ?: return
            val key = LiteralTranslationIdentifier().identify(literal)?.key ?: return
            val popup = ChooseByNamePopup.createPopup(
                project,
                TranslationGotoModel(project, key.prefix, key.suffix),
                null,
            )
            popup.invoke(
                object : ChooseByNamePopupComponent.Callback() {
                    override fun elementChosen(element: Any) {
                        val selectedKey = (element as PsiNamedElement).name ?: return
                        val insertion = selectedKey.substring(
                            key.prefix.length,
                            selectedKey.length - key.suffix.length,
                        )
                        val elementFactory = literal.getUastElementFactory(project) ?: return
                        val replacement = elementFactory.createStringLiteralExpression(insertion, element)
                            ?: return
                        descriptor.psiElement.containingFile.runWriteAction {
                            literal.replace(replacement)
                        }
                    }
                },
                ModalityState.current(),
                false,
            )
        } catch (ignored: IncorrectOperationException) {
        }
    }
 
    override fun startInWriteAction() = false
 
    override fun getFamilyName() = name
}